re base code
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / components / path / BaseForwardingPathTest.java
1 package org.openecomp.sdc.be.components.path;
2
3 import com.google.common.collect.Lists;
4 import fj.data.Either;
5 import org.junit.Before;
6 import org.openecomp.sdc.be.components.BeConfDependentTest;
7 import org.openecomp.sdc.be.components.impl.ComponentInstanceBusinessLogic;
8 import org.openecomp.sdc.be.components.impl.ServiceBusinessLogic;
9 import org.openecomp.sdc.be.components.path.beans.TitanGraphTestSetup;
10 import org.openecomp.sdc.be.components.path.utils.GraphTestUtils;
11 import org.openecomp.sdc.be.dao.api.ActionStatus;
12 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
13 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
14 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
15 import org.openecomp.sdc.be.dao.titan.TitanGraphClient;
16 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
17 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathDataDefinition;
18 import org.openecomp.sdc.be.datatypes.elements.ForwardingPathElementDataDefinition;
19 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
20 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
21 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
22 import org.openecomp.sdc.be.datatypes.enums.ResourceTypeEnum;
23 import org.openecomp.sdc.be.model.DistributionStatusEnum;
24 import org.openecomp.sdc.be.model.LifecycleStateEnum;
25 import org.openecomp.sdc.be.model.Service;
26 import org.openecomp.sdc.be.model.User;
27 import org.openecomp.sdc.be.model.category.CategoryDefinition;
28 import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
29 import org.openecomp.sdc.be.model.operations.api.IElementOperation;
30 import org.openecomp.sdc.be.model.operations.impl.UniqueIdBuilder;
31 import org.openecomp.sdc.be.tosca.CapabilityRequirementConverter;
32 import org.openecomp.sdc.be.user.Role;
33 import org.openecomp.sdc.common.util.ValidationUtils;
34 import org.openecomp.sdc.exception.ResponseFormat;
35 import org.springframework.beans.factory.annotation.Autowired;
36
37 import java.util.ArrayList;
38 import java.util.HashMap;
39 import java.util.Map;
40
41 import static org.junit.Assert.assertNull;
42 import static org.junit.Assert.assertTrue;
43
44 public abstract class BaseForwardingPathTest extends BeConfDependentTest implements ForwardingPathTestUtils {
45
46
47     protected User user;
48     private ForwardingPathDataDefinition forwardingPathDataDefinition;
49
50     @Autowired
51     protected TitanGraphClient titanGraphClient;
52
53     @Autowired
54     protected CapabilityRequirementConverter capabiltyRequirementConvertor;
55
56     @Autowired
57     protected ToscaOperationFacade toscaOperationFacade;
58
59     @Autowired
60     protected ServiceBusinessLogic bl;
61
62     @Autowired
63     protected IElementOperation elementDao;
64
65     @Autowired
66     protected ComponentInstanceBusinessLogic componentInstanceBusinessLogic;
67
68     @javax.annotation.Resource
69     protected TitanDao titanDao;
70
71     @Before
72     public void initTitan() {
73         TitanGraphTestSetup.createGraph(titanGraphClient.getGraph().left().value());
74         categoryDefinition = new CategoryDefinition();
75         categoryDefinition.setName(CATEGORY_NAME);
76     }
77
78     @Before
79     public void initUser() {
80         // User data and management
81         user = new User();
82         user.setUserId("jh0003");
83         user.setFirstName("Jimmi");
84         user.setLastName("Hendrix");
85         user.setRole(Role.ADMIN.name());
86     }
87
88
89     private CategoryDefinition categoryDefinition;
90     private static final String CATEGORY_NAME = "cat_name";
91     static final String FORWARDING_PATH_ID = "forwarding_pathId";
92     static final String HTTP_PROTOCOL = "http";
93     private static final String INSTANTIATION_TYPE = "A-la-carte";
94     private static final String CAPABILITY_NAME_1 = "CP1";
95     private static final String CAPABILITY_NAME_2 = "CP2";
96     private static final String CAPABILITY_NAME_3 = "CP3";
97     private static final String CI_NAME_1 = "CI1";
98     private static final String CI_NAME_2 = "CI2";
99     private static final String CI_NAME_3 = "CI3";
100
101
102     private void initGraph() {
103         Map<GraphPropertyEnum, Object> props = new HashMap<>();
104         props.put(GraphPropertyEnum.IS_HIGHEST_VERSION, true);
105         props.put(GraphPropertyEnum.STATE, LifecycleStateEnum.CERTIFIED.name());
106         props.put(GraphPropertyEnum.TOSCA_RESOURCE_NAME, "org.openecomp.resource.abstract.nodes.service");
107
108         GraphTestUtils.createServiceVertex(titanDao, props);
109
110         GraphVertex resourceVertex = GraphTestUtils.createResourceVertex(titanDao, props, ResourceTypeEnum.PNF);
111         resourceVertex.setJsonMetadataField(JsonPresentationFields.VERSION, "0.1");
112         Either<GraphVertex, TitanOperationStatus> vertexTitanOperationStatusEither = titanDao.updateVertex(resourceVertex);
113         assertTrue(vertexTitanOperationStatusEither.isLeft());
114     }
115
116     private Service createTestService() {
117         createCategory();
118         createServiceCategory(CATEGORY_NAME);
119         initGraph();
120         Service service = new Service();
121         service.setName("ForwardingPathTestingService");
122         service.setDescription("Just a comment.");
123         service.setTags(Lists.newArrayList(service.getName(), service.getComponentType().getValue() + service.getName() + "2"));
124         service.setContactId("as123y");
125         service.setIcon("MyIcon");
126         service.setProjectCode("414155");
127         service.setInstantiationType(INSTANTIATION_TYPE);
128         ArrayList<CategoryDefinition> categories = new ArrayList<>();
129         CategoryDefinition cd = new CategoryDefinition();
130         cd.setName(CATEGORY_NAME);
131         cd.setNormalizedName("abcde");
132         categories.add(cd);
133         service.setCategories(categories);
134         service.setLifecycleState(LifecycleStateEnum.NOT_CERTIFIED_CHECKOUT);
135         service.setDistributionStatus(DistributionStatusEnum.DISTRIBUTION_APPROVED);
136         return service;
137     }
138
139     private void createCategory() {
140         Either<CategoryDefinition, ActionStatus> category = elementDao.createCategory(categoryDefinition, NodeTypeEnum.ServiceNewCategory);
141         assertTrue("Failed to create category", category.isLeft());
142     }
143
144     private void createServiceCategory(String categoryName) {
145         GraphVertex cat = new GraphVertex(VertexTypeEnum.SERVICE_CATEGORY);
146         Map<GraphPropertyEnum, Object> metadataProperties = new HashMap<>();
147         String catId = UniqueIdBuilder.buildComponentCategoryUid(categoryName, VertexTypeEnum.SERVICE_CATEGORY);
148         cat.setUniqueId(catId);
149         metadataProperties.put(GraphPropertyEnum.UNIQUE_ID, catId);
150         metadataProperties.put(GraphPropertyEnum.LABEL, VertexTypeEnum.SERVICE_CATEGORY.getName());
151         metadataProperties.put(GraphPropertyEnum.NAME, categoryName);
152         metadataProperties.put(GraphPropertyEnum.NORMALIZED_NAME, ValidationUtils.normalizeCategoryName4Uniqueness(categoryName));
153         cat.setMetadataProperties(metadataProperties);
154         cat.updateMetadataJsonWithCurrentMetadataProperties();
155
156         Either<GraphVertex, TitanOperationStatus> catRes = titanDao.createVertex(cat);
157
158         assertTrue(catRes.isLeft());
159     }
160
161     Service initForwardPath() {
162         ForwardingPathDataDefinition forwardingPathDataDefinition = createMockPath();
163         Service service = new Service();
164         service.setUniqueId(FORWARDING_PATH_ID);
165         assertNull(service.addForwardingPath(forwardingPathDataDefinition));
166         return service;
167     }
168
169     private ForwardingPathDataDefinition createMockPath() {
170         if (forwardingPathDataDefinition != null) {
171             return forwardingPathDataDefinition;
172         }
173         forwardingPathDataDefinition = new ForwardingPathDataDefinition("Yoyo");
174         forwardingPathDataDefinition.setUniqueId(java.util.UUID.randomUUID().toString());
175         forwardingPathDataDefinition.setDestinationPortNumber("414155");
176         forwardingPathDataDefinition.setProtocol(HTTP_PROTOCOL);
177         org.openecomp.sdc.be.datatypes.elements.ListDataDefinition<org.openecomp.sdc.be.datatypes.elements.ForwardingPathElementDataDefinition> forwardingPathElementDataDefinitionListDataDefinition = new org.openecomp.sdc.be.datatypes.elements.ListDataDefinition<>();
178         forwardingPathElementDataDefinitionListDataDefinition.add(new ForwardingPathElementDataDefinition(CI_NAME_1, CI_NAME_2, CAPABILITY_NAME_1, CAPABILITY_NAME_2, "2222", "5555"));
179         forwardingPathElementDataDefinitionListDataDefinition.add(new ForwardingPathElementDataDefinition(CI_NAME_2, CI_NAME_3, CAPABILITY_NAME_2, CAPABILITY_NAME_3, "4", "44"));
180         forwardingPathDataDefinition.setPathElements(forwardingPathElementDataDefinitionListDataDefinition);
181         return forwardingPathDataDefinition;
182     }
183
184     Service createService() {
185         Either<Service, ResponseFormat> serviceCreateResult = bl.createService(createTestService(), user);
186         assertTrue("Failed to create service", serviceCreateResult.isLeft());
187         return serviceCreateResult.left().value();
188     }
189 }
190