Fixed imposter tests

This commit is contained in:
Bertie690 2025-06-21 10:57:11 -04:00
parent 50c9bcff92
commit d3381fa003
2 changed files with 47 additions and 59 deletions

View File

@ -2,9 +2,8 @@ import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";
import Phaser from "phaser"; import Phaser from "phaser";
import GameManager from "#test/testUtils/gameManager"; import GameManager from "#test/testUtils/gameManager";
import { SpeciesId } from "#enums/species-id"; import { SpeciesId } from "#enums/species-id";
import { TurnEndPhase } from "#app/phases/turn-end-phase";
import { MoveId } from "#enums/move-id"; import { MoveId } from "#enums/move-id";
import { Stat, BATTLE_STATS, EFFECTIVE_STATS } from "#enums/stat"; import { Stat } from "#enums/stat";
import { AbilityId } from "#enums/ability-id"; import { AbilityId } from "#enums/ability-id";
// TODO: Add more tests once Imposter is fully implemented // TODO: Add more tests once Imposter is fully implemented
@ -38,40 +37,24 @@ describe("Abilities - Imposter", () => {
it("should copy species, ability, gender, all stats except HP, all stat stages, moveset, and types of target", async () => { it("should copy species, ability, gender, all stats except HP, all stat stages, moveset, and types of target", async () => {
await game.classicMode.startBattle([SpeciesId.DITTO]); await game.classicMode.startBattle([SpeciesId.DITTO]);
game.move.select(MoveId.SPLASH); const player = game.field.getPlayerPokemon();
await game.phaseInterceptor.to(TurnEndPhase); const enemy = game.field.getEnemyPokemon();
const player = game.scene.getPlayerPokemon()!;
const enemy = game.scene.getEnemyPokemon()!;
expect(player.getSpeciesForm().speciesId).toBe(enemy.getSpeciesForm().speciesId); expect(player.getSpeciesForm().speciesId).toBe(enemy.getSpeciesForm().speciesId);
expect(player.getAbility()).toBe(enemy.getAbility()); expect(player.getAbility()).toBe(enemy.getAbility());
expect(player.getGender()).toBe(enemy.getGender()); expect(player.getGender()).toBe(enemy.getGender());
expect(player.getStat(Stat.HP, false)).not.toBe(enemy.getStat(Stat.HP)); const playerStats = player.getStats(false);
for (const s of EFFECTIVE_STATS) { const enemyStats = enemy.getStats(false);
expect(player.getStat(s, false)).toBe(enemy.getStat(s, false)); // HP stays the same; all other stats should carry over
} expect(playerStats[0]).not.toBe(enemyStats[0]);
expect(playerStats.slice(1)).toEqual(enemyStats.slice(1));
for (const s of BATTLE_STATS) { expect(player.getStatStages()).toEqual(enemy.getStatStages());
expect(player.getStatStage(s)).toBe(enemy.getStatStage(s));
}
const playerMoveset = player.getMoveset(); expect(player.getMoveset().map(m => m.moveId)).toEqual(player.getMoveset().map(m => m.moveId));
const enemyMoveset = player.getMoveset();
expect(playerMoveset.length).toBe(enemyMoveset.length); expect(player.getTypes()).toEqual(enemy.getTypes());
for (let i = 0; i < playerMoveset.length && i < enemyMoveset.length; i++) {
expect(playerMoveset[i]?.moveId).toBe(enemyMoveset[i]?.moveId);
}
const playerTypes = player.getTypes();
const enemyTypes = enemy.getTypes();
expect(playerTypes.length).toBe(enemyTypes.length);
for (let i = 0; i < playerTypes.length && i < enemyTypes.length; i++) {
expect(playerTypes[i]).toBe(enemyTypes[i]);
}
}); });
it("should copy in-battle overridden stats", async () => { it("should copy in-battle overridden stats", async () => {
@ -80,33 +63,37 @@ describe("Abilities - Imposter", () => {
const [karp, ditto] = game.scene.getPlayerParty(); const [karp, ditto] = game.scene.getPlayerParty();
const enemy = game.field.getEnemyPokemon(); const enemy = game.field.getEnemyPokemon();
game.field.mockAbility(ditto, AbilityId.IMPOSTER); const abMock = game.field.mockAbility(ditto, AbilityId.IMPOSTER);
const oldAtk = karp.getStat(Stat.ATK);
// Turn 1: Use power split // Turn 1: Use power split
const avgAtk = Math.floor((karp.getStat(Stat.ATK, false) + enemy.getStat(Stat.ATK, false)) / 2);
const avgSpAtk = Math.floor((karp.getStat(Stat.SPATK, false) + enemy.getStat(Stat.SPATK, false)) / 2);
game.move.use(MoveId.SPLASH); game.move.use(MoveId.SPLASH);
await game.move.forceEnemyMove(MoveId.POWER_SPLIT); await game.move.forceEnemyMove(MoveId.POWER_SPLIT);
await game.toNextTurn(); await game.toNextTurn();
expect(enemy.getStat(Stat.ATK, false)).toBe(avgAtk); const avgAtk = Math.floor((karp.getStat(Stat.ATK, false) + enemy.getStat(Stat.ATK, false)) / 2);
expect(enemy.getStat(Stat.SPATK, false)).toBe(avgSpAtk);
// Turn 2: Switch in ditto, should copy enemy stats expect(enemy.getStat(Stat.ATK, false)).toBe(avgAtk);
expect(karp.getStat(Stat.ATK, false)).toBe(avgAtk);
expect(avgAtk).not.toBe(oldAtk);
// Turn 2: Switch in ditto, should copy power split stats
game.doSwitchPokemon(1); game.doSwitchPokemon(1);
await game.move.forceEnemyMove(MoveId.SPLASH); await game.move.forceEnemyMove(MoveId.SPLASH);
await game.toNextTurn(); await game.phaseInterceptor.to("PokemonTransformPhase", false);
// reset ability mock after activating so Ditto doesn't keep on transforming into itself
abMock.mockRestore();
await game.toEndOfTurn();
expect(ditto.getStat(Stat.ATK, false)).toBe(avgAtk); expect(ditto.getStat(Stat.ATK, false)).toBe(avgAtk);
expect(ditto.getStat(Stat.SPATK, false)).toBe(avgSpAtk);
}); });
it("should set each move's pp to a maximum of 5", async () => { it("should set each move's pp to a maximum of 5", async () => {
game.override.enemyMoveset([MoveId.SWORDS_DANCE, MoveId.GROWL, MoveId.SKETCH, MoveId.RECOVER]); game.override.enemyMoveset([MoveId.SWORDS_DANCE, MoveId.GROWL, MoveId.SKETCH, MoveId.RECOVER]);
await game.classicMode.startBattle([SpeciesId.DITTO]); await game.classicMode.startBattle([SpeciesId.DITTO]);
const player = game.scene.getPlayerPokemon()!;
const player = game.field.getPlayerPokemon();
player.getMoveset().forEach(move => { player.getMoveset().forEach(move => {
// Should set correct maximum PP without touching `ppUp` // Should set correct maximum PP without touching `ppUp`
@ -127,43 +114,41 @@ describe("Abilities - Imposter", () => {
}); });
it("should persist transformed attributes across reloads", async () => { it("should persist transformed attributes across reloads", async () => {
game.override.moveset([MoveId.ABSORB]);
await game.classicMode.startBattle([SpeciesId.DITTO]); await game.classicMode.startBattle([SpeciesId.DITTO]);
const player = game.scene.getPlayerPokemon()!; const player = game.field.getPlayerPokemon();
const enemy = game.scene.getEnemyPokemon()!; const enemy = game.field.getEnemyPokemon();
game.move.select(MoveId.SPLASH); game.move.select(MoveId.SPLASH);
await game.doKillOpponents(); await game.doKillOpponents();
await game.toNextWave(); await game.toNextWave();
expect(game.scene.phaseManager.getCurrentPhase()?.constructor.name).toBe("CommandPhase"); expect(game.scene.phaseManager.getCurrentPhase()?.phaseName).toBe("CommandPhase");
expect(game.scene.currentBattle.waveIndex).toBe(2); expect(game.scene.currentBattle.waveIndex).toBe(2);
await game.reload.reloadSession(); await game.reload.reloadSession();
const playerReloaded = game.scene.getPlayerPokemon()!; const playerReloaded = game.field.getPlayerPokemon();
const playerMoveset = player.getMoveset();
expect(playerReloaded.getSpeciesForm().speciesId).toBe(enemy.getSpeciesForm().speciesId); expect(playerReloaded.getSpeciesForm().speciesId).toBe(SpeciesId.MEW);
expect(playerReloaded.getAbility()).toBe(enemy.getAbility()); expect(playerReloaded.getAbility()).toBe(enemy.getAbility());
expect(playerReloaded.getGender()).toBe(enemy.getGender()); expect(playerReloaded.getGender()).toBe(enemy.getGender());
expect(playerReloaded.getStat(Stat.HP, false)).not.toBe(enemy.getStat(Stat.HP)); const playerStats = player.getStats(false);
for (const s of EFFECTIVE_STATS) { const playerReloadedStats = player.getStats(false);
expect(playerReloaded.getStat(s, false)).toBe(enemy.getStat(s, false)); // HP stays the same; all other stats should carry over
} expect(playerReloadedStats).toEqual(playerStats);
expect(playerMoveset.length).toEqual(1); expect(playerReloaded.getMoveset()).toEqual(player.getMoveset());
expect(playerMoveset[0]?.moveId).toEqual(MoveId.SPLASH);
}); });
// TODO: This doesn't look as though it should work
it("should stay transformed with the correct form after reload", async () => { it("should stay transformed with the correct form after reload", async () => {
game.override.moveset([MoveId.ABSORB]).enemySpecies(SpeciesId.UNOWN); game.override.enemySpecies(SpeciesId.UNOWN);
await game.classicMode.startBattle([SpeciesId.DITTO]); await game.classicMode.startBattle([SpeciesId.DITTO]);
const enemy = game.scene.getEnemyPokemon()!; const player = game.field.getPlayerPokemon();
const enemy = game.field.getEnemyPokemon();
// change form // change form
enemy.species.forms[5]; enemy.species.forms[5];
@ -176,11 +161,14 @@ describe("Abilities - Imposter", () => {
expect(game.scene.phaseManager.getCurrentPhase()?.constructor.name).toBe("CommandPhase"); expect(game.scene.phaseManager.getCurrentPhase()?.constructor.name).toBe("CommandPhase");
expect(game.scene.currentBattle.waveIndex).toBe(2); expect(game.scene.currentBattle.waveIndex).toBe(2);
expect(player.getSpeciesForm().speciesId).toBe(SpeciesId.UNOWN);
expect(player.getSpeciesForm().formIndex).toBe(5);
await game.reload.reloadSession(); await game.reload.reloadSession();
const playerReloaded = game.scene.getPlayerPokemon()!; const playerReloaded = game.field.getPlayerPokemon();
expect(playerReloaded.getSpeciesForm().speciesId).toBe(enemy.getSpeciesForm().speciesId); expect(playerReloaded.getSpeciesForm().speciesId).toBe(SpeciesId.UNOWN);
expect(playerReloaded.getSpeciesForm().formIndex).toBe(enemy.getSpeciesForm().formIndex); expect(playerReloaded.getSpeciesForm().formIndex).toBe(5);
}); });
}); });

View File

@ -70,7 +70,7 @@ export class FieldHelper extends GameManagerHelper {
* @see {@linkcode vi.spyOn} * @see {@linkcode vi.spyOn}
* @see https://vitest.dev/api/mock#mockreturnvalue * @see https://vitest.dev/api/mock#mockreturnvalue
*/ */
public mockAbility(pokemon: Pokemon, ability: AbilityId): MockInstance<(baseOnly?: boolean) => Ability> { public mockAbility(pokemon: Pokemon, ability: AbilityId): MockInstance<(ignoreOverride?: boolean) => Ability> {
return vi.spyOn(pokemon, "getAbility").mockReturnValue(allAbilities[ability]); return vi.spyOn(pokemon, "getAbility").mockReturnValue(allAbilities[ability]);
} }