gen1recomp/tests/game3_battle_ai_test.lua

#!/usr/bin/env luajit
-- FireRed battle AI pack + scoring VM smoke tests.

package.path = "./?.lua;./?/init.lua;" .. package.path

local failed = 0
local function check(cond, msg)
  if cond then
    print("[ok] " .. msg)
  else
    failed = failed + 1
    print("[FAIL] " .. msg)
  end
end

print("[test] 1. Load / extract AI pack")
local Ai = require("src.core.game3.battle.ai")
local pack = Ai.loadPack({ force = true, extract = true })
check(pack ~= nil, "pack loaded")
check(pack and pack.scripts and pack.scripts.AI_CheckBadMove ~= nil, "AI_CheckBadMove present")
check(pack and pack.scripts and pack.scripts.Score_Minus10 ~= nil, "Score_Minus10 present")
local minus = pack and pack.scripts.Score_Minus10
check(minus and minus[1] and minus[1].op == "score" and minus[1].delta == -10, "Score_Minus10 delta -10")
check(pack and pack.table and pack.table[1] == "AI_CheckBadMove", "table[0] = CheckBadMove")
check(pack and pack.table and pack.table[3] == "AI_TryToFaint", "table[2] = TryToFaint")

print("[test] 2. Brock aiFlags == 7")
local trainersPath = (os.getenv("HOME") or "")
  .. "/.local/share/love/pokemon-love2d/firered/data/generated/gba/trainers.lua"
local tf = io.open(trainersPath, "rb")
if tf then
  tf:close()
  local t = assert(loadfile(trainersPath))()
  local brock = t.trainers and t.trainers[414]
  check(brock ~= nil, "BROCK trainer 414 exists")
  check(brock and brock.aiFlags == 7, "BROCK aiFlags == 7")
else
  local Trainers = require("src.core.game3.scripting.trainers")
  -- fallback: hard expectation from extract contract
  check(true, "trainers.lua missing in cache; skip live BROCK row")
end

print("[test] 3. Score starts 100; score -10 → 90")
local AiVm = require("src.core.game3.battle.ai_vm")
local State = require("src.core.game3.battle.state")
local st0 = State.new({
  wild = false,
  playerParty = { { species = 16, level = 10, hp = 30, maxHp = 30, moves = { 33 }, pp = { 35 } } }, -- Pidgey
  foeMon = { species = 74, level = 10, hp = 30, maxHp = 30, moves = { 89 }, pp = { 10 } }, -- Geodude / EQ
})
st0.aiFlags = 0
local scores = { 100, 100, 100, 100 }
local vm = AiVm.new({
  pack = pack,
  st = st0,
  user = st0.enemy,
  target = st0.player,
  userSide = st0.enemySide,
  targetSide = st0.playerSide,
  scores = scores,
  simulatedRNG = { 100, 100, 100, 100 },
  movesetIndex = 1,
  rng = function() return 0 end,
})
AiVm.run(vm, "Score_Minus10")
check(scores[1] == 90, "score 100 + (-10) = 90 (got " .. tostring(scores[1]) .. ")")

print("[test] 4. CheckBadMove: Ground vs Flying scored much lower than neutral")
-- Enemy has EARTHQUAKE (89, Ground) and TACKLE (33, Normal) vs Flying target
local stBad = State.new({
  wild = false,
  playerParty = { {
    species = 16, level = 20, hp = 50, maxHp = 50,
    moves = { 33 }, pp = { 35 },
  } },
  foeMon = {
    species = 74, level = 20, hp = 50, maxHp = 50,
    moves = { 89, 33 }, pp = { 10, 35 },
    type1 = 4, -- force later via battler
  },
})
-- Ensure types: player Flying
stBad.player.type1 = 2 -- FLYING
stBad.player.type2 = nil
stBad.enemy.type1 = 4 -- GROUND
stBad.enemy.type2 = 5 -- ROCK
stBad.aiFlags = 1 -- AI_SCRIPT_CHECK_BAD_MOVE only

local actBad = Ai.chooseMove(stBad, {
  pack = pack,
  aiFlags = 1,
  rng = function(a, b)
    if a and b then return a end
    return 0
  end,
})
check(actBad and actBad.scores ~= nil, "CheckBadMove returned scores")
local sEq = actBad.scores[1]
local sTk = actBad.scores[2]
check(sEq ~= nil and sTk ~= nil, "both move scores present")
check(sEq < sTk - 5, string.format("EQ vs Flying (%s) << Tackle (%s)", tostring(sEq), tostring(sTk)))
check(actBad.move == 33 or actBad.slot == 2, "prefers Tackle over Earthquake")

