1 files changed, 157 insertions, 0 deletions
diff --git a/map_test.go b/map_test.go
new file mode 100644
index 0000000..b829fac
--- /dev/null
+++ b/map_test.go
@@ -0,0 +1,157 @@
+package cache
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestStoreGetSet(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := &New[any, any]().Store
+		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("Not Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := &New[any, any]().Store
+		_, _, ok := store.Get([]byte("Key"))
+		assert.False(t, ok)
+	})
+
+	t.Run("Update", func(t *testing.T) {
+		t.Parallel()
+
+		store := &New[any, any]().Store
+		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)
+	})
+}
+
+func TestStoreDelete(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := &New[any, any]().Store
+		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 := &New[any, any]().Store
+		ok := store.Delete([]byte("Key"))
+		assert.False(t, ok)
+	})
+}
+
+func TestStoreClear(t *testing.T) {
+	t.Parallel()
+
+	store := &New[any, any]().Store
+	want := []byte("Value")
+	store.Set([]byte("Key"), want, 0)
+	store.Clear()
+	_, _, ok := store.Get([]byte("Key"))
+	assert.False(t, ok)
+}
+
+func TestHashMapGetSet(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := New[string, string]()
+		want := "Value"
+		store.Set("Key", want, 0)
+		got, _, err := store.Get("Key")
+		assert.Equal(t, want, got)
+		assert.NoError(t, err)
+	})
+
+	t.Run("Not Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := New[string, string]()
+		_, _, err := store.Get("Key")
+		assert.ErrorIs(t, err, ErrKeyNotFound)
+	})
+
+	t.Run("Update", func(t *testing.T) {
+		t.Parallel()
+
+		store := New[string, string]()
+		store.Set("Key", "Other", 0)
+		want := "Value"
+		store.Set("Key", want, 0)
+		got, _, err := store.Get("Key")
+		assert.Equal(t, want, got)
+		assert.NoError(t, err)
+	})
+}
+
+func TestHashMapDelete(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := New[string, string]()
+		want := "Value"
+		store.Set("Key", want, 0)
+		err := store.Delete("Key")
+		assert.NoError(t, err)
+	})
+
+	t.Run("Not Exists", func(t *testing.T) {
+		t.Parallel()
+
+		store := New[string, string]()
+		err := store.Delete("Key")
+		assert.ErrorIs(t, err, ErrKeyNotFound)
+	})
+}
+
+func BenchmarkStoreGet(b *testing.B) {
+	store := &New[any, any]().Store
+	key := []byte("Key")
+	store.Set(key, []byte("Store"), 0)
+	b.SetBytes(1)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			store.Get(key)
+		}
+	})
+}
+
+func BenchmarkStoreSet(b *testing.B) {
+	store := &New[any, any]().Store
+	key := []byte("Key")
+	b.SetBytes(1)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			store.Set(key, []byte("Store"), 0)
+		}
+	})
+}