services/keep-balance/main_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package main
   6
   7 import (
   8         "bytes"
   9         "io/ioutil"
  10         "net"
  11         "net/http"
  12         "time"
  13
  14         check "gopkg.in/check.v1"
  15 )
  16
  17 var _ = check.Suite(&mainSuite{})
  18
  19 type mainSuite struct{}
  20
  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())
  26 }
  27
  28 func (s *mainSuite) TestHTTPServer(c *check.C) {
  29         ln, err := net.Listen("tcp", ":0")
  30         if err != nil {
  31                 c.Fatal(err)
  32         }
  33         _, p, err := net.SplitHostPort(ln.Addr().String())
  34         ln.Close()
  35         config := "Clusters:\n zzzzz:\n  ManagementToken: abcdefg\n  Services: {Keepbalance: {InternalURLs: {'http://localhost:" + p + "/': {}}}}\n"
  36
  37         var stdout bytes.Buffer
  38         go runCommand("keep-balance", []string{"-config", "-"}, bytes.NewBufferString(config), &stdout, &stdout)
  39         done := make(chan struct{})
  40         go func() {
  41                 defer close(done)
  42                 for {
  43                         time.Sleep(time.Second / 10)
  44                         req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/metrics", nil)
  45                         if err != nil {
  46                                 c.Fatal(err)
  47                                 return
  48                         }
  49                         req.Header.Set("Authorization", "Bearer abcdefg")
  50                         resp, err := http.DefaultClient.Do(req)
  51                         if err != nil {
  52                                 c.Logf("error %s", err)
  53                                 continue
  54                         }
  55                         defer resp.Body.Close()
  56                         if resp.StatusCode != http.StatusOK {
  57                                 c.Logf("http status %d", resp.StatusCode)
  58                                 continue
  59                         }
  60                         buf, err := ioutil.ReadAll(resp.Body)
  61                         if err != nil {
  62                                 c.Logf("read body: %s", err)
  63                                 continue
  64                         }
  65                         c.Check(string(buf), check.Matches, `(?ms).*arvados_keepbalance_sweep_seconds_sum.*`)
  66                         return
  67                 }
  68         }()
  69         select {
  70         case <-done:
  71         case <-time.After(time.Second):
  72                 c.Log(stdout.String())
  73                 c.Fatal("timeout")
  74         }
  75
  76         // Check non-metrics URL that gets passed through to us from
  77         // service.Command
  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)
  84 }