3704b7cd9e9125f6503edb230b47797023969fd2
[aai/aai-common.git] / aai-schema-ingest / src / test / java / org / onap / aai / testutils / TestUtilConfigTranslator.java
1 /** 
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-18 AT&T Intellectual Property. 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.aai.testutils;
22
23 import org.onap.aai.setup.ConfigTranslator;
24 import org.onap.aai.setup.SchemaLocationsBean;
25 import org.onap.aai.setup.SchemaVersion;
26 import org.onap.aai.setup.SchemaVersions;
27 import org.springframework.context.annotation.PropertySource;
28
29 import java.util.ArrayList;
30 import java.util.List;
31 import java.util.Map;
32 import java.util.TreeMap;
33 @PropertySource(value = "classpath:schema-ingest.properties", ignoreResourceNotFound = true)
34 @PropertySource(value = "file:${schema.ingest.file}", ignoreResourceNotFound = true)
35
36 public class TestUtilConfigTranslator extends ConfigTranslator {
37
38     public static final SchemaVersion LATEST = new SchemaVersion("v15");
39     public TestUtilConfigTranslator(SchemaLocationsBean bean, SchemaVersions schemaVersions) {
40         super(bean, schemaVersions);
41     }
42
43     @Override
44     public Map<SchemaVersion, List<String>> getNodeFiles() {
45         List<String> files10 = new ArrayList<>();
46         files10.add("src/test/resources/oxm/test_network_v10.xml");
47         files10.add("src/test/resources/oxm/test_business_v10.xml");
48         
49         List<String> files11 = new ArrayList<>();
50         files11.add("src/test/resources/oxm/test_network_v11.xml");
51         files11.add("src/test/resources/oxm/test_business_v11.xml");
52         
53         Map<SchemaVersion, List<String>> input = new TreeMap<>();
54         input.put(new SchemaVersion("v10"), files10);
55         input.put(new SchemaVersion("v11"), files11);
56
57         List<String> files13 = new ArrayList<>();
58         files13.add("src/test/resources/oxm/business_oxm_v13.xml");
59         files13.add("src/test/resources/oxm/common_oxm_v13.xml");
60         files13.add("src/test/resources/oxm/serviceDesign_oxm_v13.xml");
61         files13.add("src/test/resources/oxm/network_oxm_v13.xml");
62     
63     input.put(new SchemaVersion("v10"), files10);
64         input.put(new SchemaVersion("v11"), files11);
65         input.put(new SchemaVersion("v13"), files13);
66         return input;
67     }
68
69     @Override
70     public Map<SchemaVersion, List<String>> getEdgeFiles() {
71         List<String> files = new ArrayList<>();
72         files.add("src/test/resources/edgeRules/test.json");
73         files.add("src/test/resources/edgeRules/test2.json");
74         files.add("src/test/resources/edgeRules/otherTestRules.json");
75         Map<SchemaVersion, List<String>> input = new TreeMap<>();
76         input.put(LATEST, files);
77         
78         List<String> files2 = new ArrayList<>();
79         files2.add("src/test/resources/edgeRules/test.json");
80         input.put(new SchemaVersion("v10"), files2);
81         
82         List<String> files3 = new ArrayList<>();
83         files3.add("src/test/resources/edgeRules/test3.json");
84         files3.add("src/test/resources/edgeRules/defaultEdgesTest.json");
85         input.put(new SchemaVersion("v11"), files3);
86         return input;
87     }
88 }
89
90