Arvados-DCO-1.1-Signed-off-by: Radhika Chippada <radhika@curoverse.com>
[arvados.git] / sdk / go / dispatch / dispatch_test.go
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: Apache-2.0
4
5 package dispatch
6
7 import (
8         "time"
9
10         "git.curoverse.com/arvados.git/sdk/go/arvados"
11         "git.curoverse.com/arvados.git/sdk/go/arvadosclient"
12         "git.curoverse.com/arvados.git/sdk/go/arvadostest"
13         . "gopkg.in/check.v1"
14 )
15
16 // Gocheck boilerplate
17 var _ = Suite(&suite{})
18
19 type suite struct{}
20
21 func (s *suite) SetUpSuite(c *C) {
22         arvadostest.StartAPI()
23 }
24
25 func (s *suite) TearDownSuite(c *C) {
26         arvadostest.StopAPI()
27 }
28
29 func (s *suite) TestTrackContainer(c *C) {
30         arv, err := arvadosclient.MakeArvadosClient()
31         c.Assert(err, Equals, nil)
32         arv.ApiToken = arvadostest.Dispatch1Token
33
34         done := make(chan bool, 1)
35         time.AfterFunc(10*time.Second, func() { done <- false })
36         d := &Dispatcher{
37                 Arv: arv,
38                 RunContainer: func(dsp *Dispatcher, ctr arvados.Container, status <-chan arvados.Container) {
39                         for ctr := range status {
40                                 c.Logf("%#v", ctr)
41                         }
42                         done <- true
43                 },
44         }
45         d.TrackContainer(arvadostest.QueuedContainerUUID)
46         c.Assert(<-done, Equals, true)
47 }