1 package org.onap.aai.sparky.viewandinspect;
3 import static org.junit.Assert.*;
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.subscription.config.SubscriptionConfig;
13 import org.onap.aai.sparky.sync.config.ElasticSearchEndpointConfig;
14 import org.onap.aai.sparky.sync.config.ElasticSearchSchemaConfig;
15 import org.onap.aai.sparky.util.OxmModelAndProcessorHelper;
16 import org.onap.aai.sparky.viewandinspect.config.VisualizationConfigs;
17 import org.onap.aai.sparky.viewandinspect.entity.QueryRequest;
18 import org.onap.aai.sparky.viewandinspect.services.BaseVisualizationService;
19 import org.onap.aai.sparky.viewandinspect.util.SchemaVisualizationTestDataBuilder;
21 public class BaseVisualizationServiceTest {
22 private ActiveInventoryAdapter mockAaiAdapter;
23 private ElasticSearchAdapter mockEsAdapter;
24 private VisualizationConfigs visualizationConfigs;
25 private SubscriptionConfig subConfig;
26 private ElasticSearchEndpointConfig endpointEConfig;
27 private ElasticSearchSchemaConfig schemaEConfig;
28 private OxmEntityLookup oxmEntityLookup;
30 private BaseVisualizationService baseVisService;
33 public void init() throws Exception {
34 this.mockAaiAdapter = Mockito.mock(ActiveInventoryAdapter.class);
35 this.mockEsAdapter = Mockito.mock(ElasticSearchAdapter.class);
36 this.visualizationConfigs = new VisualizationConfigs();
37 this.subConfig = new SubscriptionConfig();
38 this.endpointEConfig = new ElasticSearchEndpointConfig();
39 this.schemaEConfig = new ElasticSearchSchemaConfig();
40 this.oxmEntityLookup = OxmModelAndProcessorHelper.getInstance().getOxmEntityLookup();
42 OxmModelLoader modelLoader = OxmModelAndProcessorHelper.getInstance().getModelLoader();
44 this.baseVisService = new BaseVisualizationService(modelLoader, visualizationConfigs, mockAaiAdapter,
45 mockEsAdapter, endpointEConfig, schemaEConfig, 1, oxmEntityLookup, subConfig);
49 public void testAnalyzeQueryRequestBody() {
50 QueryRequest validResquest = baseVisService.analyzeQueryRequestBody(SchemaVisualizationTestDataBuilder.getQueryRequest());
51 assertEquals(SchemaVisualizationTestDataBuilder.ROOT_NODE_HASH_ID, validResquest.getHashId());
53 QueryRequest nullRequest = baseVisService.analyzeQueryRequestBody("This String should make the request return null eh!");
54 assertEquals(null, nullRequest);
58 public void testBuildVisualizationUsingGenericQuery() {
60 initializeMocksForBuildVisualizationUsingGenericQueryTest();
62 QueryRequest rootNodeQuery = baseVisService.analyzeQueryRequestBody(SchemaVisualizationTestDataBuilder.getQueryRequest());
66 private void initializeMocksForBuildVisualizationUsingGenericQueryTest() {
67 Mockito.when(mockAaiAdapter.queryActiveInventoryWithRetries(Mockito.anyString(), Mockito.anyString(), Mockito.anyInt())).thenReturn(null);