package runtime

import (
	"testing"
	"time"

	"neuralhunt/internal/data"
)

func TestPresenceSingleSession(t *testing.T) {
	s := New()
	lease1, err := s.AcquirePresence("c", "s1")
	if err != nil {
		t.Fatal(err)
	}
	if _, err := s.AcquirePresence("c", "s2"); err == nil {
		t.Fatal("expected conflict")
	}
	if !s.HasPresence("c", "s1") {
		t.Fatal("missing presence")
	}
	s.ReleasePresence("c", "s1", lease1)
	if s.ConnectedCount() != 0 {
		t.Fatal("presence not released")
	}
}

func TestPresenceReconnectOldReleaseCannotDeleteNewLease(t *testing.T) {
	s := New()
	oldLease, err := s.AcquirePresence("c", "same-session")
	if err != nil {
		t.Fatal(err)
	}
	newLease, err := s.AcquirePresence("c", "same-session")
	if err != nil {
		t.Fatal(err)
	}
	if oldLease == newLease {
		t.Fatal("expected a new lease generation")
	}

	// This is the race that caused periodic HTTP 409 responses: the old
	// websocket's defer ran after the replacement websocket had connected.
	s.ReleasePresence("c", "same-session", oldLease)
	if !s.HasPresence("c", "same-session") {
		t.Fatal("stale websocket release removed the replacement presence")
	}

	s.ReleasePresence("c", "same-session", newLease)
	if s.ConnectedCount() != 0 {
		t.Fatal("new lease was not released")
	}
}

func TestGuessHotPathAndReroll(t *testing.T) {
	s := New()
	task := data.Task{ID: "t", PublicSeed: "a", Revision: 1}
	s.InitGuess(task, "c", GuessState{})
	r, err := s.Accept(task, "c", 0, 12.5, time.Millisecond)
	if err != nil {
		t.Fatal(err)
	}
	if !r.Improved || r.State.NextSeq != 1 || r.State.GuessCount != 1 {
		t.Fatalf("bad state: %+v", r)
	}
	if _, err = s.Accept(task, "c", 1, 10, time.Hour); err != ErrRateLimited {
		t.Fatalf("want rate limit, got %v", err)
	}
	// Same seed/admin revision preserves sequence.
	task.Revision = 2
	g := s.InitGuess(task, "c", GuessState{})
	if g.NextSeq != 1 {
		t.Fatalf("sequence lost: %d", g.NextSeq)
	}
	// New seed is a reroll and reloads the persisted reset.
	task.PublicSeed = "b"
	g = s.InitGuess(task, "c", GuessState{})
	if g.NextSeq != 0 || g.BestScore != 0 {
		t.Fatalf("reroll not reset: %+v", g)
	}
}