2 * ========================LICENSE_START=================================
5 * Copyright (C) 2019 Nordix Foundation
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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===================================
21 package org.oransc.policyagent;
23 import com.google.gson.JsonObject;
24 import com.google.gson.JsonParser;
27 import java.io.IOException;
29 import java.nio.file.Files;
31 import org.junit.jupiter.api.Test;
32 import org.junit.jupiter.api.extension.ExtendWith;
33 import org.oransc.policyagent.clients.A1Client;
34 import org.oransc.policyagent.configuration.ApplicationConfig;
35 import org.oransc.policyagent.repository.ImmutablePolicyType;
36 import org.oransc.policyagent.repository.Policies;
37 import org.oransc.policyagent.repository.PolicyType;
38 import org.oransc.policyagent.repository.PolicyTypes;
39 import org.oransc.policyagent.repository.Rics;
40 import org.oransc.policyagent.utils.MockA1Client;
41 import org.springframework.boot.test.context.SpringBootTest;
42 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
43 import org.springframework.boot.test.context.TestConfiguration;
44 import org.springframework.boot.web.server.LocalServerPort;
45 import org.springframework.context.annotation.Bean;
46 import org.springframework.test.context.junit.jupiter.SpringExtension;
48 @ExtendWith(SpringExtension.class)
49 @SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT)
50 public class MockPolicyAgent {
52 static class MockApplicationConfig extends ApplicationConfig {
55 public void initialize() {
56 URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
57 loadConfigurationFromFile(url.getFile());
62 * overrides the BeanFactory
65 static class TestBeanFactory {
67 private final Rics rics = new Rics();
68 private final Policies policies = new Policies();
69 private final PolicyTypes policyTypes = new PolicyTypes();
72 public ApplicationConfig getApplicationConfig() {
73 return new MockApplicationConfig();
77 public A1Client getA1Client() {
78 PolicyTypes ricTypes = new PolicyTypes();
80 A1Client client = new MockA1Client(ricTypes);
85 public Policies getPolicies() {
90 public PolicyTypes getPolicyTypes() {
91 return this.policyTypes;
95 public Rics getRics() {
99 private static File[] getResourceFolderFiles(String folder) {
100 ClassLoader loader = Thread.currentThread().getContextClassLoader();
101 URL url = loader.getResource(folder);
102 String path = url.getPath();
103 return new File(path).listFiles();
106 private static String readFile(File file) throws IOException {
107 return new String(Files.readAllBytes(file.toPath()));
110 private void loadTypes(PolicyTypes policyTypes) {
111 File[] files = getResourceFolderFiles("policy_types/");
112 for (File file : files) {
114 String schema = readFile(file);
115 String typeName = title(schema);
116 PolicyType type = ImmutablePolicyType.builder().name(typeName).schema(schema).build();
117 policyTypes.put(type);
118 } catch (Exception e) {
119 System.out.println("Could not load json schema " + e);
129 private void keepServerAlive() {
130 System.out.println("Keeping server alive!");
132 synchronized (this) {
135 } catch (Exception ex) {
136 System.out.println("Unexpected: " + ex.toString());
140 private static String title(String jsonSchema) {
141 JsonObject parsedSchema = (JsonObject) new JsonParser().parse(jsonSchema);
142 String title = parsedSchema.get("title").getAsString();
147 public void runMock() throws Exception {