39 lines
922 B
Go
39 lines
922 B
Go
package token
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"testing"
|
|
)
|
|
|
|
func TestGenerateURLSafe(t *testing.T) {
|
|
t.Run("invalid size", func(t *testing.T) {
|
|
if _, err := GenerateURLSafe(0); err == nil {
|
|
t.Fatal("expected error for non-positive size")
|
|
}
|
|
})
|
|
|
|
t.Run("valid token decodes from raw url base64", func(t *testing.T) {
|
|
got, err := GenerateURLSafe(32)
|
|
if err != nil {
|
|
t.Fatalf("GenerateURLSafe: %v", err)
|
|
}
|
|
if got == "" {
|
|
t.Fatal("expected non-empty token")
|
|
}
|
|
raw, err := base64.RawURLEncoding.DecodeString(got)
|
|
if err != nil {
|
|
t.Fatalf("token was not raw-url-base64: %v", err)
|
|
}
|
|
if len(raw) != 32 {
|
|
t.Fatalf("expected 32 random bytes, got %d", len(raw))
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestHashSHA256(t *testing.T) {
|
|
const expected = "2cf24dba5fb0a30e26e83b2ac5b9e29e1b161e5c1fa7425e73043362938b9824"
|
|
if got := HashSHA256("hello"); got != expected {
|
|
t.Fatalf("expected %q, got %q", expected, got)
|
|
}
|
|
}
|