1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
14 "git.arvados.org/arvados.git/lib/config"
15 "git.arvados.org/arvados.git/sdk/go/arvados"
16 "git.arvados.org/arvados.git/sdk/go/arvadostest"
17 "git.arvados.org/arvados.git/sdk/go/ctxlog"
18 "github.com/ghodss/yaml"
19 check "gopkg.in/check.v1"
22 var _ = check.Suite(&mainSuite{})
24 type mainSuite struct{}
26 func (s *mainSuite) TestVersionFlag(c *check.C) {
27 var stdout, stderr bytes.Buffer
28 runCommand("keep-balance", []string{"-version"}, nil, &stdout, &stderr)
29 c.Check(stderr.String(), check.Equals, "")
30 c.Log(stdout.String())
33 func (s *mainSuite) TestHTTPServer(c *check.C) {
34 arvadostest.StartKeep(2, true)
36 ln, err := net.Listen("tcp", ":0")
40 _, p, err := net.SplitHostPort(ln.Addr().String())
41 c.Check(err, check.IsNil)
43 cfg, err := config.NewLoader(nil, ctxlog.TestLogger(c)).Load()
44 c.Assert(err, check.IsNil)
45 cluster, err := cfg.GetCluster("")
46 c.Assert(err, check.IsNil)
47 cluster.Services.Keepbalance.InternalURLs[arvados.URL{Host: "localhost:" + p, Path: "/"}] = arvados.ServiceInstance{}
48 cfg.Clusters[cluster.ClusterID] = *cluster
49 config, err := yaml.Marshal(cfg)
50 c.Assert(err, check.IsNil)
52 var stdout bytes.Buffer
53 go runCommand("keep-balance", []string{"-config", "-"}, bytes.NewBuffer(config), &stdout, &stdout)
54 done := make(chan struct{})
58 time.Sleep(time.Second / 10)
59 req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/metrics", nil)
64 req.Header.Set("Authorization", "Bearer "+cluster.ManagementToken)
65 resp, err := http.DefaultClient.Do(req)
67 c.Logf("error %s", err)
70 defer resp.Body.Close()
71 if resp.StatusCode != http.StatusOK {
72 c.Logf("http status %d", resp.StatusCode)
75 buf, err := ioutil.ReadAll(resp.Body)
77 c.Logf("read body: %s", err)
80 c.Check(string(buf), check.Matches, `(?ms).*arvados_keepbalance_sweep_seconds_sum.*`)
86 case <-time.After(time.Second):
87 c.Log(stdout.String())
90 c.Log(stdout.String())
92 // Check non-metrics URL that gets passed through to us from
94 req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/not-metrics", nil)
95 c.Assert(err, check.IsNil)
96 resp, err := http.DefaultClient.Do(req)
97 c.Check(err, check.IsNil)
98 defer resp.Body.Close()
99 c.Check(resp.StatusCode, check.Equals, http.StatusNotFound)