X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=aai-schema-ingest%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Ftestutils%2FConfigTranslatorForWiringTest.java;fp=aai-schema-ingest%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Ftestutils%2FConfigTranslatorForWiringTest.java;h=fc0a3984114e72ce9627697a44d703c011e5eb48;hb=a86d6a6644d6de3f3f814cd6e25cfe2213d5dd05;hp=5e1b6c8264b3043556c3127a477d46577a0009a0;hpb=a395fa69a28a04d0a667fe458b4f10497a1d6794;p=aai%2Faai-common.git

diff --git a/aai-schema-ingest/src/test/java/org/onap/aai/testutils/ConfigTranslatorForWiringTest.java b/aai-schema-ingest/src/test/java/org/onap/aai/testutils/ConfigTranslatorForWiringTest.java
index 5e1b6c82..fc0a3984 100644
--- a/aai-schema-ingest/src/test/java/org/onap/aai/testutils/ConfigTranslatorForWiringTest.java
+++ b/aai-schema-ingest/src/test/java/org/onap/aai/testutils/ConfigTranslatorForWiringTest.java
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * org.onap.aai
  * ================================================================================
- * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017-18 AT&T Intellectual Property. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -16,44 +16,37 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  * ============LICENSE_END=========================================================
- *
- * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  */
 
 package org.onap.aai.testutils;
 
-import java.util.ArrayList;
-import java.util.EnumMap;
-import java.util.List;
-import java.util.Map;
+import java.util.*;
 
-import org.onap.aai.setup.ConfigTranslator;
-import org.onap.aai.setup.SchemaLocationsBean;
-import org.onap.aai.setup.Version;
+import org.onap.aai.setup.*;
 
 public class ConfigTranslatorForWiringTest extends ConfigTranslator {
 
-	public ConfigTranslatorForWiringTest(SchemaLocationsBean bean) {
-		super(bean);
+	public ConfigTranslatorForWiringTest(SchemaLocationsBean bean, SchemaVersions schemaVersions) {
+		super(bean, schemaVersions);
 	}
 
 	@Override
-	public Map<Version, List<String>> getNodeFiles() {
+	public Map<SchemaVersion, List<String>> getNodeFiles() {
 		String f = bean.getNodeDirectory() + "test_business_v10.xml";
 		List<String> files = new ArrayList<>();
 		files.add(f);
-		Map<Version, List<String>> mp = new EnumMap<>(Version.class);
-		mp.put(Version.V10, files);
+		Map<SchemaVersion, List<String>> mp = new TreeMap<>();
+		mp.put(new SchemaVersion("v10"), files);
 		return mp;
 	}
 
 	@Override
-	public Map<Version, List<String>> getEdgeFiles() {
+	public Map<SchemaVersion, List<String>> getEdgeFiles() {
 		String f = bean.getEdgeDirectory() + "test.json";
 		List<String> files = new ArrayList<>();
 		files.add(f);
-		Map<Version, List<String>> mp = new EnumMap<>(Version.class);
-		mp.put(Version.V10, files);
+		Map<SchemaVersion, List<String>> mp = new TreeMap<>();
+		mp.put(new SchemaVersion("v10"), files);
 		return mp;
 	}