Merge "Return response codes from policy controller"
[nonrtric.git] / policy-agent / src / test / java / org / oransc / policyagent / ApplicationTest.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 static org.assertj.core.api.Assertions.assertThat;
24 import static org.junit.Assert.assertEquals;
25 import static org.junit.jupiter.api.Assertions.assertFalse;
26
27 import com.google.gson.Gson;
28 import com.google.gson.GsonBuilder;
29 import com.google.gson.reflect.TypeToken;
30
31 import java.net.URL;
32 import java.util.List;
33 import java.util.Vector;
34
35 import org.junit.jupiter.api.Test;
36 import org.junit.jupiter.api.extension.ExtendWith;
37 import org.oransc.policyagent.clients.A1Client;
38 import org.oransc.policyagent.configuration.ApplicationConfig;
39 import org.oransc.policyagent.configuration.ImmutableRicConfig;
40 import org.oransc.policyagent.configuration.RicConfig;
41 import org.oransc.policyagent.controllers.ImmutableServiceRegistrationInfo;
42 import org.oransc.policyagent.controllers.ImmutableServiceStatus;
43 import org.oransc.policyagent.controllers.ServiceRegistrationInfo;
44 import org.oransc.policyagent.controllers.ServiceStatus;
45 import org.oransc.policyagent.exceptions.ServiceException;
46 import org.oransc.policyagent.repository.ImmutablePolicy;
47 import org.oransc.policyagent.repository.ImmutablePolicyType;
48 import org.oransc.policyagent.repository.Policies;
49 import org.oransc.policyagent.repository.Policy;
50 import org.oransc.policyagent.repository.PolicyType;
51 import org.oransc.policyagent.repository.PolicyTypes;
52 import org.oransc.policyagent.repository.Ric;
53 import org.oransc.policyagent.repository.Rics;
54 import org.oransc.policyagent.utils.MockA1Client;
55 import org.springframework.beans.factory.annotation.Autowired;
56 import org.springframework.boot.test.context.SpringBootTest;
57 import org.springframework.boot.test.context.SpringBootTest.WebEnvironment;
58 import org.springframework.boot.test.context.TestConfiguration;
59 import org.springframework.boot.web.server.LocalServerPort;
60 import org.springframework.context.ApplicationContext;
61 import org.springframework.context.annotation.Bean;
62 import org.springframework.http.HttpStatus;
63 import org.springframework.http.ResponseEntity;
64 import org.springframework.test.context.junit.jupiter.SpringExtension;
65 import org.springframework.web.client.RestTemplate;
66
67 @ExtendWith(SpringExtension.class)
68 @SpringBootTest(webEnvironment = WebEnvironment.RANDOM_PORT)
69 public class ApplicationTest {
70     @Autowired
71     ApplicationContext context;
72
73     @Autowired
74     private Rics rics;
75
76     @Autowired
77     private Policies policies;
78
79     @Autowired
80     private PolicyTypes policyTypes;
81
82     private static Gson gson = new GsonBuilder() //
83         .serializeNulls() //
84         .create(); //
85
86     public static class MockApplicationConfig extends ApplicationConfig {
87         @Override
88         protected String getLocalConfigurationFilePath() {
89             URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
90             return url.getFile();
91         }
92     }
93
94     /**
95      * Overrides the BeanFactory.
96      */
97     @TestConfiguration
98     static class TestBeanFactory {
99         private final Rics rics = new Rics();
100         private final Policies policies = new Policies();
101         private final PolicyTypes policyTypes = new PolicyTypes();
102
103         @Bean
104         public ApplicationConfig getApplicationConfig() {
105             return new MockApplicationConfig();
106         }
107
108         @Bean
109         A1Client getA1Client() {
110             return new MockA1Client(this.policyTypes);
111         }
112
113         @Bean
114         public Policies getPolicies() {
115             return this.policies;
116         }
117
118         @Bean
119         public PolicyTypes getPolicyTypes() {
120             return this.policyTypes;
121         }
122
123         @Bean
124         public Rics getRics() {
125             return this.rics;
126         }
127     }
128
129     @LocalServerPort
130     private int port;
131
132     private final RestTemplate restTemplate = new RestTemplate();
133
134     private void reset() {
135         rics.clear();
136         policies.clear();
137         policyTypes.clear();
138         assertThat(policies.size()).isEqualTo(0);
139     }
140
141     @Test
142     public void testGetRics() throws Exception {
143         reset();
144         addRic("kista_1");
145         String url = baseUrl() + "/rics";
146         String rsp = this.restTemplate.getForObject(url, String.class);
147         System.out.println(rsp);
148         assertThat(rsp).contains("kista_1");
149
150         url = baseUrl() + "/rics?policyType=ANR";
151         rsp = this.restTemplate.getForObject(url, String.class);
152         assertThat(rsp).isEqualTo("[]");
153     }
154
155     @Test
156     public void testGetRic() throws Exception {
157         reset();
158         String url = baseUrl() + "/ric?managedElementId=kista_1";
159         String rsp = this.restTemplate.getForObject(url, String.class);
160         System.out.println(rsp);
161         assertThat(rsp).isEqualTo("ric1");
162     }
163
164     @Test
165     public void testPutPolicy() throws Exception {
166         putService("service1");
167
168         String url = baseUrl() + "/policy?type=type1&instance=instance1&ric=ric1&service=service1";
169         String json = "{}";
170         addPolicyType("type1", "ric1");
171         this.rics.getRic("ric1").setState(Ric.RicState.IDLE);
172
173         this.restTemplate.put(url, json);
174
175         Policy policy = policies.getPolicy("instance1");
176
177         assertThat(policy).isNotNull();
178         assertThat(policy.id()).isEqualTo("instance1");
179         assertThat(policy.ownerServiceName()).isEqualTo("service1");
180
181         url = baseUrl() + "/policies";
182         String rsp = this.restTemplate.getForObject(url, String.class);
183         System.out.println(rsp);
184     }
185
186     private PolicyType addPolicyType(String policyTypeName, String ricName) {
187         PolicyType type = ImmutablePolicyType.builder() //
188             .name(policyTypeName) //
189             .schema("{\"title\":\"" + policyTypeName + "\"}") //
190             .build();
191
192         policyTypes.put(type);
193         addRic(ricName).addSupportedPolicyType(type);
194         return type;
195     }
196
197     private Ric addRic(String ricName) {
198         if (rics.get(ricName) != null) {
199             return rics.get(ricName);
200         }
201         Vector<String> mes = new Vector<>();
202         RicConfig conf = ImmutableRicConfig.builder() //
203             .name(ricName) //
204             .baseUrl("baseUrl") //
205             .managedElementIds(mes) //
206             .build();
207         Ric ric = new Ric(conf);
208         this.rics.put(ric);
209         return ric;
210     }
211
212     private Policy addPolicy(String id, String typeName, String service, String ric) throws ServiceException {
213         addRic(ric);
214         Policy p = ImmutablePolicy.builder().id(id) //
215             .json("{}") //
216             .ownerServiceName(service) //
217             .ric(rics.getRic(ric)) //
218             .type(addPolicyType(typeName, ric)) //
219             .lastModified("lastModified").build();
220         policies.put(p);
221         return p;
222     }
223
224     private Policy addPolicy(String id, String typeName, String service) throws ServiceException {
225         return addPolicy(id, typeName, service, "ric");
226     }
227
228     private String baseUrl() {
229         return "http://localhost:" + port;
230     }
231
232     @Test
233     public void testGetPolicy() throws Exception {
234         String url = baseUrl() + "/policy?instance=id";
235         Policy policy = addPolicy("id", "typeName", "service1", "ric1");
236         {
237             String rsp = this.restTemplate.getForObject(url, String.class);
238             assertThat(rsp).isEqualTo(policy.json());
239         }
240         {
241             policies.remove(policy);
242             ResponseEntity<String> rsp = this.restTemplate.getForEntity(url, String.class);
243             assertThat(rsp.getStatusCodeValue()).isEqualTo(HttpStatus.NO_CONTENT.value());
244         }
245     }
246
247     @Test
248     public void testDeletePolicy() throws Exception {
249         reset();
250         String url = baseUrl() + "/policy?instance=id";
251         Policy policy = addPolicy("id", "typeName", "service1", "ric1");
252         policy.ric().setState(Ric.RicState.IDLE);
253         assertThat(policies.size()).isEqualTo(1);
254
255         this.restTemplate.delete(url);
256
257         assertThat(policies.size()).isEqualTo(0);
258     }
259
260     private static <T> List<T> parseList(String json, Class<T> clazz) {
261         if (null == json) {
262             return null;
263         }
264         return gson.fromJson(json, new TypeToken<T>() {}.getType());
265
266     }
267
268     @Test
269     public void testGetPolicySchemas() throws Exception {
270         reset();
271         addPolicyType("type1", "ric1");
272         addPolicyType("type2", "ric2");
273
274         String url = baseUrl() + "/policy_schemas";
275         String rsp = this.restTemplate.getForObject(url, String.class);
276         System.out.println("*** " + rsp);
277         assertThat(rsp).contains("type1");
278         assertThat(rsp).contains("type2");
279         assertThat(rsp).contains("title");
280
281         List<String> info = parseList(rsp, String.class);
282         assertEquals(2, info.size());
283
284         url = baseUrl() + "/policy_schemas?ric=ric1";
285         rsp = this.restTemplate.getForObject(url, String.class);
286         assertThat(rsp).contains("type1");
287         info = parseList(rsp, String.class);
288         assertEquals(1, info.size());
289     }
290
291     @Test
292     public void testGetPolicySchema() throws Exception {
293         reset();
294         addPolicyType("type1", "ric1");
295         addPolicyType("type2", "ric2");
296
297         String url = baseUrl() + "/policy_schema?id=type1";
298         String rsp = this.restTemplate.getForObject(url, String.class);
299         System.out.println(rsp);
300         assertThat(rsp).contains("type1");
301         assertThat(rsp).contains("title");
302     }
303
304     @Test
305     public void testGetPolicyTypes() throws Exception {
306         reset();
307         addPolicyType("type1", "ric1");
308         addPolicyType("type2", "ric2");
309
310         String url = baseUrl() + "/policy_types";
311         String rsp = this.restTemplate.getForObject(url, String.class);
312         assertThat(rsp).isEqualTo("[\"type2\",\"type1\"]");
313
314         url = baseUrl() + "/policy_types?ric=ric1";
315         rsp = this.restTemplate.getForObject(url, String.class);
316         assertThat(rsp).isEqualTo("[\"type1\"]");
317     }
318
319     @Test
320     public void testGetPolicies() throws Exception {
321         String url = baseUrl() + "/policies";
322         addPolicy("id1", "type1", "service1");
323         addPolicy("id2", "type2", "service2");
324
325         String rsp = this.restTemplate.getForObject(url, String.class);
326         System.out.println(rsp);
327         assertThat(rsp).contains("id1");
328         assertThat(rsp).contains("id2");
329     }
330
331     @Test
332     public void testGetPoliciesFilter() throws Exception {
333         addPolicy("id1", "type1", "service1");
334         addPolicy("id2", "type1", "service2");
335         addPolicy("id3", "type2", "service1");
336
337         String url = baseUrl() + "/policies?type=type1";
338         String rsp = this.restTemplate.getForObject(url, String.class);
339         System.out.println(rsp);
340         assertThat(rsp).contains("id1");
341         assertThat(rsp).contains("id2");
342         assertFalse(rsp.contains("id3"));
343
344         url = baseUrl() + "/policies?type=type1&service=service2";
345         rsp = this.restTemplate.getForObject(url, String.class);
346         System.out.println(rsp);
347         assertFalse(rsp.contains("id1"));
348         assertThat(rsp).contains("id2");
349         assertFalse(rsp.contains("id3"));
350     }
351
352     private String createServiceJson(String name) {
353         ServiceRegistrationInfo service = ImmutableServiceRegistrationInfo.builder() //
354             .keepAliveInterval(1) //
355             .name(name) //
356             .callbackUrl("callbackUrl") //
357             .build();
358         String json = gson.toJson(service);
359         return json;
360     }
361
362     private void putService(String name) {
363         String url = baseUrl() + "/service";
364         this.restTemplate.put(url, createServiceJson(name));
365     }
366
367     @Test
368     public void testPutAndGetService() throws Exception {
369         putService("name");
370
371         String url = baseUrl() + "/service?name=name";
372         String rsp = this.restTemplate.getForObject(url, String.class);
373         ServiceStatus status = gson.fromJson(rsp, ImmutableServiceStatus.class);
374         assertThat(status.keepAliveInterval() == 1);
375         assertThat(status.name().equals("name"));
376
377         url = baseUrl() + "/services";
378         rsp = this.restTemplate.getForObject(url, String.class);
379         assertThat(rsp.contains("name"));
380         System.out.println(rsp);
381
382         url = baseUrl() + "/service/ping";
383         this.restTemplate.put(url, "name");
384     }
385
386 }