From e034d7e04ef0f5c66e3febe297b14560dd81d4fc Mon Sep 17 00:00:00 2001
From: Wlowscha <54003515+Wlowscha@users.noreply.github.com>
Date: Sun, 13 Jul 2025 13:04:46 +0200
Subject: [PATCH] Added test override for pokeballs

---
 test/moves/payback.test.ts                |  2 +-
 test/testUtils/helpers/overridesHelper.ts | 13 +++++++++++++
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/test/moves/payback.test.ts b/test/moves/payback.test.ts
index 20e832288a3..a92d22d7b8f 100644
--- a/test/moves/payback.test.ts
+++ b/test/moves/payback.test.ts
@@ -33,7 +33,7 @@ describe("Move - Payback", () => {
       .enemySpecies(SpeciesId.DRACOVISH)
       .enemyAbility(AbilityId.BALL_FETCH)
       .enemyMoveset(MoveId.SPLASH)
-      .startingModifier([{ name: "POKEBALL", count: 5 }]);
+      .startingPokeballs({ [PokeballType.POKEBALL]: 5 });
 
     powerSpy = vi.spyOn(allMoves[MoveId.PAYBACK], "calculateBattlePower");
   });
diff --git a/test/testUtils/helpers/overridesHelper.ts b/test/testUtils/helpers/overridesHelper.ts
index 9fc13407844..d8e6bf0d9d9 100644
--- a/test/testUtils/helpers/overridesHelper.ts
+++ b/test/testUtils/helpers/overridesHelper.ts
@@ -1,7 +1,9 @@
 /** biome-ignore-start lint/correctness/noUnusedImports: tsdoc imports */
 import type { NewArenaEvent } from "#events/battle-scene";
+
 /** biome-ignore-end lint/correctness/noUnusedImports: tsdoc imports */
 
+import type { PokeballCounts } from "#app/battle-scene";
 import type { BattleStyle, RandomTrainerOverride } from "#app/overrides";
 import Overrides, { defaultOverrides } from "#app/overrides";
 import { AbilityId } from "#enums/ability-id";
@@ -134,6 +136,17 @@ export class OverridesHelper extends GameManagerHelper {
     return this;
   }
 
+  /**
+   * Override the player's starting pokeballs
+   * @param items - The items to hold
+   * @returns `this`
+   */
+  public startingPokeballs(pokeballs: PokeballCounts): this {
+    vi.spyOn(Overrides, "POKEBALL_OVERRIDE", "get").mockReturnValue({ active: true, pokeballs: pokeballs });
+    this.log("Player Pokemon starting held items set to:", { active: true, pokeballs: pokeballs });
+    return this;
+  }
+
   /**
    * Override the player pokemon's {@linkcode SpeciesId | species}
    * @param species - The {@linkcode SpeciesId | species} to set