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.stream.Collectors;
38 import org.oransc.portal.nonrtric.controlpanel.eiproducerapi.EiProducerApi;
39 import org.oransc.portal.nonrtric.controlpanel.model.JobInfo;
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();
92 result.addAll(database.getAllEiJobs());
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, "owner", "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 JobInfo i = JobInfo.builder() //
145 .jobData(instanceData) //
149 .targetUri(targetUrl) //
151 eiJobs.put(instanceId, i);
154 void putEiProducerInstance(String id, String creation_url, String deletion_url, String callback_url,
155 List<EiType> supported_types, ProducerStatusInfo status) {
156 EiProducer eiProducer = ImmutableEiProducer.builder() //
157 .ei_producer_id(id) //
158 .ei_job_creation_callback_url(creation_url) //
159 .ei_job_deletion_callback_url(deletion_url) //
160 .ei_producer_supervision_callback_url(callback_url) //
161 .supported_ei_types(supported_types) //
162 .status(status).build(); //
163 eiProducers.put(id, eiProducer);
166 EiType getEiTypeInstance(String id, Object data, List<String> producer_ids) {
167 return ImmutableEiType.builder() //
168 .ei_type_identity(id) //
169 .ei_job_data_schema(data) //
170 .ei_producer_ids(producer_ids) //
174 public void deleteEiProducerInstance(String id) {
175 eiProducers.remove(id);
178 public void deleteEiJobInstance(String id) {
182 EiProducer getEiProducerInstance(String id) throws RestClientException {
183 EiProducer i = eiProducers.get(id);
185 throw new RestClientException("Producer not found: " + id);
190 public Collection<String> getAllEiProducerIds() {
191 return Collections.unmodifiableCollection(eiProducers.keySet());
194 public Collection<EiProducer> getAllEiProducers() {
195 return eiProducers.values();
198 public List<JobInfo> getAllEiJobs() {
199 return new ArrayList<>(eiJobs.values());
203 private Map<String, EiProducer> eiProducers = new HashMap<>();
204 private Map<String, JobInfo> eiJobs = new HashMap<>();