From d4b4312dd2ed78b9d82a4157fa119c096548ba86 Mon Sep 17 00:00:00 2001 From: Ed_ Date: Thu, 7 May 2026 15:02:00 -0400 Subject: [PATCH] chore: remove debug logging and fix closure bug in test hooks --- src/app_controller.py | 11 +++++++---- tests/mock_concurrent_mma.py | 4 ++-- 2 files changed, 9 insertions(+), 6 deletions(-) diff --git a/src/app_controller.py b/src/app_controller.py index 7ce8ae1..5749a61 100644 --- a/src/app_controller.py +++ b/src/app_controller.py @@ -907,8 +907,9 @@ class AppController: if self.test_hooks_enabled and not getattr(self, "ui_manual_approve", False): if "dialog_container" in task: class AutoStepDialog: - def wait(self): return True, task.get("payload", "") - task["dialog_container"][0] = AutoStepDialog() + def __init__(self, t): self.t = t + def wait(self): return True, self.t.get("payload", "") + task["dialog_container"][0] = AutoStepDialog(task) continue dlg = MMAApprovalDialog(str(task.get("ticket_id") or ""), str(task.get("payload") or "")) self._pending_mma_approvals.append(task) @@ -928,8 +929,10 @@ class AppController: if self.test_hooks_enabled and not getattr(self, "ui_manual_approve", False): if "dialog_container" in task: class AutoSpawnDialog: - def wait(self): return {'approved': True, 'abort': False, 'prompt': task.get("prompt"), 'context_md': task.get("context_md")} - task["dialog_container"][0] = AutoSpawnDialog() + def __init__(self, t): self.t = t + def wait(self): + return {'approved': True, 'abort': False, 'prompt': self.t.get("prompt"), 'context_md': self.t.get("context_md")} + task["dialog_container"][0] = AutoSpawnDialog(task) continue spawn_dlg = MMASpawnApprovalDialog( str(task.get("ticket_id") or ""), diff --git a/tests/mock_concurrent_mma.py b/tests/mock_concurrent_mma.py index 207aca2..2c31cba 100644 --- a/tests/mock_concurrent_mma.py +++ b/tests/mock_concurrent_mma.py @@ -51,9 +51,9 @@ def main() -> None: if 'You are assigned to Ticket' in prompt: # Extract ticket ID import re - match = re.search(r'Ticket (ticket-[A-B]-1)', prompt) + match = re.search(r'Ticket (ticket-[A-Ba-b]-1)', prompt, re.IGNORECASE) tid = match.group(1) if match else "unknown" - + print(json.dumps({ "type": "message", "role": "assistant",