gogobee/internal/plugin/dnd_subclass_mage_test.go

package plugin

import (
	"testing"

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

// Phase 10 SUB2b — Evocation, Abjuration, Necromancy.

// ── Evocation ───────────────────────────────────────────────────────────

func TestEvocation_EmpoweredEvocationAddsINTMod(t *testing.T) {
	spell, _ := lookupSpell("fire_bolt") // school=evocation
	c := &DnDCharacter{
		Class: ClassMage, Subclass: SubclassEvocation, Level: 7, INT: 18, // +4 mod
	}
	mods := &CombatModifiers{SpellPreDamage: 5}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.SpellPreDamage != 9 {
		t.Errorf("Empowered Evocation L7 INT 18: SpellPreDamage = %d, want 9 (5+4)", mods.SpellPreDamage)
	}
}

func TestEvocation_EmpoweredEvocationMinBonusOne(t *testing.T) {
	spell, _ := lookupSpell("fire_bolt")
	c := &DnDCharacter{
		Class: ClassMage, Subclass: SubclassEvocation, Level: 7, INT: 8, // -1 mod
	}
	mods := &CombatModifiers{SpellPreDamage: 5}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.SpellPreDamage != 6 {
		t.Errorf("Empowered Evocation INT 8 (negative mod): SpellPreDamage = %d, want 6 (min +1)", mods.SpellPreDamage)
	}
}

func TestEvocation_EmpoweredEvocationPreL7Skipped(t *testing.T) {
	spell, _ := lookupSpell("fire_bolt")
	c := &DnDCharacter{
		Class: ClassMage, Subclass: SubclassEvocation, Level: 6, INT: 18,
	}
	mods := &CombatModifiers{SpellPreDamage: 5}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.SpellPreDamage != 5 {
		t.Errorf("Empowered Evocation should not fire pre-L7: SpellPreDamage = %d, want 5", mods.SpellPreDamage)
	}
}

func TestEvocation_EmpoweredEvocationOnlyForEvocationSpells(t *testing.T) {
	spell, _ := lookupSpell("chill_touch") // school=necromancy
	c := &DnDCharacter{
		Class: ClassMage, Subclass: SubclassEvocation, Level: 7, INT: 18,
	}
	mods := &CombatModifiers{SpellPreDamage: 5}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.SpellPreDamage != 5 {
		t.Errorf("Empowered Evocation should not fire on non-evocation school: SpellPreDamage = %d, want 5", mods.SpellPreDamage)
	}
}

// ── Abjuration ──────────────────────────────────────────────────────────

func TestAbjuration_ArcaneWardL5(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 5}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.ArcaneWardHP != 10 {
		t.Errorf("Abjuration L5 ArcaneWardHP = %d, want 10 (2× level)", mods.ArcaneWardHP)
	}
}

func TestAbjuration_ArcaneWardL7AddsProf(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 7}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	// 2*7 + prof(L7=3) = 17.
	if mods.ArcaneWardHP != 17 {
		t.Errorf("Abjuration L7 ArcaneWardHP = %d, want 17 (14 + prof 3)", mods.ArcaneWardHP)
	}
}

func TestAbjuration_PreL5NoWard(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 4}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.ArcaneWardHP != 0 {
		t.Errorf("Abjuration pre-L5 ArcaneWardHP = %d, want 0", mods.ArcaneWardHP)
	}
}

// End-to-end: a sturdy Arcane Ward should reduce damage taken when the
// engine resolves enemy attacks.
func TestSimulateCombat_ArcaneWardAbsorbsDamage(t *testing.T) {
	const trials = 600
	build := func(ward int) Combatant {
		return Combatant{
			Name: "Mage", IsPlayer: true,
			Stats: CombatStats{
				MaxHP: 60, Attack: 10, Defense: 8, Speed: 10, AttackBonus: 4, AC: 14,
				CritRate: 0.05, DodgeRate: 0.05, BlockRate: 0.05,
			},
			Mods: CombatModifiers{DamageReduct: 1.0, ArcaneWardHP: ward},
		}
	}
	enemy := Combatant{
		Name: "Brute",
		Stats: CombatStats{
			MaxHP: 80, Attack: 12, Defense: 6, Speed: 8, AC: 13, AttackBonus: 4,
			CritRate: 0.05, DodgeRate: 0.05, BlockRate: 0.05,
		},
		Mods: CombatModifiers{DamageReduct: 1.0},
	}
	plainEndHP, wardEndHP := 0, 0
	for i := 0; i < trials; i++ {
		plainEndHP += SimulateCombat(build(0), enemy, defaultCombatPhases).PlayerEndHP
		wardEndHP += SimulateCombat(build(40), enemy, defaultCombatPhases).PlayerEndHP
	}
	if wardEndHP <= plainEndHP {
		t.Errorf("Arcane Ward should improve survivability: plain=%d ward=%d (sum HP)", plainEndHP, wardEndHP)
	}
}

