1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
package org.openecomp.sdc.enrichment.impl.external.artifact;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.openecomp.core.enrichment.types.ArtifactCategory;
import org.openecomp.core.model.dao.EnrichedServiceModelDao;
import org.openecomp.core.model.types.ServiceArtifact;
import org.openecomp.core.utilities.file.FileUtils;
import org.openecomp.sdc.enrichment.EnrichmentInfo;
import org.openecomp.sdc.vendorsoftwareproduct.dao.ProcessDao;
import org.openecomp.sdc.vendorsoftwareproduct.dao.VendorSoftwareProductDao;
import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity;
import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessEntity;
import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ProcessType;
import org.openecomp.sdc.versioning.dao.types.Version;
import org.testng.Assert;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import java.io.File;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.Collection;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Mockito.atLeastOnce;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.when;
public class ProcessArtifactEnricherTest {
@Mock
ProcessDao processDaoMock;
@Mock
EnrichedServiceModelDao enrichedServiceModelDaoMock;
@Mock
VendorSoftwareProductDao vendorSoftwareProductDaoMock;
@InjectMocks
ProcessArtifactEnricher processArtifactEnricher;
@BeforeMethod(alwaysRun = true)
public void injectDoubles() {
MockitoAnnotations.initMocks(this);
}
@Test
public void testEnrichComponent() throws Exception {
String vspId = "123";
String componentId = "1111111111";
Version version = new Version();
version.setMajor(1);
version.setMinor(0);
ComponentEntity componentEntity = getComponentEntity(vspId, version, componentId);
ProcessEntity entity = new ProcessEntity(vspId, version, componentId, null);
ProcessEntity processEntity = new ProcessEntity();
processEntity.setType(ProcessType.Lifecycle_Operations);
processEntity.setVspId(vspId);
processEntity.setVersion(version);
processEntity.setComponentId(componentId);
processEntity.setArtifactName("artifact_1kb.txt");
processEntity.setArtifact(getMibByteBuffer("/mock/enrichProcess/artifact_1kb.txt"));
Collection<ComponentEntity> componentList = new ArrayList<ComponentEntity>();
componentList.add(componentEntity);
when(vendorSoftwareProductDaoMock.listComponents(vspId, version)).thenReturn(componentList);
Collection<ProcessEntity> list = new ArrayList<ProcessEntity>();
list.add(processEntity);
when(processDaoMock.list(entity)).thenReturn(list);
when(processDaoMock.get(anyObject())).thenReturn(processEntity);
EnrichmentInfo info = new EnrichmentInfo();
info.setVersion(version);
info.setKey(vspId);
processArtifactEnricher.enrich(info);
String componentName = componentEntity.getComponentCompositionData().getName();
ArgumentCaptor<ServiceArtifact> expectedServiceArtifact =
ArgumentCaptor.forClass(ServiceArtifact.class);
Mockito.verify(enrichedServiceModelDaoMock, atLeastOnce())
.storeExternalArtifact(expectedServiceArtifact.capture());
Assert
.assertEquals(expectedServiceArtifact.getValue().getName().startsWith(componentName), true);
Assert.assertEquals(expectedServiceArtifact.getValue().getName(),
componentName + File.separator + ArtifactCategory.DEPLOYMENT.getDisplayName() +
File.separator + "Lifecycle Operations" + File.separator + "artifact_1kb.txt");
}
private ComponentEntity getComponentEntity(String vspId, Version version, String componentId) {
ComponentEntity componentEntity = new ComponentEntity();
componentEntity.setId(componentId);
componentEntity.setVspId(vspId);
componentEntity.setVersion(version);
String componentName = vspId + "enrichMib_server";
String compositionData = "{\n" +
" \"name\": \"org.openecomp.resource.vfc.nodes.heat." + componentName + "\",\n" +
" \"displayName\": \"" + componentName + "\"\n" +
"}";
componentEntity.setCompositionData(compositionData);
return componentEntity;
}
private ByteBuffer getMibByteBuffer(String fileName) {
InputStream mibFile = FileUtils.getFileInputStream(this.getClass().getResource(fileName));
byte[] mibBytes = FileUtils.toByteArray(mibFile);
return ByteBuffer.wrap(mibBytes);
}
}
|
