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 }