server/auth: Add TokenSource call.

server/auth/context_test.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package auth
 
 import (
         "fmt"
         "net/http"
         "net/http/httptest"
         "net/url"
         "testing"
 
         "golang.org/x/net/context"
 
         "github.com/luci/luci-go/common/errors"
         "github.com/luci/luci-go/server/auth/identity"
         "github.com/luci/luci-go/server/router"
         . "github.com/smartystreets/goconvey/convey"
 )
 
 func TestContext(t *testing.T) {
+        t.Parallel()
+
         Convey("Works", t, func() {
                 c := context.Background()
 
                 So(getAuthenticator(c), ShouldBeNil)
                 _, err := LoginURL(c, "dest")
                 So(err, ShouldEqual, ErrNoUsersAPI)
                 _, err = LogoutURL(c, "dest")
                 So(err, ShouldEqual, ErrNoUsersAPI)
 
                 // Authenticator without UsersAPI.
@@ skipping 13 matching lines @@
                 So(err, ShouldBeNil)
                 So(dest, ShouldEqual, "http://login_url?r=dest")
                 dest, err = LogoutURL(c, "dest")
                 So(err, ShouldBeNil)
                 So(dest, ShouldEqual, "http://logout_url?r=dest")
         })
 
 }
 
 func TestContextAuthenticate(t *testing.T) {
+        t.Parallel()
+
         call := func(c context.Context, m router.MiddlewareChain, h router.Handler) *httptest.ResponseRecorder {
                 req, err := http.NewRequest("GET", "http://example.com/foo", nil)
                 So(err, ShouldBeNil)
                 w := httptest.NewRecorder()
                 router.RunMiddleware(&router.Context{
                         Context: c,
                         Writer:  w,
                         Request: req,
                 }, m, h)
                 return w
@@ skipping 39 matching lines @@
 
         Convey("Broken ID is rejected", t, func() {
                 c := prepareCtx(fakeMethod{userID: "???"})
                 rr := call(c, router.NewMiddlewareChain(Authenticate), handler)
                 So(rr.Code, ShouldEqual, 401)
                 So(rr.Body.String(), ShouldEqual, "Authentication error - auth: bad identity string \"???\"\n")
         })
 }
 
 func TestAutologin(t *testing.T) {
+        t.Parallel()
+
         call := func(c context.Context, m router.MiddlewareChain, h router.Handler) *httptest.ResponseRecorder {
                 req, err := http.NewRequest("GET", "http://example.com/foo", nil)
                 So(err, ShouldBeNil)
                 w := httptest.NewRecorder()
                 router.RunMiddleware(&router.Context{
                         Context: c,
                         Writer:  w,
                         Request: req,
                 }, m, h)
                 return w
@@ skipping 86 matching lines @@
         v := url.Values{}
         v.Set("r", dest)
         return "http://login_url?" + v.Encode(), nil
 }
 
 func (m fakeMethod) LogoutURL(c context.Context, dest string) (string, error) {
         v := url.Values{}
         v.Set("r", dest)
         return "http://logout_url?" + v.Encode(), nil
 }