// ── Necromancy ──────────────────────────────────────────────────────────

func TestNecromancy_GrimHarvestStashOnDamage(t *testing.T) {
	spell, _ := lookupSpell("chill_touch") // necrotic, school necromancy
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 5, INT: 16}
	mods := &CombatModifiers{SpellPreDamage: 8}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.GrimHarvestSlot != 1 {
		t.Errorf("cantrip Grim Harvest slot = %d, want 1 (cantrip floor)", mods.GrimHarvestSlot)
	}
	if !mods.GrimHarvestNecrotic {
		t.Error("chill_touch should mark GrimHarvestNecrotic")
	}
}

func TestNecromancy_GrimHarvestStashLeveledSpell(t *testing.T) {
	spell, _ := lookupSpell("magic_missile") // force damage; school evocation
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 5, INT: 16}
	mods := &CombatModifiers{SpellPreDamage: 12}
	applyMageSubclassSpellHooks(c, spell, 3, mods)
	if mods.GrimHarvestSlot != 3 {
		t.Errorf("magic_missile L3 Grim Harvest slot = %d, want 3", mods.GrimHarvestSlot)
	}
	if mods.GrimHarvestNecrotic {
		t.Error("magic_missile is force, not necrotic — should not set Necrotic flag")
	}
}

func TestNecromancy_GrimHarvestPreL5NoStash(t *testing.T) {
	spell, _ := lookupSpell("chill_touch")
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 4, INT: 16}
	mods := &CombatModifiers{SpellPreDamage: 8}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	if mods.GrimHarvestSlot != 0 {
		t.Errorf("pre-L5 Necromancer should not stash; slot = %d, want 0", mods.GrimHarvestSlot)
	}
}

func TestGrimHarvestHeal_NecroticTriple(t *testing.T) {
	c := &DnDCharacter{
		Class: ClassMage, Subclass: SubclassNecromancy, Level: 5,
	}
	result := CombatResult{
		PlayerWon: true,
		Events: []CombatEvent{
			{Action: "spell_cast", EnemyHP: 0},
		},
	}
	mods := CombatModifiers{GrimHarvestSlot: 2, GrimHarvestNecrotic: true}
	if got := grimHarvestHeal(c, result, mods); got != 6 {
		t.Errorf("necrotic Grim Harvest L2 = %d, want 6 (3× slot)", got)
	}
}

func TestGrimHarvestHeal_NonNecroticDouble(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 5}
	result := CombatResult{
		PlayerWon: true,
		Events:    []CombatEvent{{Action: "spell_cast", EnemyHP: 0}},
	}
	mods := CombatModifiers{GrimHarvestSlot: 3}
	if got := grimHarvestHeal(c, result, mods); got != 6 {
		t.Errorf("non-necrotic Grim Harvest L3 = %d, want 6 (2× slot)", got)
	}
}

func TestGrimHarvestHeal_OnlyOnSpellKill(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 5}
	// Spell hit but didn't kill — combat finished with regular attack.
	result := CombatResult{
		PlayerWon: true,
		Events:    []CombatEvent{{Action: "spell_cast", EnemyHP: 5}},
	}
	mods := CombatModifiers{GrimHarvestSlot: 2, GrimHarvestNecrotic: true}
	if got := grimHarvestHeal(c, result, mods); got != 0 {
		t.Errorf("Grim Harvest fires on non-spell-kill: got %d, want 0", got)
	}
}

