246 lines
9.7 KiB
Python
246 lines
9.7 KiB
Python
"""Быстрые проверки Lab046: фрагментация большого объекта и fail-closed сборка.
|
|
|
|
Тесты не трогают оборудование, не создают файлов и не запускают RF. Полный
|
|
тракт PHY (кадр Lab045 -> канал -> приём -> сборка) проверяется на маленьких
|
|
объектах, чтобы оставаться быстрым.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import struct
|
|
|
|
import numpy as np
|
|
|
|
from experiments import lab046_bpsk_object_transport as lab046
|
|
from protocol import image_fragments as fragments
|
|
|
|
|
|
# ------------------------------------------------------------ Детерминированный объект
|
|
|
|
|
|
def test_deterministic_object_is_reproducible_and_hashable() -> None:
|
|
first = lab046.make_deterministic_object(4096, seed=46)
|
|
second = lab046.make_deterministic_object(4096, seed=46)
|
|
assert first == second
|
|
assert len(first) == 4096
|
|
other = lab046.make_deterministic_object(4096, seed=47)
|
|
assert other != first
|
|
assert len(lab046.object_sha256(first)) == 64
|
|
|
|
|
|
# ------------------------------------------------------------ Фрагмент туда-обратно
|
|
|
|
|
|
def test_fragment_round_trip_exact_without_channel() -> None:
|
|
obj = lab046.make_deterministic_object(3000, seed=1)
|
|
pieces = fragments.split_image_bytes(obj, lab046.DEFAULT_OBJECT_ID, lab046.FRAGMENT_DATA_SIZE)
|
|
assert fragments.reassemble_image(pieces) == obj
|
|
|
|
|
|
def test_last_short_fragment_recovered_by_header_length() -> None:
|
|
# 3000 не кратно 244, значит последний фрагмент короче.
|
|
obj = lab046.make_deterministic_object(3000, seed=2)
|
|
pieces = fragments.split_image_bytes(obj, lab046.DEFAULT_OBJECT_ID, lab046.FRAGMENT_DATA_SIZE)
|
|
last = pieces[-1]
|
|
assert len(last.data) < lab046.FRAGMENT_DATA_SIZE
|
|
padded = lab046._pad_payload_to_packet(fragments.encode_image_fragment(last))
|
|
assert len(padded) == lab046.PACKET_PAYLOAD_SIZE
|
|
recovered = lab046.recover_fragment_from_payload(padded)
|
|
assert recovered is not None
|
|
assert recovered.data == last.data
|
|
|
|
|
|
def _make_payload(object_id: int, index: int, count: int, original_size: int, data: bytes) -> bytes:
|
|
header = struct.pack(
|
|
fragments.FRAGMENT_HEADER_FORMAT,
|
|
object_id,
|
|
index,
|
|
count,
|
|
original_size,
|
|
)
|
|
return lab046._pad_payload_to_packet(header + data)
|
|
|
|
|
|
# ------------------------------------------------------------ recover: fail-closed
|
|
|
|
|
|
def test_recover_rejects_truncated_metadata() -> None:
|
|
# Короче заголовка -> декод не проходит, объект не собирается.
|
|
assert lab046.recover_fragment_from_payload(b"\x00\x03") is None
|
|
|
|
|
|
def test_recover_rejects_impossible_index() -> None:
|
|
payload = _make_payload(lab046.DEFAULT_OBJECT_ID, index=5, count=3, original_size=600, data=b"\x01" * 200)
|
|
assert lab046.recover_fragment_from_payload(payload) is None
|
|
|
|
|
|
def test_recover_rejects_inconsistent_count_and_size() -> None:
|
|
# count=1, но original_size больше одного фрагмента -> несогласовано.
|
|
payload = _make_payload(lab046.DEFAULT_OBJECT_ID, index=0, count=1, original_size=10_000, data=b"\x01" * 244)
|
|
assert lab046.recover_fragment_from_payload(payload) is None
|
|
|
|
|
|
def test_recover_rejects_absurd_original_size_without_huge_alloc() -> None:
|
|
payload = _make_payload(lab046.DEFAULT_OBJECT_ID, index=0, count=2, original_size=2**31, data=b"\x01" * 244)
|
|
assert lab046.recover_fragment_from_payload(payload) is None
|
|
|
|
|
|
def test_expected_length_returns_none_on_bad_fields() -> None:
|
|
assert lab046.expected_fragment_data_length(0, 0, 100) is None
|
|
assert lab046.expected_fragment_data_length(-1, 3, 100) is None
|
|
assert lab046.expected_fragment_data_length(0, 1, 0) is None
|
|
|
|
|
|
# ------------------------------------------------------------ Сборщик: fail-closed
|
|
|
|
|
|
def test_assembler_rejects_foreign_object_id() -> None:
|
|
obj = lab046.make_deterministic_object(600, seed=3)
|
|
pieces = fragments.split_image_bytes(obj, lab046.DEFAULT_OBJECT_ID, lab046.FRAGMENT_DATA_SIZE)
|
|
assembler = lab046.ObjectAssembler()
|
|
assert assembler.offer(pieces[0]) == "accepted"
|
|
foreign = fragments.ImageFragment(
|
|
image_id=0xDEADBEEF,
|
|
fragment_index=pieces[1].fragment_index,
|
|
fragment_count=pieces[1].fragment_count,
|
|
original_size=pieces[1].original_size,
|
|
data=pieces[1].data,
|
|
)
|
|
assert assembler.offer(foreign) == "rejected"
|
|
assert not assembler.object_complete()
|
|
|
|
|
|
def test_assembler_identical_duplicate_ignored_conflicting_rejected() -> None:
|
|
obj = lab046.make_deterministic_object(600, seed=4)
|
|
pieces = fragments.split_image_bytes(obj, lab046.DEFAULT_OBJECT_ID, lab046.FRAGMENT_DATA_SIZE)
|
|
assembler = lab046.ObjectAssembler()
|
|
assert assembler.offer(pieces[0]) == "accepted"
|
|
assert assembler.offer(pieces[0]) == "duplicate"
|
|
conflicting = fragments.ImageFragment(
|
|
image_id=pieces[0].image_id,
|
|
fragment_index=pieces[0].fragment_index,
|
|
fragment_count=pieces[0].fragment_count,
|
|
original_size=pieces[0].original_size,
|
|
data=bytes(len(pieces[0].data)),
|
|
)
|
|
assert assembler.offer(conflicting) == "conflicting"
|
|
|
|
|
|
def test_assembler_none_is_rejected_not_raised() -> None:
|
|
assembler = lab046.ObjectAssembler()
|
|
assert assembler.offer(None) == "rejected"
|
|
assert assembler.recovered_object() is None
|
|
|
|
|
|
def test_assembler_incomplete_returns_no_object() -> None:
|
|
obj = lab046.make_deterministic_object(600, seed=5)
|
|
pieces = fragments.split_image_bytes(obj, lab046.DEFAULT_OBJECT_ID, lab046.FRAGMENT_DATA_SIZE)
|
|
assembler = lab046.ObjectAssembler()
|
|
assembler.offer(pieces[0]) # намеренно не все
|
|
assert not assembler.object_complete()
|
|
assert assembler.recovered_object() is None
|
|
|
|
|
|
# ------------------------------------------------------------ Полный тракт PHY: сценарии
|
|
|
|
|
|
def test_scenario_a_ideal_channel_exact_object() -> None:
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
result = lab046.run_object_transport(obj, channel=lab046.ideal_channel)
|
|
assert result.object_complete
|
|
assert result.sha_match
|
|
assert result.object_byte_errors == 0
|
|
assert result.crc_fail_count == 0
|
|
assert result.recovered_object_sha256 == result.object_sha256
|
|
|
|
|
|
def test_scenario_b_distributed_bit_errors_corrected() -> None:
|
|
# Честный гауссов шум: до FEC ошибки есть, после FEC ноль, объект точен.
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
channel = lab046.make_noisy_channel(noise_sigma=0.40, base_seed=46000)
|
|
result = lab046.run_object_transport(obj, channel=channel)
|
|
assert result.pre_fec_coded_bit_errors > 0
|
|
assert result.post_fec_protocol_bit_errors == 0
|
|
assert result.crc_fail_count == 0
|
|
assert result.object_complete
|
|
assert result.sha_match
|
|
assert result.object_byte_errors == 0
|
|
|
|
|
|
def test_scenario_c_out_of_order_delivery_still_exact() -> None:
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
frames = lab046.build_object_frames(obj)
|
|
order = list(reversed(range(len(frames))))
|
|
result = lab046.run_object_transport(obj, channel=lab046.ideal_channel, delivery_order=order)
|
|
assert result.object_complete
|
|
assert result.sha_match
|
|
|
|
|
|
def test_scenario_d_duplicate_fragments_still_exact() -> None:
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
result = lab046.run_object_transport(
|
|
obj,
|
|
channel=lab046.ideal_channel,
|
|
duplicate_indexes=(0, 2, 2),
|
|
)
|
|
assert result.duplicates >= 1
|
|
assert result.object_complete
|
|
assert result.sha_match
|
|
|
|
|
|
def test_scenario_e_missing_fragment_means_not_complete() -> None:
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
result = lab046.run_object_transport(
|
|
obj,
|
|
channel=lab046.ideal_channel,
|
|
drop_indexes=(1,),
|
|
)
|
|
assert not result.object_complete
|
|
assert 1 in result.missing_indexes
|
|
assert result.recovered_object_sha256 is None
|
|
assert not result.sha_match
|
|
|
|
|
|
def test_scenario_f_corrupted_packet_rejected_no_fake_object() -> None:
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
result = lab046.run_object_transport(
|
|
obj,
|
|
channel=lab046.ideal_channel,
|
|
corrupt_indexes=(0,),
|
|
)
|
|
assert result.crc_fail_count >= 1
|
|
assert not result.object_complete
|
|
assert result.recovered_object_sha256 is None
|
|
|
|
|
|
# ------------------------------------------------------------ Метрики кадра
|
|
|
|
|
|
def test_object_metrics_match_lab045_frame() -> None:
|
|
obj = lab046.make_deterministic_object(16384, seed=46)
|
|
frames = lab046.build_object_frames(obj)
|
|
assert len(frames) == 68 # ceil(16384 / 244)
|
|
assert all(f.plan.frame_symbol_count == 5724 for f in frames)
|
|
assert all(len(f.plan.packet) == 268 for f in frames)
|
|
|
|
|
|
# ------------------------------------------------------------ Нет оракула
|
|
|
|
|
|
def test_receiver_ignores_original_object_on_corruption() -> None:
|
|
# Если бы приёмник подглядывал в оригинал, повреждённый кадр «восстановился»
|
|
# бы. Проверяем, что этого не происходит: объект остаётся неполным.
|
|
obj = lab046.make_deterministic_object(2048, seed=46)
|
|
result = lab046.run_object_transport(
|
|
obj,
|
|
channel=lab046.ideal_channel,
|
|
corrupt_indexes=tuple(range(len(lab046.build_object_frames(obj)))),
|
|
)
|
|
assert result.crc_pass_count == 0
|
|
assert not result.object_complete
|
|
assert result.recovered_object_sha256 is None
|
|
|
|
|
|
def test_functional_self_check_runs() -> None:
|
|
lab046.run_functional_tests()
|