1 files changed, 155 insertions, 0 deletions
diff --git a/store_test.go b/store_test.go
new file mode 100644
index 0000000..3ad33e5
--- /dev/null
+++ b/store_test.go
@@ -0,0 +1,155 @@
+package cache
+
+import (
+	"encoding/binary"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func setupTestStore(t testing.TB) *store {
+	t.Helper()
+
+	store := &store{}
+	store.Init()
+	return store
+}
+
+func TestStoreGetSet(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		want := []byte("Value")
+		store.Set([]byte("Key"), want, 1*time.Hour)
+		got, ttl, ok := store.Get([]byte("Key"))
+		assert.Equal(t, want, got)
+
+		now := time.Now()
+		assert.WithinDuration(t, now.Add(ttl), now.Add(1*time.Hour), 1*time.Millisecond)
+		assert.True(t, ok)
+	})
+
+	t.Run("Exists TTL", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		want := []byte("Value")
+		store.Set([]byte("Key"), want, time.Nanosecond)
+		_, _, ok := store.Get([]byte("Key"))
+		assert.False(t, ok)
+	})
+
+	t.Run("Not Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		_, _, ok := store.Get([]byte("Key"))
+		assert.False(t, ok)
+	})
+
+	t.Run("Update", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		store.Set([]byte("Key"), []byte("Other"), 0)
+		want := []byte("Value")
+		store.Set([]byte("Key"), want, 0)
+		got, _, ok := store.Get([]byte("Key"))
+		assert.Equal(t, want, got)
+		assert.True(t, ok)
+	})
+
+	t.Run("Resize", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		for i := range initialBucketSize {
+			key := binary.LittleEndian.AppendUint64(nil, i)
+			store.Set(key, key, 0)
+		}
+
+		for i := range store.Length {
+			key := binary.LittleEndian.AppendUint64(nil, i)
+			_, _, ok := store.Get(key)
+			assert.True(t, ok, i)
+		}
+
+		assert.Len(t, store.Bucket, int(initialBucketSize)*2)
+
+		for i := range store.Length {
+			key := binary.LittleEndian.AppendUint64(nil, i)
+			_, _, ok := store.Get(key)
+			assert.True(t, ok, i)
+		}
+	})
+}
+
+func TestStoreDelete(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		want := []byte("Value")
+		store.Set([]byte("Key"), want, 0)
+		ok := store.Delete([]byte("Key"))
+		assert.True(t, ok)
+		_, _, ok = store.Get([]byte("Key"))
+		assert.False(t, ok)
+	})
+
+	t.Run("Not Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := setupTestStore(t)
+
+		ok := store.Delete([]byte("Key"))
+		assert.False(t, ok)
+	})
+}
+
+func TestStoreClear(t *testing.T) {
+	t.Parallel()
+
+	store := setupTestStore(t)
+
+	want := []byte("Value")
+	store.Set([]byte("Key"), want, 0)
+	store.Clear()
+	_, _, ok := store.Get([]byte("Key"))
+	assert.False(t, ok)
+}
+
+func BenchmarkStoreGet(b *testing.B) {
+	store := setupTestStore(b)
+
+	key := []byte("Key")
+	store.Set(key, []byte("Store"), 0)
+	b.ReportAllocs()
+
+	for i := 0; i < b.N; i++ {
+		store.Get(key)
+	}
+}
+
+func BenchmarkStoreSet(b *testing.B) {
+	store := setupTestStore(b)
+
+	key := []byte("Key")
+	b.ReportAllocs()
+
+	for i := 0; i < b.N; i++ {
+		store.Set(key, []byte("Store"), 0)
+	}
+}