mirror of
https://github.com/empayre/fleet.git
synced 2024-11-06 17:05:18 +00:00
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package sso
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gomodule/redigo/redis"
|
|
"github.com/kolide/fleet/server/pubsub"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newPool(t *testing.T) *redis.Pool {
|
|
if _, ok := os.LookupEnv("REDIS_TEST"); ok {
|
|
var (
|
|
addr = "127.0.0.1:6379"
|
|
password = ""
|
|
)
|
|
if a, ok := os.LookupEnv("REDIS_PORT_6379_TCP_ADDR"); ok {
|
|
addr = fmt.Sprintf("%s:6379", a)
|
|
}
|
|
|
|
p := pubsub.NewRedisPool(addr, password)
|
|
_, err := p.Get().Do("PING")
|
|
require.Nil(t, err)
|
|
return p
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func TestSessionStore(t *testing.T) {
|
|
if _, ok := os.LookupEnv("REDIS_TEST"); !ok {
|
|
t.Skip("skipping sso session store tests")
|
|
}
|
|
p := newPool(t)
|
|
require.NotNil(t, p)
|
|
defer p.Close()
|
|
store := NewSessionStore(p)
|
|
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)
|
|
}
|