134 lines
2.4 KiB
Go
134 lines
2.4 KiB
Go
// Copyright (C) 2018 Marius Schellenberger
|
|
|
|
package jwt
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestValidate(t *testing.T) {
|
|
jwt, err := New(time.Second, NewMemBlacklist(), nil)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
token := NewToken(Claims{
|
|
"sub": "1234567890",
|
|
"name": "John Doe",
|
|
"admin": true,
|
|
"fizz": "buzz",
|
|
}, nil)
|
|
err = jwt.Sign(token)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = DecodeToken("")
|
|
if err == nil {
|
|
t.Error(errors.New("token is empty"))
|
|
}
|
|
nt, err := DecodeToken(token.String())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
err = jwt.Verify(nt)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
time.Sleep(time.Second * 2)
|
|
err = jwt.Verify(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("token should be expired"))
|
|
}
|
|
|
|
err = jwt.Invalidate(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("token is expired"))
|
|
}
|
|
|
|
err = jwt.Sign(nt)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
err = jwt.Verify(nt)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
err = jwt.Invalidate(nt)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
err = jwt.Invalidate(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("double invalidate"))
|
|
}
|
|
|
|
err = jwt.Verify(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("token should be blacklisted"))
|
|
}
|
|
err = jwt.Stop()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
// should not panic
|
|
err = jwt.Stop()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestNoBlacklist(t *testing.T) {
|
|
jwt, err := New(time.Second, nil, nil)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
token := NewToken(Claims{
|
|
"sub": "1234567890",
|
|
"name": "John Doe",
|
|
"admin": true,
|
|
"fizz": "buzz",
|
|
}, nil)
|
|
err = jwt.Sign(token)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = DecodeToken("")
|
|
if err == nil {
|
|
t.Error(errors.New("token is empty"))
|
|
}
|
|
nt, err := DecodeToken(token.String())
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
err = jwt.Verify(nt)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
time.Sleep(time.Second * 2)
|
|
err = jwt.Verify(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("token should be expired"))
|
|
}
|
|
|
|
err = jwt.Invalidate(nt)
|
|
if err == nil {
|
|
t.Error(errors.New("blacklisting should be disabled"))
|
|
}
|
|
}
|
|
|
|
func TestEmptySecretReader(t *testing.T) {
|
|
_, err := New(time.Second, nil, new(bytes.Buffer))
|
|
if err == nil {
|
|
t.Error(errors.New("error should be secret reader error"))
|
|
}
|
|
}
|
|
|
|
func TestInvalidSecretReader(t *testing.T) {
|
|
_, err := New(time.Second, nil, bytes.NewBufferString("123"))
|
|
if err != ErrInvalidKeySize {
|
|
t.Error(errors.New("error should be invalid key size"))
|
|
}
|
|
}
|