services/keep-web/status_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         "encoding/json"
   9         "net/http"
  10         "net/http/httptest"
  11         "net/url"
  12
  13         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
  14         "gopkg.in/check.v1"
  15 )
  16
  17 func (s *UnitSuite) TestStatus(c *check.C) {
  18         h := handler{Config: DefaultConfig()}
  19         u, _ := url.Parse("http://keep-web.example/status.json")
  20         req := &http.Request{
  21                 Method:     "GET",
  22                 Host:       u.Host,
  23                 URL:        u,
  24                 RequestURI: u.RequestURI(),
  25         }
  26         resp := httptest.NewRecorder()
  27         h.ServeHTTP(resp, req)
  28         c.Check(resp.Code, check.Equals, http.StatusOK)
  29
  30         var status map[string]interface{}
  31         err := json.NewDecoder(resp.Body).Decode(&status)
  32         c.Check(err, check.IsNil)
  33         c.Check(status["Cache.Requests"], check.Equals, float64(0))
  34         c.Check(status["Version"], check.Not(check.Equals), "")
  35 }
  36
  37 func (s *IntegrationSuite) TestNoStatusFromVHost(c *check.C) {
  38         u, _ := url.Parse("http://" + arvadostest.FooCollection + "--keep-web.example/status.json")
  39         req := &http.Request{
  40                 Method:     "GET",
  41                 Host:       u.Host,
  42                 URL:        u,
  43                 RequestURI: u.RequestURI(),
  44                 Header: http.Header{
  45                         "Authorization": {"OAuth2 " + arvadostest.ActiveToken},
  46                 },
  47         }
  48         resp := httptest.NewRecorder()
  49         s.testServer.Handler.ServeHTTP(resp, req)
  50         c.Check(resp.Code, check.Equals, http.StatusNotFound)
  51 }