mirror of
https://github.com/MobileGL-Dev/MobileGL
synced 2026-11-07 01:53:24 +09:00
45 lines
2.0 KiB
Python
45 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Exercise the actual APK retry classifier with saved and live adb states."""
|
|
import pathlib
|
|
import subprocess
|
|
import tempfile
|
|
import unittest
|
|
|
|
SCRIPT = pathlib.Path(__file__).resolve().parents[2] / "android-plugin/trace-replay-ci.sh"
|
|
|
|
|
|
class TraceInfrastructureTest(unittest.TestCase):
|
|
def classify(self, saved="device", live="device", disconnected=False, logcat=""):
|
|
source = SCRIPT.read_text()
|
|
functions = source[source.index("record_infrastructure_reason() {"):source.index("copy_app_artifact() {")]
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root = pathlib.Path(directory)
|
|
(root / "adb-state.txt").write_text(saved)
|
|
(root / "logcat.txt").write_text(logcat)
|
|
if disconnected:
|
|
(root / "adb-disconnected.txt").write_text("offline")
|
|
shell = functions + '\nadb_device_path() { printf "%s\\n" "$LIVE"; }\n' + \
|
|
'result_root="$1"; LIVE="$2"; is_infrastructure_failure "$1"\n'
|
|
result = subprocess.run(["sh", "-c", shell, "classifier", directory, live], capture_output=True)
|
|
reason = root / "infrastructure-failure-reason.txt"
|
|
return result.returncode, reason.read_text().strip() if reason.exists() else ""
|
|
|
|
def test_device_disappears_during_diagnostics(self):
|
|
self.assertEqual(self.classify(live="offline"), (0, "device-unavailable"))
|
|
|
|
def test_polling_disconnect_survives_reconnect(self):
|
|
self.assertEqual(self.classify(disconnected=True), (0, "device-unavailable"))
|
|
|
|
def test_saved_disconnect_survives_reconnect(self):
|
|
self.assertEqual(self.classify(saved=""), (0, "device-unavailable"))
|
|
|
|
def test_online_application_failure_stays_failure(self):
|
|
self.assertEqual(self.classify(logcat="Fatal signal 11 (trace_replay)\n"), (1, ""))
|
|
|
|
def test_system_server_crash_remains_retryable(self):
|
|
self.assertEqual(self.classify(logcat="Fatal signal 11 (system_server)\n"), (0, "system-server-crash"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|