1 files changed, 48 insertions, 0 deletions
diff --git a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/DefaultEnvRewriterTest.java b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/DefaultEnvRewriterTest.java
new file mode 100644
index 00000000000..7b8ece2c6c2
--- /dev/null
+++ b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/DefaultEnvRewriterTest.java
@@ -0,0 +1,48 @@
+// Copyright 2019 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.hosted.node.admin.task.util;
+
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+import static java.nio.file.StandardCopyOption.REPLACE_EXISTING;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author bjorncs
+ */
+public class DefaultEnvRewriterTest {
+
+    @Rule
+    public TemporaryFolder temporaryFolder = new TemporaryFolder();
+
+    private static final Path EXAMPLE_FILE = Paths.get(DefaultEnvRewriterTest.class.getResource("/default-env-example.txt").getFile());
+    private static final Path EXPECTED_RESULT_FILE = Paths.get(DefaultEnvRewriterTest.class.getResource("/default-env-rewritten.txt").getFile());
+
+    @Test
+    public void default_env_is_correctly_rewritten() throws IOException {
+        Path tempFile = temporaryFolder.newFile().toPath();
+        Files.copy(EXAMPLE_FILE, tempFile, REPLACE_EXISTING);
+
+        DefaultEnvRewriter rewriter = new DefaultEnvRewriter(tempFile);
+        rewriter.addOverride("VESPA_HOSTNAME", "my-new-hostname");
+        rewriter.addFallback("VESPA_CONFIGSERVER", "new-fallback-configserver");
+        rewriter.addOverride("VESPA_TLS_CONFIG_FILE", "/override/path/to/config.file");
+
+        boolean converged = rewriter.converge();
+
+        assertTrue(converged);
+        assertEquals(Files.readString(EXPECTED_RESULT_FILE), Files.readString(tempFile));
+
+        converged = rewriter.converge();
+        assertFalse(converged);
+        assertEquals(Files.readString(EXPECTED_RESULT_FILE), Files.readString(tempFile));
+    }
+}
+\ No newline at end of file