mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-09-20 02:22:55 +00:00
Knut Ahlers
9a93463bea
in order not to throw around Twitch tokens and have tokens expire Signed-off-by: Knut Ahlers <knut@ahlers.me>
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package editortoken
|
|
|
|
import (
|
|
"crypto/ed25519"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/Luzifer/twitch-bot/v3/pkg/database"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreateToken(t *testing.T) {
|
|
dbc := database.GetTestDatabase(t)
|
|
s := New(dbc)
|
|
|
|
// Fresh database, no key stored, the key should be generated and
|
|
// stored
|
|
pk1, err := s.getSigningKey()
|
|
require.NoError(t, err)
|
|
assert.IsType(t, ed25519.PrivateKey{}, pk1)
|
|
|
|
// Now database should contain key
|
|
var dbpk ed25519.PrivateKey
|
|
err = dbc.ReadCoreMeta(coreMetaSigningKey, &dbpk)
|
|
require.Error(t, err, "Key must not be readable with plain func")
|
|
err = dbc.ReadEncryptedCoreMeta(coreMetaSigningKey, &dbpk)
|
|
require.NoError(t, err)
|
|
|
|
// When fetching the key again it should be the same as before
|
|
pk2, err := s.getSigningKey()
|
|
require.NoError(t, err)
|
|
assert.Equal(t, pk1, pk2)
|
|
assert.Equal(t, dbpk, pk2)
|
|
}
|
|
|
|
func TestTokenFlow(t *testing.T) {
|
|
dbc := database.GetTestDatabase(t)
|
|
s := New(dbc)
|
|
|
|
var (
|
|
id = "123456"
|
|
user = "example"
|
|
)
|
|
|
|
tok, expiresAt, err := s.CreateLoginToken(id, user)
|
|
require.NoError(t, err)
|
|
assert.True(t, expiresAt.After(time.Now().Add(tokenValidity-time.Minute)))
|
|
|
|
tid, tuser, texpiresAt, err := s.ValidateLoginToken(tok)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, id, tid)
|
|
assert.Equal(t, user, tuser)
|
|
assert.Equal(t, expiresAt, texpiresAt)
|
|
}
|