You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.8 KiB
109 lines
3.8 KiB
"""Parrot playback send loop and max recording timer."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from tests.harness.deterministic import DeterministicScenario, PacketSpec
|
|
from tests.harness.playback_helpers import FakePlaybackProtocol, install_reactor_capture, run_scheduled
|
|
|
|
from adn_server.application.playback_use_cases import (
|
|
_PACKET_INTERVAL_S,
|
|
_PLAYBACK_DELAY_S,
|
|
_SOURCE_MAX_S,
|
|
PlaybackUseCases,
|
|
)
|
|
from adn_server.domain import bytes_3, bytes_4
|
|
|
|
|
|
def test_send_next_packet_emits_all_packets_then_finishes() -> None:
|
|
proto = FakePlaybackProtocol()
|
|
pb = PlaybackUseCases("ECHO", get_protocol=lambda: proto)
|
|
pb._playback_busy = True
|
|
pb._playback_stream_id = bytes_4(0x77777777)
|
|
pb._playback_packets = [bytes([i]) * 55 for i in range(1, 5)]
|
|
pb._playback_index = 0
|
|
mock_reactor, scheduled = install_reactor_capture()
|
|
|
|
with patch("adn_server.application.playback_use_cases.reactor", mock_reactor):
|
|
pb._send_next_packet(proto)
|
|
while scheduled:
|
|
run_scheduled(scheduled)
|
|
|
|
assert len(proto.sent) == 4
|
|
assert pb._playback_busy is False
|
|
assert pb._playback_packets == []
|
|
assert pb._playback_index == 0
|
|
|
|
|
|
def test_start_playback_sends_first_packet_and_schedules_rest() -> None:
|
|
proto = FakePlaybackProtocol()
|
|
pb = PlaybackUseCases("ECHO", get_protocol=lambda: proto)
|
|
base = PacketSpec(dst_id=9990, stream_id=0x88888888, slot=2)
|
|
recorded = [
|
|
DeterministicScenario.voice_head_spec(base).data(),
|
|
DeterministicScenario.voice_burst_spec(base, seq=1, dtype_vseq=1).data(),
|
|
]
|
|
mock_reactor, scheduled = install_reactor_capture()
|
|
|
|
with patch("adn_server.application.playback_use_cases.reactor", mock_reactor):
|
|
pb._start_playback(
|
|
proto,
|
|
recorded,
|
|
bytes_3(base.rf_src),
|
|
bytes_4(base.peer_id),
|
|
bytes_3(base.dst_id),
|
|
2,
|
|
1.5,
|
|
)
|
|
|
|
assert len(proto.sent) == 1
|
|
assert pb._playback_index == 1
|
|
assert any(item[0] == _PACKET_INTERVAL_S for item in scheduled)
|
|
with patch("adn_server.application.playback_use_cases.reactor", mock_reactor):
|
|
while scheduled:
|
|
run_scheduled(scheduled)
|
|
|
|
assert len(proto.sent) == 2
|
|
assert pb._playback_busy is False
|
|
|
|
|
|
def test_max_duration_commits_recording_when_no_vterm() -> None:
|
|
proto = FakePlaybackProtocol()
|
|
pb = PlaybackUseCases("ECHO", get_protocol=lambda: proto)
|
|
base = PacketSpec(dst_id=9990, stream_id=0x99999999, slot=2)
|
|
pb._recording_active = True
|
|
pb.CALL_DATA = [DeterministicScenario.voice_head_spec(base).data()]
|
|
pb._record_stream = base.data()[16:20]
|
|
pb.STATUS["RX_START"] = 100.0
|
|
pb._record_ctx = {
|
|
"slot": 2,
|
|
"rf_src": bytes_3(base.rf_src),
|
|
"peer_id": bytes_4(base.peer_id),
|
|
"dst_id": bytes_3(base.dst_id),
|
|
}
|
|
mock_reactor, scheduled = install_reactor_capture()
|
|
|
|
with patch("adn_server.application.playback_use_cases.reactor", mock_reactor):
|
|
with patch("adn_server.application.playback_use_cases.time", return_value=100.0 + _SOURCE_MAX_S):
|
|
pb._on_record_max_duration(proto)
|
|
|
|
assert not pb._recording_active
|
|
assert pb._playback_busy is True
|
|
assert any(item[0] == _PLAYBACK_DELAY_S for item in scheduled)
|
|
|
|
|
|
def test_packet_interval_matches_expected() -> None:
|
|
proto = FakePlaybackProtocol()
|
|
pb = PlaybackUseCases("ECHO", get_protocol=lambda: proto)
|
|
pb._playback_busy = True
|
|
pb._playback_stream_id = bytes_4(0xAAAAAAAA)
|
|
pb._playback_packets = [b"\x01" * 55, b"\x02" * 55]
|
|
pb._playback_index = 0
|
|
mock_reactor, scheduled = install_reactor_capture()
|
|
|
|
with patch("adn_server.application.playback_use_cases.reactor", mock_reactor):
|
|
pb._send_next_packet(proto)
|
|
|
|
assert scheduled[0][0] == _PACKET_INTERVAL_S
|