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;
26 import java.io.IOException;
28 import java.nio.file.Files;
29 import org.junit.jupiter.api.Test;
30 import org.junit.jupiter.api.extension.ExtendWith;
31 import org.oransc.policyagent.configuration.ApplicationConfig;
32 import org.oransc.policyagent.repository.ImmutablePolicyType;
33 import org.oransc.policyagent.repository.Policies;
34 import org.oransc.policyagent.repository.PolicyType;
35 import org.oransc.policyagent.repository.PolicyTypes;
36 import org.oransc.policyagent.repository.Rics;
37 import org.oransc.policyagent.utils.MockA1ClientFactory;
38 import org.springframework.beans.factory.annotation.Autowired;
39 import org.springframework.boot.test.context.SpringBootTest;
40 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
41 import org.springframework.boot.test.context.TestConfiguration;
42 import org.springframework.boot.web.server.LocalServerPort;
43 import org.springframework.context.annotation.Bean;
44 import org.springframework.test.context.junit.jupiter.SpringExtension;
46 @ExtendWith(SpringExtension.class)
47 @SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT)
48 public class MockPolicyAgent {
53 static class MockApplicationConfig extends ApplicationConfig {
55 public String getLocalConfigurationFilePath() {
56 URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
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 MockA1ClientFactory getA1ClientFactory() {
78 PolicyTypes ricTypes = new PolicyTypes();
80 return new MockA1ClientFactory(ricTypes);
84 public Policies getPolicies() {
89 public PolicyTypes getPolicyTypes() {
90 return this.policyTypes;
94 public Rics getRics() {
98 private static File[] getResourceFolderFiles(String folder) {
99 ClassLoader loader = Thread.currentThread().getContextClassLoader();
100 URL url = loader.getResource(folder);
101 String path = url.getPath();
102 return new File(path).listFiles();
105 private static String readFile(File file) throws IOException {
106 return new String(Files.readAllBytes(file.toPath()));
109 private void loadTypes(PolicyTypes policyTypes) {
110 File[] files = getResourceFolderFiles("policy_types/");
111 for (File file : files) {
113 String schema = readFile(file);
114 String typeName = title(schema);
115 PolicyType type = ImmutablePolicyType.builder().name(typeName).schema(schema).build();
116 policyTypes.put(type);
117 } catch (Exception e) {
118 System.out.println("Could not load json schema " + e);
127 private void keepServerAlive() {
128 System.out.println("Keeping server alive!");
130 synchronized (this) {
133 } catch (Exception ex) {
134 System.out.println("Unexpected: " + ex.toString());
138 private static String title(String jsonSchema) {
139 JsonObject parsedSchema = (JsonObject) JsonParser.parseString(jsonSchema);
140 String title = parsedSchema.get("title").getAsString();
145 public void runMock() throws Exception {