lib/controller/localdb/link_test.go

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 package localdb
   6
   7 import (
   8         "context"
   9
  10         "git.arvados.org/arvados.git/lib/config"
  11         "git.arvados.org/arvados.git/lib/controller/rpc"
  12         "git.arvados.org/arvados.git/sdk/go/arvados"
  13         "git.arvados.org/arvados.git/sdk/go/arvadostest"
  14         "git.arvados.org/arvados.git/sdk/go/auth"
  15         "git.arvados.org/arvados.git/sdk/go/ctxlog"
  16         check "gopkg.in/check.v1"
  17 )
  18
  19 var _ = check.Suite(&LinkSuite{})
  20
  21 type LinkSuite struct {
  22         cluster  *arvados.Cluster
  23         localdb  *Conn
  24         railsSpy *arvadostest.Proxy
  25 }
  26
  27 func (s *LinkSuite) TearDownSuite(c *check.C) {
  28         // Undo any changes/additions to the user database so they
  29         // don't affect subsequent tests.
  30         arvadostest.ResetEnv()
  31         c.Check(arvados.NewClientFromEnv().RequestAndDecode(nil, "POST", "database/reset", nil, nil), check.IsNil)
  32 }
  33
  34 func (s *LinkSuite) SetUpTest(c *check.C) {
  35         cfg, err := config.NewLoader(nil, ctxlog.TestLogger(c)).Load()
  36         c.Assert(err, check.IsNil)
  37         s.cluster, err = cfg.GetCluster("")
  38         c.Assert(err, check.IsNil)
  39         s.localdb = NewConn(s.cluster)
  40         s.railsSpy = arvadostest.NewProxy(c, s.cluster.Services.RailsAPI)
  41         *s.localdb.railsProxy = *rpc.NewConn(s.cluster.ClusterID, s.railsSpy.URL, true, rpc.PassthroughTokenProvider)
  42 }
  43
  44 func (s *LinkSuite) TearDownTest(c *check.C) {
  45         s.railsSpy.Close()
  46 }
  47
  48 func (s *LinkSuite) setUpVocabulary(c *check.C, testVocabulary string) {
  49         if testVocabulary == "" {
  50                 testVocabulary = `{
  51                         "strict_tags": false,
  52                         "tags": {
  53                                 "IDTAGIMPORTANCES": {
  54                                         "strict": true,
  55                                         "labels": [{"label": "Importance"}, {"label": "Priority"}],
  56                                         "values": {
  57                                                 "IDVALIMPORTANCES1": { "labels": [{"label": "Critical"}, {"label": "Urgent"}, {"label": "High"}] },
  58                                                 "IDVALIMPORTANCES2": { "labels": [{"label": "Normal"}, {"label": "Moderate"}] },
  59                                                 "IDVALIMPORTANCES3": { "labels": [{"label": "Low"}] }
  60                                         }
  61                                 }
  62                         }
  63                 }`
  64         }
  65         voc, err := arvados.NewVocabulary([]byte(testVocabulary), []string{})
  66         c.Assert(err, check.IsNil)
  67         s.localdb.vocabularyCache = voc
  68         s.cluster.API.VocabularyPath = "foo"
  69 }
  70
  71 func (s *LinkSuite) TestLinkCreateWithProperties(c *check.C) {
  72         s.setUpVocabulary(c, "")
  73         ctx := auth.NewContext(context.Background(), &auth.Credentials{Tokens: []string{arvadostest.ActiveTokenV2}})
  74
  75         tests := []struct {
  76                 name    string
  77                 props   map[string]interface{}
  78                 success bool
  79         }{
  80                 {"Invalid prop key", map[string]interface{}{"Priority": "IDVALIMPORTANCES1"}, false},
  81                 {"Invalid prop value", map[string]interface{}{"IDTAGIMPORTANCES": "high"}, false},
  82                 {"Valid prop key & value", map[string]interface{}{"IDTAGIMPORTANCES": "IDVALIMPORTANCES1"}, true},
  83                 {"Empty properties", map[string]interface{}{}, true},
  84         }
  85         for _, tt := range tests {
  86                 c.Log(c.TestName()+" ", tt.name)
  87
  88                 lnk, err := s.localdb.LinkCreate(ctx, arvados.CreateOptions{
  89                         Select: []string{"uuid", "properties"},
  90                         Attrs: map[string]interface{}{
  91                                 "link_class": "star",
  92                                 "tail_uuid":  "zzzzz-j7d0g-publicfavorites",
  93                                 "head_uuid":  arvadostest.FooCollection,
  94                                 "properties": tt.props,
  95                         }})
  96                 if tt.success {
  97                         c.Assert(err, check.IsNil)
  98                         c.Assert(lnk.Properties, check.DeepEquals, tt.props)
  99                 } else {
 100                         c.Assert(err, check.NotNil)
 101                 }
 102         }
 103 }
 104
 105 func (s *LinkSuite) TestLinkUpdateWithProperties(c *check.C) {
 106         s.setUpVocabulary(c, "")
 107         ctx := auth.NewContext(context.Background(), &auth.Credentials{Tokens: []string{arvadostest.ActiveTokenV2}})
 108
 109         tests := []struct {
 110                 name    string
 111                 props   map[string]interface{}
 112                 success bool
 113         }{
 114                 {"Invalid prop key", map[string]interface{}{"Priority": "IDVALIMPORTANCES1"}, false},
 115                 {"Invalid prop value", map[string]interface{}{"IDTAGIMPORTANCES": "high"}, false},
 116                 {"Valid prop key & value", map[string]interface{}{"IDTAGIMPORTANCES": "IDVALIMPORTANCES1"}, true},
 117                 {"Empty properties", map[string]interface{}{}, true},
 118         }
 119         for _, tt := range tests {
 120                 c.Log(c.TestName()+" ", tt.name)
 121                 lnk, err := s.localdb.LinkCreate(ctx, arvados.CreateOptions{
 122                         Attrs: map[string]interface{}{
 123                                 "link_class": "star",
 124                                 "tail_uuid":  "zzzzz-j7d0g-publicfavorites",
 125                                 "head_uuid":  arvadostest.FooCollection,
 126                         },
 127                 })
 128                 c.Assert(err, check.IsNil)
 129                 lnk, err = s.localdb.LinkUpdate(ctx, arvados.UpdateOptions{
 130                         UUID:   lnk.UUID,
 131                         Select: []string{"uuid", "properties"},
 132                         Attrs: map[string]interface{}{
 133                                 "properties": tt.props,
 134                         }})
 135                 if tt.success {
 136                         c.Assert(err, check.IsNil)
 137                         c.Assert(lnk.Properties, check.DeepEquals, tt.props)
 138                 } else {
 139                         c.Assert(err, check.NotNil)
 140                 }
 141         }
 142 }