Fix test

ThomasFarstrike · ThomasFarstrike · commit 144f770bf392 · 2026-05-26T22:40:38.000+02:00
diff --git a/tests/test_graphical_hotspot_then_station.py b/tests/test_graphical_hotspot_then_station.py
@@ -10,7 +10,6 @@
 """
 
 import unittest
-import time
 import lvgl as lv
 import mpos.ui
 from mpos import (
@@ -28,6 +27,26 @@
 class TestGraphicalHotspotThenStation(unittest.TestCase):
     """Test hotspot start flow via the hotspot settings app."""
 
+    def _first_text_in_tree(self, node):
+        try:
+            if hasattr(node, "get_text"):
+                text = node.get_text()
+                if text:
+                    return text
+        except Exception:
+            pass
+
+        try:
+            child_count = node.get_child_count()
+        except Exception:
+            return None
+
+        for i in range(child_count):
+            text = self._first_text_in_tree(node.get_child(i))
+            if text:
+                return text
+        return None
+
     def _find_first_list_item(self, screen):
         def find_list(node):
             try:
@@ -55,9 +74,24 @@ def find_list(node):
         if wifi_list is None:
             return None
         try:
-            if wifi_list.get_child_count() < 1:
+            child_count = wifi_list.get_child_count()
+            if child_count < 1:
                 return None
-            return wifi_list.get_child(0)
+
+            for idx in range(child_count):
+                child = wifi_list.get_child(idx)
+                text = self._first_text_in_tree(child)
+                if not text:
+                    continue
+                if text == "Add network":
+                    continue
+                if "Scanning" in text:
+                    continue
+                if "ERROR TEXT" in text:
+                    continue
+                return child
+
+            return None
         except Exception:
             return None
 
@@ -118,14 +152,16 @@ def test_hotspot_start_button_enables_hotspot(self):
         print("\nWiFi screen labels (before scan wait):")
         print_screen_labels(screen)
 
-        print("\nWaiting 10 seconds for WiFi scan to finish...")
-        time.sleep(10)
+        print("\nWaiting for first discovered access point...")
+        first_item = wait_for_widget(
+            lambda: self._find_first_list_item(lv.screen_active()),
+            timeout=25,
+        )
 
         screen = lv.screen_active()
         print("\nWiFi screen labels (after scan wait):")
         print_screen_labels(screen)
 
-        first_item = self._find_first_list_item(screen)
         self.assertIsNotNone(first_item, "Could not find first WiFi access point")
 
         coords = get_widget_coords(first_item)
