1 files changed, 36 insertions, 0 deletions
diff --git a/internal/storage/replay_test.go b/internal/storage/replay_test.go
new file mode 100644
index 0000000..4d147b8
--- /dev/null
+++ b/internal/storage/replay_test.go
@@ -0,0 +1,36 @@
+package storage
+
+import (
+	"testing"
+	"time"
+)
+
+func TestReplayCacheCheckAndMarkLivesOnlyInMemory(t *testing.T) {
+	cache := NewReplayCache(40 * time.Millisecond)
+
+	replayed, err := cache.CheckAndMark("token-hash")
+	if err != nil {
+		t.Fatalf("CheckAndMark returned error: %v", err)
+	}
+	if replayed {
+		t.Fatal("first token use should not be marked as replayed")
+	}
+
+	replayed, err = cache.CheckAndMark("token-hash")
+	if err != nil {
+		t.Fatalf("CheckAndMark returned error on second use: %v", err)
+	}
+	if !replayed {
+		t.Fatal("second token use should be marked as replayed")
+	}
+
+	time.Sleep(60 * time.Millisecond)
+
+	replayed, err = cache.CheckAndMark("token-hash")
+	if err != nil {
+		t.Fatalf("CheckAndMark returned error after ttl expiry: %v", err)
+	}
+	if replayed {
+		t.Fatal("expired token entry should be forgotten from memory")
+	}
+}