diff options
Diffstat (limited to 'node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/env/DefaultEnvRewriterTest.java')
-rw-r--r-- | node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/env/DefaultEnvRewriterTest.java | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/env/DefaultEnvRewriterTest.java b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/env/DefaultEnvRewriterTest.java new file mode 100644 index 00000000000..2f24af37b71 --- /dev/null +++ b/node-admin/src/test/java/com/yahoo/vespa/hosted/node/admin/task/util/env/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.env; + +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 |