Publish

1 files changed, 100 insertions, 0 deletions

diff --git a/config-application-package/src/test/java/com/yahoo/config/model/application/provider/FilesApplicationPackageTest.java b/config-application-package/src/test/java/com/yahoo/config/model/application/provider/FilesApplicationPackageTest.java
new file mode 100644
index 00000000000..77842a693b7
--- /dev/null
+++ b/config-application-package/src/test/java/com/yahoo/config/model/application/provider/FilesApplicationPackageTest.java
@@ -0,0 +1,100 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.config.model.application.provider;
+
+import com.yahoo.config.application.TestBase;
+import com.yahoo.config.application.api.RuleConfigDeriver;
+import com.yahoo.config.application.api.ApplicationPackage;
+import com.yahoo.config.provision.Environment;
+import com.yahoo.config.provision.RegionName;
+import com.yahoo.config.provision.Zone;
+import com.yahoo.io.IOUtils;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.xml.sax.SAXException;
+
+import javax.xml.parsers.ParserConfigurationException;
+import javax.xml.transform.TransformerException;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author lulf
+ * @since 5.25
+ */
+public class FilesApplicationPackageTest {
+
+    @Rule
+    public TemporaryFolder temporaryFolder = new TemporaryFolder();
+
+    @Test
+    public void testPreprocessing() throws IOException, TransformerException, ParserConfigurationException, SAXException {
+        File appDir = temporaryFolder.newFolder();
+        IOUtils.copyDirectory(new File("src/test/resources/multienvapp"), appDir);
+        assertTrue(new File(appDir, "services.xml").exists());
+        assertTrue(new File(appDir, "hosts.xml").exists());
+        FilesApplicationPackage app = FilesApplicationPackage.fromFile(appDir);
+
+        ApplicationPackage processed = app.preprocess(new Zone(Environment.dev, RegionName.defaultName()),
+                new RuleConfigDeriver() {
+                    @Override
+                    public void derive(String ruleBaseDir, String outputDir) throws Exception {
+                    }
+                },
+                new BaseDeployLogger());
+        assertTrue(new File(appDir, ".preprocessed").exists());
+        String expectedServices = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><services xmlns:deploy=\"vespa\" xmlns:preprocess=\"properties\" version=\"1.0\">\n" +
+                "    <admin version=\"2.0\">\n" +
+                "        <adminserver hostalias=\"node0\"/>\n" +
+                "    </admin>\n" +
+                "    <content id=\"foo\" version=\"1.0\">\n" +
+                "      <redundancy>1</redundancy>\n" +
+                "      <documents>\n" +
+                "        <document mode=\"index\" type=\"music.sd\"/>\n" +
+                "      </documents>\n" +
+                "      <nodes>\n" +
+                "        <node distribution-key=\"0\" hostalias=\"node0\"/>\n" +
+                "      </nodes>\n" +
+                "    </content>\n" +
+                "    <jdisc id=\"stateless\" version=\"1.0\">\n" +
+                "      <search/>\n" +
+                "      <component bundle=\"foobundle\" class=\"MyFoo\" id=\"foo\"/>\n" +
+                "      <component bundle=\"foobundle\" class=\"TestBar\" id=\"bar\"/>\n" +
+                "      <nodes>\n" +
+                "        <node hostalias=\"node0\" baseport=\"5000\"/>\n" +
+                "      </nodes>\n" +
+                "    </jdisc>\n" +
+                "</services>";
+        TestBase.assertDocument(expectedServices, processed.getServices());
+        String expectedHosts = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?><hosts xmlns:deploy=\"vespa\" xmlns:preprocess=\"properties\">\n" +
+                "    <host name=\"bar.yahoo.com\">\n" +
+                "        <alias>node1</alias>\n" +
+                "    </host>\n" +
+                "</hosts>";
+        TestBase.assertDocument(expectedHosts, processed.getHosts());
+    }
+
+    @Test
+    public void testDeploymentXmlNotAvailable() throws IOException, TransformerException, ParserConfigurationException, SAXException {
+        File appDir = new File("src/test/resources/multienvapp");
+        assertFalse(new File(appDir, "deployment.xml").exists());
+        FilesApplicationPackage app = FilesApplicationPackage.fromFile(appDir);
+        assertFalse(app.getDeployment().isPresent());
+    }
+
+    @Test
+    public void testDeploymentXml() throws IOException, TransformerException, ParserConfigurationException, SAXException {
+        File appDir = new File("src/test/resources/app-with-deployment");
+        final File deployment = new File(appDir, "deployment.xml");
+        assertTrue(deployment.exists());
+        FilesApplicationPackage app = FilesApplicationPackage.fromFile(appDir);
+        assertTrue(app.getDeployment().isPresent());
+        assertThat(IOUtils.readAll(new FileReader(deployment)), is(IOUtils.readAll(app.getDeployment().get())));
+    }
+}
\ No newline at end of file