1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
13 "git.curoverse.com/arvados.git/lib/cmd"
14 check "gopkg.in/check.v1"
17 var _ = check.Suite(&CommandSuite{})
20 // Commands must satisfy cmd.Handler interface
21 _ cmd.Handler = dumpCommand{}
22 _ cmd.Handler = checkCommand{}
25 type CommandSuite struct{}
27 func (s *CommandSuite) TestBadArg(c *check.C) {
28 var stderr bytes.Buffer
29 code := DumpCommand.RunCommand("arvados config-dump", []string{"-badarg"}, bytes.NewBuffer(nil), bytes.NewBuffer(nil), &stderr)
30 c.Check(code, check.Equals, 2)
31 c.Check(stderr.String(), check.Matches, `(?ms)flag provided but not defined: -badarg\nUsage:\n.*`)
34 func (s *CommandSuite) TestEmptyInput(c *check.C) {
35 var stdout, stderr bytes.Buffer
36 code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, &bytes.Buffer{}, &stdout, &stderr)
37 c.Check(code, check.Equals, 1)
38 c.Check(stderr.String(), check.Matches, `config does not define any clusters\n`)
41 func (s *CommandSuite) TestCheckNoDeprecatedKeys(c *check.C) {
42 var stdout, stderr bytes.Buffer
47 MaxItemsPerResponse: 1234
49 code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
50 c.Check(code, check.Equals, 0)
51 c.Check(stdout.String(), check.Equals, "")
52 c.Check(stderr.String(), check.Equals, "")
55 func (s *CommandSuite) TestCheckDeprecatedKeys(c *check.C) {
56 var stdout, stderr bytes.Buffer
61 MaxItemsPerResponse: 1234
63 code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
64 c.Check(code, check.Equals, 1)
65 c.Check(stdout.String(), check.Matches, `(?ms).*\n\- +.*MaxItemsPerResponse: 1000\n\+ +MaxItemsPerResponse: 1234\n.*`)
68 func (s *CommandSuite) TestCheckOldKeepstoreConfigFile(c *check.C) {
69 f, err := ioutil.TempFile("", "")
70 c.Assert(err, check.IsNil)
71 defer os.Remove(f.Name())
73 io.WriteString(f, "Debug: true\n")
75 var stdout, stderr bytes.Buffer
82 code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-", "-legacy-keepstore-config", f.Name()}, bytes.NewBufferString(in), &stdout, &stderr)
83 c.Check(code, check.Equals, 1)
84 c.Check(stdout.String(), check.Matches, `(?ms).*\n\- +.*LogLevel: info\n\+ +LogLevel: debug\n.*`)
85 c.Check(stderr.String(), check.Matches, `.*you should remove the legacy keepstore config file.*\n`)
88 func (s *CommandSuite) TestCheckUnknownKey(c *check.C) {
89 var stdout, stderr bytes.Buffer
103 code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
104 c.Log(stderr.String())
105 c.Check(code, check.Equals, 1)
106 c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.Bogus1"\n.*`)
107 c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.BogusSection"\n.*`)
108 c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.API.Bogus3"\n.*`)
109 c.Check(stderr.String(), check.Matches, `(?ms).*unexpected object in config entry: Clusters.z1234.PostgreSQL.ConnectionPool"\n.*`)
112 func (s *CommandSuite) TestDumpFormatting(c *check.C) {
113 var stdout, stderr bytes.Buffer
123 http://localhost:12345: {}
125 code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
126 c.Check(code, check.Equals, 0)
127 c.Check(stdout.String(), check.Matches, `(?ms).*TimeoutBooting: 10m\n.*`)
128 c.Check(stdout.String(), check.Matches, `(?ms).*http://localhost:12345: {}\n.*`)
131 func (s *CommandSuite) TestDumpUnknownKey(c *check.C) {
132 var stdout, stderr bytes.Buffer
137 ManagementToken: secret
139 code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
140 c.Check(code, check.Equals, 0)
141 c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.UnknownKey.*`)
142 c.Check(stdout.String(), check.Matches, `(?ms)Clusters:\n z1234:\n.*`)
143 c.Check(stdout.String(), check.Matches, `(?ms).*\n *ManagementToken: secret\n.*`)
144 c.Check(stdout.String(), check.Not(check.Matches), `(?ms).*UnknownKey.*`)