From b455214421107f3ff87b8861ef5b98f138c1dcde Mon Sep 17 00:00:00 2001
From: Tiago Rodrigues <tiago.n.rodrigues@tecnico.ulisboa.pt>
Date: Thu, 24 Jul 2025 02:43:35 +0100
Subject: [PATCH] [Fix] Updated tests to handle the corfirmation box

---
 src/ui/party-ui-handler.ts         |  2 +-
 test/ui/item-manage-button.test.ts | 11 ++++++++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/ui/party-ui-handler.ts b/src/ui/party-ui-handler.ts
index 06612fbd9a8..4ccf324d976 100644
--- a/src/ui/party-ui-handler.ts
+++ b/src/ui/party-ui-handler.ts
@@ -685,6 +685,7 @@ export class PartyUiHandler extends MessageUiHandler {
 
   private doDiscard(option: PartyOption, pokemon: PlayerPokemon) {
     const itemModifiers = this.getTransferrableItemsFromPokemon(pokemon);
+    this.clearOptions();
 
     if (option === PartyOption.ALL) {
       for (let i = 0; i < itemModifiers.length; i++) {
@@ -693,7 +694,6 @@ export class PartyUiHandler extends MessageUiHandler {
     } else {
       globalScene.tryDiscardHeldItemModifier(itemModifiers[option], this.transferQuantities[option]);
     }
-    return;
   }
 
   private moveOptionCursor(button: Button.UP | Button.DOWN): boolean {
diff --git a/test/ui/item-manage-button.test.ts b/test/ui/item-manage-button.test.ts
index 26d37ba6ef1..c185447f16f 100644
--- a/test/ui/item-manage-button.test.ts
+++ b/test/ui/item-manage-button.test.ts
@@ -145,10 +145,19 @@ describe("UI - Transfer Items", () => {
         await new Promise(r => setTimeout(r, 100));
         const handler = game.scene.ui.getHandler() as PartyUiHandler;
         handler.processInput(Button.ACTION);
+        resolve();
+      });
+    });
+
+    await new Promise<void>(resolve => {
+      game.onNextPrompt("SelectModifierPhase", UiMode.PARTY, async () => {
+        await new Promise(r => setTimeout(r, 100));
+        const handler = game.scene.ui.getHandler() as PartyUiHandler;
+        handler.processInput(Button.ACTION);
+
         pokemon = game.field.getPlayerPokemon();
 
         handler.processInput(Button.CANCEL);
-
         resolve();
       });
     });