From 3a425db6a90651f890050ff3dfdb829c5f469cfa Mon Sep 17 00:00:00 2001
From: mikecarper <mikecarper@gmail.com>
Date: Sun, 18 Oct 2026 18:21:52 -0700
Subject: [PATCH] Fix V4 button fixture for display presence checks

---
 test/test_heltec_v4_wifi_setup_page.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/test_heltec_v4_wifi_setup_page.py b/test/test_heltec_v4_wifi_setup_page.py
index e0fcc19d0..6a64da502 100644
--- a/test/test_heltec_v4_wifi_setup_page.py
+++ b/test/test_heltec_v4_wifi_setup_page.py
@@ -286,6 +286,7 @@ int main() {
 #include <cassert>
 #define UI_WIFI_SETUP_HOME_PAGE 1
 #include <cstdint>
+#include <cstddef>
 unsigned long now = 4000;
 unsigned long millis() { return now; }
 struct Screen {};
@@ -303,6 +304,7 @@ struct UITask {
   Screen* msg_preview;
   unsigned long ui_started_at = 0;
   bool _startup_complete = false;
+  Screen* _display = nullptr;
   char handleLongPress(char c);
 };
 ''' + ui[start:end] + r'''
@@ -310,6 +312,7 @@ int main() {
   HomeScreen home;
   Screen reader;
   UITask task{&reader, &home, &reader};
+  task._display = &home;
   assert(task.handleLongPress('E') == 'E');
   assert(the_mesh.rescue_calls == 0);
   task.curr = &home;