package cache

import (
	"testing"
	"time"
)

func TestTTLCache_SetGet(t *testing.T) {
	c := New[string, int](time.Minute)
	c.Set("a", 42)

	v, ok := c.Get("a")
	if !ok || v != 42 {
		t.Fatalf("expected (42, true), got (%d, %v)", v, ok)
	}
}

func TestTTLCache_MissingKey(t *testing.T) {
	c := New[string, int](time.Minute)
	if _, ok := c.Get("missing"); ok {
		t.Fatal("expected miss for key that was never set")
	}
}

func TestTTLCache_Expiry(t *testing.T) {
	c := New[string, int](10 * time.Millisecond)
	c.Set("a", 1)

	if _, ok := c.Get("a"); !ok {
		t.Fatal("expected hit immediately after Set")
	}

	time.Sleep(30 * time.Millisecond)

	if _, ok := c.Get("a"); ok {
		t.Fatal("expected miss after TTL expired")
	}
}

func TestTTLCache_Delete(t *testing.T) {
	c := New[string, int](time.Minute)
	c.Set("a", 1)
	c.Delete("a")

	if _, ok := c.Get("a"); ok {
		t.Fatal("expected miss after Delete")
	}
}

func TestTTLCache_Clear(t *testing.T) {
	c := New[string, int](time.Minute)
	c.Set("a", 1)
	c.Set("b", 2)
	c.Clear()

	if _, ok := c.Get("a"); ok {
		t.Fatal("expected miss for 'a' after Clear")
	}
	if _, ok := c.Get("b"); ok {
		t.Fatal("expected miss for 'b' after Clear")
	}
}