[sdc] update to the current code base
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / components / impl / ArtifactResolverTest.java
1 package org.openecomp.sdc.be.components.impl;
2
3 import org.junit.Before;
4 import org.junit.Test;
5 import org.openecomp.sdc.be.datatypes.enums.ComponentTypeEnum;
6 import org.openecomp.sdc.be.model.ArtifactDefinition;
7 import org.openecomp.sdc.be.model.ComponentInstance;
8 import org.openecomp.sdc.be.model.Resource;
9 import org.openecomp.sdc.be.model.Service;
10
11 import java.util.Collections;
12 import java.util.Map;
13
14 import static org.junit.Assert.assertNotNull;
15 import static org.junit.Assert.assertNull;
16
17 public class ArtifactResolverTest {
18
19     private ArtifactResolverImpl testInstance = new ArtifactResolverImpl();
20     private Service service, noArtifactsService;
21     private Resource resource, noArtifactsResource;
22     private ComponentInstance componentInstance, noArtifactsInstance;
23
24     @Before
25     public void setUp() throws Exception {
26         noArtifactsService = new Service();
27         noArtifactsResource = new Resource();
28         resource = new Resource();
29         service = new Service();
30         componentInstance = new ComponentInstance();
31         noArtifactsInstance = new ComponentInstance();
32
33         ArtifactDefinition artifact1 = new ArtifactDefinition();
34         artifact1.setUniqueId("a1");
35
36         ArtifactDefinition artifact2 = new ArtifactDefinition();
37         artifact2.setUniqueId("a2");
38
39         ArtifactDefinition artifact3 = new ArtifactDefinition();
40         artifact3.setUniqueId("a3");
41
42         Map<String, ArtifactDefinition> artifact1Map = Collections.singletonMap("key1", artifact1);
43         Map<String, ArtifactDefinition> artifact2Map = Collections.singletonMap("key1", artifact2);
44         Map<String, ArtifactDefinition> artifact3Map = Collections.singletonMap("key1", artifact3);
45
46         resource.setDeploymentArtifacts(artifact1Map);
47         resource.setArtifacts(artifact2Map);
48
49         service.setDeploymentArtifacts(artifact1Map);
50         service.setArtifacts(artifact2Map);
51         service.setServiceApiArtifacts(artifact3Map);
52
53         componentInstance.setDeploymentArtifacts(artifact1Map);
54         componentInstance.setArtifacts(artifact2Map);
55     }
56
57     @Test
58     public void findArtifactOnComponent_noArtifactsOnComponent() throws Exception {
59         assertNull(testInstance.findArtifactOnComponent(noArtifactsResource, ComponentTypeEnum.RESOURCE, "someId"));
60         assertNull(testInstance.findArtifactOnComponent(noArtifactsService, ComponentTypeEnum.SERVICE, "someId"));
61     }
62
63     @Test
64     public void findArtifactOnComponent_resource() throws Exception {
65         assertNull(testInstance.findArtifactOnComponent(resource, ComponentTypeEnum.RESOURCE, "someId"));
66         assertNotNull(testInstance.findArtifactOnComponent(resource, ComponentTypeEnum.RESOURCE, "a1"));
67         assertNotNull(testInstance.findArtifactOnComponent(resource, ComponentTypeEnum.RESOURCE, "a2"));
68     }
69
70     @Test
71     public void findArtifactOnComponent_service() throws Exception {
72         assertNull(testInstance.findArtifactOnComponent(service, ComponentTypeEnum.SERVICE, "someId"));
73         assertNotNull(testInstance.findArtifactOnComponent(service, ComponentTypeEnum.SERVICE, "a1"));
74         assertNotNull(testInstance.findArtifactOnComponent(service, ComponentTypeEnum.SERVICE, "a2"));
75         assertNotNull(testInstance.findArtifactOnComponent(service, ComponentTypeEnum.SERVICE, "a3"));
76     }
77
78     @Test
79     public void findArtifactOnInstance_instanceHasNoArtifacts() throws Exception {
80         assertNull(testInstance.findArtifactOnComponentInstance(noArtifactsInstance, "someId"));
81     }
82
83     @Test
84     public void findArtifactOnInstance() throws Exception {
85         assertNull(testInstance.findArtifactOnComponentInstance(componentInstance, "someId"));
86         assertNotNull(testInstance.findArtifactOnComponentInstance(componentInstance, "a1"));
87         assertNotNull(testInstance.findArtifactOnComponentInstance(componentInstance, "a2"));
88     }
89 }