gen1recomp/tests/drivers/thrash_menu_flash_bug2252.lua

-- engine/battle/core.asm:293
--   POKEPORT_IDENTITY=red-sep04 POKEPORT_SHOT_DIR=/tmp/shots POKEPORT_DRIVER=tests/drivers/thrash_menu_flash_bug2252.lua love .
return function(game)
  local U = dofile("tests/drivers/util.lua")
  local DIR = os.getenv("POKEPORT_SHOT_DIR") or os.getenv("SHOT_DIR") or "/tmp/shots"
  local Pokemon = require("src.pokemon.Pokemon")
  local BattleState = require("src.battle.BattleState")

  local fails = 0
  local function ok(cond, line)
    print((cond and "PASS " or "FAIL ") .. line)
    if not cond then fails = fails + 1 end
  end

  U.teleport(game, "ROUTE_1", 5, 5, "down")
  local ow = game.overworld

  local function startBattle()
    love.math.setRandomSeed(2252)
    local king = Pokemon.new(game.data, "NIDOKING", 60)
    king.moves = { { id = "THRASH", pp = 20 } }
    king.stats.hp, king.hp = 999, 999
    game.save.party = { king }
    local battle = BattleState.newWild(game, "MAGIKARP", 60)
    battle.onFinish = function() end
    battle.enemy.mon.stats.hp, battle.enemy.mon.hp = 999, 999
    ow:pushBattle(battle)
    for _ = 1, 200 do
      if battle.phase == "menu" then break end
      U.tap(game, "a")
      U.wait(4)
    end
    return battle
  end

  local function selectThrash(battle)
    U.tap(game, "a"); U.wait(12)
    U.tap(game, "a")
  end

  local function locked(battle)
    return (battle.player.thrashTurns or 0) > 0
  end

  local battle = startBattle()
  ok(battle.phase == "menu", "2252 pass 1 reached the command menu")
  local startTurn = battle.turnCount or 0
  selectThrash(battle)
  local f, drainFrame, flashes, sawLocked = 0, nil, 0, false
  local lastSig, logged = nil, 0
  for _ = 1, 2400 do
    f = f + 1
    if f % 20 == 0 then U.tap(game, "a") else U.wait(1) end
    local sig = tostring(battle.phase) .. " thrash=" .. tostring(battle.player.thrashTurns)
      .. " turn=" .. tostring(battle.turnCount) .. " top=" .. tostring(game.stack:top() == battle)
    if sig ~= lastSig and logged < 80 then
      U.log("f", f, sig)
      lastSig, logged = sig, logged + 1
    end
    if locked(battle) then sawLocked = true end
    if not drainFrame and (battle.turnCount or 0) >= startTurn + 2 then drainFrame = f end
    if battle.phase == "menu" and locked(battle) then flashes = flashes + 1 end
    if battle.phase == "menu" and not locked(battle) and f > 30 then break end
  end
  U.log("turns", (battle.turnCount or 0) - startTurn, "drainFrame", drainFrame,
    "status", battle.player.mon.status, "confused", battle.player.confusedTurns)
  ok(sawLocked, "2252 player locked into thrash")
  ok(drainFrame ~= nil, "2252 locked turn resolved without input")
  ok(flashes == 0, "2252 no command menu frame while locked (" .. flashes .. ")")
  if game.stack:top() == battle then game.stack:pop() end

  if drainFrame then
    battle = startBattle()
    selectThrash(battle)
    local g, shotBefore, shotDrain, shotAfter = 0, false, false, false
    local phaseAtDrain
    while g < drainFrame + 60 do
      if not shotBefore and g >= drainFrame - 12 then
        shotBefore = U.shot(game, DIR .. "/2252_01_thrash_text_before_drain.png")
        g = g + 3
      elseif not shotDrain and g >= drainFrame then
        phaseAtDrain = battle.phase
        shotDrain = U.shot(game, DIR .. "/2252_02_drain_frame_no_menu.png")
        g = g + 3
      elseif not shotAfter and g >= drainFrame + 40 then
        shotAfter = U.shot(game, DIR .. "/2252_03_thrashing_about_text.png")
        g = g + 3
      else
        g = g + 1
        if g % 20 == 0 then U.tap(game, "a") else U.wait(1) end
      end
    end
    ok(phaseAtDrain ~= "menu", "2252 pass 2 drain frame is not the command menu")
    ok(shotBefore and shotDrain and shotAfter, "2252 shots written")
  end

  love.event.quit(fails == 0 and 0 or 1)
end