mirror of
https://github.com/espressif/esp-idf.git
synced 2026-04-27 19:13:21 +00:00
Zhou Xiao
a11bde6fd0
Add BUF_UTIL=5 to InternalSource, BufUtilPool enum, BufUtilResult TypedDict, BufUtilEntry dataclass, and name resolution helpers. Add decode branch in internal_decoder with pool/index extraction from the packed lbm_id field. Tests cover valid decode, pool/index extraction, and truncated payload handling.
108 lines
4.0 KiB
Python
108 lines
4.0 KiB
Python
# SPDX-FileCopyrightText: 2026 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
import struct
|
|
|
|
from src.backend.internal_decoder import decode_internal_frame
|
|
from src.backend.models import InternalSource
|
|
|
|
|
|
def _make_internal_payload(os_ts: int, int_src: int, sub_payload: bytes) -> bytes:
|
|
"""Build a full INTERNAL frame payload (os_ts + int_src_code + sub_payload)."""
|
|
return struct.pack('<I', os_ts) + bytes([int_src]) + sub_payload
|
|
|
|
|
|
class TestInitDone:
|
|
def test_decode_init_done(self) -> None:
|
|
payload = _make_internal_payload(os_ts=1234, int_src=0, sub_payload=b'\x03')
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['int_src'] == InternalSource.INIT_DONE
|
|
assert result['version'] == 3
|
|
assert result['os_ts_ms'] == 1234
|
|
|
|
|
|
class TestInfo:
|
|
def test_decode_info(self) -> None:
|
|
payload = _make_internal_payload(os_ts=5678, int_src=3, sub_payload=b'\x03')
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['int_src'] == InternalSource.INFO
|
|
assert result['version'] == 3
|
|
|
|
|
|
class TestEnhStat:
|
|
def test_decode_enh_stat(self) -> None:
|
|
sub = struct.pack('<BIIII', 2, 100, 5, 4096, 256) # src=2, written=100, lost=5, ...
|
|
payload = _make_internal_payload(os_ts=9999, int_src=2, sub_payload=sub)
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['int_src'] == InternalSource.ENH_STAT
|
|
assert result['src_code'] == 2
|
|
assert result['written_frame_cnt'] == 100
|
|
assert result['lost_frame_cnt'] == 5
|
|
assert result['written_bytes_cnt'] == 4096
|
|
assert result['lost_bytes_cnt'] == 256
|
|
|
|
|
|
class TestFlush:
|
|
def test_decode_flush(self) -> None:
|
|
payload = _make_internal_payload(os_ts=0, int_src=4, sub_payload=b'\x03')
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['int_src'] == InternalSource.FLUSH
|
|
assert result['version'] == 3
|
|
|
|
|
|
class TestTs:
|
|
def test_ts_ignored(self) -> None:
|
|
sub = struct.pack('<BIII', 1, 100, 200, 300) # io_level, lc_ts, esp_ts, os_ts
|
|
payload = _make_internal_payload(os_ts=0, int_src=1, sub_payload=sub)
|
|
result = decode_internal_frame(payload)
|
|
assert result is None # TS is ignored
|
|
|
|
|
|
class TestUnknown:
|
|
def test_unknown_int_src(self) -> None:
|
|
payload = _make_internal_payload(os_ts=0, int_src=99, sub_payload=b'\x00')
|
|
result = decode_internal_frame(payload)
|
|
assert result is None
|
|
|
|
|
|
class TestBufUtil:
|
|
def test_decode_buf_util(self) -> None:
|
|
# lbm_id=0x01 (pool=0, index=1), trans_cnt=4, inflight_peak=3
|
|
sub = b'\x01\x04\x03'
|
|
payload = _make_internal_payload(os_ts=7777, int_src=5, sub_payload=sub)
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['int_src'] == InternalSource.BUF_UTIL
|
|
assert result['lbm_id'] == 0x01
|
|
assert result['pool'] == 0
|
|
assert result['index'] == 1
|
|
assert result['trans_cnt'] == 4
|
|
assert result['inflight_peak'] == 3
|
|
assert result['os_ts_ms'] == 7777
|
|
|
|
def test_buf_util_pool_index_extraction(self) -> None:
|
|
# lbm_id=0x21 -> pool=2 (LL), index=1 (ll_hci)
|
|
sub = b'\x21\x04\x02'
|
|
payload = _make_internal_payload(os_ts=0, int_src=5, sub_payload=sub)
|
|
result = decode_internal_frame(payload)
|
|
assert result is not None
|
|
assert result['pool'] == 2
|
|
assert result['index'] == 1
|
|
|
|
def test_buf_util_too_short(self) -> None:
|
|
# Only 2 bytes of sub_payload (need 3 after int_src_code)
|
|
sub = b'\x00\x04'
|
|
payload = _make_internal_payload(os_ts=0, int_src=5, sub_payload=sub)
|
|
result = decode_internal_frame(payload)
|
|
assert result is None
|
|
|
|
|
|
class TestMalformed:
|
|
def test_too_short_payload(self) -> None:
|
|
result = decode_internal_frame(b'\x00\x00\x00')
|
|
assert result is None
|