blueprints-processor/plugin/assignment-provider/src/test/java/org/onap/ccsdk/features/assignment/processor/DBResourceProcessorTest.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.processor;\r
  19 \r
  20 import static org.mockito.Matchers.any;\r
  21 import java.io.File;\r
  22 import java.nio.charset.Charset;\r
  23 import java.util.ArrayList;\r
  24 import java.util.Arrays;\r
  25 import java.util.HashMap;\r
  26 import java.util.List;\r
  27 import java.util.Map;\r
  28 import org.apache.commons.io.FileUtils;\r
  29 import org.junit.Assert;\r
  30 import org.junit.Before;\r
  31 import org.junit.Test;\r
  32 import org.junit.runner.RunWith;\r
  33 import org.mockito.Matchers;\r
  34 import org.mockito.Mock;\r
  35 import org.mockito.Mockito;\r
  36 import org.mockito.MockitoAnnotations;\r
  37 import org.mockito.invocation.InvocationOnMock;\r
  38 import org.mockito.runners.MockitoJUnitRunner;\r
  39 import org.mockito.stubbing.Answer;\r
  40 import org.onap.ccsdk.features.assignment.processor.DBResourceProcessor;\r
  41 import org.onap.ccsdk.features.assignment.service.ConfigResourceAssignmentTestUtils;\r
  42 import org.onap.ccsdk.features.data.adaptor.domain.TransactionLog;\r
  43 import org.onap.ccsdk.features.data.adaptor.service.ConfigResourceService;\r
  44 import org.onap.ccsdk.features.model.ConfigModelConstant;\r
  45 import org.onap.ccsdk.features.model.data.ResourceAssignment;\r
  46 import org.onap.ccsdk.features.model.data.dict.ResourceDefinition;\r
  47 import org.onap.ccsdk.features.model.utils.TransformationUtils;\r
  48 import org.onap.ccsdk.sli.core.sli.SvcLogicContext;\r
  49 import org.onap.ccsdk.sli.core.sli.SvcLogicException;\r
  50 import com.att.eelf.configuration.EELFLogger;\r
  51 import com.att.eelf.configuration.EELFManager;\r
  52 \r
  53 @RunWith(MockitoJUnitRunner.class)\r
  54 public class DBResourceProcessorTest {\r
  55 \r
  56     private static EELFLogger logger = EELFManager.getInstance().getLogger(DBResourceProcessorTest.class);\r
  57 \r
  58     @Mock\r
  59     private ConfigResourceService configResourceService;\r
  60 \r
  61     @SuppressWarnings("unchecked")\r
  62     @Before\r
  63     public void before() {\r
  64         MockitoAnnotations.initMocks(this);\r
  65 \r
  66         try {\r
  67             Mockito.doAnswer(new Answer<Void>() {\r
  68                 @Override\r
  69                 public Void answer(InvocationOnMock invocationOnMock) throws Throwable {\r
  70                     Object[] args = invocationOnMock.getArguments();\r
  71                     if (args != null) {\r
  72                         logger.trace("Transaction info " + Arrays.asList(args));\r
  73                     }\r
  74                     return null;\r
  75                 }\r
  76             }).when(configResourceService).save(any(TransactionLog.class));\r
  77 \r
  78         } catch (SvcLogicException e) {\r
  79             e.printStackTrace();\r
  80         }\r
  81     }\r
  82 \r
  83     @Test\r
  84     public void testDbSimpleProcess() throws Exception {\r
  85         logger.info(" *******************************  testDbSimpleProcess  ***************************");\r
  86 \r
  87         Mockito.doAnswer(new Answer<List<Map<String, Object>>>() {\r
  88             @Override\r
  89             public List<Map<String, Object>> answer(InvocationOnMock invocationOnMock) throws Throwable {\r
  90                 Object[] args = invocationOnMock.getArguments();\r
  91                 List<Map<String, Object>> results = new ArrayList<>();\r
  92                 if (args != null) {\r
  93                     logger.info("Query " + Arrays.asList(args));\r
  94                     Map<String, Object> record = new HashMap<>();\r
  95                     record.put("country", "US");\r
  96                     results.add(record);\r
  97                 }\r
  98                 return results;\r
  99             }\r
 100         }).when(configResourceService).query(Matchers.anyString(), Matchers.<Map<String, Object>>any());\r
 101 \r
 102         String recipeName = "sample-recipe";\r
 103 \r
 104         String resourceassignmentContent = FileUtils.readFileToString(\r
 105                 new File("src/test/resources/mapping/db/resource-assignments-simple.json"), Charset.defaultCharset());\r
 106         List<ResourceAssignment> batchResourceAssignment =\r
 107                 TransformationUtils.getListfromJson(resourceassignmentContent, ResourceAssignment.class);\r
 108 \r
 109         String dictionaryContent = FileUtils.readFileToString(new File("src/test/resources/mapping/db/db-simple.json"),\r
 110                 Charset.defaultCharset());\r
 111         Map<String, ResourceDefinition> dictionaries =\r
 112                 ConfigResourceAssignmentTestUtils.getMapfromJson(dictionaryContent);\r
 113         DBResourceProcessor dbResourceProcessor = new DBResourceProcessor(configResourceService);\r
 114         Map<String, Object> componentContext = new HashMap<>();\r
 115         componentContext.put(ConfigModelConstant.PROPERTY_RESOURCE_ASSIGNMENTS, batchResourceAssignment);\r
 116         componentContext.put(ConfigModelConstant.PROPERTY_ACTION_NAME, recipeName);\r
 117         componentContext.put(ConfigModelConstant.PROPERTY_TEMPLATE_NAME, "sample-template");\r
 118         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARIES, dictionaries);\r
 119 \r
 120         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARY_KEY_DOT + recipeName + ".profile_name", "sample");\r
 121 \r
 122         Map<String, String> inParams = new HashMap<>();\r
 123         SvcLogicContext ctx = new SvcLogicContext();\r
 124         dbResourceProcessor.process(inParams, ctx, componentContext);\r
 125 \r
 126         logger.info(" Context " + componentContext);\r
 127 \r
 128     }\r
 129 \r
 130     @Test\r
 131     public void testDbComplexProcess() throws Exception {\r
 132         logger.info(" *******************************  testDbComplexProcess  ***************************");\r
 133 \r
 134         Mockito.doAnswer(new Answer<List<Map<String, Object>>>() {\r
 135             @Override\r
 136             public List<Map<String, Object>> answer(InvocationOnMock invocationOnMock) throws Throwable {\r
 137                 Object[] args = invocationOnMock.getArguments();\r
 138                 List<Map<String, Object>> results = new ArrayList<>();\r
 139                 if (args != null) {\r
 140                     logger.info("Query " + Arrays.asList(args));\r
 141                     Map<String, Object> record = new HashMap<>();\r
 142                     record.put("db-country", "US");\r
 143                     record.put("db-state", "NJ");\r
 144                     results.add(record);\r
 145                 }\r
 146                 return results;\r
 147             }\r
 148         }).when(configResourceService).query(Matchers.anyString(), Matchers.<Map<String, Object>>any());\r
 149 \r
 150         String recipeName = "sample-recipe";\r
 151 \r
 152         String resourceassignmentContent = FileUtils.readFileToString(\r
 153                 new File("src/test/resources/mapping/db/resource-assignments-complex.json"), Charset.defaultCharset());\r
 154         List<ResourceAssignment> batchResourceAssignment =\r
 155                 TransformationUtils.getListfromJson(resourceassignmentContent, ResourceAssignment.class);\r
 156 \r
 157         String dictionaryContent = FileUtils.readFileToString(new File("src/test/resources/mapping/db/db-complex.json"),\r
 158                 Charset.defaultCharset());\r
 159         Map<String, ResourceDefinition> dictionaries =\r
 160                 ConfigResourceAssignmentTestUtils.getMapfromJson(dictionaryContent);\r
 161         DBResourceProcessor dbResourceProcessor = new DBResourceProcessor(configResourceService);\r
 162         Map<String, Object> componentContext = new HashMap<>();\r
 163         componentContext.put(ConfigModelConstant.PROPERTY_RESOURCE_ASSIGNMENTS, batchResourceAssignment);\r
 164         componentContext.put(ConfigModelConstant.PROPERTY_ACTION_NAME, recipeName);\r
 165         componentContext.put(ConfigModelConstant.PROPERTY_TEMPLATE_NAME, "sample-template");\r
 166         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARIES, dictionaries);\r
 167         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARY_KEY_DOT + recipeName + ".profile_name", "sample");\r
 168 \r
 169         Map<String, String> inParams = new HashMap<>();\r
 170         SvcLogicContext ctx = new SvcLogicContext();\r
 171         String datatypeContent = FileUtils.readFileToString(new File("src/test/resources/mapping/db/dt-location.json"),\r
 172                 Charset.defaultCharset());\r
 173         ctx.setAttribute("data_types.dt-location", datatypeContent);\r
 174         dbResourceProcessor.process(inParams, ctx, componentContext);\r
 175 \r
 176     }\r
 177 \r
 178     @Test\r
 179     public void testDbArrayComplexProcess() throws Exception {\r
 180         logger.info(" *******************************  testDbArrayComplexProcess  ***************************");\r
 181         Mockito.doAnswer(new Answer<List<Map<String, Object>>>() {\r
 182             @Override\r
 183             public List<Map<String, Object>> answer(InvocationOnMock invocationOnMock) throws Throwable {\r
 184                 Object[] args = invocationOnMock.getArguments();\r
 185                 List<Map<String, Object>> results = new ArrayList<>();\r
 186                 if (args != null) {\r
 187                     logger.info("Query " + Arrays.asList(args));\r
 188                     Map<String, Object> record = new HashMap<>();\r
 189                     record.put("db-country", "US");\r
 190                     record.put("db-state", "NJ");\r
 191                     results.add(record);\r
 192 \r
 193                     Map<String, Object> record2 = new HashMap<>();\r
 194                     record2.put("db-country", "INDIA");\r
 195                     record2.put("db-state", "TN");\r
 196                     results.add(record2);\r
 197                 }\r
 198                 return results;\r
 199             }\r
 200         }).when(configResourceService).query(Matchers.anyString(), Matchers.<Map<String, Object>>any());\r
 201 \r
 202         String recipeName = "sample-recipe";\r
 203 \r
 204         String resourceassignmentContent = FileUtils.readFileToString(\r
 205                 new File("src/test/resources/mapping/db/resource-assignments-array.json"), Charset.defaultCharset());\r
 206         List<ResourceAssignment> batchResourceAssignment =\r
 207                 TransformationUtils.getListfromJson(resourceassignmentContent, ResourceAssignment.class);\r
 208 \r
 209         String dictionaryContent = FileUtils.readFileToString(new File("src/test/resources/mapping/db/db-array.json"),\r
 210                 Charset.defaultCharset());\r
 211         Map<String, ResourceDefinition> dictionaries =\r
 212                 ConfigResourceAssignmentTestUtils.getMapfromJson(dictionaryContent);\r
 213         DBResourceProcessor dbResourceProcessor = new DBResourceProcessor(configResourceService);\r
 214         Map<String, Object> componentContext = new HashMap<>();\r
 215         componentContext.put(ConfigModelConstant.PROPERTY_RESOURCE_ASSIGNMENTS, batchResourceAssignment);\r
 216         componentContext.put(ConfigModelConstant.PROPERTY_ACTION_NAME, recipeName);\r
 217         componentContext.put(ConfigModelConstant.PROPERTY_TEMPLATE_NAME, "sample-template");\r
 218         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARIES, dictionaries);\r
 219         componentContext.put(ConfigModelConstant.PROPERTY_DICTIONARY_KEY_DOT + recipeName + ".profile_name", "sample");\r
 220 \r
 221         Map<String, String> inParams = new HashMap<>();\r
 222         SvcLogicContext ctx = new SvcLogicContext();\r
 223         String datatypeContent = FileUtils.readFileToString(new File("src/test/resources/mapping/db/dt-location.json"),\r
 224                 Charset.defaultCharset());\r
 225         ctx.setAttribute("data_types.dt-location", datatypeContent);\r
 226         dbResourceProcessor.process(inParams, ctx, componentContext);\r
 227         logger.info("Component Context = ({})", componentContext);\r
 228         Assert.assertNotNull("faile dto populate Array Complex response ", componentContext);\r
 229 \r
 230     }\r
 231 \r
 232 }\r