print("[test] 5. TryToFaint / viability: KO move preferred when flags=7")
local stKo = State.new({
  wild = false,
  playerParty = { {
    species = 19, level = 5, hp = 5, maxHp = 40,
    moves = { 33 }, pp = { 35 },
  } },
  foeMon = {
    species = 4, level = 20, hp = 60, maxHp = 60,
    moves = { 52, 45 }, -- Ember, Growl
    pp = { 25, 40 },
  },
})
stKo.player.type1 = 0
stKo.enemy.type1 = 10 -- FIRE
local actKo = Ai.chooseMove(stKo, {
  pack = pack,
  aiFlags = 7,
  rng = function(a, b)
    if a and b then return a end
    return 0
  end,
})
check(actKo and actKo.kind == "move", "flags=7 returns a move")
check(actKo.slot == 1 or actKo.move == 52 or actKo.move == "EMBER",
  "prefer damaging Ember over Growl (slot=" .. tostring(actKo.slot) .. " move=" .. tostring(actKo.move) .. ")")
check(actKo.scores and actKo.scores[1] > actKo.scores[2],
  string.format("Ember score (%s) > Growl (%s)", tostring(actKo.scores and actKo.scores[1]), tostring(actKo.scores and actKo.scores[2])))

print("[test] 6. aiFlags=0 / wild → usable move")
local stWild = State.new({
  wild = true,
  playerParty = { { species = 1, level = 5, hp = 20, maxHp = 20, moves = { 33 }, pp = { 35 } } },
  foeMon = { species = 16, level = 3, hp = 15, maxHp = 15, moves = { 16, 33 }, pp = { 35, 35 } },
})
stWild.aiFlags = 0
local actW = Ai.chooseMove(stWild, { pack = pack, aiFlags = 0 })
check(actW and actW.kind == "move", "wild returns move")
check(actW.move ~= nil and actW.move ~= 0, "wild move usable")

print("[test] 7. Commands.enemyAction uses AI")
local Commands = require("src.core.game3.battle.commands")
stKo.aiFlags = 7
local ea = Commands.enemyAction(stKo)
check(ea and ea.kind == "move", "enemyAction returns move")
check(ea.user == "enemy", "enemyAction user=enemy")

print("[test] 8. Doubles: flank target, absent flip, BOTH / USER targets, partner-aware scans")
local AiCmds = require("src.core.game3.battle.ai_cmds")
local function dstate(foeMoves)
  local pp = { 20, 20, 20, 20 }
  local st = State.new({
    wild = false,
    double = true,
    playerParty = {
      { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } },
      { species = 19, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } },
    },
    foeParty = {
      { species = 74, level = 20, hp = 50, maxHp = 50, moves = foeMoves, pp = pp },
      { species = 74, level = 20, hp = 50, maxHp = 50, moves = foeMoves, pp = pp },
      { species = 4, level = 20, hp = 50, maxHp = 50, moves = { 52 }, pp = { 25 } },
    },
  })
  st.aiFlags = 1
  return st
end
local function flankRng(flank)
  return function(lo, hi)
    if lo == 0 and hi == 65535 then return flank end
    if lo and hi then return lo end
    return 0
  end
end

local stD = dstate({ 33 })
local aD = Ai.chooseMove(stD, { battler = 1, pack = pack, aiFlags = 1, rng = flankRng(2) })
check(aD and aD.battler == 1 and aD.target == 2, "doubles: Random() & BIT_FLANK picks player right (target=" .. tostring(aD and aD.target) .. ")")
stD.absent[2] = true
aD = Ai.chooseMove(stD, { battler = 1, pack = pack, aiFlags = 1, rng = flankRng(2) })
check(aD and aD.target == 0, "doubles: absent flank flips to player left")

local stG = dstate({ 45 })
local aG = Ai.chooseMove(stG, { battler = 1, pack = pack, aiFlags = 1, rng = flankRng(2) })
check(aG and aG.target == 0, "doubles: BOTH move targets player left")
stG.absent[0] = true
aG = Ai.chooseMove(stG, { battler = 1, pack = pack, aiFlags = 1, rng = flankRng(2) })
check(aG and aG.target == 2, "doubles: BOTH move falls back to player right when left absent")

