feat: 实现 Android 设备搜索 (#21)
This commit is contained in:
chengma
@@ -0,0 +1,109 @@
|
||||
"""ADB Android 设备搜索服务测试。"""
|
||||
|
||||
import subprocess
|
||||
import unittest
|
||||
|
||||
from src.android_device_service import (
|
||||
AndroidDeviceSearchError,
|
||||
AndroidDeviceService,
|
||||
)
|
||||
|
||||
|
||||
def completed(command, stdout="", stderr="", returncode=0):
|
||||
"""创建一条假的 ADB 命令结果。"""
|
||||
|
||||
return subprocess.CompletedProcess(command, returncode, stdout, stderr)
|
||||
|
||||
|
||||
class AndroidDeviceServiceTest(unittest.TestCase):
|
||||
def test_parse_usb_wifi_and_unavailable_devices(self):
|
||||
output = """List of devices attached
|
||||
USB-001 device product:p model:Pixel_8 device:husky transport_id:1
|
||||
192.168.0.173:5555 offline transport_id:2
|
||||
USB-002 unauthorized usb:1-2 transport_id:3
|
||||
"""
|
||||
|
||||
devices = AndroidDeviceService.parse_devices(output)
|
||||
|
||||
self.assertEqual(len(devices), 3)
|
||||
self.assertEqual(devices[0].connection_type, "USB")
|
||||
self.assertEqual(devices[0].model, "Pixel 8")
|
||||
self.assertEqual(devices[1].connection_type, "Wi-Fi")
|
||||
self.assertEqual(devices[1].status, "offline")
|
||||
self.assertEqual(devices[2].status, "unauthorized")
|
||||
|
||||
def test_search_reads_version_only_for_connected_device(self):
|
||||
commands = []
|
||||
|
||||
def runner(command, _timeout):
|
||||
commands.append(list(command))
|
||||
if command == ["adb", "devices", "-l"]:
|
||||
return completed(
|
||||
command,
|
||||
"List of devices attached\n"
|
||||
"USB-001 device model:Pixel_8 transport_id:1\n"
|
||||
"192.168.0.173:5555 offline transport_id:2\n",
|
||||
)
|
||||
return completed(command, "14\n")
|
||||
|
||||
devices = AndroidDeviceService(runner).search()
|
||||
|
||||
self.assertEqual(devices[0].android_version, "14")
|
||||
self.assertEqual(devices[1].android_version, "—")
|
||||
self.assertEqual(len(commands), 2)
|
||||
self.assertIn("ro.build.version.release", commands[1])
|
||||
|
||||
def test_missing_model_is_read_from_device(self):
|
||||
def runner(command, _timeout):
|
||||
if command == ["adb", "devices", "-l"]:
|
||||
return completed(
|
||||
command,
|
||||
"List of devices attached\nUSB-001 device transport_id:1\n",
|
||||
)
|
||||
if command[-1] == "ro.product.model":
|
||||
return completed(command, "Redmi K70\n")
|
||||
return completed(command, "15\n")
|
||||
|
||||
device = AndroidDeviceService(runner).search()[0]
|
||||
|
||||
self.assertEqual(device.model, "Redmi K70")
|
||||
self.assertEqual(device.android_version, "15")
|
||||
|
||||
def test_property_failure_keeps_discovered_device(self):
|
||||
def runner(command, _timeout):
|
||||
if command == ["adb", "devices", "-l"]:
|
||||
return completed(
|
||||
command,
|
||||
"List of devices attached\nUSB-001 device model:Phone\n",
|
||||
)
|
||||
return completed(command, stderr="device disconnected", returncode=1)
|
||||
|
||||
device = AndroidDeviceService(runner).search()[0]
|
||||
|
||||
self.assertEqual(device.serial, "USB-001")
|
||||
self.assertEqual(device.android_version, "—")
|
||||
|
||||
def test_missing_adb_has_recovery_message(self):
|
||||
def runner(_command, _timeout):
|
||||
raise FileNotFoundError("adb")
|
||||
|
||||
with self.assertRaisesRegex(AndroidDeviceSearchError, "未找到 adb"):
|
||||
AndroidDeviceService(runner).search()
|
||||
|
||||
def test_timeout_has_recovery_message(self):
|
||||
def runner(command, timeout):
|
||||
raise subprocess.TimeoutExpired(command, timeout)
|
||||
|
||||
with self.assertRaisesRegex(AndroidDeviceSearchError, "搜索设备超时"):
|
||||
AndroidDeviceService(runner).search()
|
||||
|
||||
def test_nonzero_exit_includes_adb_error(self):
|
||||
def runner(command, _timeout):
|
||||
return completed(command, stderr="ADB server didn't ACK", returncode=1)
|
||||
|
||||
with self.assertRaisesRegex(AndroidDeviceSearchError, "didn't ACK"):
|
||||
AndroidDeviceService(runner).search()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
@@ -8,13 +8,17 @@ from pathlib import Path
|
||||
|
||||
os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
|
||||
|
||||
from PyQt5.QtCore import QTimer
|
||||
from PyQt5.QtCore import Qt, QTimer
|
||||
from PyQt5.QtTest import QTest
|
||||
from PyQt5.QtWidgets import QApplication
|
||||
|
||||
from src.android_device_service import (
|
||||
AndroidDevice,
|
||||
AndroidDeviceSearchError,
|
||||
)
|
||||
from src.mock_admin_gateway import MockAdminGateway
|
||||
from src.settings_repository import SettingsRepository
|
||||
from src.settings_ui import SettingsPage
|
||||
from src.settings_ui import AndroidDeviceRow, SettingsPage
|
||||
|
||||
|
||||
class SlowMockAdminGateway(MockAdminGateway):
|
||||
@@ -25,6 +29,34 @@ class SlowMockAdminGateway(MockAdminGateway):
|
||||
return super().register_client(client, capabilities)
|
||||
|
||||
|
||||
class SequenceAndroidDeviceService:
|
||||
"""按顺序返回搜索结果，便于测试刷新行为。"""
|
||||
|
||||
def __init__(self, results):
|
||||
self._results = list(results)
|
||||
self.call_count = 0
|
||||
|
||||
def search(self, _is_cancelled=None):
|
||||
result = self._results[min(self.call_count, len(self._results) - 1)]
|
||||
self.call_count += 1
|
||||
if isinstance(result, Exception):
|
||||
raise result
|
||||
return result
|
||||
|
||||
|
||||
class SlowAndroidDeviceService:
|
||||
"""模拟较慢的 ADB 搜索。"""
|
||||
|
||||
def __init__(self, delay=0.08):
|
||||
self.delay = delay
|
||||
self.call_count = 0
|
||||
|
||||
def search(self, _is_cancelled=None):
|
||||
self.call_count += 1
|
||||
time.sleep(self.delay)
|
||||
return [AndroidDevice("USB-001", "USB", "Pixel 8", "14")]
|
||||
|
||||
|
||||
class SettingsPageEventTest(unittest.TestCase):
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
@@ -144,6 +176,151 @@ class SettingsPageEventTest(unittest.TestCase):
|
||||
self.assertEqual(page.currentDeviceStatusLabel.text(), status_before_close)
|
||||
page.deleteLater()
|
||||
|
||||
def test_search_updates_table_and_device_status(self):
|
||||
service = SequenceAndroidDeviceService(
|
||||
[[
|
||||
AndroidDevice("USB-001", "USB", "Pixel 8", "14"),
|
||||
AndroidDevice(
|
||||
"192.168.0.173:5555",
|
||||
"Wi-Fi",
|
||||
status="offline",
|
||||
),
|
||||
]]
|
||||
)
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
|
||||
page.searchButton.click()
|
||||
self.assertFalse(page.searchButton.isEnabled())
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
|
||||
model = page.deviceTableModel
|
||||
self.assertEqual(model.rowCount(), 2)
|
||||
self.assertEqual(model.device_for_serial("USB-001").model, "Pixel 8")
|
||||
self.assertEqual(
|
||||
model.device_for_serial("192.168.0.173:5555").status,
|
||||
"offline",
|
||||
)
|
||||
offline_index = model.index(
|
||||
model.row_for_serial("192.168.0.173:5555"), 0
|
||||
)
|
||||
self.assertFalse(
|
||||
model.setData(offline_index, Qt.Checked, Qt.CheckStateRole)
|
||||
)
|
||||
self.assertEqual(
|
||||
page.deviceStatusLabel.text(),
|
||||
"搜索完成：找到 2 台 Android 设备",
|
||||
)
|
||||
self.assertTrue(page.searchButton.isEnabled())
|
||||
page.eventBinder.shutdown()
|
||||
page.deleteLater()
|
||||
|
||||
def test_empty_search_keeps_placeholder_rows(self):
|
||||
service = SequenceAndroidDeviceService([[]])
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
|
||||
self.assertGreaterEqual(page.deviceTableModel.rowCount(), 1)
|
||||
self.assertIn("未找到 Android 设备", page.deviceStatusLabel.text())
|
||||
page.eventBinder.shutdown()
|
||||
page.deleteLater()
|
||||
|
||||
def test_search_failure_preserves_previous_rows(self):
|
||||
service = SequenceAndroidDeviceService(
|
||||
[AndroidDeviceSearchError("未找到 adb，请检查 PATH 后重试")]
|
||||
)
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
page.set_android_devices(
|
||||
[AndroidDeviceRow("USB-OLD", "USB", "Old Phone", "12")]
|
||||
)
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
|
||||
self.assertIsNotNone(
|
||||
page.deviceTableModel.device_for_serial("USB-OLD")
|
||||
)
|
||||
self.assertIn("搜索失败：未找到 adb", page.deviceStatusLabel.text())
|
||||
self.assertIn("设备列表未更改", page.deviceStatusLabel.text())
|
||||
page.eventBinder.shutdown()
|
||||
page.deleteLater()
|
||||
|
||||
def test_search_preserves_or_clears_checked_device_by_serial(self):
|
||||
connected = AndroidDevice("USB-001", "USB", "Pixel 8", "14")
|
||||
service = SequenceAndroidDeviceService([[connected], [connected], []])
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
page.deviceTableModel.set_checked_serial("USB-001")
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
self.assertEqual(page.deviceTableModel.checked_serial, "USB-001")
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
self.assertEqual(page.deviceTableModel.checked_serial, "")
|
||||
page.eventBinder.shutdown()
|
||||
page.deleteLater()
|
||||
|
||||
def test_slow_search_does_not_block_or_start_twice(self):
|
||||
service = SlowAndroidDeviceService()
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
timer_fired = []
|
||||
QTimer.singleShot(10, lambda: timer_fired.append(True))
|
||||
|
||||
page.searchButton.click()
|
||||
page.eventBinder._request_search()
|
||||
self._wait_until(lambda: bool(timer_fired), timeout_ms=500)
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is None)
|
||||
|
||||
self.assertTrue(timer_fired)
|
||||
self.assertEqual(service.call_count, 1)
|
||||
page.eventBinder.shutdown()
|
||||
page.deleteLater()
|
||||
|
||||
def test_shutdown_ignores_late_search_result(self):
|
||||
service = SlowAndroidDeviceService()
|
||||
page = SettingsPage(
|
||||
settings_repository=self.repository,
|
||||
admin_gateway=MockAdminGateway(),
|
||||
android_device_service=service,
|
||||
)
|
||||
|
||||
page.searchButton.click()
|
||||
self._wait_until(lambda: page.eventBinder._search_thread is not None)
|
||||
status_before_close = page.deviceStatusLabel.text()
|
||||
page.eventBinder.shutdown()
|
||||
QTest.qWait(120)
|
||||
|
||||
self.assertEqual(page.deviceStatusLabel.text(), status_before_close)
|
||||
self.assertIsNone(
|
||||
page.deviceTableModel.device_for_serial("USB-001")
|
||||
)
|
||||
page.deleteLater()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user