From f159b2e84d70dddb1df27b24e2252d8b6d94b303 Mon Sep 17 00:00:00 2001
From: Christopher Schmidt <schmidtc086@outlook.com>
Date: Fri, 30 Aug 2024 13:09:17 -0400
Subject: [PATCH] Adds unit tests for sparkly swirl

---
 src/data/move.ts                     |  2 +-
 src/test/moves/sparkly_swirl.test.ts | 87 ++++++++++++++++++++++++++++
 2 files changed, 88 insertions(+), 1 deletion(-)
 create mode 100644 src/test/moves/sparkly_swirl.test.ts

diff --git a/src/data/move.ts b/src/data/move.ts
index 70b4b0892bf..80e1addccae 100644
--- a/src/data/move.ts
+++ b/src/data/move.ts
@@ -35,8 +35,8 @@ import { PokemonHealPhase } from "#app/phases/pokemon-heal-phase";
 import { StatChangePhase } from "#app/phases/stat-change-phase";
 import { SwitchPhase } from "#app/phases/switch-phase";
 import { SwitchSummonPhase } from "#app/phases/switch-summon-phase";
+import { ShowAbilityPhase } from "#app/phases/show-ability-phase";
 import { SpeciesFormChangeRevertWeatherFormTrigger } from "./pokemon-forms";
-import { ShowAbilityPhase } from "#app/phases/show-ability-phase.js";
 import { NumberHolder } from "#app/utils";
 import { GameMode } from "#app/game-mode";
 import { applyChallenges, ChallengeType } from "./challenge";
diff --git a/src/test/moves/sparkly_swirl.test.ts b/src/test/moves/sparkly_swirl.test.ts
new file mode 100644
index 00000000000..55a3d4661d4
--- /dev/null
+++ b/src/test/moves/sparkly_swirl.test.ts
@@ -0,0 +1,87 @@
+import { allMoves } from "#app/data/move.js";
+import { StatusEffect } from "#app/enums/status-effect.js";
+import { CommandPhase } from "#app/phases/command-phase.js";
+import { Abilities } from "#enums/abilities";
+import { Moves } from "#enums/moves";
+import { Species } from "#enums/species";
+import GameManager from "#test/utils/gameManager";
+import { SPLASH_ONLY } from "#test/utils/testUtils";
+import Phaser from "phaser";
+import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
+
+describe("Moves - Sparkly Swirl", () => {
+  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
+      .enemySpecies(Species.SHUCKLE)
+      .enemyLevel(100)
+      .enemyMoveset(SPLASH_ONLY)
+      .enemyAbility(Abilities.BALL_FETCH)
+      .moveset([Moves.SPARKLY_SWIRL, Moves.SPLASH])
+      .ability(Abilities.BALL_FETCH);
+
+    vi.spyOn(allMoves[Moves.SPARKLY_SWIRL], "accuracy", "get").mockReturnValue(100);
+  });
+
+  it("should cure status effect of the user, its ally, and all party pokemon", async () => {
+    game.override
+      .battleType("double")
+      .statusEffect(StatusEffect.BURN);
+    await game.classicMode.startBattle([Species.RATTATA, Species.RATTATA, Species.RATTATA]);
+    const [leftPlayer, rightPlayer, partyPokemon] = game.scene.getParty();
+    const leftOpp = game.scene.getEnemyPokemon()!;
+
+    vi.spyOn(leftPlayer, "resetStatus");
+    vi.spyOn(rightPlayer, "resetStatus");
+    vi.spyOn(partyPokemon, "resetStatus");
+
+    game.move.select(Moves.SPARKLY_SWIRL, 0, leftOpp.getBattlerIndex());
+    await game.phaseInterceptor.to(CommandPhase);
+    game.move.select(Moves.SPLASH, 1);
+    await game.toNextTurn();
+
+    expect(leftPlayer.resetStatus).toHaveBeenCalledOnce();
+    expect(rightPlayer.resetStatus).toHaveBeenCalledOnce();
+    expect(partyPokemon.resetStatus).toHaveBeenCalledOnce();
+
+    expect(leftPlayer.status?.effect).toBeUndefined();
+    expect(rightPlayer.status?.effect).toBeUndefined();
+    expect(partyPokemon.status?.effect).toBeUndefined();
+  });
+
+  it("should not cure status effect of the target/target's allies", async () => {
+    game.override
+      .battleType("double")
+      .enemyStatusEffect(StatusEffect.BURN);
+    await game.classicMode.startBattle([Species.RATTATA, Species.RATTATA]);
+    const [leftOpp, rightOpp] = game.scene.getEnemyField();
+
+    vi.spyOn(leftOpp, "resetStatus");
+    vi.spyOn(rightOpp, "resetStatus");
+
+    game.move.select(Moves.SPARKLY_SWIRL, 0, leftOpp.getBattlerIndex());
+    await game.phaseInterceptor.to(CommandPhase);
+    game.move.select(Moves.SPLASH, 1);
+    await game.toNextTurn();
+
+    expect(leftOpp.resetStatus).toHaveBeenCalledTimes(0);
+    expect(rightOpp.resetStatus).toHaveBeenCalledTimes(0);
+
+    expect(leftOpp.status?.effect).toBeTruthy();
+    expect(rightOpp.status?.effect).toBeTruthy();
+
+    expect(leftOpp.status?.effect).toBe(StatusEffect.BURN);
+    expect(rightOpp.status?.effect).toBe(StatusEffect.BURN);
+  });
+});