Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(189)

Side by Side Diff: common/auth/auth_test.go

Issue 2951393002: [errors] de-specialize Transient in favor of Tags. (Closed)
Patch Set: copyright Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2015 The LUCI Authors. All rights reserved. 1 // Copyright 2015 The LUCI Authors. All rights reserved.
2 // Use of this source code is governed under the Apache License, Version 2.0 2 // Use of this source code is governed under the Apache License, Version 2.0
3 // that can be found in the LICENSE file. 3 // that can be found in the LICENSE file.
4 4
5 package auth 5 package auth
6 6
7 import ( 7 import (
8 "fmt"
9 "io/ioutil" 8 "io/ioutil"
10 "math/rand" 9 "math/rand"
11 "net/http" 10 "net/http"
12 "net/http/httptest" 11 "net/http/httptest"
13 "testing" 12 "testing"
14 "time" 13 "time"
15 14
16 "golang.org/x/net/context" 15 "golang.org/x/net/context"
17 "golang.org/x/oauth2" 16 "golang.org/x/oauth2"
18 17
19 "github.com/luci/luci-go/common/auth/internal" 18 "github.com/luci/luci-go/common/auth/internal"
20 "github.com/luci/luci-go/common/clock" 19 "github.com/luci/luci-go/common/clock"
21 "github.com/luci/luci-go/common/clock/testclock" 20 "github.com/luci/luci-go/common/clock/testclock"
22 "github.com/luci/luci-go/common/data/rand/mathrand" 21 "github.com/luci/luci-go/common/data/rand/mathrand"
23 "github.com/luci/luci-go/common/errors" 22 "github.com/luci/luci-go/common/errors"
23 "github.com/luci/luci-go/common/retry"
24 24
25 . "github.com/luci/luci-go/common/testing/assertions" 25 . "github.com/luci/luci-go/common/testing/assertions"
26 . "github.com/smartystreets/goconvey/convey" 26 . "github.com/smartystreets/goconvey/convey"
27 ) 27 )
28 28
29 var ( 29 var (
30 now = time.Date(2015, time.January, 1, 0, 0, 0, 0, time.UTC) 30 now = time.Date(2015, time.January, 1, 0, 0, 0, 0, time.UTC)
31 past = time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC) 31 past = time.Date(2000, time.January, 1, 0, 0, 0, 0, time.UTC)
32 future = now.Add(24 * time.Hour) 32 future = now.Add(24 * time.Hour)
33 ) 33 )
(...skipping 550 matching lines...) Expand 10 before | Expand all | Expand 10 after
584 return p.tokenToMint, nil 584 return p.tokenToMint, nil
585 } 585 }
586 return &oauth2.Token{AccessToken: "some minted token"}, nil 586 return &oauth2.Token{AccessToken: "some minted token"}, nil
587 } 587 }
588 588
589 func (p *fakeTokenProvider) RefreshToken(ctx context.Context, prev, base *oauth2 .Token) (*oauth2.Token, error) { 589 func (p *fakeTokenProvider) RefreshToken(ctx context.Context, prev, base *oauth2 .Token) (*oauth2.Token, error) {
590 p.refreshTokenCalled = true 590 p.refreshTokenCalled = true
591 p.baseTokenInRefresh = base 591 p.baseTokenInRefresh = base
592 if p.transientRefreshErrors != 0 { 592 if p.transientRefreshErrors != 0 {
593 p.transientRefreshErrors-- 593 p.transientRefreshErrors--
594 » » return nil, errors.WrapTransient(fmt.Errorf("transient error")) 594 » » return nil, errors.New("transient error", retry.Tag)
595 } 595 }
596 if p.revokedCreds { 596 if p.revokedCreds {
597 return nil, internal.ErrBadCredentials 597 return nil, internal.ErrBadCredentials
598 } 598 }
599 if p.revokedToken { 599 if p.revokedToken {
600 return nil, internal.ErrBadRefreshToken 600 return nil, internal.ErrBadRefreshToken
601 } 601 }
602 if p.tokenToRefresh != nil { 602 if p.tokenToRefresh != nil {
603 return p.tokenToRefresh, nil 603 return p.tokenToRefresh, nil
604 } 604 }
605 return &oauth2.Token{AccessToken: "some refreshed token"}, nil 605 return &oauth2.Token{AccessToken: "some refreshed token"}, nil
606 } 606 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698