From f8e6230a14c96edc64090e6bf44fd330ad7cd0dc Mon Sep 17 00:00:00 2001
From: Felix Staud <felix.staud@headwire.com>
Date: Thu, 25 Jul 2024 15:19:16 -0700
Subject: [PATCH] add OverridesHelper.starterSpecies()

---
 src/test/utils/overridesHelper.ts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/test/utils/overridesHelper.ts b/src/test/utils/overridesHelper.ts
index 7571a4eb657..c9627f38c0b 100644
--- a/src/test/utils/overridesHelper.ts
+++ b/src/test/utils/overridesHelper.ts
@@ -41,6 +41,17 @@ export class OverridesHelper {
     return this;
   }
 
+  /**
+   * Override the player (pokemon) {@linkcode Species | species}
+   * @param species the (pokemon) {@linkcode Species | species} to set
+   * @returns this
+   */
+  starterSpecies(species: Species | number): this {
+    vi.spyOn(Overrides, "STARTER_SPECIES_OVERRIDE", "get").mockReturnValue(species);
+    this.log(`Player Pokemon species set to ${Species[species]} (=${species})!`);
+    return this;
+  }
+
   /**
    * Override each wave to have or not have standard trainer battles
    * @returns this