1 package org.onap.so.adapters.audit;
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;
8 import org.junit.Before;
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;
24 @RunWith(MockitoJUnitRunner.Silent.class)
25 public class AuditDataServiceTest {
28 AuditDataService auditDataService = new AuditDataService();
31 protected RequestsDbClient requestsDbClient;
33 AuditInventory auditInventory = new AuditInventory();
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");
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());
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);
59 GraphInventoryCommonObjectMapperProvider objectMapper = new GraphInventoryCommonObjectMapperProvider();
60 String auditListString = objectMapper.getMapper().writeValueAsString(auditList);;
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);
69 auditDataService.writeStackDataToRequestDb(auditInventory, auditList);
70 Mockito.verify(requestsDbClient, Mockito.times(1)).saveRequestProcessingData(requestProcessingData);
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);
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);
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");