17014: Add mock response to users/current.

[arvados.git] / lib / controller / integration_test.go

diff --git a/lib/controller/integration_test.go b/lib/controller/integration_test.go
index 998b177d1160197eaaea1646c047dba76c749c79..a479e4476daa96b18112f53365b71d1c41c1260c 100644 (file)
--- a/lib/controller/integration_test.go
+++ b/lib/controller/integration_test.go
@@ -510,18 +510,18 @@ func (s *IntegrationSuite) TestRuntimeTokenInCR(c *check.C) {
                }
                cr, err := conn1.ContainerRequestCreate(rootctx1, arvados.CreateOptions{Attrs: rq})
                if tt.expectAToGetAValidCR {
-                       c.Assert(err, check.IsNil)
-                       c.Assert(cr, check.NotNil)
-                       c.Assert(cr.UUID, check.Not(check.Equals), "")
+                       c.Check(err, check.IsNil)
+                       c.Check(cr, check.NotNil)
+                       c.Check(cr.UUID, check.Not(check.Equals), "")
                }
 
                if tt.expectedToken == nil {
-                       break
+                       continue
                }
 
                c.Logf("cr.UUID: %s", cr.UUID)
                row := dbconn.QueryRowContext(rootctx1, `SELECT runtime_token from container_requests where uuid=$1`, cr.UUID)
-               c.Assert(row, check.NotNil)
+               c.Check(row, check.NotNil)
                var token sql.NullString
                row.Scan(&token)
                if c.Check(token.Valid, check.Equals, true) {