From c080ba0b46e2cdeaeed8f76c65bfdcd3f3704a72 Mon Sep 17 00:00:00 2001
From: snoozbuster <snoozbuster@outlook.com>
Date: Mon, 12 Aug 2024 21:40:56 -0700
Subject: [PATCH] add methods to overridesHelper for new overrides

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

diff --git a/src/test/utils/helpers/overridesHelper.ts b/src/test/utils/helpers/overridesHelper.ts
index 2e4319b211f..27509567025 100644
--- a/src/test/utils/helpers/overridesHelper.ts
+++ b/src/test/utils/helpers/overridesHelper.ts
@@ -276,6 +276,22 @@ export class OverridesHelper extends GameManagerHelper {
     return this;
   }
 
+  enemyParty(species: Species[]) {
+    vi.spyOn(Overrides, "TRAINER_PARTY_OVERRIDE", "get").mockReturnValue(species);
+    this.log("Enemy trainer party set to:", species);
+    return this;
+  }
+
+  /**
+   * Forces the AI to always switch out
+   * @returns this
+   */
+  forceTrainerSwitches() {
+    vi.spyOn(Overrides, "TRAINER_ALWAYS_SWITCHES_OVERRIDE", "get").mockReturnValue(true);
+    this.log("Trainers will always switch out");
+    return this;
+  }
+
   private log(...params: any[]) {
     console.log("Overrides:", ...params);
   }