sdk/go/dispatch/dispatch_test.go

   1 package dispatch
   2
   3 import (
   4         "git.curoverse.com/arvados.git/sdk/go/arvadosclient"
   5         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
   6
   7         . "gopkg.in/check.v1"
   8 )
   9
  10 // Gocheck boilerplate
  11 var _ = Suite(&DispatchTestSuite{})
  12
  13 type DispatchTestSuite struct{}
  14
  15 func (s *DispatchTestSuite) SetUpSuite(c *C) {
  16         arvadostest.StartAPI()
  17 }
  18
  19 func (s *DispatchTestSuite) TearDownSuite(c *C) {
  20         arvadostest.StopAPI()
  21 }
  22
  23 func (s *DispatchTestSuite) TestTrackContainer(c *C) {
  24         arv, err := arvadosclient.MakeArvadosClient()
  25         c.Assert(err, Equals, nil)
  26
  27         d := &Dispatcher{Arv: arv}
  28         d.trackers = make(map[string]*runTracker)
  29
  30         d.TrackContainer(arvadostest.QueuedContainerUuid)
  31         _, tracking := d.trackers[arvadostest.QueuedContainerUuid]
  32         c.Assert(tracking, Equals, true)
  33 }