Fixed concurrency problems
[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.configuration.ApplicationConfig;
34 import org.oransc.policyagent.repository.ImmutablePolicyType;
35 import org.oransc.policyagent.repository.Policies;
36 import org.oransc.policyagent.repository.PolicyType;
37 import org.oransc.policyagent.repository.PolicyTypes;
38 import org.oransc.policyagent.repository.Rics;
39 import org.oransc.policyagent.utils.MockA1ClientFactory;
40 import org.springframework.beans.factory.annotation.Autowired;
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     @Autowired
53     Rics rics;
54
55     static class MockApplicationConfig extends ApplicationConfig {
56         @Override
57         public String getLocalConfigurationFilePath() {
58             URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
59             return url.getFile();
60         }
61     }
62
63     /**
64      * Overrides the BeanFactory.
65      */
66     @TestConfiguration
67     static class TestBeanFactory {
68
69         private final Rics rics = new Rics();
70         private final Policies policies = new Policies();
71         private final PolicyTypes policyTypes = new PolicyTypes();
72
73         @Bean
74         public ApplicationConfig getApplicationConfig() {
75             return new MockApplicationConfig();
76         }
77
78         @Bean
79         public MockA1ClientFactory getA1ClientFactory() {
80             PolicyTypes ricTypes = new PolicyTypes();
81             loadTypes(ricTypes);
82             return new MockA1ClientFactory(ricTypes);
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     @LocalServerPort
127     private int port;
128
129     private void keepServerAlive() {
130         System.out.println("Keeping server alive!");
131         try {
132             synchronized (this) {
133                 this.wait();
134             }
135         } catch (Exception ex) {
136             System.out.println("Unexpected: " + ex.toString());
137         }
138     }
139
140     private static String title(String jsonSchema) {
141         JsonObject parsedSchema = (JsonObject) JsonParser.parseString(jsonSchema);
142         String title = parsedSchema.get("title").getAsString();
143         return title;
144     }
145
146     @Test
147     public void runMock() throws Exception {
148         keepServerAlive();
149     }
150
151 }