CorrelationId cleanup
[so.git] / bpmn / mso-infrastructure-bpmn / src / test / java / org / onap / so / bpmn / infrastructure / pnf / delegate / PnfManagementTestImpl.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.onap.so.bpmn.infrastructure.pnf.delegate;
22
23 import java.util.HashMap;
24 import java.util.Map;
25 import java.util.Objects;
26 import java.util.Optional;
27 import org.onap.aai.domain.yang.Pnf;
28 import org.onap.so.bpmn.infrastructure.pnf.management.PnfManagement;
29 import org.springframework.context.annotation.Primary;
30 import org.springframework.stereotype.Component;
31
32 @Component
33 @Primary
34 public class PnfManagementTestImpl implements PnfManagement {
35
36     public static final String ID_WITHOUT_ENTRY = "IdWithoutEntry";
37     public static final String ID_WITH_ENTRY = "idWithEntryNoIp";
38
39     private Map<String, Pnf> created = new HashMap<>();
40     private Map<String, String> serviceAndPnfRelationMap = new HashMap<>();
41
42     @Override
43     public Optional<Pnf> getEntryFor(String pnfCorrelationId) {
44         if (Objects.equals(pnfCorrelationId, ID_WITH_ENTRY)) {
45             return Optional.of(new Pnf());
46         } else {
47             return Optional.empty();
48         }
49     }
50
51     @Override
52     public void createEntry(String pnfCorrelationId, Pnf entry) {
53         created.put(pnfCorrelationId, entry);
54     }
55
56     @Override
57     public void createRelation(String serviceInstanceId, String pnfName) {
58         serviceAndPnfRelationMap.put(serviceInstanceId, pnfName);
59     }
60
61     public Map<String, Pnf> getCreated() {
62         return created;
63     }
64
65     public Map<String, String> getServiceAndPnfRelationMap() {
66         return serviceAndPnfRelationMap;
67     }
68
69     public void reset() {
70         created.clear();
71         serviceAndPnfRelationMap.clear();
72     }
73 }