Raise JUnit coverage common-be
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / be / components / impl / CapabilityTypeImportManager.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.be.components.impl;
22
23 import java.util.ArrayList;
24 import java.util.Iterator;
25 import java.util.List;
26 import java.util.Map;
27 import java.util.function.Consumer;
28
29 import javax.annotation.Resource;
30
31 import org.openecomp.sdc.be.components.impl.ImportUtils.ToscaTagNamesEnum;
32 import org.openecomp.sdc.be.config.BeEcompErrorManager;
33 import org.openecomp.sdc.be.dao.api.ActionStatus;
34 import org.openecomp.sdc.be.impl.ComponentsUtils;
35 import org.openecomp.sdc.be.model.CapabilityTypeDefinition;
36 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
37 import org.openecomp.sdc.be.model.operations.impl.CapabilityTypeOperation;
38 import org.openecomp.sdc.exception.ResponseFormat;
39 import org.slf4j.Logger;
40 import org.slf4j.LoggerFactory;
41 import org.springframework.stereotype.Component;
42
43 import fj.data.Either;
44
45 @Component("capabilityTypeImportManager")
46 public class CapabilityTypeImportManager {
47
48     private static final Logger log = LoggerFactory.getLogger(CapabilityTypeImportManager.class);
49     @Resource
50     private CapabilityTypeOperation capabilityTypeOperation;
51     @Resource
52     private ComponentsUtils componentsUtils;
53     @Resource
54     private CommonImportManager commonImportManager;
55
56     public Either<List<CapabilityTypeDefinition>, ResponseFormat> createCapabilityTypes(String capabilityYml) {
57         Either<List<CapabilityTypeDefinition>, ActionStatus> capabilityTypes = createCapabilityTypesFromYml(capabilityYml);
58         if (capabilityTypes.isRight()) {
59             ActionStatus status = capabilityTypes.right().value();
60             ResponseFormat responseFormat = componentsUtils.getResponseFormatByCapabilityType(status, null);
61             return Either.right(responseFormat);
62         }
63         return createCapabilityTypesByDao(capabilityTypes.left().value());
64
65     }
66
67     private Either<List<CapabilityTypeDefinition>, ActionStatus> createCapabilityTypesFromYml(String capabilityYml) {
68         return commonImportManager.createElementTypesFromYml(capabilityYml, (capTypeName, capTypeJsonData) -> createCapabilityType(capTypeName, capTypeJsonData));
69
70     }
71
72     private Either<List<CapabilityTypeDefinition>, ResponseFormat> createCapabilityTypesByDao(List<CapabilityTypeDefinition> capabilityTypesToCreate) {
73         List<CapabilityTypeDefinition> createdCapabilities = new ArrayList<>();
74         Either<List<CapabilityTypeDefinition>, ResponseFormat> eitherResult = Either.left(createdCapabilities);
75         Iterator<CapabilityTypeDefinition> capTypeItr = capabilityTypesToCreate.iterator();
76         boolean stopDao = false;
77         while (capTypeItr.hasNext() && !stopDao) {
78             CapabilityTypeDefinition capabilityType = capTypeItr.next();
79
80             log.info("send capabilityType {} to dao for create", capabilityType.getType());
81             Either<CapabilityTypeDefinition, StorageOperationStatus> dataModelResponse = capabilityTypeOperation.addCapabilityType(capabilityType);
82             if (dataModelResponse.isRight()) {
83                 BeEcompErrorManager.getInstance().logBeFailedAddingNodeTypeError("Create CapabilityTypes", "capability type");
84                 log.debug("failed to create capabilityType: {}", capabilityType.getType());
85                 if (dataModelResponse.right().value() != StorageOperationStatus.SCHEMA_VIOLATION) {
86                     ResponseFormat responseFormat = componentsUtils.getResponseFormatByCapabilityType(componentsUtils.convertFromStorageResponseForCapabilityType(dataModelResponse.right().value()), capabilityType);
87                     eitherResult = Either.right(responseFormat);
88                     stopDao = true;
89                 }
90
91             } else {
92                 createdCapabilities.add(capabilityType);
93             }
94             if (!capTypeItr.hasNext()) {
95                 log.info("capabilityTypes were created successfully!!!");
96             }
97
98         }
99
100         return eitherResult;
101
102     }
103
104     private CapabilityTypeDefinition createCapabilityType(String capabilityTypeName, Map<String, Object> toscaJson) {
105         CapabilityTypeDefinition capabilityType = new CapabilityTypeDefinition();
106
107         capabilityType.setType(capabilityTypeName);
108
109         // Description
110         final Consumer<String> descriptionSetter = description -> capabilityType.setDescription(description);
111         commonImportManager.setField(toscaJson, ToscaTagNamesEnum.DESCRIPTION.getElementName(), descriptionSetter);
112         // Derived From
113         final Consumer<String> derivedFromSetter = derivedFrom -> capabilityType.setDerivedFrom(derivedFrom);
114         commonImportManager.setField(toscaJson, ToscaTagNamesEnum.DERIVED_FROM.getElementName(), derivedFromSetter);
115         // Properties
116         commonImportManager.setPropertiesMap(toscaJson, (values) -> capabilityType.setProperties(values));
117
118         return capabilityType;
119     }
120
121 }