41 lines
981 B
Go
41 lines
981 B
Go
|
//go:build unit
|
||
|
|
|
||
|
|
package redissession
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/alicebob/miniredis/v2"
|
||
|
|
"github.com/redis/go-redis/v9"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStoreRoundTripAndSingleUse(t *testing.T) {
|
||
|
|
mr := miniredis.RunT(t)
|
||
|
|
rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
|
||
|
|
t.Cleanup(func() { _ = rdb.Close() })
|
||
|
|
store := New(rdb, "oauth:test", time.Minute)
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
require.NoError(t, store.Set(ctx, "sid", map[string]string{"state": "state"}))
|
||
|
|
var got map[string]string
|
||
|
|
ok, err := store.Get(ctx, "sid", &got)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, ok)
|
||
|
|
require.Equal(t, "state", got["state"])
|
||
|
|
|
||
|
|
ok, err = store.TryConsume(ctx, "sid")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, ok)
|
||
|
|
ok, err = store.TryConsume(ctx, "sid")
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.False(t, ok)
|
||
|
|
|
||
|
|
require.NoError(t, store.Delete(ctx, "sid"))
|
||
|
|
ok, err = store.Get(ctx, "sid", &got)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.False(t, ok)
|
||
|
|
}
|