X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/6c0bf267d795a3ca49c3258c9490714c9e18d333..9171d310942ea3c8e3bed3e21a1c2d0604e93ad6:/services/keep-balance/main_test.go

diff --git a/services/keep-balance/main_test.go b/services/keep-balance/main_test.go
index 157ee4574b..26002eca98 100644
--- a/services/keep-balance/main_test.go
+++ b/services/keep-balance/main_test.go
@@ -1,8 +1,20 @@
-package main
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
+package keepbalance
 
 import (
+	"bytes"
+	"io/ioutil"
+	"net"
+	"net/http"
 	"time"
 
+	"git.arvados.org/arvados.git/lib/config"
+	"git.arvados.org/arvados.git/sdk/go/arvados"
+	"git.arvados.org/arvados.git/sdk/go/arvadostest"
+	"git.arvados.org/arvados.git/sdk/go/ctxlog"
 	"github.com/ghodss/yaml"
 	check "gopkg.in/check.v1"
 )
@@ -11,32 +23,79 @@ var _ = check.Suite(&mainSuite{})
 
 type mainSuite struct{}
 
-func (s *mainSuite) TestExampleJSON(c *check.C) {
-	var config Config
-	c.Check(yaml.Unmarshal(exampleConfigFile, &config), check.IsNil)
-	c.Check(config.KeepServiceTypes, check.DeepEquals, []string{"disk"})
-	c.Check(config.Client.AuthToken, check.Equals, "xyzzy")
-	c.Check(time.Duration(config.RunPeriod), check.Equals, 600*time.Second)
+func (s *mainSuite) TestVersionFlag(c *check.C) {
+	var stdout, stderr bytes.Buffer
+	Command.RunCommand("keep-balance", []string{"-version"}, nil, &stdout, &stderr)
+	c.Check(stderr.String(), check.Equals, "")
+	c.Log(stdout.String())
+	c.Check(stdout.String(), check.Matches, `keep-balance.*\(go1.*\)\n`)
 }
 
-func (s *mainSuite) TestConfigJSONWithKeepServiceList(c *check.C) {
-	var config Config
-	c.Check(yaml.Unmarshal([]byte(`{
-		    "Client": {
-			"APIHost": "zzzzz.arvadosapi.com:443",
-			"AuthToken": "xyzzy",
-			"Insecure": false
-		    },
-		    "KeepServiceList": {
-			"items": [
-			    {"uuid":"zzzzz-bi64l-abcdefghijklmno", "service_type":"disk", "service_host":"a.zzzzz.arvadosapi.com", "service_port":12345},
-			    {"uuid":"zzzzz-bi64l-bcdefghijklmnop", "service_type":"blob", "service_host":"b.zzzzz.arvadosapi.com", "service_port":12345}
-			]
-		    },
-		    "RunPeriod": "600s"
-		}`), &config), check.IsNil)
-	c.Assert(len(config.KeepServiceList.Items), check.Equals, 2)
-	c.Check(config.KeepServiceList.Items[0].UUID, check.Equals, "zzzzz-bi64l-abcdefghijklmno")
-	c.Check(config.KeepServiceList.Items[0].ServicePort, check.Equals, 12345)
-	c.Check(config.Client.AuthToken, check.Equals, "xyzzy")
+func (s *mainSuite) TestHTTPServer(c *check.C) {
+	arvadostest.StartKeep(2, true)
+
+	ln, err := net.Listen("tcp", ":0")
+	if err != nil {
+		c.Fatal(err)
+	}
+	_, p, err := net.SplitHostPort(ln.Addr().String())
+	c.Check(err, check.IsNil)
+	ln.Close()
+	cfg, err := config.NewLoader(nil, ctxlog.TestLogger(c)).Load()
+	c.Assert(err, check.IsNil)
+	cluster, err := cfg.GetCluster("")
+	c.Assert(err, check.IsNil)
+	cluster.Services.Keepbalance.InternalURLs[arvados.URL{Host: "localhost:" + p, Path: "/"}] = arvados.ServiceInstance{}
+	cfg.Clusters[cluster.ClusterID] = *cluster
+	config, err := yaml.Marshal(cfg)
+	c.Assert(err, check.IsNil)
+
+	var stdout bytes.Buffer
+	go Command.RunCommand("keep-balance", []string{"-config", "-"}, bytes.NewBuffer(config), &stdout, &stdout)
+	done := make(chan struct{})
+	go func() {
+		defer close(done)
+		for {
+			time.Sleep(time.Second / 10)
+			req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/metrics", nil)
+			if err != nil {
+				c.Fatal(err)
+				return
+			}
+			req.Header.Set("Authorization", "Bearer "+cluster.ManagementToken)
+			resp, err := http.DefaultClient.Do(req)
+			if err != nil {
+				c.Logf("error %s", err)
+				continue
+			}
+			defer resp.Body.Close()
+			if resp.StatusCode != http.StatusOK {
+				c.Logf("http status %d", resp.StatusCode)
+				continue
+			}
+			buf, err := ioutil.ReadAll(resp.Body)
+			if err != nil {
+				c.Logf("read body: %s", err)
+				continue
+			}
+			c.Check(string(buf), check.Matches, `(?ms).*arvados_keepbalance_sweep_seconds_sum.*`)
+			return
+		}
+	}()
+	select {
+	case <-done:
+	case <-time.After(time.Second):
+		c.Log(stdout.String())
+		c.Fatal("timeout")
+	}
+	c.Log(stdout.String())
+
+	// Check non-metrics URL that gets passed through to us from
+	// service.Command
+	req, err := http.NewRequest(http.MethodGet, "http://:"+p+"/not-metrics", nil)
+	c.Assert(err, check.IsNil)
+	resp, err := http.DefaultClient.Do(req)
+	c.Check(err, check.IsNil)
+	defer resp.Body.Close()
+	c.Check(resp.StatusCode, check.Equals, http.StatusNotFound)
 }