dd993bca51dde2c9af00977c8d9700d1ac35a0ca
[so.git] /
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.infrastructure.pnf.delegate;
22
23 import java.util.HashMap;
24 import java.util.Objects;
25 import java.util.Optional;
26 import org.onap.so.bpmn.infrastructure.pnf.dmaap.DmaapClient;
27 import org.springframework.context.annotation.Primary;
28 import org.springframework.stereotype.Component;
29
30 @Component
31 @Primary
32 public class DmaapClientTestImpl implements DmaapClient {
33
34     private String pnfCorrelationId;
35     private Runnable informConsumer;
36
37     @Override
38     public void registerForUpdate(String pnfCorrelationId, Runnable informConsumer,
39             Optional<HashMap<String, String>> updateInfo) {
40         this.pnfCorrelationId = pnfCorrelationId;
41         this.informConsumer = informConsumer;
42     }
43
44     @Override
45     public Runnable unregister(String pnfCorrelationId) {
46         if (Objects.equals(this.pnfCorrelationId, pnfCorrelationId)) {
47             this.pnfCorrelationId = null;
48             Runnable informConsumer = this.informConsumer;
49             this.informConsumer = null;
50             return informConsumer;
51         }
52         return null;
53     }
54
55     public String getPnfCorrelationId() {
56         return pnfCorrelationId;
57     }
58
59     public Runnable getInformConsumer() {
60         return informConsumer;
61     }
62
63     public void sendMessage() {
64         informConsumer.run();
65     }
66
67     public boolean haveRegisteredConsumer() {
68         return pnfCorrelationId != null;
69     }
70 }