Merge from ECOMP's repository
[vid.git] / vid-app-common / src / test / java / org / onap / vid / controller / AaiServiceInstanceStandardQueryControllerTest.java
1 package org.onap.vid.controller;
2
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;
20
21 import java.util.UUID;
22
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.*;
27
28 public class AaiServiceInstanceStandardQueryControllerTest {
29
30     @InjectMocks
31     private AaiServiceInstanceStandardQueryController aaiServiceInstanceStandardQueryController;
32
33     @Mock
34     private FeatureManager featureManager;
35
36     @Mock
37     private VidService sdcService;
38
39     @Mock(answer = Answers.RETURNS_MOCKS)
40     private ServiceInstanceStandardQuery serviceInstanceStandardQuery;
41
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
45     @BeforeClass
46     public void initMocks() {
47         MockitoAnnotations.initMocks(this);
48     }
49
50     @AfterMethod
51     public void resetMocks() {
52         reset(sdcService, featureManager, serviceInstanceStandardQuery);
53     }
54
55     @Test
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
61
62         when(featureManager.isActive(Features.FLAG_PRESENT_PROVIDER_NETWORKS_ASSOCIATIONS)).thenReturn(true);
63
64         final UUID randomModelUuid = UUID.randomUUID();
65         mockServiceModel(ModelCategory.IS_5G_PROVIDER_NETWORK_MODEL, randomModelUuid);
66
67         doGetNetworksToVlansByServiceInstance(randomModelUuid);
68
69         verify(serviceInstanceStandardQuery).fetchServiceInstance(any(), any(), any());
70     }
71
72     @Test
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
79
80         when(featureManager.isActive(Features.FLAG_PRESENT_PROVIDER_NETWORKS_ASSOCIATIONS)).thenReturn(true);
81
82         final UUID randomModelUuid = UUID.randomUUID();
83         mockServiceModel(ModelCategory.OTHER, randomModelUuid);
84
85         assertThat(doGetNetworksToVlansByServiceInstance(randomModelUuid).serviceNetworks, hasSize(0));
86         verifyZeroInteractions(serviceInstanceStandardQuery);
87     }
88
89     @Test
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);
93
94         assertTrue(aaiServiceInstanceStandardQueryController.isModelOf5g(randomModelUuid));
95     }
96
97     private void mockServiceModel(ModelCategory modelCategory, UUID randomModelUuid) throws AsdcCatalogException {
98         mockServiceModel(modelCategory, randomModelUuid, VidNotions.InstantiationUI.LEGACY);
99     }
100
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)
107         );
108
109         when(sdcService.getService(randomModelUuid.toString())).thenReturn(mockedModel);
110     }
111
112     private AaiServiceInstanceStandardQueryController.VlansByNetworksHierarchy doGetNetworksToVlansByServiceInstance(UUID randomModelUuid) throws AsdcCatalogException {
113         return aaiServiceInstanceStandardQueryController.getNetworksToVlansByServiceInstance(
114                 new MockHttpServletRequest(),
115                 randomModelUuid,
116                 "my global customer id",
117                 "my service type",
118                 "my instance id");
119     }
120 }