Add tests for Disable

This commit is contained in:
Zach Day 2024-06-11 00:53:49 -04:00
parent 5a33a1bf74
commit 4a0900385b

View File

@ -0,0 +1,113 @@
import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
import GameManager from "../utils/gameManager";
import * as overrides from "#app/overrides";
import { Abilities } from "#app/data/enums/abilities.js";
import { Moves } from "#app/data/enums/moves.js";
import { Species } from "#app/data/enums/species.js";
import { CommandPhase, EnemyCommandPhase, TurnInitPhase } from "#app/phases.js";
import { Mode } from "#app/ui/ui.js";
import { getMovePosition } from "../utils/gameManagerUtils";
import {Command} from "#app/ui/command-ui-handler";
import { MoveResult } from "#app/field/pokemon.js";
import { Stat } from "#app/data/pokemon-stat.js";
import { DisabledTag } from "#app/data/battler-tags.js";
import { BattlerTagType } from "#app/data/enums/battler-tag-type.js";
describe("Moves - Disable", () => {
let phaserGame: Phaser.Game;
let game: GameManager;
/** If in CommandPhase, input a move */
const _useMove = (move?: Moves) => {
move ??= Moves.DISABLE;
game.onNextPrompt("CommandPhase", Mode.COMMAND, () => {
game.scene.ui.setMode(Mode.FIGHT, (game.scene.getCurrentPhase() as CommandPhase).getFieldIndex());
});
game.onNextPrompt("CommandPhase", Mode.FIGHT, () => {
const movePosition = getMovePosition(game.scene, 0, move);
(game.scene.getCurrentPhase() as CommandPhase).handleCommand(Command.FIGHT, movePosition, false);
});
};
beforeAll(() => {
phaserGame = new Phaser.Game({
type: Phaser.HEADLESS,
});
});
afterEach(() => {
game.phaseInterceptor.restoreOg();
});
beforeEach(() => {
game = new GameManager(phaserGame);
vi.spyOn(overrides, "SINGLE_BATTLE_OVERRIDE", "get").mockReturnValue(true);
vi.spyOn(overrides, "ABILITY_OVERRIDE", "get").mockReturnValue(Abilities.NONE);
vi.spyOn(overrides, "OPP_ABILITY_OVERRIDE", "get").mockReturnValue(Abilities.NONE);
vi.spyOn(overrides, "MOVESET_OVERRIDE", "get").mockReturnValue([Moves.DISABLE, Moves.SPLASH]);
vi.spyOn(overrides, "OPP_MOVESET_OVERRIDE", "get").mockReturnValue([Moves.SPLASH,Moves.NONE,Moves.NONE,Moves.NONE]);
});
it("DISABLE fails if enemy has no move history", async() => {
// Player goes first
await game.startBattle([
Species.PIKACHU,
Species.SHUCKLE,
]);
expect(game.scene.getParty()[0].stats[Stat.SPD]).toBeGreaterThan(game.scene.getEnemyParty()[0].stats[Stat.SPD]);
_useMove();
await game.phaseInterceptor.runFrom(EnemyCommandPhase).to(TurnInitPhase);
expect(game.scene.getParty()[0].getMoveHistory().at(0).result).toBe(MoveResult.FAIL);
}, 20000);
it("DISABLE works when user moves after enemy", async() => {
await game.startBattle([
Species.SHUCKLE,
Species.PIKACHU,
]);
expect(game.scene.getParty()[0].stats[Stat.SPD]).toBeLessThan(game.scene.getEnemyParty()[0].stats[Stat.SPD]);
_useMove();
await game.phaseInterceptor.runFrom(EnemyCommandPhase).to(CommandPhase);
expect(game.scene.getParty()[0].getMoveHistory().at(0).move).toBe(Moves.DISABLE);
expect(game.scene.getParty()[0].getMoveHistory().at(0).result).toBe(MoveResult.SUCCESS);
expect(game.scene.getEnemyParty()[0].getMoveHistory().at(0).move).toBe(Moves.SPLASH);
expect(game.scene.getEnemyParty()[0].summonData.tags).toHaveLength(1);
expect(game.scene.getEnemyParty()[0].summonData.tags[0]).toBeInstanceOf(DisabledTag);
expect((game.scene.getEnemyParty()[0].getTag(BattlerTagType.DISABLED) as DisabledTag).moveId).toBe(Moves.SPLASH);
expect(game.scene.getEnemyParty()[0].isMoveDisabled(Moves.SPLASH)).toBe(true);
_useMove();
await game.phaseInterceptor.runFrom(EnemyCommandPhase).to(CommandPhase);
expect(game.scene.getEnemyParty()[0].getLastXMoves().at(0).move).toBe(Moves.STRUGGLE);
}, 20000);
it("DISABLE interrupts target's move when user moves first", async() => {
// Player goes first
await game.startBattle([
Species.PIKACHU,
Species.SHUCKLE,
]);
_useMove(Moves.SPLASH);
await game.phaseInterceptor.runFrom(EnemyCommandPhase).to(CommandPhase);
expect(game.scene.getEnemyParty()[0].getMoveHistory()).toHaveLength(1);
// Both mons just used Splash last turn; now have player use Disable.
_useMove(Moves.DISABLE);
await game.phaseInterceptor.runFrom(EnemyCommandPhase).to(CommandPhase);
expect(game.scene.getPlayerPokemon().getMoveHistory().length === 2);
expect(game.scene.getEnemyParty()[0].isMoveDisabled(Moves.SPLASH));
expect(game.scene.getEnemyParty()[0].getMoveHistory().length === 2);
expect(game.scene.getEnemyParty()[0].getLastXMoves().at(0).result).toBe(MoveResult.FAIL);
expect(game.scene.getEnemyParty()[0].getLastXMoves().at(1).result).toBe(MoveResult.SUCCESS);
}, 20000);
});