@@ -314,6 +314,11 @@ func TestJWT(t *testing.T) {
314
314
ss , _ := jwt .NewWithClaims (method , testClaim (input )).SignedString (secret )
315
315
return ss
316
316
}
317
+ srv := createAndStartServer (t , & httpConfig {jwtSecret : []byte ("secret" )},
318
+ true , & wsConfig {Origins : []string {"*" }, jwtSecret : []byte ("secret" )})
319
+ wsUrl := fmt .Sprintf ("ws://%v" , srv .listenAddr ())
320
+ htUrl := fmt .Sprintf ("http://%v" , srv .listenAddr ())
321
+
317
322
expOk := []string {
318
323
fmt .Sprintf ("Bearer %v" , issueToken (secret , nil , testClaim {"iat" : time .Now ().Unix ()})),
319
324
fmt .Sprintf ("Bearer %v" , issueToken (secret , nil , testClaim {"iat" : time .Now ().Unix () + 4 })),
@@ -327,6 +332,14 @@ func TestJWT(t *testing.T) {
327
332
"bar" : "baz" ,
328
333
})),
329
334
}
335
+ for i , token := range expOk {
336
+ if err := wsRequest (t , wsUrl , "Authorization" , token ); err != nil {
337
+ t .Errorf ("test %d-ws, token '%v': expected ok, got %v" , i , token , err )
338
+ }
339
+ if resp := rpcRequest (t , htUrl , "Authorization" , token ); resp .StatusCode != 200 {
340
+ t .Errorf ("test %d-http, token '%v': expected ok, got %v" , i , token , resp .StatusCode )
341
+ }
342
+ }
330
343
expFail := []string {
331
344
// future
332
345
fmt .Sprintf ("Bearer %v" , issueToken (secret , nil , testClaim {"iat" : time .Now ().Unix () + 6 })),
@@ -351,19 +364,6 @@ func TestJWT(t *testing.T) {
351
364
fmt .Sprintf ("Bearer\t %v" , issueToken (secret , nil , testClaim {"iat" : time .Now ().Unix ()})),
352
365
fmt .Sprintf ("Bearer \t %v" , issueToken (secret , nil , testClaim {"iat" : time .Now ().Unix ()})),
353
366
}
354
- srv := createAndStartServer (t , & httpConfig {jwtSecret : []byte ("secret" )},
355
- true , & wsConfig {Origins : []string {"*" }, jwtSecret : []byte ("secret" )})
356
- wsUrl := fmt .Sprintf ("ws://%v" , srv .listenAddr ())
357
- htUrl := fmt .Sprintf ("http://%v" , srv .listenAddr ())
358
-
359
- for i , token := range expOk {
360
- if err := wsRequest (t , wsUrl , "Authorization" , token ); err != nil {
361
- t .Errorf ("test %d-ws, token '%v': expected ok, got %v" , i , token , err )
362
- }
363
- if resp := rpcRequest (t , htUrl , "Authorization" , token ); resp .StatusCode != 200 {
364
- t .Errorf ("test %d-http, token '%v': expected ok, got %v" , i , token , resp .StatusCode )
365
- }
366
- }
367
367
for i , token := range expFail {
368
368
if err := wsRequest (t , wsUrl , "Authorization" , token ); err == nil {
369
369
t .Errorf ("tc %d-ws, token '%v': expected not to allow, got ok" , i , token )
0 commit comments