blueprints-processor/plugin/assignment-provider/src/test/java/org/onap/ccsdk/features/assignment/service/ConfigResourceAssignmentTestUtils.java

   1 /*\r
   2  * Copyright © 2017-2018 AT&T Intellectual Property.\r
   3  * Modifications Copyright © 2018 IBM.\r
   4  * \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
   8  * \r
   9  * http://www.apache.org/licenses/LICENSE-2.0\r
  10  * \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
  16  */\r
  17 \r
  18 package org.onap.ccsdk.features.assignment.service;\r
  19 \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
  44 \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
  49 \r
  50 public class ConfigResourceAssignmentTestUtils {\r
  51 \r
  52     private static EELFLogger logger = EELFManager.getInstance().getLogger(ConfigResourceAssignmentTestUtils.class);\r
  53 \r
  54     public static void injectTransactionLogSaveMock(ConfigResourceService configResourceService) throws Exception {\r
  55         Mockito.doAnswer(new Answer<Void>() {\r
  56             @Override\r
  57             public Void answer(InvocationOnMock invocationOnMock) throws Throwable {\r
  58                 Object[] args = invocationOnMock.getArguments();\r
  59                 if (args != null) {\r
  60                     logger.trace("Transaction info " + Arrays.asList(args));\r
  61                 }\r
  62                 return null;\r
  63             }\r
  64         }).when(configResourceService).save(any(TransactionLog.class));\r
  65     }\r
  66 \r
  67     public static void injectConfigModelMock(ConfigRestAdaptorService configRestAdaptorService,\r
  68             String serviceTemplateName) throws Exception {\r
  69 \r
  70         Mockito.doAnswer(new Answer<ConfigModel>() {\r
  71             @Override\r
  72             public org.onap.ccsdk.features.model.domain.ConfigModel answer(InvocationOnMock invocationOnMock)\r
  73                     throws Throwable {\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
  77 \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
  83 \r
  84                     ConfigModelContent configModelContent = new ConfigModelContent();\r
  85                     configModelContent.setContent(modelContent);\r
  86                     configModelContent.setContentType(ConfigModelConstant.MODEL_CONTENT_TYPE_TOSCA_JSON);\r
  87 \r
  88                     List<ConfigModelContent> configModelContents = new ArrayList<>();\r
  89                     configModelContents.add(configModelContent);\r
  90 \r
  91                     String velocityDir = ConfigResourceAssignmentTestUtils.class.getClassLoader()\r
  92                             .getResource("service_templates/" + serviceTemplateName + "/velocity").getPath();\r
  93 \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
 104                     }\r
 105 \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
 111                 }\r
 112 \r
 113                 return serviceArtifact;\r
 114             }\r
 115         }).when(configRestAdaptorService).getResource(anyString(), anyString(), Matchers.any(Class.class));\r
 116     }\r
 117 \r
 118     public static void injectResourceDictionaryMock(ConfigRestAdaptorService configRestAdaptorService,\r
 119             String dictionaryFileName) throws Exception {\r
 120 \r
 121         Mockito.doAnswer(new Answer<String>() {\r
 122             @Override\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
 131                 }\r
 132                 return dictionaties;\r
 133             }\r
 134         }).when(configRestAdaptorService).postResource(Matchers.any(), Matchers.any(), Matchers.any(),\r
 135                 Matchers.any(Class.class));\r
 136     }\r
 137 \r
 138     public static void injectConfigResourceSaveMock(ConfigResourceService configResourceService) throws Exception {\r
 139         Mockito.doAnswer(new Answer<ConfigResource>() {\r
 140             @Override\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
 148                 }\r
 149                 return configResource;\r
 150             }\r
 151         }).when(configResourceService).saveConfigResource(any(ConfigResource.class));\r
 152     }\r
 153 \r
 154     public static void injectGetConfigResourceMock(ConfigResourceService configResourceService,\r
 155             ConfigResource configResource) throws Exception {\r
 156         Mockito.doAnswer(new Answer<List<ConfigResource>>() {\r
 157             @Override\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
 163                 }\r
 164                 return configResources;\r
 165             }\r
 166         }).when(configResourceService).getConfigResource(any(ConfigResource.class));\r
 167     }\r
 168 \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
 172     }\r
 173 \r
 174     public static Map<String, ResourceDefinition> getMapfromJson(String content) {\r
 175         try {\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
 180         }\r
 181         return null;\r
 182     }\r
 183 }\r