2 * Copyright © 2017-2018 AT&T Intellectual Property.
\r
3 * Modifications Copyright © 2018 IBM.
\r
5 * Licensed under the Apache License, Version 2.0 (the "License");
\r
6 * you may not use this file except in compliance with the License.
\r
7 * You may obtain a copy of the License at
\r
9 * http://www.apache.org/licenses/LICENSE-2.0
\r
11 * Unless required by applicable law or agreed to in writing, software
\r
12 * distributed under the License is distributed on an "AS IS" BASIS,
\r
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
\r
14 * See the License for the specific language governing permissions and
\r
15 * limitations under the License.
\r
18 package org.onap.ccsdk.features.assignment.service;
\r
20 import static org.mockito.Matchers.any;
\r
21 import static org.mockito.Matchers.anyString;
\r
22 import java.io.File;
\r
23 import java.nio.charset.Charset;
\r
24 import java.util.ArrayList;
\r
25 import java.util.Arrays;
\r
26 import java.util.Collection;
\r
27 import java.util.List;
\r
28 import java.util.Map;
\r
29 import org.apache.commons.io.FileUtils;
\r
30 import org.apache.commons.io.FilenameUtils;
\r
31 import org.apache.commons.io.IOUtils;
\r
32 import org.mockito.Matchers;
\r
33 import org.mockito.Mockito;
\r
34 import org.mockito.invocation.InvocationOnMock;
\r
35 import org.mockito.stubbing.Answer;
\r
36 import org.onap.ccsdk.features.data.adaptor.domain.ConfigResource;
\r
37 import org.onap.ccsdk.features.data.adaptor.domain.TransactionLog;
\r
38 import org.onap.ccsdk.features.data.adaptor.service.ConfigResourceService;
\r
39 import org.onap.ccsdk.features.model.ConfigModelConstant;
\r
40 import org.onap.ccsdk.features.model.data.dict.ResourceDefinition;
\r
41 import org.onap.ccsdk.features.model.domain.ConfigModel;
\r
42 import org.onap.ccsdk.features.model.domain.ConfigModelContent;
\r
43 import org.onap.ccsdk.features.rest.adaptor.service.ConfigRestAdaptorService;
\r
45 import com.att.eelf.configuration.EELFLogger;
\r
46 import com.att.eelf.configuration.EELFManager;
\r
47 import com.fasterxml.jackson.core.type.TypeReference;
\r
48 import com.fasterxml.jackson.databind.ObjectMapper;
\r
50 public class ConfigResourceAssignmentTestUtils {
\r
52 private static EELFLogger logger = EELFManager.getInstance().getLogger(ConfigResourceAssignmentTestUtils.class);
\r
54 public static void injectTransactionLogSaveMock(ConfigResourceService configResourceService) throws Exception {
\r
55 Mockito.doAnswer(new Answer<Void>() {
\r
57 public Void answer(InvocationOnMock invocationOnMock) throws Throwable {
\r
58 Object[] args = invocationOnMock.getArguments();
\r
60 logger.trace("Transaction info " + Arrays.asList(args));
\r
64 }).when(configResourceService).save(any(TransactionLog.class));
\r
67 public static void injectConfigModelMock(ConfigRestAdaptorService configRestAdaptorService,
\r
68 String serviceTemplateName) throws Exception {
\r
70 Mockito.doAnswer(new Answer<ConfigModel>() {
\r
72 public org.onap.ccsdk.features.model.domain.ConfigModel answer(InvocationOnMock invocationOnMock)
\r
74 Object[] args = invocationOnMock.getArguments();
\r
75 org.onap.ccsdk.features.model.domain.ConfigModel serviceArtifact = null;
\r
76 if (args != null && args.length == 3) {
\r
78 logger.info("Artifact info " + Arrays.asList(args));
\r
79 String modelContent = IOUtils.toString(
\r
80 ConfigResourceAssignmentTestUtils.class.getClassLoader().getResourceAsStream(
\r
81 "service_templates/" + serviceTemplateName + "/" + serviceTemplateName + ".json"),
\r
82 Charset.defaultCharset());
\r
84 ConfigModelContent configModelContent = new ConfigModelContent();
\r
85 configModelContent.setContent(modelContent);
\r
86 configModelContent.setContentType(ConfigModelConstant.MODEL_CONTENT_TYPE_TOSCA_JSON);
\r
88 List<ConfigModelContent> configModelContents = new ArrayList<>();
\r
89 configModelContents.add(configModelContent);
\r
91 String velocityDir = ConfigResourceAssignmentTestUtils.class.getClassLoader()
\r
92 .getResource("service_templates/" + serviceTemplateName + "/velocity").getPath();
\r
94 Collection<File> templateFiles =
\r
95 FileUtils.listFiles(new File(velocityDir), new String[] {"vtl"}, true);
\r
96 logger.info("Template Files info " + templateFiles);
\r
97 for (File templateFile : templateFiles) {
\r
98 String templateContent = FileUtils.readFileToString(templateFile, Charset.defaultCharset());
\r
99 ConfigModelContent configModelTemplateContent = new ConfigModelContent();
\r
100 configModelTemplateContent.setContent(templateContent);
\r
101 configModelTemplateContent.setName(FilenameUtils.getBaseName(templateFile.getName()));
\r
102 configModelTemplateContent.setContentType(ConfigModelConstant.MODEL_CONTENT_TYPE_TEMPLATE);
\r
103 configModelContents.add(configModelTemplateContent);
\r
106 serviceArtifact = new org.onap.ccsdk.features.model.domain.ConfigModel();
\r
107 serviceArtifact.setArtifactName(String.valueOf(args[0]));
\r
108 serviceArtifact.setArtifactVersion(String.valueOf(args[1]));
\r
109 serviceArtifact.setPublished("Y");
\r
110 serviceArtifact.setConfigModelContents(configModelContents);
\r
113 return serviceArtifact;
\r
115 }).when(configRestAdaptorService).getResource(anyString(), anyString(), Matchers.any(Class.class));
\r
118 public static void injectResourceDictionaryMock(ConfigRestAdaptorService configRestAdaptorService,
\r
119 String dictionaryFileName) throws Exception {
\r
121 Mockito.doAnswer(new Answer<String>() {
\r
123 public String answer(InvocationOnMock invocationOnMock) throws Throwable {
\r
124 Object[] args = invocationOnMock.getArguments();
\r
125 String dictionaties = "[]";
\r
126 if (args != null) {
\r
127 logger.info("getResourceDictionaryByNames " + Arrays.asList(args));
\r
128 dictionaties = IOUtils.toString(
\r
129 ConfigAssignmentNodeTest.class.getClassLoader().getResourceAsStream(dictionaryFileName),
\r
130 Charset.defaultCharset());
\r
132 return dictionaties;
\r
134 }).when(configRestAdaptorService).postResource(Matchers.any(), Matchers.any(), Matchers.any(),
\r
135 Matchers.any(Class.class));
\r
138 public static void injectConfigResourceSaveMock(ConfigResourceService configResourceService) throws Exception {
\r
139 Mockito.doAnswer(new Answer<ConfigResource>() {
\r
141 public ConfigResource answer(InvocationOnMock invocationOnMock) throws Throwable {
\r
142 Object[] args = invocationOnMock.getArguments();
\r
143 ConfigResource configResource = null;
\r
144 if (args != null) {
\r
145 configResource = (ConfigResource) args[0];
\r
146 logger.info("Config Resource Save info " + configResource);
\r
147 return configResource;
\r
149 return configResource;
\r
151 }).when(configResourceService).saveConfigResource(any(ConfigResource.class));
\r
154 public static void injectGetConfigResourceMock(ConfigResourceService configResourceService,
\r
155 ConfigResource configResource) throws Exception {
\r
156 Mockito.doAnswer(new Answer<List<ConfigResource>>() {
\r
158 public List<ConfigResource> answer(InvocationOnMock invocationOnMock) throws Throwable {
\r
159 Object[] args = invocationOnMock.getArguments();
\r
160 List<ConfigResource> configResources = new ArrayList<>();
\r
161 if (args != null) {
\r
162 configResources.add(configResource);
\r
164 return configResources;
\r
166 }).when(configResourceService).getConfigResource(any(ConfigResource.class));
\r
169 public static String getFileContent(String filePath) throws Exception {
\r
170 return IOUtils.toString(ConfigResourceAssignmentTestUtils.class.getClassLoader().getResourceAsStream(filePath),
\r
171 Charset.defaultCharset());
\r
174 public static Map<String, ResourceDefinition> getMapfromJson(String content) {
\r
176 ObjectMapper mapper = new ObjectMapper();
\r
177 return mapper.readValue(content, new TypeReference<Map<String, ResourceDefinition>>() {});
\r
178 } catch (Exception e) {
\r
179 logger.info("getMapfromJson Exception ({})", e);
\r