1 package org.onap.aai.sparky.viewandinspect;
3 import static org.junit.Assert.assertEquals;
5 import org.junit.Before;
7 import org.mockito.Mockito;
8 import org.onap.aai.sparky.config.oxm.OxmEntityLookup;
9 import org.onap.aai.sparky.config.oxm.OxmModelLoader;
10 import org.onap.aai.sparky.dal.ActiveInventoryAdapter;
11 import org.onap.aai.sparky.dal.ElasticSearchAdapter;
12 import org.onap.aai.sparky.dal.GizmoAdapter;
13 import org.onap.aai.sparky.subscription.config.SubscriptionConfig;
14 import org.onap.aai.sparky.sync.config.ElasticSearchEndpointConfig;
15 import org.onap.aai.sparky.sync.config.ElasticSearchSchemaConfig;
16 import org.onap.aai.sparky.util.OxmModelAndProcessorHelper;
17 import org.onap.aai.sparky.viewandinspect.config.VisualizationConfigs;
18 import org.onap.aai.sparky.viewandinspect.entity.QueryRequest;
19 import org.onap.aai.sparky.viewandinspect.services.BaseVisualizationService;
20 import org.onap.aai.sparky.viewandinspect.util.SchemaVisualizationTestDataBuilder;
22 public class BaseVisualizationServiceTest {
23 private ActiveInventoryAdapter mockAaiAdapter;
24 private ElasticSearchAdapter mockEsAdapter;
25 private VisualizationConfigs visualizationConfigs;
26 private SubscriptionConfig subConfig;
27 private ElasticSearchEndpointConfig endpointEConfig;
28 private ElasticSearchSchemaConfig schemaEConfig;
29 private OxmEntityLookup oxmEntityLookup;
30 private GizmoAdapter mockGizmoAdapter;
32 private BaseVisualizationService baseVisService;
35 public void init() throws Exception {
36 this.mockAaiAdapter = Mockito.mock(ActiveInventoryAdapter.class);
37 this.mockAaiAdapter = Mockito.mock(ActiveInventoryAdapter.class);
38 this.mockEsAdapter = Mockito.mock(ElasticSearchAdapter.class);
39 this.mockGizmoAdapter = Mockito.mock(GizmoAdapter.class);
40 this.visualizationConfigs = new VisualizationConfigs();
41 this.subConfig = new SubscriptionConfig();
42 this.endpointEConfig = new ElasticSearchEndpointConfig();
43 this.schemaEConfig = new ElasticSearchSchemaConfig();
44 this.oxmEntityLookup = OxmModelAndProcessorHelper.getInstance().getOxmEntityLookup();
46 OxmModelLoader modelLoader = OxmModelAndProcessorHelper.getInstance().getModelLoader();
48 this.baseVisService = new BaseVisualizationService(modelLoader, visualizationConfigs,
49 mockAaiAdapter, mockGizmoAdapter, mockEsAdapter, endpointEConfig, schemaEConfig, 1,
50 oxmEntityLookup, subConfig);
54 public void testAnalyzeQueryRequestBody() {
55 QueryRequest validResquest = baseVisService.analyzeQueryRequestBody(SchemaVisualizationTestDataBuilder.getQueryRequest());
56 assertEquals(SchemaVisualizationTestDataBuilder.ROOT_NODE_HASH_ID, validResquest.getHashId());
58 QueryRequest nullRequest = baseVisService.analyzeQueryRequestBody("This String should make the request return null eh!");
59 assertEquals(null, nullRequest);
63 public void testBuildVisualizationUsingGenericQuery() {
65 initializeMocksForBuildVisualizationUsingGenericQueryTest();
67 QueryRequest rootNodeQuery = baseVisService.analyzeQueryRequestBody(SchemaVisualizationTestDataBuilder.getQueryRequest());
71 private void initializeMocksForBuildVisualizationUsingGenericQueryTest() {
72 Mockito.when(mockAaiAdapter.queryActiveInventoryWithRetries(Mockito.anyString(), Mockito.anyString(), Mockito.anyInt())).thenReturn(null);