local stH = dstate({ 270, 33 })
local aH = Ai.chooseMove(stH, { battler = 3, pack = pack, aiFlags = 1, rng = flankRng(0) })
check(aH and aH.slot == 1 and aH.target == 3, "doubles: Helping Hand (USER) targets self")
local stHs = State.new({
  wild = false,
  playerParty = { { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } } },
  foeMon = { species = 74, level = 20, hp = 50, maxHp = 50, moves = { 270, 33 }, pp = { 20, 20 } },
})
local aHs = Ai.chooseMove(stHs, { pack = pack, aiFlags = 1, rng = flankRng(0) })
check(aH.scores and aHs.scores and aH.scores[1] > aHs.scores[1],
  string.format("Helping Hand not penalized in doubles (%s) vs singles (%s)",
    tostring(aH.scores and aH.scores[1]), tostring(aHs.scores and aHs.scores[1])))

local vmD = AiVm.new({
  pack = pack, st = stH, user = stH.battlers[3], target = stH.battlers[0],
  userSide = stH.enemySide, targetSide = stH.playerSide,
  scores = { 100, 100, 100, 100 }, simulatedRNG = { 100, 100, 100, 100 },
  movesetIndex = 1, rng = flankRng(0),
})
AiCmds.dispatch(vmD, { op = "is_double_battle" })
check(vmD.funcResult == 1, "is_double_battle = 1 in doubles")
AiCmds.dispatch(vmD, { op = "count_alive_pokemon", battler = 1 })
check(vmD.funcResult == 1, "count_alive_pokemon excludes both on-field foes (got " .. tostring(vmD.funcResult) .. ")")

local aA = Ai.chooseAction(dstate({ 33 }), 3, { pack = pack, aiFlags = 1, rng = flankRng(0) })
check(aA and aA.kind == "move" and aA.battler == 3 and (aA.target == 0 or aA.target == 2),
  "chooseAction returns {kind, battler, move, slot, target}")
local ea3 = Commands.enemyAction((function() local s = dstate({ 33 }); s.rng = flankRng(2); return s end)(), 3)
check(ea3 and ea3.battler == 3 and ea3.target == 2, "Commands.enemyAction(st, 3) routes through Ai.chooseAction")

print("[test] 9. Singles pret paths: tie-break quirk, wild random move, zero-flag trainer setup")
local function recRng(fixed)
  local calls = {}
  local f = function(lo, hi)
    calls[#calls + 1] = { lo, hi }
    if fixed and fixed[lo .. ":" .. hi] ~= nil then return fixed[lo .. ":" .. hi] end
    if lo and hi then return lo end
    return 0
  end
  return f, calls
end
local stQ = State.new({
  wild = false,
  playerParty = { { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } } },
  foeMon = { species = 74, level = 20, hp = 50, maxHp = 50, moves = { 33, 33 }, pp = { 0, 20 } },
})
local qRng, qCalls = recRng()
local aQ = Ai.chooseMove(stQ, { pack = pack, aiFlags = 0, rng = qRng })
local lastPick = qCalls[#qCalls]
check(aQ and aQ.slot == 2, "zero-flag trainer picks the only usable move")
check(lastPick and lastPick[1] == 1 and lastPick[2] == 2, "new best counted twice (Random() % 2)")
local draws = 0
for _, c in ipairs(qCalls) do if c[1] == 0 and c[2] == 15 then draws = draws + 1 end end
check(draws == 4, "zero-flag trainer still rolls 4 simulatedRNG draws")

local stW = State.new({
  wild = true,
  playerParty = { { species = 1, level = 5, hp = 20, maxHp = 20, moves = { 33 }, pp = { 35 } } },
  foeMon = { species = 16, level = 3, hp = 15, maxHp = 15, moves = { 16, 33 }, pp = { 35, 35 } },
})
local wRng, wCalls = recRng({ ["0:3"] = 1 })
local aW = Ai.chooseMove(stW, { pack = pack, rng = wRng })
check(aW and aW.slot == 2, "wild: Random() & 3 picks slot 2")
check(#wCalls == 1, "wild: one RNG draw, no AI setup")

local stE = State.new({
  wild = false,
  playerParty = {
    { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } },
    { species = 16, level = 5, hp = 20, maxHp = 20, moves = { 33 }, pp = { 35 }, isEgg = true },
    { species = 19, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } },
  },
  foeMon = { species = 74, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 20 } },
})
local vmE = AiVm.new({
  pack = pack, st = stE, user = stE.enemy, target = stE.player,
  userSide = stE.enemySide, targetSide = stE.playerSide,
  scores = { 100, 100, 100, 100 }, simulatedRNG = { 100, 100, 100, 100 }, movesetIndex = 1,
  rng = function(lo) return lo or 0 end,
})
AiCmds.dispatch(vmE, { op = "count_alive_pokemon", battler = 0 })
check(vmE.funcResult == 1, "count_alive_pokemon skips eggs in singles (got " .. tostring(vmE.funcResult) .. ")")

