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.mock;
23 import com.google.gson.GsonBuilder;
25 import java.io.BufferedReader;
26 import java.io.InputStream;
27 import java.io.InputStreamReader;
28 import java.lang.invoke.MethodHandles;
29 import java.util.ArrayList;
30 import java.util.Arrays;
31 import java.util.Collection;
32 import java.util.Collections;
33 import java.util.HashMap;
34 import java.util.List;
36 import java.util.Optional;
37 import java.util.stream.Collectors;
39 import org.oransc.portal.nonrtric.controlpanel.eiproducerapi.EiProducerApi;
40 import org.slf4j.Logger;
41 import org.slf4j.LoggerFactory;
42 import org.springframework.boot.test.context.TestConfiguration;
43 import org.springframework.context.annotation.Bean;
44 import org.springframework.http.HttpStatus;
45 import org.springframework.http.ResponseEntity;
46 import org.springframework.web.client.RestClientException;
49 * Creates a mock implementation of the policy controller client API.
52 public class EnrichmentControllerMockConfiguration {
54 private static final Logger logger = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
56 private static com.google.gson.Gson gson = new GsonBuilder() //
61 public EiProducerApi eiProducerApi() {
62 MockEiProducerApi apiClient = new MockEiProducerApi();
66 class MockEiProducerApi implements EiProducerApi {
67 private final Database database = new Database();
70 public ResponseEntity<String> getAllEiProducerIds() {
71 List<String> result = new ArrayList<>();
72 result.addAll(database.getAllEiProducerIds());
73 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
76 public ResponseEntity<String> getAllEiProducers() {
77 EiProducers result = new EiProducers();
78 result.addAll(database.getAllEiProducers());
79 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
83 public ResponseEntity<String> getEiProducer(String eiProducerId) {
84 EiProducer result = database.getEiProducerInstance(eiProducerId);
85 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
89 public ResponseEntity<String> getEiJobsForOneEiProducer(String eiProducerId) {
90 EiJobs result = new EiJobs();
91 List<EiJob> inst = database.getEiJobsForOneEiProducer(Optional.of(eiProducerId));
93 return new ResponseEntity<>(gson.toJson(result), HttpStatus.OK);
97 public ResponseEntity<String> getEiProducerStatus(String eiProducerId) {
98 EiProducer result = database.getEiProducerInstance(eiProducerId);
99 return new ResponseEntity<>(gson.toJson(result.status()), HttpStatus.OK);
107 List<EiType> supported_types = new ArrayList<EiType>();
109 // Create EiType instance
110 String schema = getStringFromFile("ei-type-1.json");
111 EiType eiType1 = getEiTypeInstance("type1", schema, Arrays.asList("prod-1"));
112 supported_types.add(eiType1);
114 // Create EiType instance
115 schema = getStringFromFile("ei-type-2.json");
116 EiType eiType2 = getEiTypeInstance("type2", schema, Arrays.asList("prod-1"));
117 supported_types.add(eiType2);
119 // Create EiProducer instance
120 putEiProducerInstance("prod-1", "http://example.com/", "http://example.com/", "http://example.com/",
121 supported_types, new ProducerStatusInfo(ProducerStatusInfo.OperationalState.ENABLED));
123 // Create EiJob instance
124 schema = getStringFromFile("job-1.json");
125 putEiJobInstance("type1", "job1", schema, "prod-1", "http://example.com/");
128 private String getStringFromFile(String path) {
130 InputStream inputStream =
131 MethodHandles.lookup().lookupClass().getClassLoader().getResourceAsStream(path);
132 return new BufferedReader(new InputStreamReader(inputStream)).lines().collect(Collectors.joining("\n"));
133 } catch (Exception e) {
134 logger.error("Cannot read file :" + path, e);
139 String normalize(String str) {
140 return str.replace('\n', ' ');
143 void putEiJobInstance(String typeId, String instanceId, Object instanceData, String owner, String targetUrl) {
144 EiJob i = ImmutableEiJob.builder() //
145 .ei_job_data(instanceData) //
146 .ei_job_identity(instanceId) //
148 .ei_type_identity(typeId) //
149 .target_uri(targetUrl) //
150 .status("ENABLED") //
152 eiJobs.put(instanceId, i);
155 void putEiProducerInstance(String id, String creation_url, String deletion_url, String callback_url,
156 List<EiType> supported_types, ProducerStatusInfo status) {
157 EiProducer eiProducer = ImmutableEiProducer.builder() //
158 .ei_producer_id(id) //
159 .ei_job_creation_callback_url(creation_url) //
160 .ei_job_deletion_callback_url(deletion_url) //
161 .ei_producer_supervision_callback_url(callback_url) //
162 .supported_ei_types(supported_types) //
163 .status(status).build(); //
164 eiProducers.put(id, eiProducer);
167 EiType getEiTypeInstance(String id, Object data, List<String> producer_ids) {
168 return ImmutableEiType.builder() //
169 .ei_type_identity(id) //
170 .ei_job_data_schema(data) //
171 .ei_producer_ids(producer_ids) //
175 public void deleteEiProducerInstance(String id) {
176 eiProducers.remove(id);
179 public void deleteEiJobInstance(String id) {
183 EiProducer getEiProducerInstance(String id) throws RestClientException {
184 EiProducer i = eiProducers.get(id);
186 throw new RestClientException("Producer not found: " + id);
191 public Collection<String> getAllEiProducerIds() {
192 return Collections.unmodifiableCollection(eiProducers.keySet());
195 public Collection<EiProducer> getAllEiProducers() {
196 return eiProducers.values();
199 public Collection<EiJob> getAllEiJobs() {
200 return eiJobs.values();
203 public List<EiJob> getEiJobsForOneEiProducer(Optional<String> eiProducerId) {
204 List<EiJob> result = new ArrayList<>();
205 for (EiJob i : eiJobs.values()) {
206 if (eiProducerId.isPresent()) {
207 if (i.owner().equals(eiProducerId.get())) {
218 private Map<String, EiProducer> eiProducers = new HashMap<>();
219 private Map<String, EiJob> eiJobs = new HashMap<>();