From f0cbe8b679dc281d647e231829e8254f1e58c52d Mon Sep 17 00:00:00 2001
From: xsn34kzx <xsn34kzx@gmail.com>
Date: Thu, 22 Aug 2024 20:30:46 -0400
Subject: [PATCH] Adjust Last X Item Unit Test

---
 src/test/items/temp_stat_stage_booster.test.ts | 11 ++++++-----
 src/test/utils/helpers/overridesHelper.ts      |  6 ++++++
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/src/test/items/temp_stat_stage_booster.test.ts b/src/test/items/temp_stat_stage_booster.test.ts
index 7fd62fec20e..f02a6c6e741 100644
--- a/src/test/items/temp_stat_stage_booster.test.ts
+++ b/src/test/items/temp_stat_stage_booster.test.ts
@@ -7,7 +7,6 @@ import { Moves } from "#app/enums/moves.js";
 import { TurnEndPhase } from "#app/phases/turn-end-phase.js";
 import { SPLASH_ONLY } from "../utils/testUtils";
 import { Abilities } from "#app/enums/abilities.js";
-import Overrides from "#app/overrides";
 import { TempStatStageBoosterModifier } from "#app/modifier/modifier.js";
 import { Mode } from "#app/ui/ui.js";
 import { Button } from "#app/enums/buttons.js";
@@ -39,8 +38,8 @@ describe("Items - Temporary Stat Stage Boosters", () => {
     game.override.battleType("single");
     game.override.enemySpecies(Species.MEW);
     game.override.enemyMoveset(SPLASH_ONLY);
-    game.override.enemyAbility(Abilities.PICKUP);
-    game.override.moveset([ Moves.TACKLE, Moves.HONE_CLAWS, Moves.BELLY_DRUM ]);
+    game.override.enemyAbility(Abilities.BALL_FETCH);
+    game.override.moveset([ Moves.TACKLE, Moves.SPLASH, Moves.HONE_CLAWS, Moves.BELLY_DRUM ]);
     game.override.startingModifier([{ name: "TEMP_STAT_STAGE_BOOSTER", type: Stat.ATK }]);
   });
 
@@ -102,12 +101,14 @@ describe("Items - Temporary Stat Stage Boosters", () => {
 
   it("should renew how many battles are left of existing booster when picking up new booster of same type", async() => {
     game.override.startingLevel(200);
-    vi.spyOn(Overrides, "ITEM_REWARD_OVERRIDE", "get").mockReturnValue([{ name: "TEMP_STAT_STAGE_BOOSTER", type: Stat.ATK }]);
+    game.override.itemRewards([{ name: "TEMP_STAT_STAGE_BOOSTER", type: Stat.ATK }]);
     await game.startBattle([
       Species.PIKACHU
     ]);
 
-    game.move.select(Moves.TACKLE);
+    game.move.select(Moves.SPLASH);
+
+    await game.doKillOpponents();
 
     await game.phaseInterceptor.to(BattleEndPhase);
 
diff --git a/src/test/utils/helpers/overridesHelper.ts b/src/test/utils/helpers/overridesHelper.ts
index d5eaee003db..24010304c42 100644
--- a/src/test/utils/helpers/overridesHelper.ts
+++ b/src/test/utils/helpers/overridesHelper.ts
@@ -281,6 +281,12 @@ export class OverridesHelper extends GameManagerHelper {
     return this;
   }
 
+  itemRewards(items: ModifierOverride[]) {
+    vi.spyOn(Overrides, "ITEM_REWARD_OVERRIDE", "get").mockReturnValue(items);
+    this.log("Item rewards set to:", items);
+    return this;
+  }
+
   private log(...params: any[]) {
     console.log("Overrides:", ...params);
   }