lib/controller/router/router_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package router
   6
   7 import (
   8         "bytes"
   9         "encoding/json"
  10         "io"
  11         "net/http"
  12         "net/http/httptest"
  13         "os"
  14         "testing"
  15
  16         "git.curoverse.com/arvados.git/sdk/go/arvados"
  17         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
  18         check "gopkg.in/check.v1"
  19 )
  20
  21 // Gocheck boilerplate
  22 func Test(t *testing.T) {
  23         check.TestingT(t)
  24 }
  25
  26 var _ = check.Suite(&RouterSuite{})
  27
  28 type RouterSuite struct {
  29         rtr *router
  30 }
  31
  32 func (s *RouterSuite) SetUpTest(c *check.C) {
  33         cluster := &arvados.Cluster{
  34                 TLS: arvados.TLS{Insecure: true},
  35         }
  36         arvadostest.SetServiceURL(&cluster.Services.RailsAPI, "https://"+os.Getenv("ARVADOS_TEST_API_HOST"))
  37         s.rtr = New(cluster)
  38 }
  39
  40 func (s *RouterSuite) TearDownTest(c *check.C) {
  41         err := arvados.NewClientFromEnv().RequestAndDecode(nil, "POST", "database/reset", nil, nil)
  42         c.Check(err, check.IsNil)
  43 }
  44
  45 func (s *RouterSuite) doRequest(c *check.C, token, method, path string, hdrs http.Header, body io.Reader) (*http.Request, *httptest.ResponseRecorder, map[string]interface{}) {
  46         req := httptest.NewRequest(method, path, body)
  47         for k, v := range hdrs {
  48                 req.Header[k] = v
  49         }
  50         req.Header.Set("Authorization", "Bearer "+token)
  51         rw := httptest.NewRecorder()
  52         s.rtr.ServeHTTP(rw, req)
  53         c.Logf("response body: %s", rw.Body.String())
  54         var jresp map[string]interface{}
  55         err := json.Unmarshal(rw.Body.Bytes(), &jresp)
  56         c.Check(err, check.IsNil)
  57         return req, rw, jresp
  58 }
  59
  60 func (s *RouterSuite) TestCollectionParams(c *check.C) {
  61         token := arvadostest.ActiveTokenV2
  62
  63         _, rw, jresp := s.doRequest(c, token, "GET", `/arvados/v1/collections?include_trash=true`, nil, nil)
  64         c.Check(rw.Code, check.Equals, http.StatusOK)
  65         c.Check(jresp["items_available"], check.FitsTypeOf, float64(0))
  66
  67         _, rw, jresp = s.doRequest(c, token, "GET", `/arvados/v1/collections`, nil, bytes.NewBufferString(`{"include_trash":true}`))
  68         c.Check(rw.Code, check.Equals, http.StatusOK)
  69         c.Check(jresp["items"], check.FitsTypeOf, []interface{}{})
  70
  71         _, rw, jresp = s.doRequest(c, token, "POST", `/arvados/v1/collections`, nil, bytes.NewBufferString(`ensure_unique_name=true`))
  72         c.Check(rw.Code, check.Equals, http.StatusOK)
  73         c.Check(jresp["uuid"], check.FitsTypeOf, "")
  74
  75         _, rw, jresp = s.doRequest(c, token, "POST", `/arvados/v1/collections?ensure_unique_name=true`, nil, nil)
  76         c.Check(rw.Code, check.Equals, http.StatusOK)
  77         c.Check(jresp["uuid"], check.FitsTypeOf, "")
  78 }
  79
  80 func (s *RouterSuite) TestContainerList(c *check.C) {
  81         token := arvadostest.ActiveTokenV2
  82
  83         _, rw, jresp := s.doRequest(c, token, "GET", `/arvados/v1/containers?limit=0`, nil, nil)
  84         c.Check(rw.Code, check.Equals, http.StatusOK)
  85         c.Check(jresp["items_available"], check.FitsTypeOf, float64(0))
  86         c.Check(jresp["items_available"].(float64) > 2, check.Equals, true)
  87         c.Check(jresp["items"], check.HasLen, 0)
  88
  89         _, rw, jresp = s.doRequest(c, token, "GET", `/arvados/v1/containers?limit=2&select=["uuid","command"]`, nil, nil)
  90         c.Check(rw.Code, check.Equals, http.StatusOK)
  91         c.Check(jresp["items_available"], check.FitsTypeOf, float64(0))
  92         c.Check(jresp["items_available"].(float64) > 2, check.Equals, true)
  93         c.Check(jresp["items"], check.HasLen, 2)
  94         item0 := jresp["items"].([]interface{})[0].(map[string]interface{})
  95         c.Check(item0["uuid"], check.HasLen, 27)
  96         c.Check(item0["command"], check.FitsTypeOf, []interface{}{})
  97         c.Check(item0["command"].([]interface{})[0], check.FitsTypeOf, "")
  98         c.Check(item0["mounts"], check.IsNil)
  99
 100         _, rw, jresp = s.doRequest(c, token, "GET", `/arvados/v1/containers`, nil, nil)
 101         c.Check(rw.Code, check.Equals, http.StatusOK)
 102         c.Check(jresp["items_available"], check.FitsTypeOf, float64(0))
 103         c.Check(jresp["items_available"].(float64) > 2, check.Equals, true)
 104         avail := int(jresp["items_available"].(float64))
 105         c.Check(jresp["items"], check.HasLen, avail)
 106         item0 = jresp["items"].([]interface{})[0].(map[string]interface{})
 107         c.Check(item0["uuid"], check.HasLen, 27)
 108         c.Check(item0["command"], check.FitsTypeOf, []interface{}{})
 109         c.Check(item0["command"].([]interface{})[0], check.FitsTypeOf, "")
 110         c.Check(item0["mounts"], check.NotNil)
 111 }
 112
 113 func (s *RouterSuite) TestContainerLock(c *check.C) {
 114         uuid := arvadostest.QueuedContainerUUID
 115         token := arvadostest.ActiveTokenV2
 116         _, rw, jresp := s.doRequest(c, token, "POST", "/arvados/v1/containers/"+uuid+"/lock", nil, nil)
 117         c.Check(rw.Code, check.Equals, http.StatusOK)
 118         c.Check(jresp["uuid"], check.HasLen, 27)
 119         c.Check(jresp["state"], check.Equals, "Locked")
 120         _, rw, jresp = s.doRequest(c, token, "POST", "/arvados/v1/containers/"+uuid+"/lock", nil, nil)
 121         c.Check(rw.Code, check.Equals, http.StatusUnprocessableEntity)
 122         c.Check(rw.Body.String(), check.Not(check.Matches), `.*"uuid":.*`)
 123         _, rw, jresp = s.doRequest(c, token, "POST", "/arvados/v1/containers/"+uuid+"/unlock", nil, nil)
 124         c.Check(rw.Code, check.Equals, http.StatusOK)
 125         c.Check(jresp["uuid"], check.HasLen, 27)
 126         c.Check(jresp["state"], check.Equals, "Queued")
 127         c.Check(jresp["environment"], check.IsNil)
 128         _, rw, jresp = s.doRequest(c, token, "POST", "/arvados/v1/containers/"+uuid+"/unlock", nil, nil)
 129         c.Check(rw.Code, check.Equals, http.StatusUnprocessableEntity)
 130         c.Check(jresp["uuid"], check.IsNil)
 131 }
 132
 133 func (s *RouterSuite) TestSelectParam(c *check.C) {
 134         uuid := arvadostest.QueuedContainerUUID
 135         token := arvadostest.ActiveTokenV2
 136         for _, sel := range [][]string{
 137                 {"uuid", "command"},
 138                 {"uuid", "command", "uuid"},
 139                 {"", "command", "uuid"},
 140         } {
 141                 j, err := json.Marshal(sel)
 142                 c.Assert(err, check.IsNil)
 143                 _, rw, resp := s.doRequest(c, token, "GET", "/arvados/v1/containers/"+uuid+"?select="+string(j), nil, nil)
 144                 c.Check(rw.Code, check.Equals, http.StatusOK)
 145
 146                 c.Check(resp["uuid"], check.HasLen, 27)
 147                 c.Check(resp["command"], check.HasLen, 2)
 148                 c.Check(resp["mounts"], check.IsNil)
 149                 _, hasMounts := resp["mounts"]
 150                 c.Check(hasMounts, check.Equals, false)
 151         }
 152 }