redis_test.go

package redis

import (
	"context"
	"testing"
	"time"

	. "github.com/smartystreets/goconvey/convey"
)

const (
	addr = "localhost:6379"
	db   = 15
)

func TestStore(t *testing.T) {
	mstore := NewRedisStore(
		&Options{
			Addr: addr,
			DB:   db,
		},
	)
	defer mstore.Close()

	Convey("Test redis storage operation", t, func() {
		store, err := mstore.Create(context.Background(), "test_redis_store", 10)
		So(err, ShouldBeNil)
		foo, ok := store.Get("foo")
		So(ok, ShouldBeFalse)
		So(foo, ShouldBeNil)

		store.Set("foo", "bar")
		store.Set("foo2", "bar2")
		err = store.Save()
		So(err, ShouldBeNil)

		foo, ok = store.Get("foo")
		So(ok, ShouldBeTrue)
		So(foo, ShouldEqual, "bar")

		foo = store.Delete("foo")
		So(foo, ShouldEqual, "bar")

		foo, ok = store.Get("foo")
		So(ok, ShouldBeFalse)
		So(foo, ShouldBeNil)

		foo2, ok := store.Get("foo2")
		So(ok, ShouldBeTrue)
		So(foo2, ShouldEqual, "bar2")

		err = store.Flush()
		So(err, ShouldBeNil)

		foo2, ok = store.Get("foo2")
		So(ok, ShouldBeFalse)
		So(foo2, ShouldBeNil)
	})
}

func TestManagerStore(t *testing.T) {
	mstore := NewRedisStore(
		&Options{
			Addr: addr,
			DB:   db,
		},
	)
	defer mstore.Close()

	Convey("Test redis-based storage management operations", t, func() {
		sid := "test_manager_store"
		store, err := mstore.Create(context.Background(), sid, 10)
		So(store, ShouldNotBeNil)
		So(err, ShouldBeNil)

		store.Set("foo", "bar")
		err = store.Save()
		So(err, ShouldBeNil)

		store, err = mstore.Update(context.Background(), sid, 10)
		So(store, ShouldNotBeNil)
		So(err, ShouldBeNil)

		foo, ok := store.Get("foo")
		So(ok, ShouldBeTrue)
		So(foo, ShouldEqual, "bar")

		newsid := "test_manager_store2"
		store, err = mstore.Refresh(context.Background(), sid, newsid, 10)
		So(store, ShouldNotBeNil)
		So(err, ShouldBeNil)

		foo, ok = store.Get("foo")
		So(ok, ShouldBeTrue)
		So(foo, ShouldEqual, "bar")

		exists, err := mstore.Check(context.Background(), sid)
		So(exists, ShouldBeFalse)
		So(err, ShouldBeNil)

		err = mstore.Delete(context.Background(), newsid)
		So(err, ShouldBeNil)

		exists, err = mstore.Check(context.Background(), newsid)
		So(exists, ShouldBeFalse)
		So(err, ShouldBeNil)
	})
}

func TestStoreWithExpired(t *testing.T) {
	mstore := NewRedisStore(
		&Options{
			Addr: addr,
			DB:   db,
		},
	)
	defer mstore.Close()

	Convey("Test Redis Store Expiration", t, func() {
		sid := "test_store_expired"
		store, err := mstore.Create(context.Background(), sid, 1)
		So(store, ShouldNotBeNil)
		So(err, ShouldBeNil)

		store.Set("foo", "bar")
		err = store.Save()
		So(err, ShouldBeNil)

		store, err = mstore.Update(context.Background(), sid, 1)
		So(store, ShouldNotBeNil)
		So(err, ShouldBeNil)

		foo, ok := store.Get("foo")
		So(foo, ShouldEqual, "bar")
		So(ok, ShouldBeTrue)

		time.Sleep(time.Second * 2)

		exists, err := mstore.Check(context.Background(), sid)
		So(err, ShouldBeNil)
		So(exists, ShouldBeFalse)
	})
}