mirror of
https://github.com/commaai/agnos-builder.git
synced 2026-10-28 00:53:48 +08:00
Adeeb Shihadeh
51 lines
2.1 KiB
Python
51 lines
2.1 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
spec = importlib.util.spec_from_file_location('hotplug', Path(__file__).resolve().parents[1] / 'userspace/root/usr/comma/hotplug.py')
|
|
hotplug = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(hotplug)
|
|
|
|
|
|
class TestHotplug(unittest.TestCase):
|
|
def test_modem_reenumeration(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
dev = root / 'dev'
|
|
dev.mkdir()
|
|
usb = root / 'usb-device'
|
|
interface = usb / '1-1:1.2'
|
|
tty = interface / 'ttyUSB7'
|
|
tty.mkdir(parents=True)
|
|
(usb / 'idVendor').write_text('2c7c')
|
|
(usb / 'idProduct').write_text('6007')
|
|
(interface / 'bInterfaceNumber').write_text('02')
|
|
with patch.object(hotplug, 'DEV', dev), patch.object(hotplug, 'permissions') as perms:
|
|
hotplug.handle(tty, 'ttyUSB7', 'tty')
|
|
self.assertEqual((dev / 'modem_at0').readlink(), dev / 'ttyUSB7')
|
|
perms.assert_called_with(dev / 'ttyUSB7', 'dialout', 0o660)
|
|
# Remove has no sysfs attributes left; still remove the stale alias.
|
|
hotplug.handle(root / 'gone', 'ttyUSB7', 'tty', True)
|
|
self.assertFalse((dev / 'modem_at0').is_symlink())
|
|
hotplug.handle(tty, 'ttyUSB9', 'tty')
|
|
self.assertEqual((dev / 'modem_at0').readlink(), dev / 'ttyUSB9')
|
|
# A late removal for the old device cannot erase the new alias.
|
|
hotplug.handle(root / 'gone', 'ttyUSB7', 'tty', True)
|
|
self.assertEqual((dev / 'modem_at0').readlink(), dev / 'ttyUSB9')
|
|
|
|
def test_panda_modes(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
path = Path(tmp)
|
|
for vendor, product in [('bbaa', 'ddee'), ('bbaa', 'ddcc'), ('0483', 'df11')]:
|
|
(path / 'idVendor').write_text(vendor)
|
|
(path / 'idProduct').write_text(product)
|
|
with patch.object(hotplug, 'permissions') as perms:
|
|
hotplug.handle(path, 'bus/usb/001/002', 'usb')
|
|
perms.assert_called_with(Path('/dev/bus/usb/001/002'), 'root', 0o666)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|