-- A trainer script's post-battle text must finish before a level evolution.
-- Its start_battle row resumes the script after OverworldState:afterBattle,
-- so this covers the same handoff used by trainer and Rocket encounters.

package.path = "./?.lua;./?/init.lua;" .. package.path
if not _G.love then _G.love = require("tests.love_stub") end

local Data = require("src.core.Data")
if not Data.maps then Data:load() end
local S = require("tests.harness").suite("parity trainer evolution order")
local check = S.check

local Game = require("src.core.Game")
local SaveData = require("src.core.SaveData")
local StateStack = require("src.core.StateStack")
local ScriptRunner = require("src.script.ScriptRunner")
local OverworldState = require("src.world.OverworldController")
local Pokemon = require("src.pokemon.Pokemon")
require("src.render.Font").load(Data)

Game.data = Data
Game.save = SaveData.newGame()
Game.stack = StateStack; StateStack:init()
Game.renderer = { worldViewSize = function() return 160, 144 end }

local caterpie = Pokemon.new(Data, "CATERPIE", 7)
Game.save.party = { caterpie }

local fakeBattle
local originalBattleState = package.loaded["src.battle.BattleState"]
package.loaded["src.battle.BattleState"] = {
  newTrainer = function()
    fakeBattle = { leveledUp = { [caterpie] = true } }
    return fakeBattle
  end,
}

local overworld = setmetatable({}, { __index = OverworldState })
overworld:enter("ROUTE_1", 5, 5, "down")
local runner = ScriptRunner.new(Game, overworld)
runner:run({
  { "start_battle", "trainer", "OPP_YOUNGSTER", 1 },
  { "show_text", "THE TRAINER TALKS AFTER THE FIGHT" },
  { "show_text", "THE TRAINER HAS MORE TO SAY" },
})

check(fakeBattle ~= nil, "trainer battle starts and yields the script")
Game.stack:pop()
fakeBattle.onFinish("win")

package.loaded["src.battle.BattleState"] = originalBattleState

local function stateHas(state, text)
  for _, page in ipairs(state.pages or {}) do
    for _, line in ipairs(page) do
      if type(line) == "string" and line:find(text, 1, true) then return true end
    end
  end
  return false
end

local afterText = Game.stack:top()
check(stateHas(afterText, "THE TRAINER TALKS"),
  "the trainer's after-battle text is shown first")
check(#Game.stack.states == 1,
  "the evolution screen waits until trainer after-text closes")

Game.stack:pop()
afterText.onDone()

local moreText = Game.stack:top()
check(stateHas(moreText, "MORE TO SAY"),
  "all trainer after-text finishes before evolution")
check(#Game.stack.states == 1,
  "the evolution still waits for the last trainer text")

Game.stack:pop()
moreText.onDone()

-- The evolution runs as the EvolutionState cutscene screen now (the
-- evolve_mon.asm sequence lives in src/ui/EvolutionState.lua), not a bare
-- "evolving!" text box, so assert the screen itself took the stack.
local evolution = Game.stack:top()
check(evolution ~= nil
      and (evolution.screenId == "EvolutionState"
           or stateHas(evolution, "evolving")),
  "the level evolution starts after trainer after-text closes")

S.finish()