Incorporate the ECOMP SDC Artefact Generator code
[aai/babel.git] / src / test / java / org / onap / aai / babel / service / CsarToXmlConverterTest.java
index 5c1e213..c8080e9 100644 (file)
@@ -2,8 +2,8 @@
  * ============LICENSE_START=======================================================
  * org.onap.aai
  * ================================================================================
- * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
- * Copyright © 2017 European Software Marketing Ltd.
+ * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
+ * Copyright © 2017-2018 European Software Marketing Ltd.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * 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.babel.service;
 
-import java.io.BufferedReader;
-import java.io.ByteArrayInputStream;
+import static org.junit.Assert.assertThat;
+
+import java.io.File;
 import java.io.IOException;
-import java.io.InputStreamReader;
-import java.net.URL;
-import java.util.Base64;
+import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.stream.Collectors;
-import org.custommonkey.xmlunit.XMLTestCase;
+import org.hamcrest.BaseMatcher;
+import org.hamcrest.Description;
+import org.hamcrest.Matcher;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.rules.ExpectedException;
-import org.junit.runner.RunWith;
 import org.onap.aai.babel.csar.CsarConverterException;
 import org.onap.aai.babel.csar.CsarToXmlConverter;
 import org.onap.aai.babel.service.data.BabelArtifact;
 import org.onap.aai.babel.util.ArtifactTestUtils;
-import org.onap.aai.babel.xml.generator.ModelGenerator;
 import org.onap.aai.babel.xml.generator.XmlArtifactGenerationException;
-import org.powermock.core.classloader.annotations.PrepareForTest;
-import org.powermock.modules.junit4.PowerMockRunner;
-import org.xml.sax.SAXException;
+import org.onap.aai.babel.xml.generator.data.GeneratorConstants;
 
 /**
  * Tests {@link CsarToXmlConverter}
  */
