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 }