lib/config/cmd_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package config
   6
   7 import (
   8         "bytes"
   9
  10         "git.curoverse.com/arvados.git/lib/cmd"
  11         check "gopkg.in/check.v1"
  12 )
  13
  14 var _ = check.Suite(&CommandSuite{})
  15
  16 var (
  17         // Commands must satisfy cmd.Handler interface
  18         _ cmd.Handler = dumpCommand{}
  19         _ cmd.Handler = checkCommand{}
  20 )
  21
  22 type CommandSuite struct{}
  23
  24 func (s *CommandSuite) TestBadArg(c *check.C) {
  25         var stderr bytes.Buffer
  26         code := DumpCommand.RunCommand("arvados config-dump", []string{"-badarg"}, bytes.NewBuffer(nil), bytes.NewBuffer(nil), &stderr)
  27         c.Check(code, check.Equals, 2)
  28         c.Check(stderr.String(), check.Matches, `(?ms)flag provided but not defined: -badarg\nUsage:\n.*`)
  29 }
  30
  31 func (s *CommandSuite) TestEmptyInput(c *check.C) {
  32         var stdout, stderr bytes.Buffer
  33         code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, &bytes.Buffer{}, &stdout, &stderr)
  34         c.Check(code, check.Equals, 1)
  35         c.Check(stderr.String(), check.Matches, `config does not define any clusters\n`)
  36 }
  37
  38 func (s *CommandSuite) TestCheckNoDeprecatedKeys(c *check.C) {
  39         var stdout, stderr bytes.Buffer
  40         in := `
  41 Clusters:
  42  z1234:
  43   API:
  44     MaxItemsPerResponse: 1234
  45 `
  46         code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
  47         c.Check(code, check.Equals, 0)
  48         c.Check(stdout.String(), check.Equals, "")
  49         c.Check(stderr.String(), check.Equals, "")
  50 }
  51
  52 func (s *CommandSuite) TestCheckDeprecatedKeys(c *check.C) {
  53         var stdout, stderr bytes.Buffer
  54         in := `
  55 Clusters:
  56  z1234:
  57   RequestLimits:
  58     MaxItemsPerResponse: 1234
  59 `
  60         code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
  61         c.Check(code, check.Equals, 1)
  62         c.Check(stdout.String(), check.Matches, `(?ms).*\n\- +.*MaxItemsPerResponse: 1000\n\+ +MaxItemsPerResponse: 1234\n.*`)
  63 }
  64
  65 func (s *CommandSuite) TestCheckUnknownKey(c *check.C) {
  66         var stdout, stderr bytes.Buffer
  67         in := `
  68 Clusters:
  69  z1234:
  70   Bogus1: foo
  71   BogusSection:
  72     Bogus2: foo
  73   API:
  74     Bogus3:
  75      Bogus4: true
  76   PostgreSQL:
  77     ConnectionPool:
  78       {Bogus5: true}
  79 `
  80         code := CheckCommand.RunCommand("arvados config-check", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
  81         c.Log(stderr.String())
  82         c.Check(code, check.Equals, 1)
  83         c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.Bogus1\n.*`)
  84         c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.BogusSection\n.*`)
  85         c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.API.Bogus3\n.*`)
  86         c.Check(stderr.String(), check.Matches, `(?ms).*unexpected object in config entry: Clusters.z1234.PostgreSQL.ConnectionPool\n.*`)
  87 }
  88
  89 func (s *CommandSuite) TestDumpFormatting(c *check.C) {
  90         var stdout, stderr bytes.Buffer
  91         in := `
  92 Clusters:
  93  z1234:
  94   Containers:
  95    CloudVMs:
  96     TimeoutBooting: 600s
  97   Services:
  98    Controller:
  99     InternalURLs:
 100      http://localhost:12345: {}
 101 `
 102         code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
 103         c.Check(code, check.Equals, 0)
 104         c.Check(stdout.String(), check.Matches, `(?ms).*TimeoutBooting: 10m\n.*`)
 105         c.Check(stdout.String(), check.Matches, `(?ms).*http://localhost:12345: {}\n.*`)
 106 }
 107
 108 func (s *CommandSuite) TestDumpUnknownKey(c *check.C) {
 109         var stdout, stderr bytes.Buffer
 110         in := `
 111 Clusters:
 112  z1234:
 113   UnknownKey: foobar
 114   ManagementToken: secret
 115 `
 116         code := DumpCommand.RunCommand("arvados config-dump", []string{"-config", "-"}, bytes.NewBufferString(in), &stdout, &stderr)
 117         c.Check(code, check.Equals, 0)
 118         c.Check(stderr.String(), check.Matches, `(?ms).*deprecated or unknown config entry: Clusters.z1234.UnknownKey.*`)
 119         c.Check(stdout.String(), check.Matches, `(?ms)Clusters:\n  z1234:\n.*`)
 120         c.Check(stdout.String(), check.Matches, `(?ms).*\n *ManagementToken: secret\n.*`)
 121         c.Check(stdout.String(), check.Not(check.Matches), `(?ms).*UnknownKey.*`)
 122 }