mirror of
https://github.com/empayre/fleet.git
synced 2024-11-06 17:05:18 +00:00
44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package sso
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fleetdm/fleet/v4/server/datastore/redis/redistest"
|
|
"github.com/fleetdm/fleet/v4/server/fleet"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSessionStore(t *testing.T) {
|
|
runTest := func(t *testing.T, pool fleet.RedisPool) {
|
|
store := NewSessionStore(pool)
|
|
require.NotNil(t, store)
|
|
// Create session that lives for 1 second.
|
|
err := store.create("request123", "https://originalurl.com", "some metadata", 1)
|
|
require.Nil(t, err)
|
|
sess, err := store.Get("request123")
|
|
require.Nil(t, err)
|
|
require.NotNil(t, sess)
|
|
assert.Equal(t, "https://originalurl.com", sess.OriginalURL)
|
|
assert.Equal(t, "some metadata", sess.Metadata)
|
|
// Wait a little bit more than one second, session should no longer be present.
|
|
time.Sleep(1100 * time.Millisecond)
|
|
sess, err = store.Get("request123")
|
|
assert.Equal(t, ErrSessionNotFound, err)
|
|
assert.Nil(t, sess)
|
|
}
|
|
|
|
t.Run("standalone", func(t *testing.T) {
|
|
p := redistest.SetupRedis(t, false, false, false)
|
|
require.NotNil(t, p)
|
|
runTest(t, p)
|
|
})
|
|
|
|
t.Run("cluster", func(t *testing.T) {
|
|
p := redistest.SetupRedis(t, true, false, false)
|
|
require.NotNil(t, p)
|
|
runTest(t, p)
|
|
})
|
|
}
|