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         public void initialize() {
  56             URL url = MockApplicationConfig.class.getClassLoader().getResource("test_application_configuration.json");
  57             loadConfigurationFromFile(url.getFile());
  58         }
  59     }
  60
  61     /**
  62      * overrides the BeanFactory
  63      */
  64     @TestConfiguration
  65     static class TestBeanFactory {
  66
  67         private final Rics rics = new Rics();
  68         private final Policies policies = new Policies();
  69         private final PolicyTypes policyTypes = new PolicyTypes();
  70
  71         @Bean
  72         public ApplicationConfig getApplicationConfig() {
  73             return new MockApplicationConfig();
  74         }
  75
  76         @Bean
  77         public A1Client getA1Client() {
  78             PolicyTypes ricTypes = new PolicyTypes();
  79             loadTypes(ricTypes);
  80             A1Client client = new MockA1Client(ricTypes);
  81             return client;
  82         }
  83
  84         @Bean
  85         public Policies getPolicies() {
  86             return this.policies;
  87         }
  88
  89         @Bean
  90         public PolicyTypes getPolicyTypes() {
  91             return this.policyTypes;
  92         }
  93
  94         @Bean
  95         public Rics getRics() {
  96             return this.rics;
  97         }
  98
  99         private static File[] getResourceFolderFiles(String folder) {
 100             ClassLoader loader = Thread.currentThread().getContextClassLoader();
 101             URL url = loader.getResource(folder);
 102             String path = url.getPath();
 103             return new File(path).listFiles();
 104         }
 105
 106         private static String readFile(File file) throws IOException {
 107             return new String(Files.readAllBytes(file.toPath()));
 108         }
 109
 110         private void loadTypes(PolicyTypes policyTypes) {
 111             File[] files = getResourceFolderFiles("policy_types/");
 112             for (File file : files) {
 113                 try {
 114                     String schema = readFile(file);
 115                     String typeName = title(schema);
 116                     PolicyType type = ImmutablePolicyType.builder().name(typeName).schema(schema).build();
 117                     policyTypes.put(type);
 118                 } catch (Exception e) {
 119                     System.out.println("Could not load json schema " + e);
 120                 }
 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) new JsonParser().parse(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 }