pokerogue/test/abilities/perish_body.test.ts

import { Abilities } from "#enums/abilities";
import { Moves } from "#enums/moves";
import { Species } from "#enums/species";
import GameManager from "#test/testUtils/gameManager";
import Phaser from "phaser";
import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";

describe("Abilities - Perish Song", () => {
  let phaserGame: Phaser.Game;
  let game: GameManager;

  beforeAll(() => {
    phaserGame = new Phaser.Game({
      type: Phaser.HEADLESS,
    });
  });

  afterEach(() => {
    game.phaseInterceptor.restoreOg();
  });

  beforeEach(() => {
    game = new GameManager(phaserGame);
    game.override.battleStyle("single");
    game.override.disableCrits();

    game.override.enemySpecies(Species.MAGIKARP);
    game.override.enemyAbility(Abilities.BALL_FETCH);

    game.override.starterSpecies(Species.CURSOLA);
    game.override.ability(Abilities.PERISH_BODY);
    game.override.moveset([Moves.SPLASH]);
  });

  it("should trigger when hit with damaging move", async () => {
    game.override.enemyMoveset([Moves.AQUA_JET]);
    await game.classicMode.startBattle();
    const cursola = game.scene.getPlayerPokemon();
    const magikarp = game.scene.getEnemyPokemon();

    game.move.select(Moves.SPLASH);
    await game.toNextTurn();

    expect(cursola?.summonData.tags[0].turnCount).toBe(3);
    expect(magikarp?.summonData.tags[0].turnCount).toBe(3);
  });

  it("should trigger even when fainting", async () => {
    game.override.enemyMoveset([Moves.AQUA_JET]).enemyLevel(100).startingLevel(1);
    await game.classicMode.startBattle([Species.CURSOLA, Species.FEEBAS]);
    const magikarp = game.scene.getEnemyPokemon();

    game.move.select(Moves.SPLASH);
    game.doSelectPartyPokemon(1);
    await game.toNextTurn();

    expect(magikarp?.summonData.tags[0].turnCount).toBe(3);
  });

  it("should not activate if attacker already has perish song", async () => {
    game.override.enemyMoveset([Moves.PERISH_SONG, Moves.AQUA_JET, Moves.SPLASH]);
    await game.classicMode.startBattle([Species.FEEBAS, Species.CURSOLA]);
    const feebas = game.scene.getPlayerPokemon();
    const magikarp = game.scene.getEnemyPokemon();

    game.move.select(Moves.SPLASH);
    await game.forceEnemyMove(Moves.PERISH_SONG);
    await game.toNextTurn();

    expect(feebas?.summonData.tags[0].turnCount).toBe(3);
    expect(magikarp?.summonData.tags[0].turnCount).toBe(3);

    game.doSwitchPokemon(1);
    await game.forceEnemyMove(Moves.SPLASH);
    await game.toNextTurn();

    const cursola = game.scene.getPlayerPokemon();
    expect(cursola?.summonData.tags.length).toBe(0);
    expect(magikarp?.summonData.tags[0].turnCount).toBe(2);

    game.move.select(Moves.SPLASH);
    await game.forceEnemyMove(Moves.AQUA_JET);
    await game.toNextTurn();

    expect(cursola?.summonData.tags.length).toBe(0);
    expect(magikarp?.summonData.tags[0].turnCount).toBe(1);
  });

  it("should activate if cursola already has perish song, but not reset its counter", async () => {
    game.override.enemyMoveset([Moves.PERISH_SONG, Moves.AQUA_JET, Moves.SPLASH]);
    game.override.moveset([Moves.WHIRLWIND, Moves.SPLASH]);
    game.override.startingWave(5);
    await game.classicMode.startBattle([Species.CURSOLA]);
    const cursola = game.scene.getPlayerPokemon();

    game.move.select(Moves.WHIRLWIND);
    await game.forceEnemyMove(Moves.PERISH_SONG);
    await game.toNextTurn();

    const magikarp = game.scene.getEnemyPokemon();
    expect(cursola?.summonData.tags[0].turnCount).toBe(3);
    expect(magikarp?.summonData.tags.length).toBe(0);

    game.move.select(Moves.SPLASH);
    await game.forceEnemyMove(Moves.AQUA_JET);
    await game.toNextTurn();

    expect(cursola?.summonData.tags[0].turnCount).toBe(2);
    expect(magikarp?.summonData.tags.length).toBe(1);
    expect(magikarp?.summonData.tags[0].turnCount).toBe(3);
  });
});