aboutsummaryrefslogtreecommitdiffstats
path: root/vespa-osgi-testrunner/src/test/java/com/yahoo/vespa/testrunner/VespaCliTestRunnerTest.java
blob: a3e6203f6458f6901ea46451f0ecfbbbdff83f88 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.testrunner;

import ai.vespa.hosted.api.TestConfig;
import com.yahoo.config.provision.SystemName;
import org.junit.jupiter.api.Test;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.List;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

/**
 * @author jonmv
 */
class VespaCliTestRunnerTest {

    @Test
    void cloud() throws IOException {
        Path temp = Files.createTempDirectory("vespa-cli-test-runner-test-");
        temp.toFile().deleteOnExit();
        Path tests = Files.createDirectory(temp.resolve("tests"));
        Path artifacts = Files.createDirectory(temp.resolve("artifacts"));
        VespaCliTestRunner runner = new VespaCliTestRunner(artifacts, tests, Files.createDirectory(temp.resolve("vespa")));

        Path systemTests = Files.createDirectory(tests.resolve("system-test"));
        TestConfig testConfig = testConfig(SystemName.PublicCd);
        assertNull(runner.testRunProcessBuilder(TestRunner.Suite.STAGING_TEST, testConfig));

        ProcessBuilder builder = runner.testRunProcessBuilder(TestRunner.Suite.SYSTEM_TEST, testConfig);
        assertEquals(List.of("vespa", "test", systemTests.toAbsolutePath().toString(),
                             "--application", "t.a.i",
                             "--zone", "dev.aws-us-east-1c",
                             "--target", "cloud"),
                     builder.command());
        assertTrue(builder.environment().containsKey("CI"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_CLOUD_CI"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_HOME"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_CACHE_DIR"));
        assertEquals("publiccd", builder.environment().get("VESPA_CLI_CLOUD_SYSTEM"));
        assertEquals("{\"endpoints\":[{\"cluster\":\"default\",\"url\":\"https://dev.endpoint:443/\"}]}",
                     builder.environment().get("VESPA_CLI_ENDPOINTS"));
        assertEquals(artifacts.resolve("key").toAbsolutePath().toString(),
                     builder.environment().get("VESPA_CLI_DATA_PLANE_KEY_FILE"));
        assertEquals(artifacts.resolve("cert").toAbsolutePath().toString(),
                     builder.environment().get("VESPA_CLI_DATA_PLANE_CERT_FILE"));
    }

    @Test
    void hosted() throws IOException {
        Path temp = Files.createTempDirectory("vespa-cli-test-runner-test-");
        temp.toFile().deleteOnExit();
        Path tests = Files.createDirectory(temp.resolve("tests"));
        Path artifacts = Files.createDirectory(temp.resolve("artifacts"));
        Path vespaHome = Files.createDirectory(temp.resolve("vespa"));
        Path keyFile = vespaHome.resolve("var/vespa/sia/keys/my.domain.foo.key.pem");
        Path certFile = vespaHome.resolve("var/vespa/sia/certs/my.domain.foo.cert.pem");
        Files.createDirectories(keyFile.getParent());
        Files.createFile(keyFile);
        VespaCliTestRunner runner = new VespaCliTestRunner(artifacts, tests, vespaHome);

        Path systemTests = Files.createDirectory(tests.resolve("system-test"));
        TestConfig testConfig = testConfig(SystemName.cd);
        assertNull(runner.testRunProcessBuilder(TestRunner.Suite.STAGING_TEST, testConfig));

        ProcessBuilder builder = runner.testRunProcessBuilder(TestRunner.Suite.SYSTEM_TEST, testConfig);
        assertEquals(List.of("vespa", "test", systemTests.toAbsolutePath().toString(),
                             "--application", "t.a.i",
                             "--zone", "dev.aws-us-east-1c",
                             "--target", "hosted"),
                     builder.command());
        assertTrue(builder.environment().containsKey("CI"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_CLOUD_CI"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_HOME"));
        assertTrue(builder.environment().containsKey("VESPA_CLI_CACHE_DIR"));
        assertEquals("cd", builder.environment().get("VESPA_CLI_CLOUD_SYSTEM"));
        assertEquals("{\"endpoints\":[{\"cluster\":\"default\",\"url\":\"https://dev.endpoint:443/\"}]}",
                     builder.environment().get("VESPA_CLI_ENDPOINTS"));
        assertEquals(keyFile.toString(),
                     builder.environment().get("VESPA_CLI_DATA_PLANE_KEY_FILE"));
        assertEquals(certFile.toString(),
                     builder.environment().get("VESPA_CLI_DATA_PLANE_CERT_FILE"));
    }

    private static TestConfig testConfig(SystemName system) {
        return TestConfig.fromJson(("{\n" +
                                    "  \"application\": \"t:a:i\",\n" +
                                    "  \"zone\": \"dev.aws-us-east-1c\",\n" +
                                    "  \"system\": \"" + system.value() + "\",\n" +
                                    "  \"isCI\": true,\n" +
                                    "  \"zoneEndpoints\": {\n" +
                                    "    \"dev.aws-us-east-1c\": {\n" +
                                    "      \"default\": \"https://dev.endpoint:443/\"\n" +
                                    "    },\n" +
                                    "    \"prod.aws-us-east-1a\": {\n" +
                                    "      \"default\": \"https://prod.endpoint:443/\"\n" +
                                    "    }\n" +
                                    "  },\n" +
                                    "  \"clusters\": {\n" +
                                    "    \"prod.aws-us-east-1c\": [\n" +
                                    "      \"documents\"\n" +
                                    "    ]\n" +
                                    "  }\n" +
                                    "}\n").getBytes(StandardCharsets.UTF_8));
    }

}