From 73538df5d07d1729c3f2e5dd55407767f21c5d50 Mon Sep 17 00:00:00 2001
From: Christopher Schmidt <schmidtc086@outlook.com>
Date: Sun, 26 Jan 2025 22:25:48 -0500
Subject: [PATCH] Add tests for reviver seed

---
 src/test/items/reviver_seed.test.ts | 117 ++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 src/test/items/reviver_seed.test.ts

diff --git a/src/test/items/reviver_seed.test.ts b/src/test/items/reviver_seed.test.ts
new file mode 100644
index 00000000000..da32799132e
--- /dev/null
+++ b/src/test/items/reviver_seed.test.ts
@@ -0,0 +1,117 @@
+import { allMoves } from "#app/data/move";
+import { BattlerTagType } from "#app/enums/battler-tag-type";
+import type { PokemonInstantReviveModifier } from "#app/modifier/modifier";
+import { Abilities } from "#enums/abilities";
+import { Moves } from "#enums/moves";
+import { Species } from "#enums/species";
+import GameManager from "#test/utils/gameManager";
+import Phaser from "phaser";
+import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
+
+describe("Items - Reviver Seed", () => {
+  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
+      .moveset([ Moves.SPLASH, Moves.TACKLE, Moves.ENDURE ])
+      .ability(Abilities.BALL_FETCH)
+      .battleType("single")
+      .disableCrits()
+      .enemySpecies(Species.MAGIKARP)
+      .enemyAbility(Abilities.BALL_FETCH)
+      .startingHeldItems([{ name: "REVIVER_SEED" }])
+      .enemyHeldItems([{ name: "REVIVER_SEED" }])
+      .enemyMoveset(Moves.SPLASH);
+    vi.spyOn(allMoves[Moves.SHEER_COLD], "accuracy", "get").mockReturnValue(100);
+    vi.spyOn(allMoves[Moves.LEECH_SEED], "accuracy", "get").mockReturnValue(100);
+    vi.spyOn(allMoves[Moves.WHIRLPOOL], "accuracy", "get").mockReturnValue(100);
+    vi.spyOn(allMoves[Moves.WILL_O_WISP], "accuracy", "get").mockReturnValue(100);
+  });
+
+  it.each([
+    { moveType: "Special Move", move: Moves.WATER_GUN },
+    { moveType: "Physical Move", move: Moves.TACKLE },
+    { moveType: "Fixed Damage Move", move: Moves.SEISMIC_TOSS },
+    { moveType: "Final Gambit", move: Moves.FINAL_GAMBIT },
+    { moveType: "Counter", move: Moves.COUNTER },
+    { moveType: "OHKO", move: Moves.SHEER_COLD }
+  ])("should activate the holder's reviver seed from a $moveType", async ({ move }) => {
+    game.override
+      .enemyLevel(100)
+      .startingLevel(1)
+      .enemyMoveset(move);
+    await game.classicMode.startBattle([ Species.MAGIKARP, Species.FEEBAS ]);
+    const player = game.scene.getPlayerPokemon()!;
+    player.damageAndUpdate(player.hp - 1);
+
+    const reviverSeed = player.getHeldItems()[0] as PokemonInstantReviveModifier;
+    vi.spyOn(reviverSeed, "apply");
+
+    game.move.select(Moves.TACKLE);
+    await game.phaseInterceptor.to("BerryPhase");
+
+    expect(reviverSeed.apply).toHaveReturnedWith(true); // Reviver Seed triggers
+    expect(player.isFainted()).toBeFalsy();
+  });
+
+  it("should activate the holder's reviver seed from confusion self-hit", async () => {
+    game.override
+      .enemyLevel(1)
+      .startingLevel(100)
+      .enemyMoveset(Moves.SPLASH);
+    await game.classicMode.startBattle([ Species.MAGIKARP, Species.FEEBAS ]);
+    const player = game.scene.getPlayerPokemon()!;
+    player.damageAndUpdate(player.hp - 1);
+    player.addTag(BattlerTagType.CONFUSED, 3);
+
+    const reviverSeed = player.getHeldItems()[0] as PokemonInstantReviveModifier;
+    vi.spyOn(reviverSeed, "apply");
+
+    vi.spyOn(player, "randSeedInt").mockReturnValue(0); // Force confusion self-hit
+    game.move.select(Moves.TACKLE);
+    await game.phaseInterceptor.to("BerryPhase");
+
+    expect(reviverSeed.apply).toHaveReturnedWith(true); // Reviver Seed triggers
+    expect(player.isFainted()).toBeFalsy();
+  });
+
+
+  //Need to fix some of tests, something wrong with the enemy fainting, wrong phase being chosen.. not sure
+  it.each([
+    //{ moveType: "Damaging Move Chip Damage", move: Moves.SALT_CURE },
+    //{ moveType: "Chip Damage", move: Moves.LEECH_SEED },
+    //{ moveType: "Trapping Chip Damage", move: Moves.WHIRLPOOL },
+    { moveType: "Status Effect Damage", move: Moves.WILL_O_WISP },
+    { moveType: "Weather", move: Moves.SANDSTORM }
+  ])("should not activate the holder's reviver seed from $moveType", async ({ move }) => {
+    game.override
+      .enemyLevel(1)
+      .startingLevel(100)
+      .enemySpecies(Species.MAGIKARP)
+      .moveset(move)
+      .enemyMoveset(Moves.SPLASH);
+    await game.classicMode.startBattle([ Species.MAGIKARP, Species.FEEBAS ]);
+    const enemy = game.scene.getEnemyPokemon()!;
+    enemy.damageAndUpdate(enemy.hp - 1);
+
+    const enemySeed = enemy.getHeldItems()[0] as PokemonInstantReviveModifier;
+    vi.spyOn(enemySeed, "apply");
+
+    game.move.select(move);
+    await game.phaseInterceptor.to("BerryPhase");
+
+    expect(enemy.isFainted()).toBeTruthy();
+  });
+});