gogobee/internal/plugin/combat_engine_party_test.go

package plugin

import (
	"math/rand/v2"
	"testing"

	"maunium.net/go/mautrix/id"
)

// seededRNG gives each test its own deterministic stream, so none of these can
// join the flaky set that rode the package global before P4 seeded them.
func seededRNG(seed uint64) *rand.Rand {
	return rand.New(rand.NewPCG(seed, seed^0x9e3779b9))
}

// The load-bearing claim of P6e: a one-seat roster draws from the RNG in exactly
// the pre-roster order, so SimulateCombat is the degenerate case of the N-body
// engine and the d8prereq_corpus baselines still compare. TestCombatCharacterization
// is the real proof (57 scenarios); this pins the delegation itself.
func TestSimulateCombat_IsTheOneSeatPartyCase(t *testing.T) {
	for seed := uint64(1); seed <= 40; seed++ {
		solo := simulateCombatWithRNG(basePlayer(), baseEnemy(), dungeonCombatPhases, seededRNG(seed))
		party := simulatePartyWithRNG([]Combatant{basePlayer()}, baseEnemy(), dungeonCombatPhases, seededRNG(seed))

		if len(party.Seats) != 1 {
			t.Fatalf("seed %d: one player seated %d seats", seed, len(party.Seats))
		}
		got := party.Seats[0]
		if got.PlayerWon != solo.PlayerWon || got.PlayerEndHP != solo.PlayerEndHP ||
			got.EnemyEndHP != solo.EnemyEndHP || got.TotalRounds != solo.TotalRounds ||
			got.TimedOut != solo.TimedOut {
			t.Fatalf("seed %d: one-seat party diverged from solo\n solo=%+v\nparty=%+v", seed, solo, got)
		}
		if len(got.Events) != len(solo.Events) {
			t.Fatalf("seed %d: event count %d != solo %d", seed, len(got.Events), len(solo.Events))
		}
		for i := range solo.Events {
			if got.Events[i] != solo.Events[i] {
				t.Fatalf("seed %d: event %d differs\n solo=%+v\nparty=%+v", seed, i, solo.Events[i], got.Events[i])
			}
		}
	}
}

// A solo fight never stamps a seat, because seat 0 is the zero value and the
// field is omitempty. If this regresses, the golden file moves.
func TestSimulateCombat_SoloEventsCarryNoSeat(t *testing.T) {
	res := simulateCombatWithRNG(basePlayer(), baseEnemy(), dungeonCombatPhases, seededRNG(7))
	for i, e := range res.Events {
		if e.Seat != 0 {
			t.Fatalf("event %d (%s/%s) stamped seat %d in a solo fight", i, e.Phase, e.Action, e.Seat)
		}
	}
}

// The whole point of P6e. Three characters swinging at one monster must land more
// player attacks per round than one character does — before this, the members
// stood in the doorway while the leader fought.
func TestSimulateParty_EverySeatSwings(t *testing.T) {
	tank := baseEnemy()
	tank.Stats.MaxHP = 5000 // outlast the phase clock so we can count swings

	countSwings := func(events []CombatEvent, seat int) int {
		n := 0
		for _, e := range events {
			if e.Actor == "player" && e.Seat == seat && e.Roll > 0 {
				n++
			}
		}
		return n
	}

	res := simulatePartyWithRNG(
		[]Combatant{basePlayer(), basePlayer(), basePlayer()}, tank, dungeonCombatPhases, seededRNG(11))

	if len(res.Seats) != 3 {
		t.Fatalf("seated %d, want 3", len(res.Seats))
	}
	for seat := range 3 {
		if got := countSwings(res.Events, seat); got == 0 {
			t.Fatalf("seat %d never swung — the party is not fighting together", seat)
		}
	}
}

// A member going down is not the end of the fight. Only an empty roster is.
// This is the bug P7 measured from the outside: every party loss was the leader
// dying alone, because a member was never in the fight to begin with.
func TestSimulateParty_DownedMemberDoesNotEndTheFight(t *testing.T) {
	glass := basePlayer()
	glass.Stats.MaxHP = 1
	glass.Stats.Defense = 0

	brute := basePlayer()
	brute.Stats.MaxHP = 4000

	killer := baseEnemy()
	killer.Stats.MaxHP = 3000
	killer.Stats.Attack = 40

	res := simulatePartyWithRNG([]Combatant{brute, glass}, killer, dungeonCombatPhases, seededRNG(3))

	if res.Seats[1].PlayerEndHP > 0 {
		t.Skip("the glass cannon survived this seed; nothing to assert")
	}
	if !res.AnySurvivor() {
		t.Fatalf("both seats down — pick a seed where the brute lives")
	}
	if res.TotalRounds <= 1 {
		t.Fatalf("fight ended in %d round(s) when a member fell; the roster should have fought on",
			res.TotalRounds)
	}
	// The brute must have kept swinging after the member fell.
	lastMemberEvent, lastBruteEvent := -1, -1
	for i, e := range res.Events {
		if e.Seat == 1 && e.Actor == "player" {
			lastMemberEvent = i
		}
		if e.Seat == 0 && e.Actor == "player" && e.Roll > 0 {
			lastBruteEvent = i
		}
	}
	if lastBruteEvent < lastMemberEvent {
		t.Fatalf("the fight stopped when seat 1 fell (last brute swing %d, last member event %d)",
			lastBruteEvent, lastMemberEvent)
	}
}

