1 files changed, 91 insertions, 0 deletions
diff --git a/src/test/java/org/openecomp/schema/RelationshipSchemaLoaderTest.java b/src/test/java/org/openecomp/schema/RelationshipSchemaLoaderTest.java
new file mode 100644
index 0000000..85ef2fe
--- /dev/null
+++ b/src/test/java/org/openecomp/schema/RelationshipSchemaLoaderTest.java
@@ -0,0 +1,91 @@
+package org.openecomp.schema;
+
+import static org.junit.Assert.*;
+import edu.emory.mathcs.backport.java.util.Arrays;
+import org.junit.Before;
+import org.junit.Test;
+import org.openecomp.crud.exception.CrudException;
+
+import java.io.File;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.junit.Assert.*;
+
+public class RelationshipSchemaLoaderTest {
+
+    @Before
+    public void init() {
+        ClassLoader classLoader = getClass().getClassLoader();
+        File dir = new File(classLoader.getResource( "model").getFile());
+        System.setProperty("CONFIG_HOME", dir.getParent());
+        RelationshipSchemaLoader.resetVersionContextMap();
+    }
+
+    @Test
+    public void loadModels() throws Exception {
+        RelationshipSchemaLoader.loadModels();
+        assertFalse( RelationshipSchemaLoader.getVersionContextMap().keySet().isEmpty());
+    }
+
+    @Test
+    public void loadModelsWithAVersion() throws Exception {
+        RelationshipSchemaLoader.loadModels("v11");
+        assertEquals(1, RelationshipSchemaLoader.getVersionContextMap().keySet().size());
+        assertEquals("v11",  RelationshipSchemaLoader.getLatestSchemaVersion());
+    }
+
+    @Test
+    public void getSchemaForVersion() throws Exception {
+        RelationshipSchemaLoader.loadModels("v11");
+        String version = RelationshipSchemaLoader.getLatestSchemaVersion();
+        RelationshipSchema g = RelationshipSchemaLoader.getSchemaForVersion(version);
+        assertNotNull(g.lookupRelationType("has"));
+    }
+
+    @Test
+    public void getSchemaForVersionFail() throws Exception {
+        RelationshipSchemaLoader.loadModels();
+        try {
+            RelationshipSchemaLoader.getSchemaForVersion("v1");
+        } catch (CrudException e) {
+            assertEquals(404, e.getHttpStatus().getStatusCode());
+        }
+    }
+
+    @Test
+    public void setVersionContextMap() throws Exception {
+        ArrayList<String> jsonString = new ArrayList<String>();
+        String rules = "{" +
+                "\"rules\": [" +
+                "{" +
+                "\"from\": \"availability-zone\"," +
+                "\"to\": \"complex\"," +
+                "\"label\": \"groupsResourcesIn\"," +
+                "\"direction\": \"OUT\"," +
+                "\"multiplicity\": \"Many2Many\"," +
+                "\"contains-other-v\": \"NONE\"," +
+                "\"delete-other-v\": \"NONE\"," +
+                "\"SVC-INFRA\": \"NONE\"," +
+                "\"prevent-delete\": \"!${direction}\"" +
+                "}]}";
+        String props = "{" +
+                "  \"isParent\":\"java.lang.Boolean\"," +
+                "  \"isParent-REV\":\"java.lang.Boolean\"," +
+                "  \"usesResource\":\"java.lang.Boolean\"," +
+                "  \"usesResource-REV\":\"java.lang.Boolean\"," +
+                "  \"SVC-INFRA\":\"java.lang.Boolean\"," +
+                "  \"SVC-INFRA-REV\":\"java.lang.Boolean\"," +
+                "  \"hasDelTarget\":\"java.lang.Boolean\"," +
+                "  \"hasDelTarget-REV\":\"java.lang.Boolean\"" +
+                "}";
+        jsonString.add(rules);
+        jsonString.add(props);
+        RelationshipSchema nRs = new RelationshipSchema(jsonString);
+        Map<String, RelationshipSchema> versionMap = new HashMap<>();
+        versionMap.put("v1", nRs);
+        RelationshipSchemaLoader.setVersionContextMap(versionMap);
+        assertNotNull(RelationshipSchemaLoader.getSchemaForVersion("v1").lookupRelationType("groupsResourcesIn"));
+    }
+}
+\ No newline at end of file