bpmn/MSOInfrastructureBPMN/src/test/java/org/openecomp/mso/bpmn/infrastructure/pnf/delegate/AaiConnectionTestImpl.java

   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.openecomp.mso.bpmn.infrastructure.pnf.delegate;
  22
  23 import java.io.IOException;
  24 import java.util.HashMap;
  25 import java.util.LinkedList;
  26 import java.util.List;
  27 import java.util.Map;
  28 import java.util.Objects;
  29 import java.util.Optional;
  30 import org.onap.aai.domain.yang.Pnf;
  31 import org.openecomp.mso.bpmn.infrastructure.pnf.implementation.AaiConnection;
  32
  33 public class AaiConnectionTestImpl implements AaiConnection {
  34
  35     public static final String ID_WITH_ENTRY_AND_IP = "idWithEntryAndIp";
  36     public static final String ID_WITH_IP_V6 = "idWithIpV6";
  37     public static final String ID_WITHOUT_ENTRY = "IdWithoutEntry";
  38     public static final String ID_WITH_ENTRY_NO_IP = "idWithEntryNoIp";
  39     public static final String DEFAULT_IP = "1.2.3.4";
  40     public static final String DEFAULT_IP_V6 = "2001:db8::ff00:42:8329";
  41
  42     private Map<String, Pnf> created = new HashMap<>();
  43
  44     @Override
  45     public Optional<Pnf> getEntryFor(String correlationId) throws IOException {
  46         if (Objects.equals(correlationId, ID_WITH_ENTRY_AND_IP)) {
  47             Pnf pnf = new Pnf();
  48             pnf.setIpaddressV4Oam(DEFAULT_IP);
  49             return Optional.of(pnf);
  50         } else if (Objects.equals(correlationId, ID_WITH_IP_V6)) {
  51             Pnf pnf = new Pnf();
  52             pnf.setIpaddressV6Oam(DEFAULT_IP_V6);
  53             return Optional.of(pnf);
  54         } else if (Objects.equals(correlationId, ID_WITH_ENTRY_NO_IP)) {
  55             return Optional.of(new Pnf());
  56         } else {
  57             return Optional.empty();
  58         }
  59     }
  60
  61     @Override
  62     public void createEntry(String correlationId, Pnf entry) throws IOException {
  63         created.put(correlationId, entry);
  64     }
  65
  66     public Map<String, Pnf> getCreated() {
  67         return created;
  68     }
  69
  70     public void reset() {
  71         created.clear();
  72     }
  73 }