Scope Indicator WiFi assertion to its connection helper

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)