6ccf162c9904803d04a16eca599d8e8f11706146
[ccsdk/oran.git] /
1 /*-
2  * ========================LICENSE_START=================================
3  * ONAP : ccsdk oran
4  * ======================================================================
5  * Copyright (C) 2019-2020 Nordix Foundation. All rights reserved.
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.onap.ccsdk.oran.a1policymanagementservice;
22
23 import static org.awaitility.Awaitility.await;
24
25 import com.google.gson.JsonObject;
26 import com.google.gson.JsonParser;
27
28 import java.io.File;
29 import java.io.IOException;
30 import java.net.URL;
31 import java.nio.file.Files;
32 import java.time.Instant;
33
34 import org.junit.jupiter.api.Test;
35 import org.junit.jupiter.api.extension.ExtendWith;
36 import org.onap.ccsdk.oran.a1policymanagementservice.configuration.ApplicationConfig;
37 import org.onap.ccsdk.oran.a1policymanagementservice.repository.ImmutablePolicy;
38 import org.onap.ccsdk.oran.a1policymanagementservice.repository.ImmutablePolicyType;
39 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Policies;
40 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Policy;
41 import org.onap.ccsdk.oran.a1policymanagementservice.repository.PolicyType;
42 import org.onap.ccsdk.oran.a1policymanagementservice.repository.PolicyTypes;
43 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Ric;
44 import org.onap.ccsdk.oran.a1policymanagementservice.repository.Rics;
45 import org.onap.ccsdk.oran.a1policymanagementservice.utils.MockA1ClientFactory;
46 import org.slf4j.Logger;
47 import org.slf4j.LoggerFactory;
48 import org.springframework.beans.factory.annotation.Autowired;
49 import org.springframework.boot.test.context.SpringBootTest;
50 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
51 import org.springframework.boot.test.context.TestConfiguration;
52 import org.springframework.boot.web.server.LocalServerPort;
53 import org.springframework.context.annotation.Bean;
54 import org.springframework.test.context.TestPropertySource;
55 import org.springframework.test.context.junit.jupiter.SpringExtension;
56 import org.springframework.util.StringUtils;
57
58 @ExtendWith(SpringExtension.class)
59 @SpringBootTest(webEnvironment = WebEnvironment.DEFINED_PORT)
60 @TestPropertySource(
61     properties = { //
62         "server.ssl.key-store=./config/keystore.jks", //
63         "app.webclient.trust-store=./config/truststore.jks"})
64 class MockPolicyManagementService {
65     private static final Logger logger = LoggerFactory.getLogger(MockPolicyManagementService.class);
66
67     @Autowired
68     Rics rics;
69
70     @Autowired
71     Policies policies;
72
73     @Autowired
74     PolicyTypes policyTypes;
75
76     @Autowired
77     ApplicationConfig applicationConfig;
78
79     static class MockApplicationConfig extends ApplicationConfig {
80         @Override
81         public String getLocalConfigurationFilePath() {
82             URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
83             return url.getFile();
84         }
85     }
86
87     /**
88      * Overrides the BeanFactory.
89      */
90     @TestConfiguration
91     static class TestBeanFactory {
92
93         private final Rics rics = new Rics();
94         private final Policies policies = new Policies();
95         private final PolicyTypes policyTypes = new PolicyTypes();
96
97         @Bean
98         public ApplicationConfig getApplicationConfig() {
99             return new MockApplicationConfig();
100         }
101
102         @Bean
103         public MockA1ClientFactory getA1ClientFactory() {
104             PolicyTypes ricTypes = new PolicyTypes();
105             loadTypes(ricTypes);
106             return new MockA1ClientFactory(ricTypes);
107         }
108
109         @Bean
110         public Policies getPolicies() {
111             return this.policies;
112         }
113
114         @Bean
115         public PolicyTypes getPolicyTypes() {
116             return this.policyTypes;
117         }
118
119         @Bean
120         public Rics getRics() {
121             return this.rics;
122         }
123
124         private static File[] getResourceFolderFiles(String folder) {
125             return getFile(folder).listFiles();
126         }
127
128         private static String readFile(File file) throws IOException {
129             return new String(Files.readAllBytes(file.toPath()));
130         }
131
132         private void loadTypes(PolicyTypes policyTypes) {
133             File[] files = getResourceFolderFiles("policy_types/");
134             for (File file : files) {
135                 try {
136                     String schema = readFile(file);
137                     String typeName = title(schema);
138                     PolicyType type = ImmutablePolicyType.builder().id(typeName).schema(schema).build();
139                     policyTypes.put(type);
140                 } catch (Exception e) {
141                     logger.error("Could not load json schema ", e);
142                 }
143             }
144             policyTypes.put(ImmutablePolicyType.builder().id("").schema("{}").build());
145         }
146     }
147
148     private static File getFile(String path) {
149         ClassLoader loader = Thread.currentThread().getContextClassLoader();
150         URL url = loader.getResource(path);
151         return new File(url.getPath());
152     }
153
154     @LocalServerPort
155     private int port;
156
157     private void keepServerAlive() throws InterruptedException, IOException {
158         waitForConfigurationToBeLoaded();
159         loadInstances();
160         logger.info("Keeping server alive!");
161         synchronized (this) {
162             this.wait();
163         }
164     }
165
166     private void waitForConfigurationToBeLoaded() throws IOException {
167         String json = getConfigJsonFromFile();
168         try {
169             int noOfRicsInConfigFile = StringUtils.countOccurrencesOf(json, "baseUrl");
170             await().until(() -> rics.size() == noOfRicsInConfigFile);
171         } catch (Exception e) {
172             logger.info("Loaded rics: {}, and no of rics in config file: {} never matched!", rics.size(),
173                 StringUtils.countOccurrencesOf(json, "baseUrl"));
174         }
175     }
176
177     private static String title(String jsonSchema) {
178         JsonObject parsedSchema = (JsonObject) JsonParser.parseString(jsonSchema);
179         String title = parsedSchema.get("title").getAsString();
180         return title;
181     }
182
183     private void loadInstances() throws IOException {
184         PolicyType unnamedPolicyType = policyTypes.get("");
185         Ric ric = rics.get("ric1");
186         String json = getConfigJsonFromFile();
187
188         Policy policy = ImmutablePolicy.builder() //
189             .id("typelessPolicy") //
190             .json(json) //
191             .ownerServiceId("MockPolicyManagementService") //
192             .ric(ric) //
193             .type(unnamedPolicyType) //
194             .lastModified(Instant.now()) //
195             .isTransient(false) //
196             .build();
197         this.policies.put(policy);
198     }
199
200     private String getConfigJsonFromFile() throws IOException {
201         File jsonFile = getFile("test_application_configuration.json");
202         String json = new String(Files.readAllBytes(jsonFile.toPath()));
203         return json;
204     }
205
206     @Test
207     @SuppressWarnings("squid:S2699") // Tests should include assertions. This test is only for keeping the server
208                                      // alive, so it will only be confusing to add an assertion.
209     void runMock() throws Exception {
210         keepServerAlive();
211     }
212
213 }