diff --git a/test/test_companion_offline_delivery.py b/test/test_companion_offline_delivery.py
index 5503b91f8..f102731da 100644
--- a/test/test_companion_offline_delivery.py
+++ b/test/test_companion_offline_delivery.py
@@ -49,7 +49,8 @@ struct MyMesh {
uint8_t out_frame[MAX_FRAME_SIZE]{};
int offline_queue_len=0, offline_queue_head=0;
#if defined(OTA_SHARED_COMPANION_QUEUE)
- mesh::BorrowableFrameBuffer offline_queue;
+@OTA_RETAINED@
+ mesh::BorrowableFrameBuffer offline_queue;
#elif ONE_KEY_DM_SHARED_OFFLINE_QUEUE
Frame offline_queue[OFFLINE_QUEUE_SIZE]{};
unsigned held_dm_count=15;
@@ -337,6 +338,10 @@ printf '%s\n' "$PLATFORMIO_BUILD_FLAGS" "$PLATFORMIO_BUILD_UNFLAGS"
source, re.MULTILINE))
native = (ROOT / "test/test_serial_mode_switch/test_serial_mode_switch.cpp").read_text()
harness = HARNESS.replace("@METHODS@", methods).replace("@CONSTANTS@", constants)
+ retained = re.search(
+ r"static constexpr unsigned OTA_RETAINED_OFFLINE_FRAMES = \d+;", header)
+ self.assertIsNotNone(retained)
+ harness = harness.replace("@OTA_RETAINED@", retained[0])
harness = harness.replace("@STREAM@", extract_braced(native, "class BufferStream") + ";")
harness = harness.replace("@FRAME@", extract_braced(header, "struct Frame {") + ";")
harness = harness.replace("@SYNC@", extract_braced(source, "if (cmd_frame[0] == CMD_SYNC_NEXT_MESSAGE)"))
diff --git a/test/test_firmware_ram.py b/test/test_firmware_ram.py
index 837e40a86..899f8efd3 100644
--- a/test/test_firmware_ram.py
+++ b/test/test_firmware_ram.py
@@ -1259,7 +1259,10 @@ int main() {
def test_affected_esp32_full_overlay_keeps_documented_capacity_and_transports(self):
for target in ("Heltec_v3_companion_radio_full", "Xiao_C3_companion_radio_full",
+ "Heltec_ct62_companion_radio_full",
"heltec_tracker_v2_companion_radio_full_femon",
+ "heltec_tracker_v2_companion_radio_full_femoff",
+ "generic_espnow_companion_radio_full",
"Heltec_Wireless_Paper_companion_radio_full"):
result = subprocess.run(["bash", "-c", '''
source build.sh
@@ -1280,7 +1283,16 @@ printf '%s\\n' "$PLATFORMIO_BUILD_FLAGS"
self.assertNotIn("-DOFFLINE_QUEUE_SIZE=", result.stdout)
else:
self.assertIn("-DOFFLINE_QUEUE_SIZE=224", result.stdout)
- self.assertNotIn("-DOTA_SHARED_COMPANION_QUEUE", result.stdout)
+ if target in ("Xiao_C3_companion_radio_full", "Heltec_ct62_companion_radio_full",
+ "heltec_tracker_v2_companion_radio_full_femon",
+ "heltec_tracker_v2_companion_radio_full_femoff"):
+ self.assertIn("-DOTA_SHARED_COMPANION_QUEUE=1", result.stdout)
+ self.assertIn("-DONE_KEY_DM_COMPACT_HELD=1", result.stdout)
+ else:
+ self.assertNotIn("-DOTA_SHARED_COMPANION_QUEUE", result.stdout)
+ if target in ("Heltec_v3_companion_radio_full", "generic_espnow_companion_radio_full",
+ "Heltec_Wireless_Paper_companion_radio_full"):
+ self.assertNotIn("-DONE_KEY_DM_COMPACT_HELD", result.stdout)
self.assertIn("-DWIFI_OTA_SEEDER=1", result.stdout)
self.assertIn("-DBLE_PIN_CODE=123456", result.stdout)