✏ refactor storage

sqlite3/sqlite3_test.go

@@ -1,3 +1,5 @@
+// +build sqlite3
+
 package sqlite3
 
 import (
@@ -8,31 +10,113 @@ import (
 	_ "github.com/mattn/go-sqlite3"
 )
 
-func Test_New(t *testing.T) {
-	New()
+var testStore *Storage
+
+func init() {
+	testConfig := ConfigDefault
+	testConfig.Database = ":memory:"
+
+	testStore = New(testConfig)
 }
 
-func Test_Sqlite_Get_Set(t *testing.T) {
-	s := New(Config{
-		Database: ":memory:",
-	})
-	err := s.Set("fiber-10k-stars?", []byte("yes!"), time.Duration(time.Hour*1))
-	utils.AssertEqual(t, nil, err)
+func Test_Redis_Set(t *testing.T) {
+	var (
+		store = testStore
+		key   = "john"
+		val   = []byte("doe")
+	)
 
-	b, err := s.Get("fiber-10k-stars?")
+	err := store.Set(key, val, 0)
 	utils.AssertEqual(t, nil, err)
-	utils.AssertEqual(t, []byte("yes!"), b)
 }
 
-func Test_Sqlite_Expiration(t *testing.T) {
-	s := New(Config{
-		Database: ":memory:",
-	})
+func Test_Redis_Get(t *testing.T) {
+	var (
+		store = testStore
+		key   = "john"
+		val   = []byte("doe")
+	)
 
-	err := s.Set("fiber-20k-stars?", []byte("yes!"), time.Duration(time.Nanosecond/2))
+	err := store.Set(key, val, 0)
 	utils.AssertEqual(t, nil, err)
 
-	b, err := s.Get("fiber-220k-stars?")
+	result, err := store.Get(key)
 	utils.AssertEqual(t, nil, err)
-	utils.AssertEqual(t, true, b == nil)
+	utils.AssertEqual(t, val, result)
+}
+
+func Test_Redis_Set_Expiration(t *testing.T) {
+	var (
+		store = testStore
+		key   = "john"
+		val   = []byte("doe")
+		exp   = 500 * time.Millisecond
+	)
+
+	err := store.Set(key, val, exp)
+	utils.AssertEqual(t, nil, err)
+
+	time.Sleep(1 * time.Second)
+
+}
+
+func Test_Redis_Get_Expired(t *testing.T) {
+	var (
+		store = testStore
+		key   = "john"
+	)
+
+	result, err := store.Get(key)
+	utils.AssertEqual(t, ErrNotExist, err)
+	utils.AssertEqual(t, true, len(result) == 0)
+}
+
+func Test_Redis_Get_NotExist(t *testing.T) {
+	var store = testStore
+
+	result, err := store.Get("notexist")
+	utils.AssertEqual(t, ErrNotExist, err)
+	utils.AssertEqual(t, true, len(result) == 0)
+}
+
+func Test_Redis_Delete(t *testing.T) {
+	var (
+		store = testStore
+		key   = "john"
+		val   = []byte("doe")
+	)
+
+	err := store.Set(key, val, 0)
+	utils.AssertEqual(t, nil, err)
+
+	err = store.Delete(key)
+	utils.AssertEqual(t, nil, err)
+
+	result, err := store.Get(key)
+	utils.AssertEqual(t, ErrNotExist, err)
+	utils.AssertEqual(t, true, len(result) == 0)
+}
+
+func Test_Redis_Clear(t *testing.T) {
+	var (
+		store = testStore
+		val   = []byte("doe")
+	)
+
+	err := store.Set("john1", val, 0)
+	utils.AssertEqual(t, nil, err)
+
+	err = store.Set("john2", val, 0)
+	utils.AssertEqual(t, nil, err)
+
+	err = store.Clear()
+	utils.AssertEqual(t, nil, err)
+
+	result, err := store.Get("john1")
+	utils.AssertEqual(t, ErrNotExist, err)
+	utils.AssertEqual(t, true, len(result) == 0)
+
+	result, err = store.Get("john2")
+	utils.AssertEqual(t, ErrNotExist, err)
+	utils.AssertEqual(t, true, len(result) == 0)
 }