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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
|
// Code generated by tools/cmd/genjwa/main.go. DO NOT EDIT
package jwa_test
import (
"testing"
"github.com/lestrrat-go/jwx/v2/jwa"
"github.com/stretchr/testify/assert"
)
func TestKeyType(t *testing.T) {
t.Parallel()
t.Run(`accept jwa constant EC`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(jwa.EC), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.EC, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept the string EC`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept("EC"), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.EC, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept fmt.Stringer for EC`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(stringer{src: "EC"}), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.EC, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`stringification for EC`, func(t *testing.T) {
t.Parallel()
if !assert.Equal(t, "EC", jwa.EC.String(), `stringified value matches`) {
return
}
})
t.Run(`accept jwa constant OKP`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(jwa.OKP), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OKP, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept the string OKP`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept("OKP"), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OKP, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept fmt.Stringer for OKP`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(stringer{src: "OKP"}), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OKP, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`stringification for OKP`, func(t *testing.T) {
t.Parallel()
if !assert.Equal(t, "OKP", jwa.OKP.String(), `stringified value matches`) {
return
}
})
t.Run(`accept jwa constant OctetSeq`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(jwa.OctetSeq), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OctetSeq, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept the string oct`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept("oct"), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OctetSeq, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept fmt.Stringer for oct`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(stringer{src: "oct"}), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.OctetSeq, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`stringification for oct`, func(t *testing.T) {
t.Parallel()
if !assert.Equal(t, "oct", jwa.OctetSeq.String(), `stringified value matches`) {
return
}
})
t.Run(`accept jwa constant RSA`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(jwa.RSA), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.RSA, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept the string RSA`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept("RSA"), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.RSA, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`accept fmt.Stringer for RSA`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(stringer{src: "RSA"}), `accept is successful`) {
return
}
if !assert.Equal(t, jwa.RSA, dst, `accepted value should be equal to constant`) {
return
}
})
t.Run(`stringification for RSA`, func(t *testing.T) {
t.Parallel()
if !assert.Equal(t, "RSA", jwa.RSA.String(), `stringified value matches`) {
return
}
})
t.Run(`do not accept invalid constant InvalidKeyType`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.Error(t, dst.Accept(jwa.InvalidKeyType), `accept should fail`) {
return
}
})
t.Run(`bail out on random integer value`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.Error(t, dst.Accept(1), `accept should fail`) {
return
}
})
t.Run(`do not accept invalid (totally made up) string value`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.Error(t, dst.Accept(`totallyInvalidValue`), `accept should fail`) {
return
}
})
t.Run(`check list of elements`, func(t *testing.T) {
t.Parallel()
var expected = map[jwa.KeyType]struct{}{
jwa.EC: {},
jwa.OKP: {},
jwa.OctetSeq: {},
jwa.RSA: {},
}
for _, v := range jwa.KeyTypes() {
if _, ok := expected[v]; !assert.True(t, ok, `%s should be in the expected list`, v) {
return
}
delete(expected, v)
}
if !assert.Len(t, expected, 0) {
return
}
})
}
// Note: this test can NOT be run in parallel as it uses options with global effect.
func TestKeyTypeCustomAlgorithm(t *testing.T) {
// These subtests can NOT be run in parallel as options with global effect change.
customAlgorithm := jwa.KeyType("custom-algorithm")
// Unregister the custom algorithm, in case tests fail.
t.Cleanup(func() {
jwa.UnregisterKeyType(customAlgorithm)
})
t.Run(`with custom algorithm registered`, func(t *testing.T) {
jwa.RegisterKeyType(customAlgorithm)
t.Run(`accept variable used to register custom algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(customAlgorithm), `accept is successful`) {
return
}
assert.Equal(t, customAlgorithm, dst, `accepted value should be equal to variable`)
})
t.Run(`accept the string custom-algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(`custom-algorithm`), `accept is successful`) {
return
}
assert.Equal(t, customAlgorithm, dst, `accepted value should be equal to variable`)
})
t.Run(`accept fmt.Stringer for custom-algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
if !assert.NoError(t, dst.Accept(stringer{src: `custom-algorithm`}), `accept is successful`) {
return
}
assert.Equal(t, customAlgorithm, dst, `accepted value should be equal to variable`)
})
})
t.Run(`with custom algorithm deregistered`, func(t *testing.T) {
jwa.UnregisterKeyType(customAlgorithm)
t.Run(`reject variable used to register custom algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
assert.Error(t, dst.Accept(customAlgorithm), `accept failed`)
})
t.Run(`reject the string custom-algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
assert.Error(t, dst.Accept(`custom-algorithm`), `accept failed`)
})
t.Run(`reject fmt.Stringer for custom-algorithm`, func(t *testing.T) {
t.Parallel()
var dst jwa.KeyType
assert.Error(t, dst.Accept(stringer{src: `custom-algorithm`}), `accept failed`)
})
})
}
|