-@RunWith(PowerMockRunner.class)
-@PrepareForTest(ModelGenerator.class)
-public class CsarToXmlConverterTest extends XMLTestCase {
+public class CsarToXmlConverterTest {
 
-    private static final String NAME = "the_name_of_the_csar_file.csar";
-    private static final String VERSION = "v1";
+    private static final String ARTIFACT_GENERATOR_CONFIG = "artifact-generator.properties";
+    private static final String CSAR_FOLDER = "compressedArtifacts";
+    private static final String VALID_CSAR_FILE = "service-SdWanServiceTest-csar.csar";
+    private static final String INCORRECT_CSAR_NAME = "the_name_of_the_csar_file.csar";
+    private static final String SERVICE_VERSION = "1.0";
 
     private CsarToXmlConverter converter;
 
+    static {
+        if (System.getProperty("AJSC_HOME") == null) {
+            System.setProperty("AJSC_HOME", ".");
+        }
+    }
+
     @Rule
     public ExpectedException exception = ExpectedException.none();
+    private ArtifactTestUtils artifactTestUtils;
 
     @Before
-    public void setUp() {
+    public void setup() {
+        System.setProperty(GeneratorConstants.PROPERTY_ARTIFACT_GENERATOR_CONFIG_FILE,
+                CsarToXmlConverterTest.class.getClassLoader().getResource(ARTIFACT_GENERATOR_CONFIG).getPath());
         converter = new CsarToXmlConverter();
-        URL url = CsarToXmlConverterTest.class.getClassLoader().getResource("artifact-generator.properties");
-        System.setProperty("artifactgenerator.config", url.getPath());
+        artifactTestUtils = new ArtifactTestUtils();
     }
 
     @After
@@ -83,25 +86,22 @@ public class CsarToXmlConverterTest extends XMLTestCase {
 
     @Test(expected = NullPointerException.class)
     public void generateXmlFromCsar_missingName() throws CsarConverterException, IOException {
-        byte[] csarArchive = new ArtifactTestUtils().loadResource("compressedArtifacts/service-VscpaasTest-csar.csar");
-        converter.generateXmlFromCsar(csarArchive, null, null);
+        converter.generateXmlFromCsar(getCsar(VALID_CSAR_FILE), null, null);
     }
 
     @Test(expected = NullPointerException.class)
     public void generateXmlFromCsar_missingVersion() throws CsarConverterException, IOException {
-        byte[] csarArchive = new ArtifactTestUtils().loadResource("compressedArtifacts/service-VscpaasTest-csar.csar");
-        converter.generateXmlFromCsar(csarArchive, NAME, null);
+        converter.generateXmlFromCsar(getCsar(VALID_CSAR_FILE), INCORRECT_CSAR_NAME, null);
     }
 
     @Test(expected = CsarConverterException.class)
     public void generateXmlFromCsar_noPayloadExists() throws CsarConverterException {
-        converter.generateXmlFromCsar(new byte[0], NAME, VERSION);
+        converter.generateXmlFromCsar(new byte[0], INCORRECT_CSAR_NAME, SERVICE_VERSION);
     }
 
     @Test(expected = CsarConverterException.class)
     public void generateXmlFromCsar_csarFileHasNoYmlFiles() throws CsarConverterException, IOException {
-        byte[] csarArchive = new ArtifactTestUtils().loadResource("compressedArtifacts/noYmlFilesArchive.zip");
-        converter.generateXmlFromCsar(csarArchive, "noYmlFilesArchive.zip", VERSION);
+        converter.generateXmlFromCsar(getCsar("noYmlFilesArchive.zip"), "noYmlFilesArchive.zip", SERVICE_VERSION);
     }
 
     @Test
@@ -110,61 +110,55 @@ public class CsarToXmlConverterTest extends XMLTestCase {
         exception.expect(CsarConverterException.class);
         exception.expectMessage("Cannot generate artifacts. artifactgenerator.config system property not configured");
 
-        byte[] csarArchive =
-                new ArtifactTestUtils().loadResource("compressedArtifacts/service-SdWanServiceTest-csar.csar");
-
         // Unset the required system property
-        System.clearProperty("artifactgenerator.config");
-        converter.generateXmlFromCsar(csarArchive, VERSION, "service-SdWanServiceTest-csar.csar");
+        System.clearProperty(GeneratorConstants.PROPERTY_ARTIFACT_GENERATOR_CONFIG_FILE);
+        converter.generateXmlFromCsar(getCsar(VALID_CSAR_FILE), VALID_CSAR_FILE, SERVICE_VERSION);
     }
 
     @Test
     public void generateXmlFromCsar() throws CsarConverterException, IOException, XmlArtifactGenerationException {
-        byte[] csarArchive =
-                new ArtifactTestUtils().loadResource("compressedArtifacts/service-SdWanServiceTest-csar.csar");
-
         Map<String, String> expectedXmlFiles = createExpectedXmlFiles();
         List<BabelArtifact> generatedArtifacts =
-                converter.generateXmlFromCsar(csarArchive, VERSION, "service-SdWanServiceTest-csar.csar");
-
-        generatedArtifacts.forEach(ga -> {
-            try {
+                converter.generateXmlFromCsar(getCsar(VALID_CSAR_FILE), VALID_CSAR_FILE, SERVICE_VERSION);
 
-                String x1 = expectedXmlFiles.get(ga.getName());
+        generatedArtifacts
+                .forEach(ga -> assertThat("The content of " + ga.getName() + " must match the expected content",
+                        ga.getPayload(), matches(expectedXmlFiles.get(ga.getName()))));
+    }
 
-                String x2 = bytesToString(ga.getPayload());
+    public Matcher<String> matches(final String expected) {
+        return new BaseMatcher<String>() {
+            protected String theExpected = expected;
 
-                assertXMLEqual("The content of " + ga.getName() + " must match the expected content", x1, x2);
+            @Override
+            public boolean matches(Object o) {
+                return artifactTestUtils.compareXMLStrings((String) o, theExpected);
+            }
 
-            } catch (SAXException | IOException e) {
-                fail("There was an Exception parsing the XML: "+e.getMessage());
+            @Override
+            public void describeTo(Description description) {
+                description.appendText(theExpected.toString());
             }
-        });
+        };
     }
 
-    public String bytesToString(byte[] source) {
-        ByteArrayInputStream bis = new ByteArrayInputStream(Base64.getDecoder().decode(source));
-
-        String result = new BufferedReader(new InputStreamReader(bis)).lines().collect(Collectors.joining("\n"));
-
-        return result;
-
+    private byte[] getCsar(String csarFileName) throws IOException {
+        return artifactTestUtils.loadResource(CSAR_FOLDER + File.separator + csarFileName);
     }
 
     private Map<String, String> createExpectedXmlFiles() throws IOException {
-        Map<String, String> xml = new HashMap<>();
-
-        ArtifactTestUtils utils = new ArtifactTestUtils();
-
-        String[] filesToLoad =
-                {"AAI-SD-WAN-Service-Test-service-1.0.xml", "AAI-SdWanTestVsp..DUMMY..module-0-resource-2.xml",
-                        "AAI-Tunnel_XConnTest-resource-2.0.xml", "AAI-SD-WAN-Test-VSP-resource-1.0.xml"};
+        Map<String, String> xmlMap = new HashMap<>();
 
-        for (String s : filesToLoad) {
-            xml.put(s, utils.loadResourceAsString("generatedXml/" + s));
+        List<String> filesToLoad = new ArrayList<>();
+        filesToLoad.add("AAI-SD-WAN-Service-Test-service-1.0.xml");
+        filesToLoad.add("AAI-SdWanTestVsp..DUMMY..module-0-resource-2.xml");
+        filesToLoad.add("AAI-Tunnel_XConnTest-resource-2.0.xml");
+        filesToLoad.add("AAI-SD-WAN-Test-VSP-resource-1.0.xml");
 
+        for (String filename : filesToLoad) {
+            xmlMap.put(filename, artifactTestUtils.loadResourceAsString("generatedXml" + File.separator + filename));
         }
 
-        return xml;
+        return xmlMap;
     }
 }