func TestGrimHarvestHeal_LossNoHeal(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 5}
	result := CombatResult{
		PlayerWon: false,
		Events:    []CombatEvent{{Action: "spell_cast", EnemyHP: 0}},
	}
	mods := CombatModifiers{GrimHarvestSlot: 2}
	if got := grimHarvestHeal(c, result, mods); got != 0 {
		t.Errorf("Grim Harvest fires on loss: got %d, want 0", got)
	}
}

func TestPersistDnDPostCombatSubclass_GrimHarvestApplies(t *testing.T) {
	setupAuditTestDB(t)
	uid := id.UserID("@grim_heal:example")
	if err := createAdvCharacter(uid, "grim"); err != nil {
		t.Fatal(err)
	}
	c := &DnDCharacter{
		UserID: uid, Race: RaceHuman, Class: ClassMage, Subclass: SubclassNecromancy,
		Level: 5, STR: 8, DEX: 12, CON: 12, INT: 16, WIS: 10, CHA: 10,
		HPMax: 30, HPCurrent: 10, ArmorClass: 12,
	}
	if err := SaveDnDCharacter(c); err != nil {
		t.Fatal(err)
	}
	result := CombatResult{
		PlayerWon: true,
		Events:    []CombatEvent{{Action: "spell_cast", EnemyHP: 0}},
	}
	mods := CombatModifiers{GrimHarvestSlot: 2, GrimHarvestNecrotic: true}
	if err := persistDnDPostCombatSubclass(c, false, result, mods); err != nil {
		t.Fatal(err)
	}
	got, _ := LoadDnDCharacter(uid)
	if got.HPCurrent != 16 {
		t.Errorf("HP after Grim Harvest = %d, want 16 (10 + 6)", got.HPCurrent)
	}
}

// ── Phase 10 SUB3b-i — Mage L10/L15 ─────────────────────────────────────

func TestEvocation_OverchannelL10MaximizesLeveledDamage(t *testing.T) {
	spell, _ := lookupSpell("magic_missile") // school evocation, leveled
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassEvocation, Level: 10, INT: 18}
	mods := &CombatModifiers{SpellPreDamage: 20}
	applyMageSubclassSpellHooks(c, spell, 3, mods)
	// magic_missile.school is "evocation", so Empowered Evocation also fires
	// at L10 (gated >=L7): +INT mod 4 → SpellPreDamage = 24, then Overchannel
	// ×1.5 → 36.
	if mods.SpellPreDamage != 36 {
		t.Errorf("Overchannel L10 leveled SpellPreDamage = %d, want 36 ((20+4)*3/2)", mods.SpellPreDamage)
	}
}

func TestEvocation_OverchannelSkipsCantrip(t *testing.T) {
	spell, _ := lookupSpell("fire_bolt") // school evocation, cantrip (Level 0)
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassEvocation, Level: 10, INT: 18}
	mods := &CombatModifiers{SpellPreDamage: 20}
	applyMageSubclassSpellHooks(c, spell, 0, mods)
	// Empowered Evocation +4, Overchannel skipped (slot < 1).
	if mods.SpellPreDamage != 24 {
		t.Errorf("Overchannel should skip cantrip (slot 0): SpellPreDamage = %d, want 24", mods.SpellPreDamage)
	}
}

func TestEvocation_OverchannelSkipsAboveSlot5(t *testing.T) {
	spell, _ := lookupSpell("magic_missile")
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassEvocation, Level: 10, INT: 18}
	mods := &CombatModifiers{SpellPreDamage: 20}
	applyMageSubclassSpellHooks(c, spell, 6, mods)
	// 5e Overchannel caps at slot 5; slot 6 just gets Empowered Evocation.
	if mods.SpellPreDamage != 24 {
		t.Errorf("Overchannel should not fire on slot 6: SpellPreDamage = %d, want 24", mods.SpellPreDamage)
	}
}

