16306: Merge branch 'master'

[arvados.git] / services / keep-balance / collection_test.go

diff --git a/services/keep-balance/collection_test.go b/services/keep-balance/collection_test.go
index e6925c4afd7d32765100f5253f89493a648f12c5..3ab9d07b2e2ed6bcc7220ae17aad4e6e7a665855 100644 (file)
--- a/services/keep-balance/collection_test.go
+++ b/services/keep-balance/collection_test.go
@@ -5,10 +5,11 @@
 package main
 
 import (
+       "context"
        "sync"
        "time"
 
-       "git.curoverse.com/arvados.git/sdk/go/arvados"
+       "git.arvados.org/arvados.git/sdk/go/arvados"
        check "gopkg.in/check.v1"
 )
 
@@ -29,7 +30,7 @@ func (s *integrationSuite) TestIdenticalTimestamps(c *check.C) {
                        longestStreak := 0
                        var lastMod time.Time
                        sawUUID := make(map[string]bool)
-                       err := EachCollection(s.client, pageSize, func(c arvados.Collection) error {
+                       err := EachCollection(context.Background(), s.client, pageSize, func(c arvados.Collection) error {
                                if c.ModifiedAt.IsZero() {
                                        return nil
                                }