test(hmac_soft_jtag): check jtag connection status properly

This commit is contained in:
Erhan Kurubas 2025-03-04 13:30:51 +01:00
parent 1153981bc8
commit adf4822a31

View File

@ -11,16 +11,18 @@ from pytest_embedded_idf import IdfDut
from pytest_embedded_idf.utils import idf_parametrize from pytest_embedded_idf.utils import idf_parametrize
def run_gdb_test(dut: IdfDut) -> None: def run_gdb_test(dut: IdfDut) -> str:
with open(os.path.join(dut.logdir, 'ocd.txt'), 'w', encoding='utf-8') as ocd_log, pexpect.spawn( with open(os.path.join(dut.logdir, 'ocd.txt'), 'w', encoding='utf-8') as ocd_log, pexpect.spawn(
f'openocd -f board/esp32c6-builtin.cfg', timeout=60, logfile=ocd_log, encoding='utf-8', codec_errors='ignore' f'openocd -f board/esp32c6-builtin.cfg', timeout=60, logfile=ocd_log, encoding='utf-8', codec_errors='ignore'
) as p: ) as p:
try: try:
p.expect(re.compile(r'JTAG tap: esp32c6.cpu tap/device found'), timeout=5) p.expect(re.compile(r'JTAG tap: esp32c6.tap0 tap/device found'), timeout=5)
logging.info('JTAG is enabled.') logging.info('JTAG is enabled.')
return 'enabled'
except pexpect.TIMEOUT: except pexpect.ExceptionPexpect:
logging.info('JTAG is disabled') logging.info('JTAG is disabled')
return 'disabled'
finally: finally:
p.terminate() p.terminate()
@ -33,16 +35,16 @@ def test_jtag_re_enable(dut: IdfDut) -> None:
dut.expect_exact('esp32c6>', timeout=30) dut.expect_exact('esp32c6>', timeout=30)
logging.info('Initially:') logging.info('Initially:')
run_gdb_test(dut) assert run_gdb_test(dut) == 'disabled'
logging.info('After calling enable_jtag:') logging.info('After calling enable_jtag:')
# The following token data is generated using the HMAC key: # The following token data is generated using the HMAC key:
# {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32} # {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32}
dut.write('enable_jtag b2a49b1cce1be922bb7e431277413e3e8e6c3e8e6e17625c50ac66a9a857949b') dut.write('enable_jtag b2a49b1cce1be922bb7e431277413e3e8e6c3e8e6e17625c50ac66a9a857949b')
dut.expect('JTAG re-enablement workflow performed', timeout=30) dut.expect('JTAG re-enablement workflow performed', timeout=30)
run_gdb_test(dut) assert run_gdb_test(dut) == 'enabled'
logging.info('After calling disable_jtag:') logging.info('After calling disable_jtag:')
dut.write('disable_jtag') dut.write('disable_jtag')
dut.expect('JTAG disabled temporarily', timeout=30) dut.expect('JTAG disabled temporarily', timeout=30)
run_gdb_test(dut) assert run_gdb_test(dut) == 'disabled'