// The enemy swings at one seat per round, not at everyone.
func TestSimulateParty_EnemyStrikesOneSeatPerRound(t *testing.T) {
	tank := baseEnemy()
	tank.Stats.MaxHP = 5000

	res := simulatePartyWithRNG(
		[]Combatant{basePlayer(), basePlayer(), basePlayer()}, tank, dungeonCombatPhases, seededRNG(23))

	perRound := map[int]map[int]bool{}
	for _, e := range res.Events {
		if e.Actor != "enemy" || e.Roll == 0 {
			continue
		}
		if perRound[e.Round] == nil {
			perRound[e.Round] = map[int]bool{}
		}
		perRound[e.Round][e.Seat] = true
	}
	if len(perRound) == 0 {
		t.Fatal("the enemy never attacked")
	}
	for round, seats := range perRound {
		if len(seats) != 1 {
			t.Fatalf("round %d: enemy attack rolls landed on %d seats, want 1", round, len(seats))
		}
	}
}

// Over many seeds the enemy must spread its attention across the roster, or
// "targeting" is really "always seat 0" and members take no risk.
func TestSimulateParty_EnemySpreadsItsTargetsAcrossTheRoster(t *testing.T) {
	hit := map[int]bool{}
	for seed := uint64(1); seed <= 30; seed++ {
		tank := baseEnemy()
		tank.Stats.MaxHP = 5000
		res := simulatePartyWithRNG(
			[]Combatant{basePlayer(), basePlayer(), basePlayer()}, tank, dungeonCombatPhases, seededRNG(seed))
		for _, e := range res.Events {
			if e.Actor == "enemy" && e.Roll > 0 {
				hit[e.Seat] = true
			}
		}
	}
	for seat := range 3 {
		if !hit[seat] {
			t.Fatalf("seat %d was never targeted across 30 seeds — the enemy is not really choosing", seat)
		}
	}
}

// Per-seat close-out reads its own events, not the party's. If eventsForSeat
// leaked, a member's potions would be burned for the leader's heals.
func TestEventsForSeat_PartitionsTheLog(t *testing.T) {
	events := []CombatEvent{
		{Action: "heal_item", Seat: 0},
		{Action: "heal_item", Seat: 1},
		{Action: "heal_item", Seat: 1},
		{Action: "regen_tick"}, // unstamped: reads as the leader's
	}
	if got := len(eventsForSeat(events, 0)); got != 2 {
		t.Fatalf("seat 0 saw %d events, want 2", got)
	}
	if got := countHealEventsFired(CombatResult{Events: eventsForSeat(events, 1)}); got != 2 {
		t.Fatalf("seat 1 fired %d heal items, want 2", got)
	}
	if got := countHealEventsFired(CombatResult{Events: eventsForSeat(events, 0)}); got != 1 {
		t.Fatalf("seat 0 fired %d heal items, want 1", got)
	}
}

// A party that wins with a casualty is a win, and the casualty is still a
// casualty. partySurvivors is what the close-out splits on.
func TestPartySurvivors_SplitsOnHPNotOnOutcome(t *testing.T) {
	res := PartyCombatResult{
		PlayerWon: true,
		Seats: []CombatResult{
			{PlayerEndHP: 12},
			{PlayerEndHP: 0},
			{PlayerEndHP: 3},
		},
	}
	up, down := partySurvivors(res, []id.UserID{"@a:x", "@b:x", "@c:x"})
	if len(up) != 2 || len(down) != 1 {
		t.Fatalf("split %d up / %d down, want 2/1", len(up), len(down))
	}
	if down[0].String() != "@b:x" {
		t.Fatalf("wrong casualty: %s", down[0])
	}
	if !res.AnySurvivor() {
		t.Fatal("AnySurvivor said nobody lived")
	}
}