X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/2db0c3a44d8f12e3a566c796bdecaa1de150b155..6b9cb6715bb7c04b2a425bc28b41a51f8a273a7a:/services/ws/permission_test.go

diff --git a/services/ws/permission_test.go b/services/ws/permission_test.go
index afaab0ec0f..2a22eae609 100644
--- a/services/ws/permission_test.go
+++ b/services/ws/permission_test.go
@@ -1,8 +1,14 @@
-package main
+// Copyright (C) The Arvados Authors. All rights reserved.
+//
+// SPDX-License-Identifier: AGPL-3.0
+
+package ws
 
 import (
-	"git.curoverse.com/arvados.git/sdk/go/arvados"
-	"git.curoverse.com/arvados.git/sdk/go/arvadostest"
+	"context"
+
+	"git.arvados.org/arvados.git/sdk/go/arvados"
+	"git.arvados.org/arvados.git/sdk/go/arvadostest"
 	check "gopkg.in/check.v1"
 )
 
@@ -11,26 +17,30 @@ var _ = check.Suite(&permSuite{})
 type permSuite struct{}
 
 func (s *permSuite) TestCheck(c *check.C) {
-	pc := newPermChecker(*(arvados.NewClientFromEnv())).(*cachingPermChecker)
+	client := arvados.NewClientFromEnv()
+	// Disable auto-retry
+	client.Timeout = 0
+
+	pc := newPermChecker(client).(*cachingPermChecker)
 	setToken := func(label, token string) {
 		c.Logf("...%s token %q", label, token)
 		pc.SetToken(token)
 	}
 	wantError := func(uuid string) {
 		c.Log(uuid)
-		ok, err := pc.Check(uuid)
+		ok, err := pc.Check(context.Background(), uuid)
 		c.Check(ok, check.Equals, false)
 		c.Check(err, check.NotNil)
 	}
 	wantYes := func(uuid string) {
 		c.Log(uuid)
-		ok, err := pc.Check(uuid)
+		ok, err := pc.Check(context.Background(), uuid)
 		c.Check(ok, check.Equals, true)
 		c.Check(err, check.IsNil)
 	}
 	wantNo := func(uuid string) {
 		c.Log(uuid)
-		ok, err := pc.Check(uuid)
+		ok, err := pc.Check(context.Background(), uuid)
 		c.Check(ok, check.Equals, false)
 		c.Check(err, check.IsNil)
 	}
@@ -63,7 +73,7 @@ func (s *permSuite) TestCheck(c *check.C) {
 	pc.SetToken(arvadostest.ActiveToken)
 
 	c.Log("...network error")
-	pc.Client.APIHost = "127.0.0.1:discard"
+	pc.ac.APIHost = "127.0.0.1:9"
 	wantError(arvadostest.UserAgreementCollection)
 	wantError(arvadostest.FooBarDirCollection)