70d94052e1d5462d5633e56716ce539236c23a61
[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.io.IOException;
24 import java.util.HashMap;
25 import java.util.Map;
26 import java.util.Objects;
27 import java.util.Optional;
28
29 import org.onap.aai.domain.yang.Pnf;
30 import org.onap.so.bpmn.infrastructure.pnf.implementation.AaiConnection;
31 import org.springframework.context.annotation.Primary;
32 import org.springframework.stereotype.Component;
33
34 @Component
35 @Primary
36 public class AaiConnectionTestImpl implements AaiConnection {
37
38     public static final String ID_WITHOUT_ENTRY = "IdWithoutEntry";
39     public static final String ID_WITH_ENTRY = "idWithEntryNoIp";
40
41     private Map<String, Pnf> created = new HashMap<>();
42
43     @Override
44     public Optional<Pnf> getEntryFor(String correlationId) throws IOException {
45         if (Objects.equals(correlationId, ID_WITH_ENTRY)) {
46             return Optional.of(new Pnf());
47         } else {
48             return Optional.empty();
49         }
50     }
51
52     @Override
53     public void createEntry(String correlationId, Pnf entry) throws IOException {
54         created.put(correlationId, entry);
55     }
56
57     public Map<String, Pnf> getCreated() {
58         return created;
59     }
60
61     public void reset() {
62         created.clear();
63     }
64 }