bpmn/so-bpmn-infrastructure-common/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  * Modifications Copyright (c) 2019 Samsung
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *      http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.so.bpmn.vcpe.scripts
  24
  25 import org.mockito.invocation.InvocationOnMock
  26 import org.mockito.stubbing.Answer
  27
  28 class MapSetter implements Answer<Void> {
  29     final Map<String, Object> map;
  30
  31     public MapSetter() {
  32         map = new HashMap<>();
  33     }
  34
  35     public MapSetter(Map<String, Object> map) {
  36         this.map = map;
  37     }
  38
  39     public Map<String, Object> getMap() {
  40         return map;
  41     }
  42
  43     @Override
  44     public Void answer(InvocationOnMock invocation) throws Throwable {
  45         map.put((String) invocation.getArgument(0), invocation.getArgument(1))
  46         return null
  47     }
  48
  49 }