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