sdk/go/arvadosclient/arvadosclient_test.go

   1 package arvadosclient
   2
   3 import (
   4         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
   5         . "gopkg.in/check.v1"
   6         "net/http"
   7         "os"
   8         "testing"
   9         "time"
  10 )
  11
  12 // Gocheck boilerplate
  13 func Test(t *testing.T) {
  14         TestingT(t)
  15 }
  16
  17 var _ = Suite(&ServerRequiredSuite{})
  18 var _ = Suite(&UnitSuite{})
  19
  20 // Tests that require the Keep server running
  21 type ServerRequiredSuite struct{}
  22
  23 func (s *ServerRequiredSuite) SetUpSuite(c *C) {
  24         arvadostest.StartAPI()
  25         arvadostest.StartKeep(2, false)
  26 }
  27
  28 func (s *ServerRequiredSuite) SetUpTest(c *C) {
  29         arvadostest.ResetEnv()
  30 }
  31
  32 func (s *ServerRequiredSuite) TestMakeArvadosClientSecure(c *C) {
  33         os.Setenv("ARVADOS_API_HOST_INSECURE", "")
  34         kc, err := MakeArvadosClient()
  35         c.Assert(err, Equals, nil)
  36         c.Check(kc.ApiServer, Equals, os.Getenv("ARVADOS_API_HOST"))
  37         c.Check(kc.ApiToken, Equals, os.Getenv("ARVADOS_API_TOKEN"))
  38         c.Check(kc.ApiInsecure, Equals, false)
  39 }
  40
  41 func (s *ServerRequiredSuite) TestMakeArvadosClientInsecure(c *C) {
  42         os.Setenv("ARVADOS_API_HOST_INSECURE", "true")
  43         kc, err := MakeArvadosClient()
  44         c.Assert(err, Equals, nil)
  45         c.Check(kc.ApiInsecure, Equals, true)
  46         c.Check(kc.ApiServer, Equals, os.Getenv("ARVADOS_API_HOST"))
  47         c.Check(kc.ApiToken, Equals, os.Getenv("ARVADOS_API_TOKEN"))
  48         c.Check(kc.Client.Transport.(*http.Transport).TLSClientConfig.InsecureSkipVerify, Equals, true)
  49 }
  50
  51 func (s *ServerRequiredSuite) TestGetInvalidUUID(c *C) {
  52         arv, err := MakeArvadosClient()
  53
  54         getback := make(Dict)
  55         err = arv.Get("collections", "", nil, &getback)
  56         c.Assert(err, Equals, ErrInvalidArgument)
  57         c.Assert(len(getback), Equals, 0)
  58
  59         err = arv.Get("collections", "zebra-moose-unicorn", nil, &getback)
  60         c.Assert(err, Equals, ErrInvalidArgument)
  61         c.Assert(len(getback), Equals, 0)
  62
  63         err = arv.Get("collections", "acbd18db4cc2f85cedef654fccc4a4d8", nil, &getback)
  64         c.Assert(err, Equals, ErrInvalidArgument)
  65         c.Assert(len(getback), Equals, 0)
  66 }
  67
  68 func (s *ServerRequiredSuite) TestGetValidUUID(c *C) {
  69         arv, err := MakeArvadosClient()
  70
  71         getback := make(Dict)
  72         err = arv.Get("collections", "zzzzz-4zz18-abcdeabcdeabcde", nil, &getback)
  73         c.Assert(err, FitsTypeOf, APIServerError{})
  74         c.Assert(err.(APIServerError).HttpStatusCode, Equals, http.StatusNotFound)
  75         c.Assert(len(getback), Equals, 0)
  76
  77         err = arv.Get("collections", "acbd18db4cc2f85cedef654fccc4a4d8+3", nil, &getback)
  78         c.Assert(err, FitsTypeOf, APIServerError{})
  79         c.Assert(err.(APIServerError).HttpStatusCode, Equals, http.StatusNotFound)
  80         c.Assert(len(getback), Equals, 0)
  81 }
  82
  83 func (s *ServerRequiredSuite) TestInvalidResourceType(c *C) {
  84         arv, err := MakeArvadosClient()
  85
  86         getback := make(Dict)
  87         err = arv.Get("unicorns", "zzzzz-zebra-unicorn7unicorn", nil, &getback)
  88         c.Assert(err, FitsTypeOf, APIServerError{})
  89         c.Assert(err.(APIServerError).HttpStatusCode, Equals, http.StatusNotFound)
  90         c.Assert(len(getback), Equals, 0)
  91
  92         err = arv.Update("unicorns", "zzzzz-zebra-unicorn7unicorn", nil, &getback)
  93         c.Assert(err, FitsTypeOf, APIServerError{})
  94         c.Assert(err.(APIServerError).HttpStatusCode, Equals, http.StatusNotFound)
  95         c.Assert(len(getback), Equals, 0)
  96
  97         err = arv.List("unicorns", nil, &getback)
  98         c.Assert(err, FitsTypeOf, APIServerError{})
  99         c.Assert(err.(APIServerError).HttpStatusCode, Equals, http.StatusNotFound)
 100         c.Assert(len(getback), Equals, 0)
 101 }
 102
 103 func (s *ServerRequiredSuite) TestCreatePipelineTemplate(c *C) {
 104         for _, idleConnections := range []bool{
 105                 false,
 106                 true,
 107         } {
 108                 arv, err := MakeArvadosClient()
 109
 110                 if idleConnections {
 111                         arv.lastClosedIdlesAt = time.Now().Add(-time.Minute)
 112                 }
 113
 114                 getback := make(Dict)
 115                 err = arv.Create("pipeline_templates",
 116                         Dict{"pipeline_template": Dict{
 117                                 "name": "tmp",
 118                                 "components": Dict{
 119                                         "c1": map[string]string{"script": "script1"},
 120                                         "c2": map[string]string{"script": "script2"}}}},
 121                         &getback)
 122                 c.Assert(err, Equals, nil)
 123                 c.Assert(getback["name"], Equals, "tmp")
 124                 c.Assert(getback["components"].(map[string]interface{})["c2"].(map[string]interface{})["script"], Equals, "script2")
 125
 126                 uuid := getback["uuid"].(string)
 127
 128                 getback = make(Dict)
 129                 err = arv.Get("pipeline_templates", uuid, nil, &getback)
 130                 c.Assert(err, Equals, nil)
 131                 c.Assert(getback["name"], Equals, "tmp")
 132                 c.Assert(getback["components"].(map[string]interface{})["c1"].(map[string]interface{})["script"], Equals, "script1")
 133
 134                 getback = make(Dict)
 135                 err = arv.Update("pipeline_templates", uuid,
 136                         Dict{
 137                                 "pipeline_template": Dict{"name": "tmp2"}},
 138                         &getback)
 139                 c.Assert(err, Equals, nil)
 140                 c.Assert(getback["name"], Equals, "tmp2")
 141
 142                 c.Assert(getback["uuid"].(string), Equals, uuid)
 143                 getback = make(Dict)
 144                 err = arv.Delete("pipeline_templates", uuid, nil, &getback)
 145                 c.Assert(err, Equals, nil)
 146                 c.Assert(getback["name"], Equals, "tmp2")
 147         }
 148 }
 149
 150 func (s *ServerRequiredSuite) TestErrorResponse(c *C) {
 151         arv, _ := MakeArvadosClient()
 152
 153         getback := make(Dict)
 154
 155         {
 156                 err := arv.Create("logs",
 157                         Dict{"log": Dict{"bogus_attr": "foo"}},
 158                         &getback)
 159                 c.Assert(err, ErrorMatches, "arvados API server error: .*")
 160                 c.Assert(err, ErrorMatches, ".*unknown attribute: bogus_attr.*")
 161                 c.Assert(err, FitsTypeOf, APIServerError{})
 162                 c.Assert(err.(APIServerError).HttpStatusCode, Equals, 422)
 163         }
 164
 165         {
 166                 err := arv.Create("bogus",
 167                         Dict{"bogus": Dict{}},
 168                         &getback)
 169                 c.Assert(err, ErrorMatches, "arvados API server error: .*")
 170                 c.Assert(err, ErrorMatches, ".*Path not found.*")
 171                 c.Assert(err, FitsTypeOf, APIServerError{})
 172                 c.Assert(err.(APIServerError).HttpStatusCode, Equals, 404)
 173         }
 174 }
 175
 176 func (s *ServerRequiredSuite) TestAPIDiscovery_Get_defaultCollectionReplication(c *C) {
 177         arv, err := MakeArvadosClient()
 178         value, err := arv.Discovery("defaultCollectionReplication")
 179         c.Assert(err, IsNil)
 180         c.Assert(value, NotNil)
 181 }
 182
 183 func (s *ServerRequiredSuite) TestAPIDiscovery_Get_noSuchParameter(c *C) {
 184         arv, err := MakeArvadosClient()
 185         value, err := arv.Discovery("noSuchParameter")
 186         c.Assert(err, NotNil)
 187         c.Assert(value, IsNil)
 188 }
 189
 190 type UnitSuite struct{}
 191
 192 func (s *UnitSuite) TestUUIDMatch(c *C) {
 193         c.Assert(UUIDMatch("zzzzz-tpzed-000000000000000"), Equals, true)
 194         c.Assert(UUIDMatch("zzzzz-zebra-000000000000000"), Equals, true)
 195         c.Assert(UUIDMatch("00000-00000-zzzzzzzzzzzzzzz"), Equals, true)
 196         c.Assert(UUIDMatch("ZEBRA-HORSE-AFRICANELEPHANT"), Equals, false)
 197         c.Assert(UUIDMatch(" zzzzz-tpzed-000000000000000"), Equals, false)
 198         c.Assert(UUIDMatch("d41d8cd98f00b204e9800998ecf8427e"), Equals, false)
 199         c.Assert(UUIDMatch("d41d8cd98f00b204e9800998ecf8427e+0"), Equals, false)
 200         c.Assert(UUIDMatch(""), Equals, false)
 201 }
 202
 203 func (s *UnitSuite) TestPDHMatch(c *C) {
 204         c.Assert(PDHMatch("zzzzz-tpzed-000000000000000"), Equals, false)
 205         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e"), Equals, false)
 206         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e+0"), Equals, true)
 207         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e+12345"), Equals, true)
 208         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e 12345"), Equals, false)
 209         c.Assert(PDHMatch("D41D8CD98F00B204E9800998ECF8427E+12345"), Equals, false)
 210         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e+12345 "), Equals, false)
 211         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e+abcdef"), Equals, false)
 212         c.Assert(PDHMatch("da39a3ee5e6b4b0d3255bfef95601890afd80709"), Equals, false)
 213         c.Assert(PDHMatch("da39a3ee5e6b4b0d3255bfef95601890afd80709+0"), Equals, false)
 214         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427+12345"), Equals, false)
 215         c.Assert(PDHMatch("d41d8cd98f00b204e9800998ecf8427e+12345\n"), Equals, false)
 216         c.Assert(PDHMatch("+12345"), Equals, false)
 217         c.Assert(PDHMatch(""), Equals, false)
 218 }