mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-07-15 22:32:11 +08:00
firestar5683
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import pytest
|
|
|
|
from tools.agnos.patch_system_reset_image import (
|
|
COMMA_SH_DISPLAY_WAIT_PATCH_MARKER,
|
|
comma_sh_has_expected_display_wait,
|
|
format_debugfs_mode,
|
|
patch_comma_sh_display_wait,
|
|
)
|
|
|
|
|
|
ORIGINAL_DISPLAY_WAIT = b'''#!/usr/bin/env bash
|
|
echo "waiting for magic"
|
|
for i in {1..200}; do
|
|
if systemctl is-active --quiet magic && [ -S /tmp/drmfd.sock ]; then
|
|
break
|
|
fi
|
|
sleep 0.1
|
|
done
|
|
|
|
if systemctl is-active --quiet magic && [ -S /tmp/drmfd.sock ]; then
|
|
echo "magic ready after ${SECONDS}s"
|
|
else
|
|
echo "timed out waiting for magic, ${SECONDS}s"
|
|
fi
|
|
|
|
exec /data/continue.sh
|
|
'''
|
|
|
|
|
|
def test_patch_comma_sh_display_wait_uses_available_display_service():
|
|
patched = patch_comma_sh_display_wait(ORIGINAL_DISPLAY_WAIT)
|
|
|
|
assert COMMA_SH_DISPLAY_WAIT_PATCH_MARKER.encode() in patched
|
|
assert b"systemctl cat magic.service" in patched
|
|
assert b"systemctl is-active --quiet magic" in patched
|
|
assert b"systemctl is-active --quiet weston-ready" in patched
|
|
assert b"[ -S /var/tmp/weston/wayland-0 ]" in patched
|
|
assert comma_sh_has_expected_display_wait(patched)
|
|
assert patch_comma_sh_display_wait(patched) == patched
|
|
|
|
|
|
def test_patch_comma_sh_display_wait_rejects_unknown_layout():
|
|
with pytest.raises(RuntimeError, match="display readiness wait"):
|
|
patch_comma_sh_display_wait(b"#!/usr/bin/env bash\nexec /data/continue.sh\n")
|
|
|
|
|
|
@pytest.mark.parametrize(("mode", "expected"), [
|
|
("100775", "0100775"),
|
|
("100644", "0100644"),
|
|
("040755", "040755"),
|
|
("120777", "0120777"),
|
|
])
|
|
def test_format_debugfs_mode(mode, expected):
|
|
assert format_debugfs_mode(mode) == expected
|