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
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package cmd
import (
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"github.com/vespa-engine/vespa/client/go/vespa"
)
func TestConfig(t *testing.T) {
assertConfigCommandErr(t, "Error: invalid option or value: \"foo\": \"bar\"\n", "config", "set", "foo", "bar")
assertConfigCommand(t, "foo = <unset>\n", "config", "get", "foo")
assertConfigCommand(t, "target = local\n", "config", "get", "target")
assertConfigCommand(t, "", "config", "set", "target", "hosted")
assertConfigCommand(t, "target = hosted\n", "config", "get", "target")
assertConfigCommand(t, "", "config", "set", "target", "cloud")
assertConfigCommand(t, "target = cloud\n", "config", "get", "target")
assertConfigCommand(t, "", "config", "set", "target", "http://127.0.0.1:8080")
assertConfigCommand(t, "", "config", "set", "target", "https://127.0.0.1")
assertConfigCommand(t, "target = https://127.0.0.1\n", "config", "get", "target")
assertEnvConfigCommand(t, "api-key-file = /tmp/private.key\n", []string{"VESPA_CLI_API_KEY_FILE=/tmp/private.key"}, "config", "get", "api-key-file")
assertConfigCommand(t, "", "config", "set", "api-key-file", "/tmp/private.key")
assertConfigCommand(t, "api-key-file = /tmp/private.key\n", "config", "get", "api-key-file")
assertConfigCommandErr(t, "Error: invalid application: \"foo\"\n", "config", "set", "application", "foo")
assertConfigCommand(t, "application = <unset>\n", "config", "get", "application")
assertConfigCommand(t, "", "config", "set", "application", "t1.a1.i1")
assertConfigCommand(t, "application = t1.a1.i1\n", "config", "get", "application")
assertConfigCommand(t, "api-key-file = /tmp/private.key\napplication = t1.a1.i1\ncolor = auto\nquiet = false\ntarget = https://127.0.0.1\nwait = 0\n", "config", "get")
assertConfigCommand(t, "", "config", "set", "wait", "60")
assertConfigCommandErr(t, "Error: wait option must be an integer >= 0, got \"foo\"\n", "config", "set", "wait", "foo")
assertConfigCommand(t, "wait = 60\n", "config", "get", "wait")
assertConfigCommand(t, "", "config", "set", "quiet", "true")
assertConfigCommand(t, "", "config", "set", "quiet", "false")
}
func assertConfigCommand(t *testing.T, expected string, args ...string) {
assertEnvConfigCommand(t, expected, nil, args...)
}
func assertEnvConfigCommand(t *testing.T, expected string, env []string, args ...string) {
cli, stdout, _ := newTestCLI(t, env...)
err := cli.Run(args...)
assert.Nil(t, err)
assert.Equal(t, expected, stdout.String())
}
func assertConfigCommandErr(t *testing.T, expected string, args ...string) {
cli, _, stderr := newTestCLI(t)
err := cli.Run(args...)
assert.NotNil(t, err)
assert.Equal(t, expected, stderr.String())
}
func TestUseAPIKey(t *testing.T) {
cli, _, _ := newTestCLI(t)
assert.False(t, cli.config.useAPIKey(cli, vespa.PublicSystem, "t1"))
cli.config.set(apiKeyFileFlag, "/tmp/foo")
assert.True(t, cli.config.useAPIKey(cli, vespa.PublicSystem, "t1"))
cli.config.set(apiKeyFileFlag, "")
cli, _, _ = newTestCLI(t, "VESPA_CLI_API_KEY=foo")
assert.True(t, cli.config.useAPIKey(cli, vespa.PublicSystem, "t1"))
// Test deprecated functionality
authContent := `
{
"version": 1,
"providers": {
"auth0": {
"version": 1,
"systems": {
"public": {
"access_token": "...",
"scopes": ["openid", "offline_access"],
"expires_at": "2030-01-01T01:01:01.000001+01:00"
}
}
}
}
}`
cli, _, _ = newTestCLI(t, "VESPA_CLI_CLOUD_SYSTEM=public")
_, err := os.Create(filepath.Join(cli.config.homeDir, "t2.api-key.pem"))
require.Nil(t, err)
assert.True(t, cli.config.useAPIKey(cli, vespa.PublicSystem, "t2"))
require.Nil(t, ioutil.WriteFile(filepath.Join(cli.config.homeDir, "auth.json"), []byte(authContent), 0600))
assert.False(t, cli.config.useAPIKey(cli, vespa.PublicSystem, "t2"))
}
|
