sdk/go/health/handler_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: Apache-2.0
   4
   5 package health
   6
   7 import (
   8         "encoding/json"
   9         "errors"
  10         "net/http"
  11         "net/http/httptest"
  12         "net/url"
  13         "testing"
  14
  15         check "gopkg.in/check.v1"
  16 )
  17
  18 // Gocheck boilerplate
  19 var _ = check.Suite(&Suite{})
  20
  21 func Test(t *testing.T) {
  22         check.TestingT(t)
  23 }
  24
  25 type Suite struct{}
  26
  27 const (
  28         goodToken = "supersecret"
  29         badToken  = "pwn"
  30 )
  31
  32 func (s *Suite) TestPassFailRefuse(c *check.C) {
  33         h := &Handler{
  34                 Token:  goodToken,
  35                 Prefix: "/_health/",
  36                 Routes: Routes{
  37                         "success": func() error { return nil },
  38                         "miracle": func() error { return errors.New("unimplemented") },
  39                 },
  40         }
  41
  42         resp := httptest.NewRecorder()
  43         h.ServeHTTP(resp, s.request("/_health/ping", goodToken))
  44         s.checkHealthy(c, resp)
  45
  46         resp = httptest.NewRecorder()
  47         h.ServeHTTP(resp, s.request("/_health/success", goodToken))
  48         s.checkHealthy(c, resp)
  49
  50         resp = httptest.NewRecorder()
  51         h.ServeHTTP(resp, s.request("/_health/miracle", goodToken))
  52         s.checkUnhealthy(c, resp)
  53
  54         resp = httptest.NewRecorder()
  55         h.ServeHTTP(resp, s.request("/_health/miracle", badToken))
  56         c.Check(resp.Code, check.Equals, http.StatusForbidden)
  57
  58         resp = httptest.NewRecorder()
  59         h.ServeHTTP(resp, s.request("/_health/miracle", ""))
  60         c.Check(resp.Code, check.Equals, http.StatusUnauthorized)
  61 }
  62
  63 func (s *Suite) TestPingOverride(c *check.C) {
  64         var ok bool
  65         h := &Handler{
  66                 Token: goodToken,
  67                 Routes: Routes{
  68                         "ping": func() error {
  69                                 ok = !ok
  70                                 if ok {
  71                                         return nil
  72                                 } else {
  73                                         return errors.New("good error")
  74                                 }
  75                         },
  76                 },
  77         }
  78         resp := httptest.NewRecorder()
  79         h.ServeHTTP(resp, s.request("/ping", goodToken))
  80         s.checkHealthy(c, resp)
  81
  82         resp = httptest.NewRecorder()
  83         h.ServeHTTP(resp, s.request("/ping", goodToken))
  84         s.checkUnhealthy(c, resp)
  85 }
  86
  87 func (s *Suite) TestZeroValue(c *check.C) {
  88         resp := httptest.NewRecorder()
  89         (&Handler{}).ServeHTTP(resp, s.request("/ping", goodToken))
  90         c.Check(resp.Code, check.Equals, http.StatusNotFound)
  91 }
  92
  93 func (s *Suite) request(path, token string) *http.Request {
  94         u, _ := url.Parse("http://foo.local" + path)
  95         req := &http.Request{
  96                 Method:     "GET",
  97                 Host:       u.Host,
  98                 URL:        u,
  99                 RequestURI: u.RequestURI(),
 100         }
 101         if token != "" {
 102                 req.Header = http.Header{
 103                         "Authorization": {"Bearer " + token},
 104                 }
 105         }
 106         return req
 107 }
 108
 109 func (s *Suite) checkHealthy(c *check.C, resp *httptest.ResponseRecorder) {
 110         c.Check(resp.Code, check.Equals, http.StatusOK)
 111         c.Check(resp.Body.String(), check.Equals, `{"health":"OK"}`+"\n")
 112 }
 113
 114 func (s *Suite) checkUnhealthy(c *check.C, resp *httptest.ResponseRecorder) {
 115         c.Check(resp.Code, check.Equals, http.StatusOK)
 116         var result map[string]interface{}
 117         err := json.Unmarshal(resp.Body.Bytes(), &result)
 118         c.Assert(err, check.IsNil)
 119         c.Check(result["health"], check.Equals, "ERROR")
 120         c.Check(result["error"].(string), check.Not(check.Equals), "")
 121 }