/
token_test.go
84 lines (69 loc) · 1.93 KB
/
token_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// SPDX-FileCopyrightText: 2020 Comcast Cable Communications Management, LLC
// SPDX-License-Identifier: Apache-2.0
package bascule
import (
"context"
"errors"
"testing"
"github.com/stretchr/testify/suite"
)
type TokenParsersSuite struct {
TestSuite
}
func (suite *TokenParsersSuite) assertUnsupportedScheme(scheme Scheme, err error) {
var use *UnsupportedSchemeError
if suite.ErrorAs(err, &use) {
suite.Equal(scheme, use.Scheme)
}
}
func (suite *TokenParsersSuite) testParseEmpty() {
var tp TokenParsers
// legal, but will always fail
token, err := tp.Parse(context.Background(), suite.testCredentials())
suite.Nil(token)
suite.assertUnsupportedScheme(testScheme, err)
}
func (suite *TokenParsersSuite) testParseUnsupported() {
var tp TokenParsers
tp.Register(
Scheme("Supported"),
TokenParserFunc(
func(context.Context, Credentials) (Token, error) {
suite.Fail("TokenParser should not have been called")
return nil, nil
},
),
)
token, err := tp.Parse(context.Background(), suite.testCredentials())
suite.Nil(token)
suite.assertUnsupportedScheme(testScheme, err)
}
func (suite *TokenParsersSuite) testParseSupported() {
var (
expectedErr = errors.New("expected Parse error")
testCtx = suite.testContext()
testCredentials = suite.testCredentials()
)
var tp TokenParsers
tp.Register(
testCredentials.Scheme,
TokenParserFunc(
func(ctx context.Context, c Credentials) (Token, error) {
suite.Equal(testCtx, ctx)
suite.Equal(testCredentials, c)
return suite.testToken(), expectedErr
},
),
)
token, err := tp.Parse(testCtx, testCredentials)
suite.Equal(suite.testToken(), token)
suite.Same(expectedErr, err)
}
func (suite *TokenParsersSuite) TestParse() {
suite.Run("Empty", suite.testParseEmpty)
suite.Run("Unsupported", suite.testParseUnsupported)
suite.Run("Supported", suite.testParseSupported)
}
func TestTokenParsers(t *testing.T) {
suite.Run(t, new(TokenParsersSuite))
}