1 package org.onap.aai.sparky.viewandinspect.services;
 
   3 import static org.junit.Assert.assertEquals;
 
   5 import java.security.SecureRandom;
 
   6 import java.util.ArrayList;
 
   7 import java.util.HashSet;
 
   9 import java.util.concurrent.ExecutorService;
 
  11 import org.junit.Before;
 
  12 import org.junit.Test;
 
  13 import org.mockito.Mockito;
 
  14 import org.onap.aai.cl.api.Logger;
 
  15 import org.onap.aai.cl.eelf.LoggerFactory;
 
  16 import org.onap.aai.cl.mdc.MdcContext;
 
  17 import org.onap.aai.sparky.config.oxm.OxmEntityLookup;
 
  18 import org.onap.aai.sparky.config.oxm.OxmModelProcessor;
 
  19 import org.onap.aai.sparky.dal.GizmoAdapter;
 
  20 import org.onap.aai.sparky.dal.rest.config.RestEndpointConfig;
 
  21 import org.onap.aai.sparky.util.NodeUtils;
 
  22 import org.onap.aai.sparky.viewandinspect.config.VisualizationConfigs;
 
  24 public class BaseGizmoVisualizationContextTest {
 
  28   private static SecureRandom secureRandom = new SecureRandom();
 
  29   private static Logger LOG =
 
  30       LoggerFactory.getInstance().getLogger(BaseGizmoVisualizationContextTest.class);
 
  32   private BaseGizmoVisualizationContext baseGizmoVisualizationContext;
 
  33   private ExecutorService graphExecutorService;
 
  34   private VisualizationConfigs visualizationConfig;
 
  36   private OxmEntityLookup oxmEntityLookup;
 
  38   private GizmoAdapter gizmoAdapter;
 
  39   private RestEndpointConfig gizmoRestEndPointConfig;
 
  42   public void init() throws Exception {
 
  43     long contextId = secureRandom.nextLong();
 
  44     baseGizmoVisualizationContext = new BaseGizmoVisualizationContext(contextId, gizmoAdapter,
 
  45         graphExecutorService, visualizationConfig, oxmEntityLookup);
 
  47     graphExecutorService = NodeUtils.createNamedExecutor("SLNC-WORKER", 5, LOG);
 
  48     visualizationConfig = new VisualizationConfigs();
 
  50     ArrayList<String> shallowEntities = new ArrayList<String>();
 
  51     shallowEntities.add("cloud-region");
 
  53     visualizationConfig.setShallowEntities(shallowEntities);
 
  54     visualizationConfig.setMaxSelfLinkTraversalDepth(2);
 
  56     oxmEntityLookup = new OxmEntityLookup();
 
  58     gizmoAdapter = Mockito.mock(GizmoAdapter.class);
 
  61     Set<OxmModelProcessor> processors = new HashSet<OxmModelProcessor>();
 
  63     processors.add(oxmEntityLookup);
 
  67     gizmoRestEndPointConfig = new RestEndpointConfig();
 
  68     gizmoRestEndPointConfig.setNumRequestRetries(5);
 
  70     Mockito.when(gizmoAdapter.getEndpointConfig()).thenReturn(gizmoRestEndPointConfig);
 
  72     MdcContext.initialize("" + secureRandom.nextLong(), "AAI-UI", "", "partner-name",
 
  79   public void validateBasicConstruction() throws Exception {
 
  81     long contextId = secureRandom.nextLong();
 
  83     baseGizmoVisualizationContext = new BaseGizmoVisualizationContext(contextId, gizmoAdapter,
 
  84         graphExecutorService, visualizationConfig, oxmEntityLookup);
 
  86     assertEquals(contextId, baseGizmoVisualizationContext.getContextId());