82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
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)
|
|
}
|
|
}
|