X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/ee908b109a95ea962425b435a8a317231829b115..a02b821d78d93bd814d8bf2b8b532b8940e93ecf:/lib/controller/rpc/conn_test.go

diff --git a/lib/controller/rpc/conn_test.go b/lib/controller/rpc/conn_test.go
index f43cc1ddee..eee8db9ac8 100644
--- a/lib/controller/rpc/conn_test.go
+++ b/lib/controller/rpc/conn_test.go
@@ -24,7 +24,11 @@ func Test(t *testing.T) {
 
 var _ = check.Suite(&RPCSuite{})
 
-const contextKeyTestTokens = "testTokens"
+type key int
+
+const (
+	contextKeyTestTokens key = iota
+)
 
 type RPCSuite struct {
 	log  logrus.FieldLogger
@@ -46,9 +50,8 @@ func (s *RPCSuite) TestLogin(c *check.C) {
 	opts := arvados.LoginOptions{
 		ReturnTo: "https://foo.example.com/bar",
 	}
-	resp, err := s.conn.Login(s.ctx, opts)
-	c.Check(err, check.IsNil)
-	c.Check(resp.RedirectLocation, check.Equals, "/auth/joshid?return_to="+url.QueryEscape(","+opts.ReturnTo))
+	_, err := s.conn.Login(s.ctx, opts)
+	c.Check(err.(*arvados.TransactionError).StatusCode, check.Equals, 404)
 }
 
 func (s *RPCSuite) TestLogout(c *check.C) {
@@ -58,7 +61,7 @@ func (s *RPCSuite) TestLogout(c *check.C) {
 	}
 	resp, err := s.conn.Logout(s.ctx, opts)
 	c.Check(err, check.IsNil)
-	c.Check(resp.RedirectLocation, check.Equals, "http://localhost:3002/users/sign_out?redirect_uri="+url.QueryEscape(opts.ReturnTo))
+	c.Check(resp.RedirectLocation, check.Equals, opts.ReturnTo)
 }
 
 func (s *RPCSuite) TestCollectionCreate(c *check.C) {