Merge "Fix default homing"
[so.git] / adapters / mso-openstack-adapters / src / test / java / org / onap / so / adapters / audit / AuditDataServiceTest.java
1 package org.onap.so.adapters.audit;
2
3 import static org.junit.Assert.assertEquals;
4 import static org.mockito.ArgumentMatchers.any;
5 import static org.mockito.Mockito.when;
6 import java.util.ArrayList;
7 import java.util.List;
8 import org.junit.Before;
9 import org.junit.Test;
10 import org.junit.runner.RunWith;
11 import org.mockito.InjectMocks;
12 import org.mockito.Mock;
13 import org.mockito.Mockito;
14 import org.mockito.junit.MockitoJUnitRunner;
15 import org.onap.aai.domain.yang.Vserver;
16 import org.onap.so.audit.beans.AuditInventory;
17 import org.onap.so.client.graphinventory.GraphInventoryCommonObjectMapperProvider;
18 import org.onap.so.db.request.beans.RequestProcessingData;
19 import org.onap.so.db.request.client.RequestsDbClient;
20 import org.onap.so.objects.audit.AAIObjectAudit;
21 import org.onap.so.objects.audit.AAIObjectAuditList;
22 import com.fasterxml.jackson.core.JsonProcessingException;
23
24 @RunWith(MockitoJUnitRunner.Silent.class)
25 public class AuditDataServiceTest {
26
27     @InjectMocks
28     AuditDataService auditDataService = new AuditDataService();
29
30     @Mock
31     protected RequestsDbClient requestsDbClient;
32
33     AuditInventory auditInventory = new AuditInventory();
34
35     @Before
36     public void before() throws JsonProcessingException {
37         auditInventory.setCloudOwner("testCloudOwner");
38         auditInventory.setCloudRegion("testLcpCloudRegionId");
39         auditInventory.setHeatStackName("testVfModuleName1");
40         auditInventory.setVfModuleId("testVnfModuleId");
41         auditInventory.setTenantId("testTenantId");
42         auditInventory.setGenericVnfId("testVnfId1");
43     }
44
45     @Test
46     public void testWriteStackDataToRequestDb() throws Exception {
47         Mockito.doReturn(new ArrayList<RequestProcessingData>()).when(requestsDbClient)
48                 .getRequestProcessingDataByGroupingIdAndNameAndTag(Mockito.any(), Mockito.any(), Mockito.any());
49         Mockito.doNothing().when(requestsDbClient).saveRequestProcessingData(Mockito.any());
50
51         AAIObjectAuditList auditList = new AAIObjectAuditList();
52         auditList.setHeatStackName("testHeatStackName");
53         AAIObjectAudit audit = new AAIObjectAudit();
54         Vserver vserver = new Vserver();
55         vserver.setVserverId("testVserverId");
56         audit.setAaiObject(vserver);
57         auditList.getAuditList().add(audit);
58
59         GraphInventoryCommonObjectMapperProvider objectMapper = new GraphInventoryCommonObjectMapperProvider();
60         String auditListString = objectMapper.getMapper().writeValueAsString(auditList);;
61
62         RequestProcessingData requestProcessingData = new RequestProcessingData();
63         requestProcessingData.setSoRequestId(auditInventory.getMsoRequestId());
64         requestProcessingData.setGroupingId(auditInventory.getVfModuleId());
65         requestProcessingData.setName(auditInventory.getHeatStackName());
66         requestProcessingData.setTag("AuditStackData");
67         requestProcessingData.setValue(auditListString);
68
69         auditDataService.writeStackDataToRequestDb(auditInventory, auditList);
70         Mockito.verify(requestsDbClient, Mockito.times(1)).saveRequestProcessingData(requestProcessingData);
71     }
72
73     @Test
74     public void testGetStackDataToRequestDb() throws Exception {
75         AAIObjectAuditList auditList = new AAIObjectAuditList();
76         auditList.setHeatStackName("testHeatStackName");
77         AAIObjectAudit audit = new AAIObjectAudit();
78         Vserver vserver = new Vserver();
79         vserver.setVserverId("testVserverId");
80         audit.setAaiObject(vserver);
81         auditList.getAuditList().add(audit);
82         GraphInventoryCommonObjectMapperProvider objectMapper = new GraphInventoryCommonObjectMapperProvider();
83         String auditListString = objectMapper.getMapper().writeValueAsString(audit);
84
85         List<RequestProcessingData> list = new ArrayList();
86         RequestProcessingData requestProcessingData = new RequestProcessingData();
87         requestProcessingData.setId(234321432);
88         requestProcessingData.setGroupingId("testVfModuleId");
89         requestProcessingData.setName("heatStackName");
90         requestProcessingData.setTag("AuditStackData");
91         requestProcessingData.setValue(auditListString);
92         list.add(requestProcessingData);
93
94         Mockito.doReturn(list).when(requestsDbClient).getRequestProcessingDataByGroupingIdAndNameAndTag(Mockito.any(),
95                 Mockito.any(), Mockito.any());
96         auditDataService.getStackDataFromRequestDb(auditInventory);
97         Mockito.verify(requestsDbClient, Mockito.times(1)).getRequestProcessingDataByGroupingIdAndNameAndTag(
98                 "testVnfModuleId", "testVfModuleName1", "AuditStackData");
99     }
100
101 }