9aa0842bb384a8d801da2698d9e093864e199239
[sdc.git] /
1 package org.openecomp.sdc.be.model.operations.impl;
2
3 import org.junit.Before;
4 import org.junit.Test;
5 import org.openecomp.sdc.be.dao.graph.datatype.GraphEdge;
6 import org.openecomp.sdc.be.dao.neo4j.GraphEdgePropertiesDictionary;
7 import org.openecomp.sdc.be.model.ComponentInstance;
8 import org.openecomp.sdc.be.model.operations.api.ToscaDefinitionPathCalculator;
9
10 import java.util.Collections;
11 import java.util.List;
12
13 import static org.junit.Assert.assertEquals;
14
15 public class ToscaDefinitionPathCalculatorTest {
16
17     private static final String INSTANCE_ID = "123";
18     private static final String OWNER_ID = "321";
19
20     private ToscaDefinitionPathCalculator toscaDefinitionPathCalculator;
21
22
23     @Before
24     public void setUp() throws Exception {
25         toscaDefinitionPathCalculator = new ToscaDefinitionPathCalculatorImpl();
26     }
27
28     @Test
29     public void calculatePath_ownerAndComponentInstanceEqual() throws Exception {
30         ComponentInstance instance = getComponentInstance(INSTANCE_ID);
31         GraphEdge edge = createEdgeWithOwner(INSTANCE_ID);
32         List<String> definitionPath = toscaDefinitionPathCalculator.calculateToscaDefinitionPath(instance, edge);
33         assertEquals(1, definitionPath.size());
34         assertEquals(INSTANCE_ID, definitionPath.get(0));
35     }
36
37     @Test
38     public void calculatePath() throws Exception {
39         ComponentInstance instance = getComponentInstance(INSTANCE_ID);
40         GraphEdge edge = createEdgeWithOwner(OWNER_ID);
41         List<String> definitionPath = toscaDefinitionPathCalculator.calculateToscaDefinitionPath(instance, edge);
42         assertEquals(2, definitionPath.size());
43         assertEquals(INSTANCE_ID, definitionPath.get(0));
44         assertEquals(OWNER_ID, definitionPath.get(1));
45
46     }
47
48
49     private ComponentInstance getComponentInstance(String instanceId) {
50         ComponentInstance instance = new ComponentInstance();
51         instance.setUniqueId(instanceId);
52         return instance;
53     }
54
55     private GraphEdge createEdgeWithOwner(String owner) {
56         GraphEdge edge = new GraphEdge();
57         edge.setProperties(Collections.singletonMap(GraphEdgePropertiesDictionary.OWNER_ID.getProperty(), owner));
58         return edge;
59     }
60
61 }