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

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

diff --git a/src/test/utils/overridesHelper.ts b/src/test/utils/overridesHelper.ts
index 00de71240b4..689f6e86be6 100644
--- a/src/test/utils/overridesHelper.ts
+++ b/src/test/utils/overridesHelper.ts
@@ -1,3 +1,4 @@
+import { StatusEffect } from "#app/data/status-effect.js";
 import { Weather, WeatherType } from "#app/data/weather";
 import { Abilities } from "#app/enums/abilities.js";
 import { Biome } from "#app/enums/biome";
@@ -242,6 +243,17 @@ export class OverridesHelper {
     return this;
   }
 
+  /**
+   * Override the enemy (pokemon) {@linkcode StatusEffect | status-effect}
+   * @param statusEffect the {@linkcode StatusEffect | status-effect} to set
+   * @returns
+   */
+  enemyStatusEffect(statusEffect: StatusEffect): this {
+    vi.spyOn(Overrides, "OPP_STATUS_OVERRIDE", "get").mockReturnValue(statusEffect);
+    this.log(`Enemy Pokemon status-effect set to ${StatusEffect[statusEffect]} (=${statusEffect})!`);
+    return this;
+  }
+
   private log(...params: any[]) {
     console.log("Overrides:", ...params);
   }