X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/da0aee751d8cb039c9b6b85a03e7d62cb973e3b3..bf4166939c77771642af846cb5372efc8a78659a:/services/keep-balance/collection_test.go

diff --git a/services/keep-balance/collection_test.go b/services/keep-balance/collection_test.go
index 629ff3a140..3ab9d07b2e 100644
--- a/services/keep-balance/collection_test.go
+++ b/services/keep-balance/collection_test.go
@@ -1,10 +1,15 @@
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
 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"
 )
 
@@ -25,24 +30,24 @@ func (s *integrationSuite) TestIdenticalTimestamps(c *check.C) {
 			longestStreak := 0
 			var lastMod time.Time
 			sawUUID := make(map[string]bool)
-			err := EachCollection(&s.config.Client, pageSize, func(c arvados.Collection) error {
-				got[trial] = append(got[trial], c.UUID)
-				if c.ModifiedAt == nil {
+			err := EachCollection(context.Background(), s.client, pageSize, func(c arvados.Collection) error {
+				if c.ModifiedAt.IsZero() {
 					return nil
 				}
 				if sawUUID[c.UUID] {
 					// dup
 					return nil
 				}
+				got[trial] = append(got[trial], c.UUID)
 				sawUUID[c.UUID] = true
-				if lastMod == *c.ModifiedAt {
+				if lastMod == c.ModifiedAt {
 					streak++
 					if streak > longestStreak {
 						longestStreak = streak
 					}
 				} else {
 					streak = 0
-					lastMod = *c.ModifiedAt
+					lastMod = c.ModifiedAt
 				}
 				return nil
 			}, nil)