Merge "Changed in config will add and recover Rics"
[nonrtric.git] / policy-agent / src / test / java / org / oransc / policyagent / MockPolicyAgent.java
1 /*-
2  * ========================LICENSE_START=================================
3  * O-RAN-SC
4  * %%
5  * Copyright (C) 2019 Nordix Foundation
6  * %%
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
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
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===================================
19  */
20
21 package org.oransc.policyagent;
22
23 import com.google.gson.JsonObject;
24 import com.google.gson.JsonParser;
25
26 import java.io.File;
27 import java.io.IOException;
28 import java.net.URL;
29 import java.nio.file.Files;
30
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;
47
48 @ExtendWith(SpringExtension.class)
49 @SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT)
50 public class MockPolicyAgent {
51
52     static class MockApplicationConfig extends ApplicationConfig {
53
54         @Override
55         protected String getLocalConfigurationFilePath() {
56             URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
57             return url.getFile();
58         }
59
60     }
61
62     /**
63      * overrides the BeanFactory
64      */
65     @TestConfiguration
66     static class TestBeanFactory {
67
68         private final Rics rics = new Rics();
69         private final Policies policies = new Policies();
70         private final PolicyTypes policyTypes = new PolicyTypes();
71
72         @Bean
73         public ApplicationConfig getApplicationConfig() {
74             return new MockApplicationConfig();
75         }
76
77         @Bean
78         public A1Client getA1Client() {
79             PolicyTypes ricTypes = new PolicyTypes();
80             loadTypes(ricTypes);
81             A1Client client = new MockA1Client(ricTypes);
82             return client;
83         }
84
85         @Bean
86         public Policies getPolicies() {
87             return this.policies;
88         }
89
90         @Bean
91         public PolicyTypes getPolicyTypes() {
92             return this.policyTypes;
93         }
94
95         @Bean
96         public Rics getRics() {
97             return this.rics;
98         }
99
100         private static File[] getResourceFolderFiles(String folder) {
101             ClassLoader loader = Thread.currentThread().getContextClassLoader();
102             URL url = loader.getResource(folder);
103             String path = url.getPath();
104             return new File(path).listFiles();
105         }
106
107         private static String readFile(File file) throws IOException {
108             return new String(Files.readAllBytes(file.toPath()));
109         }
110
111         private void loadTypes(PolicyTypes policyTypes) {
112             File[] files = getResourceFolderFiles("policy_types/");
113             for (File file : files) {
114                 try {
115                     String schema = readFile(file);
116                     String typeName = title(schema);
117                     PolicyType type = ImmutablePolicyType.builder().name(typeName).schema(schema).build();
118                     policyTypes.put(type);
119                 } catch (Exception e) {
120                     System.out.println("Could not load json schema " + e);
121                 }
122             }
123         }
124
125     }
126
127     @LocalServerPort
128     private int port;
129
130     private void keepServerAlive() {
131         System.out.println("Keeping server alive!");
132         try {
133             synchronized (this) {
134                 this.wait();
135             }
136         } catch (Exception ex) {
137             System.out.println("Unexpected: " + ex.toString());
138         }
139     }
140
141     private static String title(String jsonSchema) {
142         JsonObject parsedSchema = (JsonObject) new JsonParser().parse(jsonSchema);
143         String title = parsedSchema.get("title").getAsString();
144         return title;
145     }
146
147     @Test
148     public void runMock() throws Exception {
149         keepServerAlive();
150     }
151
152 }