G2 support

tests/drivers/gold_heal_mart_shots.lua

@@ -0,0 +1,152 @@
+-- The Pokecenter heal machine and the POKeMART shelf, in the running game.
+--
+--   POKEPORT_IDENTITY=goldshopfix POKEPORT_GAME=gold POKEPORT_GOLD_HOUR=10 \
+--     POKEPORT_DRIVER=tests/drivers/gold_heal_mart_shots.lua \
+--     perl -e 'alarm 300; exec @ARGV' \
+--     python3 -c "import pty; pty.spawn(['love','.'])"
+--
+-- Two things this is watching for:
+--   * PokecenterNurseScript's `special HealMachineAnim` used to be a bare
+--     (wrong) sfx: the balls, the flashing and MUSIC_HEAL all have to appear,
+--     ON the machine at the counter's left end, and the nurse's next line
+--     must wait for the last flash
+--   * a mart clerk's shelf came from data/generated/marts.lua being absent,
+--     so every shop opened empty: Cherrygrove must stock its four items with
+--     ItemAttributes prices and a purchase must move money and the PACK
+--
+-- Shots land in /tmp/gold-heal-mart.
+local U = require("tests.drivers.util")
+
+return function(game)
+  local out = os.getenv("POKEPORT_SHOT_DIR") or "/tmp/gold-heal-mart"
+  local fails = 0
+
+  local function ok(cond, msg)
+    if cond then print("[healmart] ok   " .. msg)
+    else fails = fails + 1 print("[healmart] FAIL " .. msg) end
+    return cond
+  end
+
+  local function tap(button, frames)
+    game.input.pressQueue[#game.input.pressQueue + 1] = button
+    game.input.state[button] = true
+    U.wait(2)
+    game.input.state[button] = false
+    U.wait(frames or 4)
+  end
+
+  U.wait(45)
+  local w = game.world
+  assert(w and w.map, "gold world did not boot")
+  local save = game.save
+
+  -- A party worth three balls on the machine, all hurt so the heal is real.
+  local Mon = require("src.battle.gen2.Mon")
+  save.party = {}
+  for _, species in ipairs({ "CYNDAQUIL", "PIDGEY", "RATTATA" }) do
+    local mon = Mon.new(game.data, species, 10)
+    mon.hp = 1
+    save.party[#save.party + 1] = mon
+  end
+  save.player = save.player or {}
+  save.player.money = 5000
+
+  -- ------------------------------------------------------------- the nurse
+  w:setMap("CHERRYGROVE_POKECENTER_1F", 3, 3, "up")
+  U.wait(20)
+  U.shot(game, out .. "/00-pokecenter.png")
+
+  -- Talk across the counter, then hold A through the greeting and the
+  -- yes/no (YES is the default), stopping the moment the machine starts.
+  tap("a", 6)
+  local sawAnim, sawBalls, sawFlash, sawJingle = false, 0, false, false
+  local Music = require("src.core.Music")
+  for _ = 1, 60 * 30 do
+    local ha = w.healAnim
+    if ha then
+      sawAnim = true
+      if ha.lit > sawBalls then
+        sawBalls = ha.lit
+        U.shot(game, ("%s/01-ball-%d.png"):format(out, ha.lit))
+      end
+      if ha.phase == "flash" and not sawFlash and ha.rotation ~= 0 then
+        sawFlash = true
+        U.shot(game, out .. "/02-flash.png")
+      end
+      if Music.current() == "Music_HealPokemon" then sawJingle = true end
+      U.wait(1)
+    elseif sawAnim then
+      break
+    else
+      tap("a", 2)
+    end
+  end
+  ok(sawAnim, "the heal machine animation ran")
+  ok(sawBalls == 3, "one ball per party member landed (" .. sawBalls .. ")")
+  ok(sawFlash, "the machine flashed its palette")
+  ok(sawJingle, "MUSIC_HEAL played over the flashing")
+  U.shot(game, out .. "/03-after-flash.png")
+
+  -- The script is still mid-conversation ("thank you for waiting"); page out.
+  for _ = 1, 40 do
+    if not w:busy() then break end
+    tap("a", 4)
+  end
+  local healed = true
+  for _, mon in ipairs(save.party) do
+    if (mon.hp or 0) < (mon.maxHp or 1) then healed = false end
+  end
+  ok(healed, "the party left the counter at full HP")
+  U.shot(game, out .. "/04-healed.png")
+
+  -- -------------------------------------------------------------- the mart
+  w:setMap("CHERRYGROVE_MART", 2, 3, "left")
+  U.wait(20)
+  tap("a", 6)
+  -- Page the welcome line until the BUY/SELL/QUIT screen owns the stack.
+  local mart
+  for _ = 1, 120 do
+    local top = game.stack and game.stack:top()
+    if top and top.martType then mart = top break end
+    tap("a", 3)
+  end
+  if not ok(mart ~= nil, "the clerk opened the mart screen") then
+    U.shot(game, out .. "/05-no-mart.png")
+    print(("[healmart] %d failures"):format(fails))
+    love.event.quit(fails == 0 and 0 or 1)
+    return
+  end
+  U.shot(game, out .. "/05-mart-top.png")
+
+  ok(#mart.entries == 4, "Cherrygrove stocks four items ("
+    .. #mart.entries .. ")")
+  ok(mart.entries[1] and mart.entries[1].id == "POTION"
+    and mart.entries[1].price == 300,
+    "POTION at the ROM's own 300 leads the shelf")
+
+  tap("a", 6) -- BUY
+  U.shot(game, out .. "/06-buy-list.png")
+  tap("a", 6) -- pick POTION -> quantity
+  tap("up", 4) -- x2
+  U.shot(game, out .. "/07-quantity.png")
+  tap("a", 6) -- how many -> confirm
+  U.shot(game, out .. "/08-confirm.png")
+  tap("a", 8) -- YES
+  tap("a", 8) -- "Here you are" page
+  U.shot(game, out .. "/09-bought.png")
+
+  ok((save.inventory and save.inventory.POTION) == 2,
+    "two POTIONs landed in the PACK")
+  ok(save.player.money == 5000 - 600,
+    "the till took 600 (money " .. tostring(save.player.money) .. ")")
+
+  -- Leave: B out of the list, then QUIT + the come-again line.
+  tap("b", 6)
+  tap("b", 6)
+  tap("a", 6)
+  tap("a", 10)
+  U.shot(game, out .. "/10-outside.png")
+
+  print(("[healmart] %d failures"):format(fails))
+  love.event.quit(fails == 0 and 0 or 1)
+end