gen1recomp/tests/drivers/ss_anne_captain_2189.lua

-- scripts/SSAnneCaptainsRoom.asm:5-10,45-68
--   POKEPORT_SHOT_DIR=/tmp/shots POKEPORT_VERSION=yellow \
--     POKEPORT_DRIVER=tests/drivers/ss_anne_captain_2189.lua love .
return function(game)
  local U = dofile("tests/drivers/util.lua")
  local Pokemon = require("src.pokemon.Pokemon")
  local Music = require("src.core.Music")

  local SHOT_DIR = os.getenv("POKEPORT_SHOT_DIR") or os.getenv("SHOT_DIR")
                   or "/tmp/shots"
  local failed = false
  local function check(label, ok)
    U.log(ok and "PASS" or "FAIL", label)
    if not ok then failed = true end
    return ok
  end
  local function quit()
    U.wait(2)
    love.event.quit(failed and 1 or 0)
    while true do coroutine.yield() end
  end

  local MAP = "SS_ANNE_CAPTAINS_ROOM"

  game.save.party = { Pokemon.new(game.data, "PIKACHU", 12) }
  game.save.player.name = "bryan"
  game.save.options = game.save.options or {}
  game.save.options.textSpeed = 1
  game.save.flags = game.save.flags or {}
  game.save.flags.EVENT_GOT_HM01 = nil
  game.save.flags.EVENT_RUBBED_CAPTAINS_BACK = nil

  U.teleport(game, MAP, 4, 3, "up")
  U.wait(20)

  local ow = game.overworld
  if not check("captain's room loaded", ow and ow.map and ow.map.id == MAP) then
    quit()
  end
  check("the map script set the no-face bit", ow.noNpcFacePlayer == true)

  local captain
  for _, npc in ipairs(ow.npcs or {}) do
    if npc.def and npc.def.sprite == "SPRITE_CAPTAIN" then captain = npc end
  end
  captain = captain or (ow.npcs and ow.npcs[1])
  if not check("the captain is on the map", captain ~= nil) then quit() end
  check("he starts facing up", captain.facing == "up")

  U.tap(game, "a")
  U.wait(20)
  local box = game.stack:top()
  check("talking to him opens a text box", box ~= ow)
  check("and he keeps his back to the player", captain.facing == "up")
  U.shot(game, SHOT_DIR .. "/2189_01_captain_back_turned.png")

  local jingle = false
  for _ = 1, 20 do
    if Music.oneShotPlaying() then jingle = true break end
    U.tap(game, "a")
    for _ = 1, 40 do
      if Music.oneShotPlaying() then break end
      U.wait(1)
    end
  end
  jingle = jingle or Music.oneShotPlaying()
  check("MUSIC_PKMN_HEALED plays", jingle)
  check("the rub box is still up while it plays", game.stack:top() ~= ow)
  check("and he is still facing up", captain.facing == "up")
  U.shot(game, SHOT_DIR .. "/2189_02_rub_box_holds_jingle.png")

  local function boxHas(node, needle, depth)
    depth = depth or 0
    if type(node) == "string" then return node:find(needle, 1, true) ~= nil end
    if type(node) ~= "table" or depth > 4 then return false end
    for _, v in pairs(node) do
      if boxHas(v, needle, depth + 1) then return true end
    end
    return false
  end
  local function betterBoxUp()
    local top = game.stack:top()
    return top ~= ow and top and boxHas(top.pages, "much better")
  end

  local rubBox = game.stack:top()
  local turnedEarly = false
  for _ = 1, 1200 do
    if betterBoxUp() then break end
    if game.stack:top() == rubBox and captain.facing ~= "up" then
      turnedEarly = true
    end
    U.wait(1)
  end
  check("he stays back-turned until the rub box is gone", not turnedEarly)
  check("the I-feel-much-better box opens", betterBoxUp())
  check("and he turns to face the player as it opens", captain.facing == "down")
  U.wait(30)
  U.still(game, SHOT_DIR .. "/2189_03_captain_faces_player_feel_much_better.png")

  captain.timer = 128
  local stayed = true
  for _ = 1, 400 do
    if game.stack:top() == ow then break end
    if captain.facing ~= "down" then stayed = false end
    U.tap(game, "a")
    for _ = 1, 6 do
      if game.stack:top() == ow then break end
      if captain.facing ~= "down" then stayed = false end
      U.wait(1)
    end
  end
  check("he keeps facing the player through the rest of the talk", stayed)
  check("the conversation closes", game.stack:top() == ow)
  check("HM01 handed over", game.save.flags.EVENT_GOT_HM01 == true)
  check("EVENT_RUBBED_CAPTAINS_BACK set",
        game.save.flags.EVENT_RUBBED_CAPTAINS_BACK == true)

  -- engine/overworld/movement.asm:193,262
  local back
  for i = 1, 140 do
    U.wait(1)
    if captain.facing == "up" then back = i break end
  end
  check("then his STAY, UP pin turns him back within 130 frames",
        back ~= nil and back <= 130)

  quit()
end