From 0d0a0f187ea94be0cc1f5f0613e4e8ab53a6a932 Mon Sep 17 00:00:00 2001 From: mikecarper Date: Fri, 16 Oct 2026 05:02:56 -0700 Subject: [PATCH] Scope Indicator WiFi assertion to its connection helper --- test/test_indicator_render_profile.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/test/test_indicator_render_profile.py b/test/test_indicator_render_profile.py index ea2cb303a..f3b0f2e64 100644 --- a/test/test_indicator_render_profile.py +++ b/test/test_indicator_render_profile.py @@ -7,6 +7,8 @@ import subprocess import tempfile import unittest +from cpp_source import body + ROOT = Path(__file__).resolve().parents[1] BUILD = ROOT / "build_legacy.sh" @@ -213,8 +215,7 @@ int main() { self.assertIn("noteCompanionWiFiDisplayState(state)", page) main = (ROOT / "examples/companion_radio/main.cpp").read_text() - helper = main[main.index("bool isCompanionWiFiConnected()"): - main.index("bool toggleCompanionWiFi()")] + helper = body(main, "bool isCompanionWiFiConnected()") self.assertNotIn("!companion_wifi_active", helper) self.assertIn("esp_wifi_sta_get_ap_info", helper)