mirror of
https://github.com/smittix/intercept.git
synced 2026-07-14 04:28:10 -07:00
style: apply ruff-format to entire codebase
First-time run of ruff-format via pre-commit hook normalises quote style, trailing commas, and whitespace across 188 Python files. No logic changes. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
James Smith
+88
-87
@@ -7,6 +7,7 @@ import pytest
|
||||
try:
|
||||
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms
|
||||
from cryptography.hazmat.primitives.ciphers import modes as cipher_modes
|
||||
|
||||
HAS_CRYPTOGRAPHY = True
|
||||
except ImportError:
|
||||
HAS_CRYPTOGRAPHY = False
|
||||
@@ -34,10 +35,10 @@ class TestResolveRPA:
|
||||
"""
|
||||
# The ah() function: encrypt(IRK, 0x00..00 || prand) then take last 3 bytes
|
||||
|
||||
irk = b'\x00' * 16
|
||||
irk = b"\x00" * 16
|
||||
# Choose prand with upper 2 bits = 01 (resolvable)
|
||||
prand = bytes([0x40, 0x00, 0x01])
|
||||
plaintext = b'\x00' * 13 + prand
|
||||
plaintext = b"\x00" * 13 + prand
|
||||
c = Cipher(algorithms.AES(irk), cipher_modes.ECB())
|
||||
enc = c.encryptor()
|
||||
encrypted = enc.update(plaintext) + enc.finalize()
|
||||
@@ -45,41 +46,41 @@ class TestResolveRPA:
|
||||
|
||||
# Build address: prand || hash
|
||||
addr_bytes = prand + hash_bytes
|
||||
address = ':'.join(f'{b:02X}' for b in addr_bytes)
|
||||
address = ":".join(f"{b:02X}" for b in addr_bytes)
|
||||
|
||||
assert resolve_rpa(irk, address) is True
|
||||
|
||||
def test_resolve_rpa_invalid_address(self):
|
||||
"""Test RPA resolution with non-matching address."""
|
||||
irk = b'\x00' * 16
|
||||
irk = b"\x00" * 16
|
||||
# Non-resolvable address (upper 2 bits != 01)
|
||||
assert resolve_rpa(irk, 'FF:FF:FF:FF:FF:FF') is False
|
||||
assert resolve_rpa(irk, "FF:FF:FF:FF:FF:FF") is False
|
||||
|
||||
@pytest.mark.skipif(not HAS_CRYPTOGRAPHY, reason="cryptography not installed")
|
||||
def test_resolve_rpa_wrong_irk(self):
|
||||
"""Test RPA resolution with wrong IRK."""
|
||||
irk = b'\x00' * 16
|
||||
irk = b"\x00" * 16
|
||||
prand = bytes([0x40, 0x00, 0x01])
|
||||
plaintext = b'\x00' * 13 + prand
|
||||
plaintext = b"\x00" * 13 + prand
|
||||
c = Cipher(algorithms.AES(irk), cipher_modes.ECB())
|
||||
enc = c.encryptor()
|
||||
encrypted = enc.update(plaintext) + enc.finalize()
|
||||
hash_bytes = encrypted[13:16]
|
||||
addr_bytes = prand + hash_bytes
|
||||
address = ':'.join(f'{b:02X}' for b in addr_bytes)
|
||||
address = ":".join(f"{b:02X}" for b in addr_bytes)
|
||||
|
||||
# Different IRK should fail
|
||||
wrong_irk = b'\x01' * 16
|
||||
wrong_irk = b"\x01" * 16
|
||||
assert resolve_rpa(wrong_irk, address) is False
|
||||
|
||||
def test_resolve_rpa_short_address(self):
|
||||
"""Test with invalid short address."""
|
||||
irk = b'\x00' * 16
|
||||
assert resolve_rpa(irk, 'AA:BB') is False
|
||||
irk = b"\x00" * 16
|
||||
assert resolve_rpa(irk, "AA:BB") is False
|
||||
|
||||
def test_resolve_rpa_empty(self):
|
||||
"""Test with empty inputs."""
|
||||
assert resolve_rpa(b'\x00' * 16, '') is False
|
||||
assert resolve_rpa(b"\x00" * 16, "") is False
|
||||
|
||||
|
||||
class TestDistanceEstimator:
|
||||
@@ -108,82 +109,82 @@ class TestDistanceEstimator:
|
||||
assert d_indoor < d_free
|
||||
|
||||
def test_proximity_band_immediate(self):
|
||||
assert DistanceEstimator.proximity_band(0.5) == 'IMMEDIATE'
|
||||
assert DistanceEstimator.proximity_band(0.5) == "IMMEDIATE"
|
||||
|
||||
def test_proximity_band_near(self):
|
||||
assert DistanceEstimator.proximity_band(3.0) == 'NEAR'
|
||||
assert DistanceEstimator.proximity_band(3.0) == "NEAR"
|
||||
|
||||
def test_proximity_band_far(self):
|
||||
assert DistanceEstimator.proximity_band(10.0) == 'FAR'
|
||||
assert DistanceEstimator.proximity_band(10.0) == "FAR"
|
||||
|
||||
|
||||
class TestLocateTarget:
|
||||
"""Test target matching."""
|
||||
|
||||
def test_match_by_mac(self):
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = 'AA:BB:CC:DD:EE:FF'
|
||||
device.device_id = "other"
|
||||
device.address = "AA:BB:CC:DD:EE:FF"
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_match_by_mac_case_insensitive(self):
|
||||
target = LocateTarget(mac_address='aa:bb:cc:dd:ee:ff')
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = 'AA:BB:CC:DD:EE:FF'
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_match_by_mac_without_separators(self):
|
||||
target = LocateTarget(mac_address='aabbccddeeff')
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = 'AA:BB:CC:DD:EE:FF'
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
def test_match_by_mac_case_insensitive(self):
|
||||
target = LocateTarget(mac_address="aa:bb:cc:dd:ee:ff")
|
||||
device = MagicMock()
|
||||
device.device_id = "other"
|
||||
device.address = "AA:BB:CC:DD:EE:FF"
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_match_by_mac_without_separators(self):
|
||||
target = LocateTarget(mac_address="aabbccddeeff")
|
||||
device = MagicMock()
|
||||
device.device_id = "other"
|
||||
device.address = "AA:BB:CC:DD:EE:FF"
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_match_by_name_pattern(self):
|
||||
target = LocateTarget(name_pattern='iPhone')
|
||||
target = LocateTarget(name_pattern="iPhone")
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = '00:00:00:00:00:00'
|
||||
device.device_id = "other"
|
||||
device.address = "00:00:00:00:00:00"
|
||||
device.name = "John's iPhone 15"
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_no_match(self):
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = '11:22:33:44:55:66'
|
||||
device.device_id = "other"
|
||||
device.address = "11:22:33:44:55:66"
|
||||
device.name = None
|
||||
assert target.matches(device) is False
|
||||
|
||||
def test_match_by_device_id(self):
|
||||
target = LocateTarget(device_id='my-device-123')
|
||||
target = LocateTarget(device_id="my-device-123")
|
||||
device = MagicMock()
|
||||
device.device_id = 'my-device-123'
|
||||
device.address = '00:00:00:00:00:00'
|
||||
device.device_id = "my-device-123"
|
||||
device.address = "00:00:00:00:00:00"
|
||||
device.name = None
|
||||
assert target.matches(device) is True
|
||||
|
||||
def test_to_dict(self):
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF', known_name='Test')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF", known_name="Test")
|
||||
d = target.to_dict()
|
||||
assert d['mac_address'] == 'AA:BB:CC:DD:EE:FF'
|
||||
assert d['known_name'] == 'Test'
|
||||
assert d["mac_address"] == "AA:BB:CC:DD:EE:FF"
|
||||
assert d["known_name"] == "Test"
|
||||
|
||||
|
||||
class TestLocateSession:
|
||||
"""Test locate session lifecycle."""
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_start_stop(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session = LocateSession(target, Environment.OUTDOOR)
|
||||
session.start()
|
||||
|
||||
@@ -194,22 +195,22 @@ class TestLocateSession:
|
||||
assert session.active is False
|
||||
mock_scanner.remove_device_callback.assert_called_once()
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
@patch('utils.bt_locate.get_current_position')
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
@patch("utils.bt_locate.get_current_position")
|
||||
def test_detection_creates_trail_point(self, mock_gps, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
mock_gps.return_value = None # No GPS
|
||||
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session = LocateSession(target, Environment.OUTDOOR)
|
||||
session.start()
|
||||
|
||||
# Simulate device callback
|
||||
device = MagicMock()
|
||||
device.device_id = 'test'
|
||||
device.address = 'AA:BB:CC:DD:EE:FF'
|
||||
device.name = 'Test Device'
|
||||
device.device_id = "test"
|
||||
device.address = "AA:BB:CC:DD:EE:FF"
|
||||
device.name = "Test Device"
|
||||
device.rssi_current = -65
|
||||
|
||||
session._on_device(device)
|
||||
@@ -218,48 +219,48 @@ class TestLocateSession:
|
||||
assert len(session.trail) == 1
|
||||
assert session.trail[0].rssi == -65
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_non_matching_device_ignored(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session = LocateSession(target, Environment.OUTDOOR)
|
||||
session.start()
|
||||
|
||||
device = MagicMock()
|
||||
device.device_id = 'other'
|
||||
device.address = '11:22:33:44:55:66'
|
||||
device.device_id = "other"
|
||||
device.address = "11:22:33:44:55:66"
|
||||
device.name = None
|
||||
device.rssi_current = -70
|
||||
|
||||
session._on_device(device)
|
||||
assert session.detection_count == 0
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_get_status(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session = LocateSession(target, Environment.FREE_SPACE)
|
||||
session.start()
|
||||
|
||||
status = session.get_status()
|
||||
assert status['active'] is True
|
||||
assert status['environment'] == 'FREE_SPACE'
|
||||
assert status['detection_count'] == 0
|
||||
assert status["active"] is True
|
||||
assert status["environment"] == "FREE_SPACE"
|
||||
assert status["detection_count"] == 0
|
||||
|
||||
|
||||
class TestModuleLevelSessionManagement:
|
||||
class TestModuleLevelSessionManagement:
|
||||
"""Test module-level session functions."""
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_start_and_get_session(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
target = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session = start_locate_session(target)
|
||||
|
||||
assert get_locate_session() is session
|
||||
@@ -268,32 +269,32 @@ class TestModuleLevelSessionManagement:
|
||||
stop_locate_session()
|
||||
assert get_locate_session() is None
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
def test_start_replaces_existing_session(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_start_replaces_existing_session(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
target1 = LocateTarget(mac_address='AA:BB:CC:DD:EE:FF')
|
||||
target1 = LocateTarget(mac_address="AA:BB:CC:DD:EE:FF")
|
||||
session1 = start_locate_session(target1)
|
||||
|
||||
target2 = LocateTarget(mac_address='11:22:33:44:55:66')
|
||||
target2 = LocateTarget(mac_address="11:22:33:44:55:66")
|
||||
session2 = start_locate_session(target2)
|
||||
|
||||
assert get_locate_session() is session2
|
||||
assert session1.active is False
|
||||
assert session2.active is True
|
||||
|
||||
stop_locate_session()
|
||||
|
||||
@patch('utils.bt_locate.get_bluetooth_scanner')
|
||||
def test_start_raises_when_scanner_cannot_start(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_scanner.is_scanning = False
|
||||
mock_scanner.start_scan.return_value = False
|
||||
status = MagicMock()
|
||||
status.error = 'No adapter'
|
||||
mock_scanner.get_status.return_value = status
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
with pytest.raises(RuntimeError):
|
||||
start_locate_session(LocateTarget(mac_address='AA:BB:CC:DD:EE:FF'))
|
||||
assert session2.active is True
|
||||
|
||||
stop_locate_session()
|
||||
|
||||
@patch("utils.bt_locate.get_bluetooth_scanner")
|
||||
def test_start_raises_when_scanner_cannot_start(self, mock_get_scanner):
|
||||
mock_scanner = MagicMock()
|
||||
mock_scanner.is_scanning = False
|
||||
mock_scanner.start_scan.return_value = False
|
||||
status = MagicMock()
|
||||
status.error = "No adapter"
|
||||
mock_scanner.get_status.return_value = status
|
||||
mock_get_scanner.return_value = mock_scanner
|
||||
|
||||
with pytest.raises(RuntimeError):
|
||||
start_locate_session(LocateTarget(mac_address="AA:BB:CC:DD:EE:FF"))
|
||||
|
||||
Reference in New Issue
Block a user