bpmn/so-bpmn-infrastructure-common/src/test/java/org/onap/so/bpmn/infrastructure/pnf/delegate/KafkaClientTestImpl.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP - SO
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * Copyright (C) 2019 Nokia.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *      http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.so.bpmn.infrastructure.pnf.delegate;
  23
  24 import java.util.Objects;
  25 import org.onap.so.bpmn.infrastructure.pnf.kafka.KafkaClient;
  26
  27 public class KafkaClientTestImpl implements KafkaClient {
  28
  29     private String pnfCorrelationId;
  30     private Runnable informConsumer;
  31
  32     @Override
  33     public void registerForUpdate(String pnfCorrelationId, Runnable informConsumer) {
  34         this.pnfCorrelationId = pnfCorrelationId;
  35         this.informConsumer = informConsumer;
  36     }
  37
  38     @Override
  39     public Runnable unregister(String pnfCorrelationId) {
  40         if (Objects.equals(this.pnfCorrelationId, pnfCorrelationId)) {
  41             this.pnfCorrelationId = null;
  42             Runnable informConsumer = this.informConsumer;
  43             this.informConsumer = null;
  44             return informConsumer;
  45         }
  46         return null;
  47     }
  48
  49     String getPnfCorrelationId() {
  50         return pnfCorrelationId;
  51     }
  52
  53     Runnable getInformConsumer() {
  54         return informConsumer;
  55     }
  56
  57     boolean haveRegisteredConsumer() {
  58         return pnfCorrelationId != null;
  59     }
  60 }