X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/6029ffdc433d69f7fe139b0e23673fa12715e413..b64a7596b2fb456de13aa885dbe14a817fd8d5b2:/sdk/go/dispatch/dispatch_test.go

diff --git a/sdk/go/dispatch/dispatch_test.go b/sdk/go/dispatch/dispatch_test.go
index fb827fcee2..25a4d2b879 100644
--- a/sdk/go/dispatch/dispatch_test.go
+++ b/sdk/go/dispatch/dispatch_test.go
@@ -1,33 +1,47 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: Apache-2.0
+
 package dispatch
 
 import (
-	"git.curoverse.com/arvados.git/sdk/go/arvadosclient"
-	"git.curoverse.com/arvados.git/sdk/go/arvadostest"
+	"time"
 
+	"git.arvados.org/arvados.git/sdk/go/arvados"
+	"git.arvados.org/arvados.git/sdk/go/arvadosclient"
+	"git.arvados.org/arvados.git/sdk/go/arvadostest"
 	. "gopkg.in/check.v1"
 )
 
 // Gocheck boilerplate
-var _ = Suite(&DispatchTestSuite{})
+var _ = Suite(&suite{})
 
-type DispatchTestSuite struct{}
+type suite struct{}
 
-func (s *DispatchTestSuite) SetUpSuite(c *C) {
+func (s *suite) SetUpSuite(c *C) {
 	arvadostest.StartAPI()
 }
 
-func (s *DispatchTestSuite) TearDownSuite(c *C) {
+func (s *suite) TearDownSuite(c *C) {
 	arvadostest.StopAPI()
 }
 
-func (s *DispatchTestSuite) TestTrackContainer(c *C) {
+func (s *suite) TestTrackContainer(c *C) {
 	arv, err := arvadosclient.MakeArvadosClient()
 	c.Assert(err, Equals, nil)
-
-	d := &Dispatcher{Arv: arv}
-	d.trackers = make(map[string]*runTracker)
-
-	d.TrackContainer(arvadostest.QueuedContainerUuid)
-	_, tracking := d.trackers[arvadostest.QueuedContainerUuid]
-	c.Assert(tracking, Equals, true)
+	arv.ApiToken = arvadostest.Dispatch1Token
+
+	done := make(chan bool, 1)
+	time.AfterFunc(10*time.Second, func() { done <- false })
+	d := &Dispatcher{
+		Arv: arv,
+		RunContainer: func(dsp *Dispatcher, ctr arvados.Container, status <-chan arvados.Container) {
+			for ctr := range status {
+				c.Logf("%#v", ctr)
+			}
+			done <- true
+		},
+	}
+	d.TrackContainer(arvadostest.QueuedContainerUUID)
+	c.Assert(<-done, Equals, true)
 }