2 * ========================LICENSE_START=================================
5 * Copyright (C) 2019 Nordix Foundation
6 * Modifications Copyright (C) 2020 Nordix Foundation
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
12 * http://www.apache.org/licenses/LICENSE-2.0
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===================================
21 package org.oransc.portal.nonrtric.controlpanel.config;
23 import com.google.gson.GsonBuilder;
24 import java.io.BufferedReader;
25 import java.io.InputStream;
26 import java.io.InputStreamReader;
27 import java.lang.invoke.MethodHandles;
28 import java.util.ArrayList;
29 import java.util.Collection;
30 import java.util.HashMap;
31 import java.util.List;
33 import java.util.Optional;
34 import java.util.Vector;
35 import java.util.stream.Collectors;
36 import org.oransc.portal.nonrtric.controlpanel.model.ImmutablePolicyInfo;
37 import org.oransc.portal.nonrtric.controlpanel.model.PolicyInfo;
38 import org.oransc.portal.nonrtric.controlpanel.model.PolicyInstances;
39 import org.oransc.portal.nonrtric.controlpanel.model.PolicyType;
40 import org.oransc.portal.nonrtric.controlpanel.model.PolicyTypes;
41 import org.oransc.portal.nonrtric.controlpanel.policyagentapi.PolicyAgentApi;
42 import org.slf4j.Logger;
43 import org.slf4j.LoggerFactory;
44 import org.springframework.boot.test.context.TestConfiguration;
45 import org.springframework.context.annotation.Bean;
46 import org.springframework.http.HttpStatus;
47 import org.springframework.http.ResponseEntity;
48 import org.springframework.web.client.RestClientException;
51 * Creates a mock implementation of the policy controller client API.
54 public class PolicyControllerMockConfiguration {
56 private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
58 private static com.google.gson.Gson gson = new GsonBuilder() //
63 public PolicyAgentApi policyAgentApi() {
64 MockPolicyAgentApi apiClient = new MockPolicyAgentApi();
68 class MockPolicyAgentApi implements PolicyAgentApi {
69 private final Database database = new Database();
72 public ResponseEntity<Object> getPolicyInstance(String id) {
73 return new ResponseEntity<>(database.getInstance(id), HttpStatus.OK);
77 public ResponseEntity<String> putPolicy(String policyTypeIdString, String policyInstanceId, Object json,
79 database.putInstance(policyTypeIdString, policyInstanceId, json, ric);
80 return new ResponseEntity<>(HttpStatus.OK);
84 public ResponseEntity<String> deletePolicy(String policyInstanceId) {
85 database.deleteInstance(policyInstanceId);
86 return new ResponseEntity<>(HttpStatus.OK);
90 public ResponseEntity<String> getAllPolicyTypes() {
91 PolicyTypes result = new PolicyTypes();
92 result.addAll(database.getTypes());
93 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
97 public ResponseEntity<String> getPolicyInstancesForType(String type) {
98 PolicyInstances result = new PolicyInstances();
99 List<PolicyInfo> inst = database.getInstances(Optional.of(type));
101 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
105 public ResponseEntity<String> getRicsSupportingType(String typeName) {
106 Vector<String> res = new Vector<>();
110 return new ResponseEntity<>(gson.toJson(res), HttpStatus.OK);
117 PolicyType policyType = new PolicyType("", "{}");
118 types.put("", policyType);
120 String schema = getStringFromFile("demo-policy-schema-1.json");
121 policyType = new PolicyType("type2", schema);
122 types.put("type2", policyType);
124 schema = getStringFromFile("demo-policy-schema-2.json");
125 policyType = new PolicyType("type3", schema);
126 types.put("type3", policyType);
128 schema = getStringFromFile("demo-policy-schema-3.json");
129 policyType = new PolicyType("type4", schema);
130 types.put("type4", policyType);
132 putInstance("", "123", "{\"data\":\"data\"}", "ric_1");
135 private String getStringFromFile(String path) {
137 InputStream inputStream =
138 MethodHandles.lookup().lookupClass().getClassLoader().getResourceAsStream(path);
139 return new BufferedReader(new InputStreamReader(inputStream)).lines().collect(Collectors.joining("\n"));
140 } catch (Exception e) {
141 logger.error("Cannot read file :" + path, e);
146 String normalize(String str) {
147 return str.replace('\n', ' ');
150 private String getTimeStampUTC() {
151 return java.time.Instant.now().toString();
154 void putInstance(String typeId, String instanceId, Object instanceData, String ric) {
155 PolicyInfo i = ImmutablePolicyInfo.builder().json(instanceData).lastModified(getTimeStampUTC())
156 .id(instanceId).ric(ric).service("service").type(typeId).build();
157 instances.put(instanceId, i);
160 public void deleteInstance(String instanceId) {
161 instances.remove(instanceId);
164 Object getInstance(String id) throws RestClientException {
165 PolicyInfo i = instances.get(id);
167 throw new RestClientException("Type not found: " + id);
172 public Collection<PolicyType> getTypes() {
173 return types.values();
176 public List<PolicyInfo> getInstances(Optional<String> typeId) {
177 ArrayList<PolicyInfo> result = new ArrayList<>();
178 for (PolicyInfo i : instances.values()) {
179 if (typeId.isPresent()) {
180 if (i.type().equals(typeId.get())) {
191 private Map<String, PolicyType> types = new HashMap<>();
192 private Map<String, PolicyInfo> instances = new HashMap<>();