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 {
version.setMinor(0);
ComponentEntity componentEntity = getComponentEntity(vspId, version, componentId);
- setMockToEnrichComponent(vspId, componentId, version);
ProcessEntity entity = new ProcessEntity(vspId, version, componentId, null);
ProcessEntity processEntity = new ProcessEntity();
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);
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);
}
- private void setMockToEnrichComponent(String vspId, String componentId, Version version) {
- ProcessEntity returnedArtifact = new ProcessEntity();
- returnedArtifact.setVspId(vspId);
- returnedArtifact.setVersion(version);
- returnedArtifact.setComponentId(componentId);
- returnedArtifact.setArtifactName("artifact_1kb.txt");
- returnedArtifact.setArtifact(getMibByteBuffer("/mock/enrichProcess/artifact_1kb.txt"));
-
- Mockito.when(processDaoMock.get(anyObject()))
- .thenReturn(returnedArtifact);
- Mockito.doNothing().when(enrichedServiceModelDaoMock).storeExternalArtifact(anyObject());
- }
-
private ComponentEntity getComponentEntity(String vspId, Version version, String componentId) {
ComponentEntity componentEntity = new ComponentEntity();
componentEntity.setId(componentId);