Upgrade SDC from Titan to Janus Graph
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / components / impl / DataTypesServiceTest.java
1 package org.openecomp.sdc.be.components.impl;
2
3 import fj.data.Either;
4 import junit.framework.Assert;
5 import org.junit.Before;
6 import org.junit.Test;
7
8 import static org.mockito.ArgumentMatchers.any;
9 import static org.mockito.ArgumentMatchers.eq;
10 import static org.mockito.Mockito.when;
11 import org.mockito.Mockito;
12 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
13 import org.openecomp.sdc.be.impl.ComponentsUtils;
14 import org.openecomp.sdc.be.model.DataTypeDefinition;
15 import org.openecomp.sdc.be.model.cache.ApplicationDataTypeCache;
16
17
18 import java.util.HashMap;
19 import java.util.Map;
20
21 public class DataTypesServiceTest {
22     ApplicationDataTypeCache applicationDataTypeCache = Mockito.mock(ApplicationDataTypeCache.class);
23     ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
24
25     DataTypesService dataTypesService = new DataTypesService(componentsUtils);
26     Map<String, DataTypeDefinition> mapreturn = new HashMap<>();
27     JanusGraphOperationStatus janusGraphOperationStatus = JanusGraphOperationStatus.NOT_FOUND;
28     Either<Map<String, DataTypeDefinition>, JanusGraphOperationStatus> allDataTypes;
29
30     @Before
31     public void setup() {
32         mapreturn.put("Demo",new DataTypeDefinition());
33         allDataTypes = Either.left(mapreturn);
34         when(applicationDataTypeCache.getAll()).thenReturn(allDataTypes);
35
36     }
37
38     @Test
39     public void getAllDataTypes_success() {
40         Assert.assertEquals(true,dataTypesService.getAllDataTypes(applicationDataTypeCache).isLeft());
41     }
42
43     @Test
44     public void getAllDataTypes_failure() {
45         allDataTypes = Either.right(janusGraphOperationStatus);
46         when(applicationDataTypeCache.getAll()).thenReturn(allDataTypes);
47         Assert.assertEquals(true,dataTypesService.getAllDataTypes(applicationDataTypeCache).isRight());
48     }
49
50 }