From b58fb9351eb3e8e838caa4c6327163a214855d59 Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Fri, 28 Mar 2025 11:45:35 +0100
Subject: [PATCH] fixup! chore(tests): adapt testing framework to session based

---
 tests/device_handler.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/device_handler.py b/tests/device_handler.py
index 9edf0b560d..a3dc61314d 100644
--- a/tests/device_handler.py
+++ b/tests/device_handler.py
@@ -53,6 +53,7 @@ class BackgroundDeviceHandler:
     def _configure_client(self, client: "Client") -> None:
         self.client = client
         self.client.ui = NullUI  # type: ignore [NullUI is OK UI]
+        self.client.button_callback = self.client.ui.button_request
         self.client.watch_layout(True)
 
     def run_with_session(