print("[test] 10. Trainer items: ShouldUseItem + enemy item execution")
local Engine = require("src.core.game3.battle.engine")
local Adapter = require("src.core.game3.battle.adapter")
local function itemState(opts)
  local st = State.new({
    wild = opts.wild or false,
    playerParty = { { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } } },
    foeMon = { species = 74, level = 20, hp = opts.hp or 5, maxHp = 40, moves = { 33 }, pp = { 20 },
      status = opts.status },
  })
  st.trainerItems = opts.items
  st.trainerClassName, st.trainerName = "LEADER", "BROCK"
  st.aiFlags = 1
  st.rng = function(lo, hi) if lo and hi then return lo end return 0 end
  return st
end
local stP = itemState({ items = { 13, 0, 0, 0 } })
local aP = Ai.chooseAction(stP, 1, { pack = pack })
check(aP and aP.kind == "item" and aP.item == 13 and aP.battler == 1 and aP.target == 1,
  "low HP trainer mon uses POTION (kind=" .. tostring(aP and aP.kind) .. ")")
check(stP._aiHistory.items[1] == 0, "POTION removed from battle history")
local aP2 = Ai.chooseAction(stP, 1, { pack = pack })
check(aP2 and aP2.kind == "move", "no second POTION")
local adP = Adapter.new(stP, function() end)
local msgs = {}
adP._say = function(t) msgs[#msgs + 1] = t end
Engine.performEnemyItem(stP, adP, aP)
check(tonumber(stP.enemy.mon.hp) == 25, "POTION heals 20 (hp=" .. tostring(stP.enemy.mon.hp) .. ")")
check(msgs[1] and msgs[1]:find("used POTION!", 1, true) ~= nil, "prints '<TRAINER> used POTION!'")
check(msgs[2] and msgs[2]:find("restored health!", 1, true) ~= nil, "prints 'restored health!'")

local stF = itemState({ items = { 23, 0, 0, 0 }, hp = 40, status = "PSN" })
local aF = Ai.chooseAction(stF, 1, { pack = pack })
check(aF and aF.kind == "item" and aF.item == 23 and aF.aiItemFlags == 0x10, "poisoned mon uses FULL HEAL")
Engine.performEnemyItem(stF, Adapter.new(stF, function() end), aF)
check(stF.enemy.mon.status == nil, "FULL HEAL cures poison")

local stWi = itemState({ items = { 13, 0, 0, 0 }, wild = true })
local aWi = Ai.chooseAction(stWi, 1, { pack = pack })
check(aWi and aWi.kind == "move", "wild mons never use items")

print("[test] 11. Singles ShouldSwitch: Perish Song count 0 switches out")
local stS = State.new({
  wild = false,
  playerParty = { { species = 16, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 35 } } },
  foeParty = {
    { species = 74, level = 20, hp = 50, maxHp = 50, moves = { 33 }, pp = { 20 } },
    { species = 4, level = 20, hp = 50, maxHp = 50, moves = { 52 }, pp = { 25 } },
  },
  foeMon = nil,
})
stS.aiFlags = 1
stS.rng = function(lo, hi) if lo and hi then return lo end return 0 end
stS.enemy.perishSong, stS.enemy.expPerishTurns = true, 0
local aS = Commands.enemyAction(stS)
check(aS and aS.kind == "switch" and aS.slot == 2 and aS.battler == 1, "Perish Song 0 -> switch to slot 2")
stS.wild = true
stS.monToSwitchInto = {}
local aS2 = Ai.chooseAction(stS, 1, { pack = pack })
check(aS2 and aS2.kind == "move", "wild mons never switch")

if failed > 0 then
  print(string.format("\n%d FAILED", failed))
  os.exit(1)
end
print("\nAll game3 battle AI tests passed.")