From eb1d195866c7759457db8e894707a9b4d6b677cf Mon Sep 17 00:00:00 2001
From: snoozbuster <snoozbuster@outlook.com>
Date: Mon, 12 Aug 2024 21:40:21 -0700
Subject: [PATCH] add shorthand for setting a moveset to all one move

---
 src/test/utils/helpers/overridesHelper.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/test/utils/helpers/overridesHelper.ts b/src/test/utils/helpers/overridesHelper.ts
index dbcb02825f2..2e4319b211f 100644
--- a/src/test/utils/helpers/overridesHelper.ts
+++ b/src/test/utils/helpers/overridesHelper.ts
@@ -111,7 +111,10 @@ export class OverridesHelper extends GameManagerHelper {
    * @param moveset the {@linkcode Moves | moves}set to set
    * @returns this
    */
-  moveset(moveset: Moves[]): this {
+  moveset(moveset: Moves[] | Moves): this {
+    if (!Array.isArray(moveset)) {
+      moveset = new Array(4).fill(moveset);
+    }
     vi.spyOn(Overrides, "MOVESET_OVERRIDE", "get").mockReturnValue(moveset);
     const movesetStr = moveset.map((moveId) => Moves[moveId]).join(", ");
     this.log(`Player Pokemon moveset set to ${movesetStr} (=[${moveset.join(", ")}])!`);
@@ -230,7 +233,10 @@ export class OverridesHelper extends GameManagerHelper {
    * @param moveset the {@linkcode Moves | moves}set to set
    * @returns this
    */
-  enemyMoveset(moveset: Moves[]): this {
+  enemyMoveset(moveset: Moves[] | Moves): this {
+    if (!Array.isArray(moveset)) {
+      moveset = new Array(4).fill(moveset);
+    }
     vi.spyOn(Overrides, "OPP_MOVESET_OVERRIDE", "get").mockReturnValue(moveset);
     const movesetStr = moveset.map((moveId) => Moves[moveId]).join(", ");
     this.log(`Enemy Pokemon moveset set to ${movesetStr} (=[${moveset.join(", ")}])!`);