wehub-resource-sync
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
# SPDX-FileCopyrightText: 2022-2026 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Unlicense OR CC0-1.0
|
|
import logging
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
|
|
import pytest
|
|
from pytest_embedded import Dut
|
|
from pytest_embedded_idf.utils import idf_parametrize
|
|
|
|
|
|
@idf_parametrize('config', ['default'], indirect=['config'])
|
|
@idf_parametrize(
|
|
'target,markers',
|
|
[
|
|
(
|
|
'supported_targets',
|
|
(pytest.mark.generic, pytest.mark.temp_skip(targets=['esp32c2'], reason='must have 4MB')),
|
|
),
|
|
('esp32c2', (pytest.mark.generic, pytest.mark.flash_4mb)),
|
|
],
|
|
indirect=['target'],
|
|
)
|
|
def test_otatool_example(dut: Dut) -> None:
|
|
# Verify factory firmware
|
|
dut.expect('OTA Tool Example')
|
|
dut.expect('Example end')
|
|
|
|
# Close connection to DUT
|
|
dut.serial.close()
|
|
|
|
# Allow the OS to fully release the serial port. pytest-embedded's
|
|
# QueueFeederThread may still hold the port FD when close() returns.
|
|
time.sleep(2)
|
|
|
|
script_path = os.path.join(str(os.getenv('IDF_PATH')), 'examples', 'system', 'ota', 'otatool', 'otatool_example.py')
|
|
binary_path = ''
|
|
|
|
for flash_file in dut.app.flash_files:
|
|
if 'otatool.bin' in flash_file[1]:
|
|
binary_path = flash_file[1]
|
|
break
|
|
|
|
# Retry the subprocess to handle transient serial port contention.
|
|
# The otatool_example.py subprocess opens the serial port independently
|
|
# via esptool, and may fail if pytest-embedded's QueueFeederThread has
|
|
# not fully released the port file descriptor yet.
|
|
last_err = None
|
|
for attempt in range(3):
|
|
try:
|
|
subprocess.check_call([sys.executable, script_path, '--binary', binary_path])
|
|
return
|
|
except subprocess.CalledProcessError as e:
|
|
last_err = e
|
|
logging.warning('otatool subprocess attempt %d/3 failed: %s', attempt + 1, e)
|
|
time.sleep(3)
|
|
|
|
assert last_err is not None
|
|
raise last_err
|