services/keep-web/cache_test.go

   1 package main
   2
   3 import (
   4         "git.curoverse.com/arvados.git/sdk/go/arvadosclient"
   5         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
   6         "gopkg.in/check.v1"
   7 )
   8
   9 func (s *UnitSuite) TestCache(c *check.C) {
  10         arv, err := arvadosclient.MakeArvadosClient()
  11         c.Assert(err, check.Equals, nil)
  12
  13         cache := DefaultConfig().Cache
  14
  15         // Hit the same collection 5 times using the same token. Only
  16         // the first req should cause an API call; the next 4 should
  17         // hit all caches.
  18         arv.ApiToken = arvadostest.AdminToken
  19         for i := 0; i < 5; i++ {
  20                 coll, err := cache.Get(arv, arvadostest.FooCollection, false)
  21                 c.Check(err, check.Equals, nil)
  22                 c.Assert(coll, check.NotNil)
  23                 c.Check(coll["portable_data_hash"], check.Equals, arvadostest.FooPdh)
  24                 c.Check(coll["manifest_text"].(string)[:2], check.Equals, ". ")
  25         }
  26         c.Check(cache.Stats().Requests, check.Equals, uint64(5))
  27         c.Check(cache.Stats().CollectionHits, check.Equals, uint64(4))
  28         c.Check(cache.Stats().PermissionHits, check.Equals, uint64(4))
  29         c.Check(cache.Stats().PDHHits, check.Equals, uint64(4))
  30         c.Check(cache.Stats().APICalls, check.Equals, uint64(1))
  31
  32         // Hit the same collection 2 more times, this time requesting
  33         // it by PDH and using a different token. The first req should
  34         // miss the permission cache. Both reqs should hit the
  35         // Collection cache and skip the API lookup.
  36         arv.ApiToken = arvadostest.ActiveToken
  37         for i := 0; i < 2; i++ {
  38                 coll, err := cache.Get(arv, arvadostest.FooPdh, false)
  39                 c.Check(err, check.Equals, nil)
  40                 c.Assert(coll, check.NotNil)
  41                 c.Check(coll["portable_data_hash"], check.Equals, arvadostest.FooPdh)
  42                 c.Check(coll["manifest_text"].(string)[:2], check.Equals, ". ")
  43         }
  44         c.Check(cache.Stats().Requests, check.Equals, uint64(5+2))
  45         c.Check(cache.Stats().CollectionHits, check.Equals, uint64(4+2))
  46         c.Check(cache.Stats().PermissionHits, check.Equals, uint64(4+1))
  47         c.Check(cache.Stats().PDHHits, check.Equals, uint64(4+0))
  48         c.Check(cache.Stats().APICalls, check.Equals, uint64(1+1))
  49
  50         // Alternating between two collections N times should produce
  51         // only 2 more API calls.
  52         arv.ApiToken = arvadostest.AdminToken
  53         for i := 0; i < 20; i++ {
  54                 var target string
  55                 if i%2 == 0 {
  56                         target = arvadostest.HelloWorldCollection
  57                 } else {
  58                         target = arvadostest.FooBarDirCollection
  59                 }
  60                 _, err := cache.Get(arv, target, false)
  61                 c.Check(err, check.Equals, nil)
  62         }
  63         c.Check(cache.Stats().Requests, check.Equals, uint64(5+2+20))
  64         c.Check(cache.Stats().CollectionHits, check.Equals, uint64(4+2+18))
  65         c.Check(cache.Stats().PermissionHits, check.Equals, uint64(4+1+18))
  66         c.Check(cache.Stats().PDHHits, check.Equals, uint64(4+0+18))
  67         c.Check(cache.Stats().APICalls, check.Equals, uint64(1+1+2))
  68 }
  69
  70 func (s *UnitSuite) TestCacheForceReloadByPDH(c *check.C) {
  71         arv, err := arvadosclient.MakeArvadosClient()
  72         c.Assert(err, check.Equals, nil)
  73
  74         cache := DefaultConfig().Cache
  75
  76         for _, forceReload := range []bool{false, true, false, true} {
  77                 _, err := cache.Get(arv, arvadostest.FooPdh, forceReload)
  78                 c.Check(err, check.Equals, nil)
  79         }
  80
  81         c.Check(cache.Stats().Requests, check.Equals, uint64(4))
  82         c.Check(cache.Stats().CollectionHits, check.Equals, uint64(3))
  83         c.Check(cache.Stats().PermissionHits, check.Equals, uint64(1))
  84         c.Check(cache.Stats().PDHHits, check.Equals, uint64(0))
  85         c.Check(cache.Stats().APICalls, check.Equals, uint64(3))
  86 }
  87
  88 func (s *UnitSuite) TestCacheForceReloadByUUID(c *check.C) {
  89         arv, err := arvadosclient.MakeArvadosClient()
  90         c.Assert(err, check.Equals, nil)
  91
  92         cache := DefaultConfig().Cache
  93
  94         for _, forceReload := range []bool{false, true, false, true} {
  95                 _, err := cache.Get(arv, arvadostest.FooCollection, forceReload)
  96                 c.Check(err, check.Equals, nil)
  97         }
  98
  99         c.Check(cache.Stats().Requests, check.Equals, uint64(4))
 100         c.Check(cache.Stats().CollectionHits, check.Equals, uint64(3))
 101         c.Check(cache.Stats().PermissionHits, check.Equals, uint64(1))
 102         c.Check(cache.Stats().PDHHits, check.Equals, uint64(3))
 103         c.Check(cache.Stats().APICalls, check.Equals, uint64(3))
 104 }