func TestEvocation_OverchannelPreL10Skipped(t *testing.T) {
	spell, _ := lookupSpell("magic_missile")
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassEvocation, Level: 9, INT: 18}
	mods := &CombatModifiers{SpellPreDamage: 20}
	applyMageSubclassSpellHooks(c, spell, 3, mods)
	if mods.SpellPreDamage != 24 {
		t.Errorf("Overchannel pre-L10 should not fire: SpellPreDamage = %d, want 24", mods.SpellPreDamage)
	}
}

func TestAbjuration_SpellResistanceL10ReducesDamage(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 10}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.DamageReduct >= 1.0 || mods.DamageReduct < 0.91 || mods.DamageReduct > 0.93 {
		t.Errorf("Abjuration L10 DamageReduct = %f, want ~0.92", mods.DamageReduct)
	}
}

func TestAbjuration_SpellResistancePreL10Skipped(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 9}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.DamageReduct != 1.0 {
		t.Errorf("Abjuration pre-L10 DamageReduct = %f, want 1.0", mods.DamageReduct)
	}
}

func TestAbjuration_SpellReflectionL15(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 15}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.ReflectNext < 0.29 || mods.ReflectNext > 0.31 {
		t.Errorf("Abjuration L15 ReflectNext = %f, want ~0.30", mods.ReflectNext)
	}
}

func TestAbjuration_SpellReflectionPreL15Skipped(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassAbjuration, Level: 14}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.ReflectNext != 0 {
		t.Errorf("Abjuration pre-L15 ReflectNext = %f, want 0", mods.ReflectNext)
	}
}

func TestNecromancy_CommandUndeadL10AmplifiesHarvest(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 10}
	result := CombatResult{
		PlayerWon: true,
		Events:    []CombatEvent{{Action: "spell_cast", EnemyHP: 0}},
	}
	// Non-necrotic at L10: 3× slot (was 2×).
	mods := CombatModifiers{GrimHarvestSlot: 3}
	if got := grimHarvestHeal(c, result, mods); got != 9 {
		t.Errorf("L10 non-necrotic Grim Harvest = %d, want 9 (3× slot)", got)
	}
	// Necrotic at L10: 4× slot (was 3×).
	mods = CombatModifiers{GrimHarvestSlot: 2, GrimHarvestNecrotic: true}
	if got := grimHarvestHeal(c, result, mods); got != 8 {
		t.Errorf("L10 necrotic Grim Harvest = %d, want 8 (4× slot)", got)
	}
}

func TestNecromancy_ImprovedUndeadThrallsL15SetsPetAttack(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 15, CON: 14} // +2 mod
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.PetAttackProc < 0.29 || mods.PetAttackProc > 0.31 {
		t.Errorf("Necromancer L15 PetAttackProc = %f, want ~0.30", mods.PetAttackProc)
	}
	if mods.PetAttackDmg != 8 { // 6 + CON mod 2
		t.Errorf("Necromancer L15 CON 14 PetAttackDmg = %d, want 8 (6 + 2)", mods.PetAttackDmg)
	}
}

func TestNecromancy_ImprovedUndeadThrallsPreL15Skipped(t *testing.T) {
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 14, CON: 14}
	mods := &CombatModifiers{DamageReduct: 1.0}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.PetAttackProc != 0 || mods.PetAttackDmg != 0 {
		t.Errorf("Necromancer pre-L15 should not summon thrall: proc=%f dmg=%d", mods.PetAttackProc, mods.PetAttackDmg)
	}
}

func TestNecromancy_ImprovedUndeadThrallsRespectsExistingHigherProc(t *testing.T) {
	// If a stronger pet (e.g. an adventure pet) is already in slot, don't downgrade.
	c := &DnDCharacter{Class: ClassMage, Subclass: SubclassNecromancy, Level: 15, CON: 14}
	mods := &CombatModifiers{DamageReduct: 1.0, PetAttackProc: 0.5, PetAttackDmg: 12}
	applySubclassPassives(&CombatStats{}, mods, c)
	if mods.PetAttackProc != 0.5 {
		t.Errorf("thrall overwrote stronger PetAttackProc: got %f, want 0.5", mods.PetAttackProc)
	}
	if mods.PetAttackDmg != 12 {
		t.Errorf("thrall overwrote stronger PetAttackDmg: got %d, want 12", mods.PetAttackDmg)
	}
}