bpmn/so-bpmn-infrastructure-flows/src/test/groovy/org/onap/so/bpmn/vcpe/scripts/MapSetter.groovy

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   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.onap.so.bpmn.vcpe.scripts;
  22
  23 import java.util.HashMap;
  24 import java.util.Map;
  25
  26 import org.mockito.invocation.InvocationOnMock;
  27 import org.mockito.stubbing.Answer;
  28
  29 class MapSetter implements Answer<Void> {
  30         final Map<String,Object> map;
  31
  32         public MapSetter() {
  33                 map = new HashMap<>();
  34         }
  35
  36         public MapSetter(Map<String,Object> map) {
  37                 this.map = map;
  38         }
  39
  40         public Map<String, Object> getMap() {
  41                 return map;
  42         }
  43
  44         @Override
  45         public Void answer(InvocationOnMock invocation) throws Throwable {
  46                 map.put(invocation.getArgumentAt(0, String.class), invocation.getArgumentAt(1, Object.class));
  47                 return null;
  48         }
  49
  50 }