X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Forg%2Fonap%2Faai%2Fbabel%2Fcsar%2Fvnfcatalog%2FVnfVendorImageExtractorTest.java;h=a6f98b8607f0cc9440d450dbf64741f0d5951973;hb=refs%2Fchanges%2F95%2F64995%2F1;hp=b1229b4830092655c50b20796b1918b751de64c3;hpb=6b2de4b2de03890d0f0ba0663244ef1a98457309;p=aai%2Fbabel.git

diff --git a/src/test/java/org/onap/aai/babel/csar/vnfcatalog/VnfVendorImageExtractorTest.java b/src/test/java/org/onap/aai/babel/csar/vnfcatalog/VnfVendorImageExtractorTest.java
index b1229b4..a6f98b8 100644
--- a/src/test/java/org/onap/aai/babel/csar/vnfcatalog/VnfVendorImageExtractorTest.java
+++ b/src/test/java/org/onap/aai/babel/csar/vnfcatalog/VnfVendorImageExtractorTest.java
@@ -21,6 +21,7 @@
 
 package org.onap.aai.babel.csar.vnfcatalog;
 
+import static org.hamcrest.CoreMatchers.containsString;
 import static org.hamcrest.CoreMatchers.equalTo;
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.hamcrest.CoreMatchers.is;
@@ -28,6 +29,9 @@ import static org.hamcrest.CoreMatchers.nullValue;
 import static org.junit.Assert.assertThat;
 
 import java.io.IOException;
+import java.util.HashMap;
+import java.util.LinkedHashMap;
+import java.util.List;
 import org.junit.Test;
 import org.onap.aai.babel.service.data.BabelArtifact;
 import org.onap.aai.babel.service.data.BabelArtifact.ArtifactType;
@@ -84,4 +88,53 @@ public class VnfVendorImageExtractorTest {
         assertThat(artifact.getPayload(),
                 is(equalTo(new ArtifactTestUtils().getRequestJson("vnfVendorImageConfigurations.json"))));
     }
+
+    @Test
+    public void testSoftwareVersions() throws ToscaToCatalogException {
+        VnfVendorImageExtractor extractor = new VnfVendorImageExtractor();
+        SdcToscaHelper helper = new SdcToscaHelper();
+
+        List<String> versions;
+        try {
+            versions = extractor.extractSoftwareVersions(helper.buildMappings());
+            assertThat(versions.size(), is(0));
+        } catch (ToscaToCatalogException e) {
+            assertThat(e.getMessage(), containsString("No software versions"));
+        }
+
+        helper.addNodeTemplate();
+        try {
+            versions = extractor.extractSoftwareVersions(helper.buildMappings());
+            assertThat(versions.size(), is(0));
+        } catch (ToscaToCatalogException e) {
+            assertThat(e.getMessage(), containsString("No software versions"));
+        }
+
+        helper.addNodeTemplate("string");
+        try {
+            versions = extractor.extractSoftwareVersions(helper.buildMappings());
+            assertThat(versions.size(), is(0));
+        } catch (ClassCastException e) {
+            assertThat(e.getMessage(), containsString("java.lang.String"));
+        }
+
+        HashMap<String, Object> images = new LinkedHashMap<>();
+        images.put("image", "string");
+        helper.addNodeTemplate(images);
+        try {
+            versions = extractor.extractSoftwareVersions(helper.buildMappings());
+            assertThat(versions.size(), is(1));
+        } catch (ClassCastException e) {
+            assertThat(e.getMessage(), containsString("java.lang.String"));
+        }
+
+        HashMap<String, Object> image = new LinkedHashMap<>();
+        image.put("software_version", "1.2.3");
+        images.put("image", image);
+        helper = new SdcToscaHelper();
+        helper.addNodeTemplate(images);
+        versions = extractor.extractSoftwareVersions(helper.buildMappings());
+        assertThat(versions.size(), is(1));
+        assertThat(versions.get(0), is("1.2.3"));
+    }
 }