mirror of https://github.com/netbirdio/netbird.git
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
b64 "encoding/base64"
|
|
"hash/crc32"
|
|
"math/big"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/netbirdio/netbird/base62"
|
|
)
|
|
|
|
func TestPAT_GenerateToken_Hashing(t *testing.T) {
|
|
hashedToken, plainToken, _ := generateNewToken()
|
|
expectedToken := sha256.Sum256([]byte(plainToken))
|
|
encodedExpectedToken := b64.StdEncoding.EncodeToString(expectedToken[:])
|
|
assert.Equal(t, hashedToken, encodedExpectedToken)
|
|
}
|
|
|
|
func TestPAT_GenerateToken_Prefix(t *testing.T) {
|
|
_, plainToken, _ := generateNewToken()
|
|
fourCharPrefix := plainToken[:4]
|
|
assert.Equal(t, PATPrefix, fourCharPrefix)
|
|
}
|
|
|
|
func TestPAT_GenerateToken_Checksum(t *testing.T) {
|
|
_, plainToken, _ := generateNewToken()
|
|
tokenWithoutPrefix := strings.Split(plainToken, "_")[1]
|
|
if len(tokenWithoutPrefix) != 36 {
|
|
t.Fatal("Token has wrong length")
|
|
}
|
|
secret := tokenWithoutPrefix[:len(tokenWithoutPrefix)-6]
|
|
tokenCheckSum := tokenWithoutPrefix[len(tokenWithoutPrefix)-6:]
|
|
|
|
var i big.Int
|
|
i.SetString(secret, 62)
|
|
expectedChecksum := crc32.ChecksumIEEE([]byte(secret))
|
|
actualChecksum, err := base62.Decode(tokenCheckSum)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, expectedChecksum, actualChecksum)
|
|
}
|