1 package org.onap.vid.controller;
3 import org.mockito.Answers;
4 import org.mockito.InjectMocks;
5 import org.mockito.Mock;
6 import org.mockito.MockitoAnnotations;
7 import org.onap.vid.aai.util.ServiceInstanceStandardQuery;
8 import org.onap.vid.asdc.AsdcCatalogException;
9 import org.onap.vid.model.Service;
10 import org.onap.vid.model.ServiceModel;
11 import org.onap.vid.model.VidNotions;
12 import org.onap.vid.model.VidNotions.ModelCategory;
13 import org.onap.vid.properties.Features;
14 import org.onap.vid.services.VidService;
15 import org.springframework.mock.web.MockHttpServletRequest;
16 import org.testng.annotations.AfterMethod;
17 import org.testng.annotations.BeforeClass;
18 import org.testng.annotations.Test;
19 import org.togglz.core.manager.FeatureManager;
21 import java.util.UUID;
23 import static org.hamcrest.Matchers.hasSize;
24 import static org.junit.Assert.assertThat;
25 import static org.junit.Assert.assertTrue;
26 import static org.mockito.Mockito.*;
28 public class AaiServiceInstanceStandardQueryControllerTest {
31 private AaiServiceInstanceStandardQueryController aaiServiceInstanceStandardQueryController;
34 private FeatureManager featureManager;
37 private VidService sdcService;
39 @Mock(answer = Answers.RETURNS_MOCKS)
40 private ServiceInstanceStandardQuery serviceInstanceStandardQuery;
42 //Don't use initMocks with @BeforeMethod
43 //because AaiServiceInstanceStandardQueryController contains final members that can not be injected twice
44 //See https://stackoverflow.com/questions/20046210/mockito-injectmocks-strange-behaviour-with-final-fields?answertab=active#tab-top
46 public void initMocks() {
47 MockitoAnnotations.initMocks(this);
51 public void resetMocks() {
52 reset(sdcService, featureManager, serviceInstanceStandardQuery);
56 public void getNetworksToVlansByServiceInstance_given5G_PROVIDER_NETWORK_aaiIsAccessed() throws AsdcCatalogException {
57 // - turn on FLAG_PRESENT_PROVIDER_NETWORKS
58 // - mock an model with 5G_PROVIDER_NETWORK
59 // - request it's AAI network->vlan mapping
60 // - assert that AAI was accessed
62 when(featureManager.isActive(Features.FLAG_PRESENT_PROVIDER_NETWORKS_ASSOCIATIONS)).thenReturn(true);
64 final UUID randomModelUuid = UUID.randomUUID();
65 mockServiceModel(ModelCategory.IS_5G_PROVIDER_NETWORK_MODEL, randomModelUuid);
67 doGetNetworksToVlansByServiceInstance(randomModelUuid);
69 verify(serviceInstanceStandardQuery).fetchServiceInstance(any(), any(), any());
73 public void getNetworksToVlansByServiceInstance_givenNon5G_PROVIDER_NETWORK_aaiIsNotAccessed() throws AsdcCatalogException {
74 // - turn on FLAG_PRESENT_PROVIDER_NETWORKS
75 // - mock an model without 5G_PROVIDER_NETWORK (i.e. OTHER)
76 // - request it's AAI network->vlan mapping
77 // - assert that AAI was not accessed
78 // - empty result was responded
80 when(featureManager.isActive(Features.FLAG_PRESENT_PROVIDER_NETWORKS_ASSOCIATIONS)).thenReturn(true);
82 final UUID randomModelUuid = UUID.randomUUID();
83 mockServiceModel(ModelCategory.OTHER, randomModelUuid);
85 assertThat(doGetNetworksToVlansByServiceInstance(randomModelUuid).serviceNetworks, hasSize(0));
86 verifyZeroInteractions(serviceInstanceStandardQuery);
90 public void isModelOf5g_givenServiceWithFabricConfiguration_returnTrue() throws AsdcCatalogException {
91 final UUID randomModelUuid = UUID.randomUUID();
92 mockServiceModel(ModelCategory.IS_5G_FABRIC_CONFIGURATION_MODEL, randomModelUuid, VidNotions.InstantiationUI.SERVICE_WITH_FABRIC_CONFIGURATION);
94 assertTrue(aaiServiceInstanceStandardQueryController.isModelOf5g(randomModelUuid));
97 private void mockServiceModel(ModelCategory modelCategory, UUID randomModelUuid) throws AsdcCatalogException {
98 mockServiceModel(modelCategory, randomModelUuid, VidNotions.InstantiationUI.LEGACY);
101 private void mockServiceModel(ModelCategory modelCategory, UUID randomModelUuid, VidNotions.InstantiationUI instantiationUI) throws AsdcCatalogException {
102 ServiceModel mockedModel = mock(ServiceModel.class);
103 Service mockedService = mock(Service.class);
104 when(mockedModel.getService()).thenReturn(mockedService);
105 when(mockedService.getVidNotions()).thenReturn(
106 new VidNotions(instantiationUI, modelCategory, VidNotions.InstantiationUI.LEGACY)
109 when(sdcService.getService(randomModelUuid.toString())).thenReturn(mockedModel);
112 private AaiServiceInstanceStandardQueryController.VlansByNetworksHierarchy doGetNetworksToVlansByServiceInstance(UUID randomModelUuid) throws AsdcCatalogException {
113 return aaiServiceInstanceStandardQueryController.getNetworksToVlansByServiceInstance(
114 new MockHttpServletRequest(),
116 "my global customer id",