blob: 59931a1a117be97c11e140856d00b25010117d27 (
plain)
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
|
package org.openecomp.sdc.enrichment.impl.external.artifact;
public class ProcessArtifactEnricherTest {
/*@Mock
ProcessDao processDaoMock;
@Mock
EnrichedServiceModelDao enrichedServiceModelDaoMock;
@Mock
ComponentDao componentDaoMock;
@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<>();
componentList.add(componentEntity);
when(componentDaoMock.list(anyObject())).thenReturn(componentList);
Collection<ProcessEntity> list = new ArrayList<>();
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) {
byte[] mibBytes = FileUtils.readViaInputStream(this.getClass().getResource(fileName),
stream -> FileUtils.toByteArray(stream));
return ByteBuffer.wrap(mibBytes);
}*/
}
|