From 5d7a3d28ffed2f6e896db8b2b1414cb403ebb5a8 Mon Sep 17 00:00:00 2001
From: Bertie690 <taylormw163@gmail.com>
Date: Fri, 1 Aug 2025 21:39:09 -0400
Subject: [PATCH] dddddd

---
 test/items/lock-capsule.test.ts | 24 ++++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/test/items/lock-capsule.test.ts b/test/items/lock-capsule.test.ts
index 17ca8c28aa8..01552a4db37 100644
--- a/test/items/lock-capsule.test.ts
+++ b/test/items/lock-capsule.test.ts
@@ -1,7 +1,8 @@
 import { AbilityId } from "#enums/ability-id";
 import { ModifierTier } from "#enums/modifier-tier";
 import { MoveId } from "#enums/move-id";
-import type { SelectModifierPhase } from "#phases/select-modifier-phase";
+import { UiMode } from "#enums/ui-mode";
+import { SelectModifierPhase } from "#phases/select-modifier-phase";
 import { GameManager } from "#test/test-utils/game-manager";
 import Phaser from "phaser";
 import { afterEach, beforeAll, beforeEach, describe, expect, it } from "vitest";
@@ -33,17 +34,20 @@ describe("Items - Lock Capsule", () => {
 
   it("doesn't set the cost of common tier items to 0", async () => {
     await game.classicMode.startBattle();
-    game.scene.phaseManager.clearAllPhases();
-    game.scene.phaseManager.unshiftNew("SelectModifierPhase", 0, undefined, {
-      guaranteedModifierTiers: [ModifierTier.COMMON, ModifierTier.COMMON, ModifierTier.COMMON],
-      fillRemaining: false,
+    game.scene.phaseManager.overridePhase(
+      new SelectModifierPhase(0, undefined, {
+        guaranteedModifierTiers: [ModifierTier.COMMON, ModifierTier.COMMON, ModifierTier.COMMON],
+        fillRemaining: false,
+      }),
+    );
+
+    game.onNextPrompt("SelectModifierPhase", UiMode.MODIFIER_SELECT, () => {
+      const selectModifierPhase = game.scene.phaseManager.getCurrentPhase() as SelectModifierPhase;
+      const rerollCost = selectModifierPhase.getRerollCost(true);
+      expect(rerollCost).toBe(150);
     });
 
     game.doSelectModifier();
-    await game.phaseInterceptor.to("SelectModifierPhase", false);
-
-    const selectModifierPhase = game.scene.phaseManager.getCurrentPhase() as SelectModifierPhase;
-    const rerollCost = selectModifierPhase.getRerollCost(true);
-    expect(rerollCost).toBe(150);
+    await game.phaseInterceptor.to("SelectModifierPhase");
   });
 });