1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
14 check "gopkg.in/check.v1"
17 var _ = check.Suite(&mainSuite{})
19 type mainSuite struct{}
21 func (s *mainSuite) TestVersionFlag(c *check.C) {
22 var stdout, stderr bytes.Buffer
23 runCommand("keep-balance", []string{"-version"}, nil, &stdout, &stderr)
24 c.Check(stderr.String(), check.Equals, "")
25 c.Log(stdout.String())
28 func (s *mainSuite) TestHTTPServer(c *check.C) {
29 ln, err := net.Listen("tcp", ":0")
33 _, p, err := net.SplitHostPort(ln.Addr().String())
35 config := "Clusters:\n zzzzz:\n ManagementToken: abcdefg\n Services: {Keepbalance: {InternalURLs: {'http://localhost:" + p + "/': {}}}}\n"
37 var stdout bytes.Buffer
38 go runCommand("keep-balance", []string{"-config", "-"}, bytes.NewBufferString(config), &stdout, &stdout)
39 done := make(chan struct{})
43 time.Sleep(time.Second / 10)
44 req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/metrics", nil)
49 req.Header.Set("Authorization", "Bearer abcdefg")
50 resp, err := http.DefaultClient.Do(req)
52 c.Logf("error %s", err)
55 defer resp.Body.Close()
56 if resp.StatusCode != http.StatusOK {
57 c.Logf("http status %d", resp.StatusCode)
60 buf, err := ioutil.ReadAll(resp.Body)
62 c.Logf("read body: %s", err)
65 c.Check(string(buf), check.Matches, `(?ms).*arvados_keepbalance_sweep_seconds_sum.*`)
71 case <-time.After(time.Second):
72 c.Log(stdout.String())
76 // Check non-metrics URL that gets passed through to us from
78 req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/not-metrics", nil)
79 c.Assert(err, check.IsNil)
80 resp, err := http.DefaultClient.Do(req)
81 c.Check(err, check.IsNil)
82 defer resp.Body.Close()
83 c.Check(resp.StatusCode, check.Equals, http.StatusNotFound)