From 97376d5c70e54d9edb32da153793f96a6988e30f Mon Sep 17 00:00:00 2001
From: Felix Staud <felix.staud@headwire.com>
Date: Thu, 25 Jul 2024 15:54:48 -0700
Subject: [PATCH] add OverridesHelper.starterForms

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

diff --git a/src/test/utils/overridesHelper.ts b/src/test/utils/overridesHelper.ts
index 5767cac4914..ad19547b992 100644
--- a/src/test/utils/overridesHelper.ts
+++ b/src/test/utils/overridesHelper.ts
@@ -63,6 +63,20 @@ export class OverridesHelper {
     return this;
   }
 
+  /**
+   * Override the player (pokemons) forms
+   * @param forms the (pokemon) forms to set
+   * @returns this
+   */
+  starterForms(forms: Partial<Record<Species, number>>): this {
+    vi.spyOn(Overrides, "STARTER_FORM_OVERRIDES", "get").mockReturnValue(forms);
+    const formsStr = Object.entries(forms)
+      .map(([speciesId, formIndex]) => `${Species[speciesId]}=${formIndex}`)
+      .join(", ");
+    this.log(`Player Pokemon form set to: ${formsStr}!`);
+    return this;
+  }
+
   /**
    * Override the player (pokemon) {@linkcode Abilities | ability}
    * @param ability the (pokemon) {@linkcode Abilities | ability} to set