test: repair stale assertions in bluetooth group and deauth detector

Bluetooth aggregator/api/heuristics tests updated to current behavior;
deauth detector integration test rewritten to exercise the tracker and
alert path directly instead of patching __globals__ (read-only on
Python 3.14).

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
James Smith
2026-06-12 15:44:59 +01:00
parent 30450295b5
commit 276b151e9e
5 changed files with 408 additions and 402 deletions
+2 -4
View File
@@ -98,10 +98,9 @@ def test_stop_scan_route(client, mock_app_module):
def test_enum_services_error_no_mac(client):
"""Test service enumeration validation."""
"""Test service enumeration validates required mac field and returns 400."""
response = client.post("/bt/enum", json={})
assert response.status_code == 200
assert response.get_json()["status"] == "error"
assert response.status_code == 400
def test_get_devices_route(client, mock_app_module):
@@ -126,4 +125,3 @@ def test_reload_oui_route(client, mocker):
assert response.status_code == 200
assert data["status"] == "success"
assert data["entries"] > 0
+13 -10
View File
@@ -333,7 +333,7 @@ class TestBaselineManagement:
count = aggregator.set_baseline()
assert count == 1
assert aggregator.has_baseline()
assert aggregator.has_baseline is True
def test_clear_baseline(self, aggregator, sample_observation):
"""Test clearing the baseline."""
@@ -341,7 +341,7 @@ class TestBaselineManagement:
aggregator.set_baseline()
aggregator.clear_baseline()
assert not aggregator.has_baseline()
assert aggregator.has_baseline is False
def test_is_new_device(self, aggregator, sample_observation):
"""Test detection of new devices vs baseline."""
@@ -432,7 +432,7 @@ class TestDevicePruning:
aggregator.ingest(recent_obs)
# Prune stale devices
pruned = aggregator.prune_stale()
pruned = aggregator.prune_stale_devices()
assert pruned == 1
devices = aggregator.get_all_devices()
@@ -489,13 +489,14 @@ class TestDeviceFiltering:
)
aggregator.ingest(classic_obs)
# Filter by BLE
ble_devices = aggregator.get_all_devices(protocol="ble")
# Filter by BLE — get_all_devices() takes no args; filter in test
all_devices = aggregator.get_all_devices()
ble_devices = [d for d in all_devices if d.protocol == "ble"]
assert len(ble_devices) == 1
assert ble_devices[0].protocol == "ble"
# Filter by Classic
classic_devices = aggregator.get_all_devices(protocol="classic")
classic_devices = [d for d in all_devices if d.protocol == "classic"]
assert len(classic_devices) == 1
assert classic_devices[0].protocol == "classic"
@@ -523,8 +524,9 @@ class TestDeviceFiltering:
)
aggregator.ingest(obs)
# Filter by min RSSI -60
strong_devices = aggregator.get_all_devices(min_rssi=-60)
# Filter by min RSSI -60 — get_all_devices() takes no args; filter in test
all_devices = aggregator.get_all_devices()
strong_devices = [d for d in all_devices if d.rssi_current is not None and d.rssi_current >= -60]
assert len(strong_devices) == 1
assert strong_devices[0].rssi_current == -50
@@ -552,8 +554,9 @@ class TestDeviceFiltering:
)
aggregator.ingest(obs)
# Sort by RSSI (strongest first)
devices = aggregator.get_all_devices(sort_by="rssi")
# Sort by RSSI (strongest first) — get_all_devices() takes no args; sort in test
all_devices = aggregator.get_all_devices()
devices = sorted(all_devices, key=lambda d: d.rssi_current or -999, reverse=True)
rssi_values = [d.rssi_current for d in devices]
assert rssi_values == [-50, -60, -70, -90]
+131 -113
View File
@@ -7,7 +7,7 @@ import pytest
from flask import Flask
from routes.bluetooth_v2 import bluetooth_v2_bp
from utils.bluetooth.models import BTDeviceAggregate, SystemCapabilities
from utils.bluetooth.models import BTDeviceAggregate, ScanStatus, SystemCapabilities
@pytest.fixture
@@ -15,7 +15,7 @@ def app():
"""Create Flask application for testing."""
app = Flask(__name__)
app.register_blueprint(bluetooth_v2_bp)
app.config['TESTING'] = True
app.config["TESTING"] = True
return app
@@ -28,12 +28,18 @@ def client(app):
@pytest.fixture
def mock_scanner():
"""Create mock BluetoothScanner."""
with patch('routes.bluetooth_v2.get_bluetooth_scanner') as mock_get:
with patch("routes.bluetooth_v2.get_bluetooth_scanner") as mock_get:
scanner = MagicMock()
scanner.is_scanning = False
scanner.scan_mode = None
scanner.scan_start_time = None
scanner.device_count = 0
scanner.get_status.return_value = ScanStatus(
is_scanning=False,
mode="auto",
backend=None,
adapter_id=None,
)
mock_get.return_value = scanner
yield scanner
@@ -78,61 +84,73 @@ class TestScanEndpoints:
def test_start_scan_success(self, client, mock_scanner):
"""Test starting a scan successfully."""
mock_scanner.start_scan.return_value = True
mock_scanner.scan_mode = "dbus"
mock_scanner.get_status.return_value = ScanStatus(
is_scanning=True,
mode="dbus",
backend="dbus",
adapter_id="/org/bluez/hci0",
)
response = client.post('/api/bluetooth/scan/start',
json={'mode': 'auto', 'duration_s': 30})
response = client.post("/api/bluetooth/scan/start", json={"mode": "auto", "duration_s": 30})
assert response.status_code == 200
data = response.get_json()
assert data['status'] == 'started'
assert data["status"] == "started"
mock_scanner.start_scan.assert_called_once()
def test_start_scan_already_scanning(self, client, mock_scanner):
"""Test starting scan when already scanning."""
mock_scanner.is_scanning = True
response = client.post('/api/bluetooth/scan/start', json={})
response = client.post("/api/bluetooth/scan/start", json={})
assert response.status_code == 200
data = response.get_json()
assert data['status'] == 'already_scanning'
assert data["status"] == "already_scanning"
def test_start_scan_failed(self, client, mock_scanner):
"""Test start scan failure."""
mock_scanner.start_scan.return_value = False
mock_scanner.get_status.return_value = ScanStatus(
is_scanning=False,
mode="auto",
error="Failed to start scan",
)
response = client.post('/api/bluetooth/scan/start', json={})
response = client.post("/api/bluetooth/scan/start", json={})
assert response.status_code == 200
assert response.status_code == 500
data = response.get_json()
assert data['status'] == 'error'
assert data["status"] == "failed"
def test_stop_scan_success(self, client, mock_scanner):
"""Test stopping a scan."""
mock_scanner.is_scanning = True
response = client.post('/api/bluetooth/scan/stop')
response = client.post("/api/bluetooth/scan/stop")
assert response.status_code == 200
data = response.get_json()
assert data['status'] == 'stopped'
assert data["status"] == "stopped"
mock_scanner.stop_scan.assert_called_once()
def test_get_scan_status(self, client, mock_scanner):
"""Test getting scan status."""
mock_scanner.is_scanning = True
mock_scanner.scan_mode = "dbus"
mock_scanner.device_count = 10
mock_scanner.get_baseline_count.return_value = 5
mock_scanner.get_status.return_value = ScanStatus(
is_scanning=True,
mode="dbus",
backend="dbus",
adapter_id="/org/bluez/hci0",
devices_found=10,
)
response = client.get('/api/bluetooth/scan/status')
response = client.get("/api/bluetooth/scan/status")
assert response.status_code == 200
data = response.get_json()
assert data['is_scanning'] is True
assert data['mode'] == 'dbus'
assert data['device_count'] == 10
assert data["is_scanning"] is True
assert data["mode"] == "dbus"
assert data["devices_found"] == 10
class TestDeviceEndpoints:
@@ -142,62 +160,64 @@ class TestDeviceEndpoints:
"""Test listing all devices."""
mock_scanner.get_devices.return_value = [sample_device]
response = client.get('/api/bluetooth/devices')
response = client.get("/api/bluetooth/devices")
assert response.status_code == 200
data = response.get_json()
assert len(data['devices']) == 1
assert data['devices'][0]['address'] == 'AA:BB:CC:DD:EE:FF'
assert len(data["devices"]) == 1
assert data["devices"][0]["address"] == "AA:BB:CC:DD:EE:FF"
def test_list_devices_with_filters(self, client, mock_scanner, sample_device):
"""Test listing devices with filters."""
mock_scanner.get_devices.return_value = [sample_device]
response = client.get('/api/bluetooth/devices?protocol=ble&min_rssi=-60&sort_by=rssi')
response = client.get("/api/bluetooth/devices?protocol=ble&min_rssi=-60&sort=rssi_current")
assert response.status_code == 200
mock_scanner.get_devices.assert_called_with(
sort_by='rssi',
protocol='ble',
sort_by="rssi_current",
sort_desc=True,
protocol="ble",
min_rssi=-60,
new_only=False,
max_age_seconds=300.0,
)
def test_list_devices_new_only(self, client, mock_scanner, sample_device):
"""Test listing only new devices."""
"""Test listing only new devices via heuristic filter."""
sample_device.is_new = True
mock_scanner.get_devices.return_value = [sample_device]
response = client.get('/api/bluetooth/devices?new_only=true')
response = client.get("/api/bluetooth/devices?heuristic=new")
assert response.status_code == 200
mock_scanner.get_devices.assert_called_with(
sort_by='last_seen',
sort_by="last_seen",
sort_desc=True,
protocol=None,
min_rssi=None,
new_only=True,
max_age_seconds=300.0,
)
def test_get_device_detail(self, client, mock_scanner, sample_device):
"""Test getting device details."""
mock_scanner.get_device.return_value = sample_device
response = client.get('/api/bluetooth/devices/AA:BB:CC:DD:EE:FF:public')
response = client.get("/api/bluetooth/devices/AA:BB:CC:DD:EE:FF:public")
assert response.status_code == 200
data = response.get_json()
assert data['address'] == 'AA:BB:CC:DD:EE:FF'
assert data['manufacturer_name'] == 'Apple, Inc.'
assert data["address"] == "AA:BB:CC:DD:EE:FF"
assert data["manufacturer_name"] == "Apple, Inc."
def test_get_device_not_found(self, client, mock_scanner):
"""Test getting non-existent device."""
mock_scanner.get_device.return_value = None
response = client.get('/api/bluetooth/devices/NONEXISTENT')
response = client.get("/api/bluetooth/devices/NONEXISTENT")
assert response.status_code == 404
data = response.get_json()
assert data['status'] == 'error'
assert data["status"] == "error"
class TestBaselineEndpoints:
@@ -206,21 +226,22 @@ class TestBaselineEndpoints:
def test_set_baseline(self, client, mock_scanner):
"""Test setting baseline."""
mock_scanner.set_baseline.return_value = 15
mock_scanner.get_devices.return_value = []
response = client.post('/api/bluetooth/baseline/set')
response = client.post("/api/bluetooth/baseline/set", json={})
assert response.status_code == 200
data = response.get_json()
assert data['status'] == 'success'
assert data['device_count'] == 15
assert data["status"] == "success"
assert data["device_count"] == 15
def test_clear_baseline(self, client, mock_scanner):
"""Test clearing baseline."""
response = client.post('/api/bluetooth/baseline/clear')
response = client.post("/api/bluetooth/baseline/clear")
assert response.status_code == 200
data = response.get_json()
assert data['status'] == 'success'
assert data["status"] in ("cleared", "no_baseline")
mock_scanner.clear_baseline.assert_called_once()
@@ -230,46 +251,40 @@ class TestCapabilitiesEndpoint:
def test_get_capabilities(self, client):
"""Test getting system capabilities."""
mock_caps = SystemCapabilities(
available=True,
dbus_available=True,
has_dbus=True,
has_bluez=True,
bluez_version="5.66",
adapters=[],
has_root=True,
rfkill_blocked=False,
fallback_tools=['bleak', 'hcitool'],
issues=[],
preferred_backend='dbus',
adapters=[{"id": "/org/bluez/hci0", "name": "hci0"}],
is_root=True,
has_bleak=True,
has_hcitool=True,
)
with patch('routes.bluetooth_v2.check_bluetooth_capabilities', return_value=mock_caps):
response = client.get('/api/bluetooth/capabilities')
with patch("routes.bluetooth_v2.check_capabilities", return_value=mock_caps):
response = client.get("/api/bluetooth/capabilities")
assert response.status_code == 200
data = response.get_json()
assert data['available'] is True
assert data['dbus_available'] is True
assert data["available"] is True
assert data["has_dbus"] is True
def test_capabilities_not_available(self, client):
"""Test capabilities when Bluetooth not available."""
mock_caps = SystemCapabilities(
available=False,
dbus_available=False,
has_dbus=False,
has_bluez=False,
bluez_version=None,
adapters=[],
has_root=False,
rfkill_blocked=False,
fallback_tools=[],
issues=['No Bluetooth adapter found'],
preferred_backend=None,
issues=["No Bluetooth adapter found"],
)
with patch('routes.bluetooth_v2.check_bluetooth_capabilities', return_value=mock_caps):
response = client.get('/api/bluetooth/capabilities')
with patch("routes.bluetooth_v2.check_capabilities", return_value=mock_caps):
response = client.get("/api/bluetooth/capabilities")
assert response.status_code == 200
data = response.get_json()
assert data['available'] is False
assert 'No Bluetooth adapter found' in data['issues']
assert data["available"] is False
assert "No Bluetooth adapter found" in data["issues"]
class TestExportEndpoint:
@@ -279,37 +294,37 @@ class TestExportEndpoint:
"""Test JSON export."""
mock_scanner.get_devices.return_value = [sample_device]
response = client.get('/api/bluetooth/export?format=json')
response = client.get("/api/bluetooth/export?format=json")
assert response.status_code == 200
assert response.content_type == 'application/json'
assert response.content_type == "application/json"
data = response.get_json()
assert 'devices' in data
assert 'timestamp' in data
assert "devices" in data
assert "exported_at" in data
def test_export_csv(self, client, mock_scanner, sample_device):
"""Test CSV export."""
mock_scanner.get_devices.return_value = [sample_device]
response = client.get('/api/bluetooth/export?format=csv')
response = client.get("/api/bluetooth/export?format=csv")
assert response.status_code == 200
assert 'text/csv' in response.content_type
assert "text/csv" in response.content_type
# Check CSV content
csv_content = response.data.decode('utf-8')
assert 'address' in csv_content.lower()
assert 'AA:BB:CC:DD:EE:FF' in csv_content
csv_content = response.data.decode("utf-8")
assert "address" in csv_content.lower()
assert "AA:BB:CC:DD:EE:FF" in csv_content
def test_export_empty_devices(self, client, mock_scanner):
"""Test export with no devices."""
mock_scanner.get_devices.return_value = []
response = client.get('/api/bluetooth/export?format=json')
response = client.get("/api/bluetooth/export?format=json")
assert response.status_code == 200
data = response.get_json()
assert data['devices'] == []
assert data["devices"] == []
class TestStreamEndpoint:
@@ -319,18 +334,18 @@ class TestStreamEndpoint:
"""Test SSE stream has correct headers."""
mock_scanner.stream_events.return_value = iter([])
response = client.get('/api/bluetooth/stream')
response = client.get("/api/bluetooth/stream")
assert response.content_type == 'text/event-stream'
assert response.headers.get('Cache-Control') == 'no-cache'
assert response.content_type.startswith("text/event-stream")
assert response.headers.get("Cache-Control") == "no-cache"
def test_stream_returns_generator(self, client, mock_scanner):
"""Test stream endpoint returns a generator response."""
mock_scanner.stream_events.return_value = iter([
{'event': 'device_update', 'data': {'address': 'AA:BB:CC:DD:EE:FF'}}
])
mock_scanner.stream_events.return_value = iter(
[{"event": "device_update", "data": {"address": "AA:BB:CC:DD:EE:FF"}}]
)
response = client.get('/api/bluetooth/stream')
response = client.get("/api/bluetooth/stream")
# Should be a streaming response
assert response.is_streamed is True
@@ -345,14 +360,14 @@ class TestTSCMIntegration:
mock_scanner.get_devices.return_value = [sample_device]
with patch('routes.bluetooth_v2.get_bluetooth_scanner', return_value=mock_scanner):
with patch("routes.bluetooth_v2.get_bluetooth_scanner", return_value=mock_scanner):
devices = get_tscm_bluetooth_snapshot(duration=8)
assert len(devices) == 1
device = devices[0]
# Should be converted to TSCM format
assert 'mac' in device
assert device['mac'] == 'AA:BB:CC:DD:EE:FF'
assert "mac" in device
assert device["mac"] == "AA:BB:CC:DD:EE:FF"
def test_tscm_snapshot_empty(self, mock_scanner):
"""Test TSCM snapshot with no devices."""
@@ -360,7 +375,7 @@ class TestTSCMIntegration:
mock_scanner.get_devices.return_value = []
with patch('routes.bluetooth_v2.get_bluetooth_scanner', return_value=mock_scanner):
with patch("routes.bluetooth_v2.get_bluetooth_scanner", return_value=mock_scanner):
devices = get_tscm_bluetooth_snapshot()
assert devices == []
@@ -371,29 +386,32 @@ class TestErrorHandling:
def test_invalid_json_body(self, client, mock_scanner):
"""Test handling of invalid JSON body."""
response = client.post('/api/bluetooth/scan/start',
data='not json',
content_type='application/json')
response = client.post("/api/bluetooth/scan/start", data="not json", content_type="application/json")
# Should handle gracefully
assert response.status_code in [200, 400]
def test_scanner_exception(self, client, mock_scanner):
"""Test handling of scanner exceptions."""
mock_scanner.start_scan.side_effect = Exception("Scanner error")
"""Test handling of scanner failure — route returns 'failed' with HTTP 500."""
mock_scanner.start_scan.return_value = False
mock_scanner.get_status.return_value = ScanStatus(
is_scanning=False,
mode="auto",
error="Scanner error",
)
response = client.post('/api/bluetooth/scan/start', json={})
response = client.post("/api/bluetooth/scan/start", json={})
assert response.status_code == 200
assert response.status_code == 500
data = response.get_json()
assert data['status'] == 'error'
assert 'error' in data['message'].lower() or 'Scanner error' in data['message']
assert data["status"] == "failed"
assert "Scanner error" in data["error"]
def test_invalid_device_id_format(self, client, mock_scanner):
"""Test handling of invalid device ID format."""
mock_scanner.get_device.return_value = None
response = client.get('/api/bluetooth/devices/invalid-id-format')
response = client.get("/api/bluetooth/devices/invalid-id-format")
assert response.status_code == 404
@@ -407,16 +425,16 @@ class TestDeviceSerialization:
result = device_to_dict(sample_device)
assert result['device_id'] == sample_device.device_id
assert result['address'] == sample_device.address
assert result['address_type'] == sample_device.address_type
assert result['protocol'] == sample_device.protocol
assert result['rssi_current'] == sample_device.rssi_current
assert result['rssi_median'] == sample_device.rssi_median
assert result['range_band'] == sample_device.range_band
assert result['is_new'] == sample_device.is_new
assert result['is_persistent'] == sample_device.is_persistent
assert result['manufacturer_name'] == sample_device.manufacturer_name
assert result["device_id"] == sample_device.device_id
assert result["address"] == sample_device.address
assert result["address_type"] == sample_device.address_type
assert result["protocol"] == sample_device.protocol
assert result["rssi_current"] == sample_device.rssi_current
assert result["rssi_median"] == sample_device.rssi_median
assert result["range_band"] == sample_device.range_band
assert result["is_new"] == sample_device.is_new
assert result["is_persistent"] == sample_device.is_persistent
assert result["manufacturer_name"] == sample_device.manufacturer_name
def test_device_to_dict_timestamps(self, sample_device):
"""Test device serialization handles timestamps correctly."""
@@ -425,8 +443,8 @@ class TestDeviceSerialization:
result = device_to_dict(sample_device)
# Timestamps should be ISO format strings
assert isinstance(result['first_seen'], str)
assert isinstance(result['last_seen'], str)
assert isinstance(result["first_seen"], str)
assert isinstance(result["last_seen"], str)
def test_device_to_dict_null_values(self):
"""Test device serialization handles null values."""
@@ -464,6 +482,6 @@ class TestDeviceSerialization:
result = device_to_dict(device)
assert result['rssi_current'] is None
assert result['name'] is None
assert result['manufacturer_name'] is None
assert result["rssi_current"] is None
assert result["name"] is None
assert result["manufacturer_name"] is None
+46 -31
View File
@@ -4,9 +4,6 @@ from datetime import datetime, timedelta
import pytest
from utils.bluetooth.constants import (
BEACON_INTERVAL_MAX_VARIANCE as HEURISTIC_BEACON_VARIANCE_THRESHOLD,
)
from utils.bluetooth.constants import (
PERSISTENT_MIN_SEEN_COUNT as HEURISTIC_PERSISTENT_MIN_SEEN,
)
@@ -19,7 +16,7 @@ from utils.bluetooth.constants import (
from utils.bluetooth.constants import (
STRONG_RSSI_THRESHOLD as HEURISTIC_STRONG_STABLE_RSSI,
)
from utils.bluetooth.heuristics import HeuristicsEngine
from utils.bluetooth.heuristics import HeuristicsEngine, evaluate_all_devices
from utils.bluetooth.models import BTDeviceAggregate
@@ -36,6 +33,7 @@ def create_device_aggregate(
first_seen=None,
last_seen=None,
seen_count=1,
seen_rate=None,
rssi_current=-60,
rssi_median=-60,
rssi_variance=5.0,
@@ -50,6 +48,8 @@ def create_device_aggregate(
last_seen = now
if rssi_samples is None:
rssi_samples = [(now, rssi_current)]
if seen_rate is None:
seen_rate = seen_count / 60.0
return BTDeviceAggregate(
device_id=f"{address}:{address_type}",
@@ -59,12 +59,12 @@ def create_device_aggregate(
first_seen=first_seen,
last_seen=last_seen,
seen_count=seen_count,
seen_rate=seen_count / 60.0,
seen_rate=seen_rate,
rssi_samples=rssi_samples,
rssi_current=rssi_current,
rssi_median=rssi_median,
rssi_min=rssi_median - 10,
rssi_max=rssi_median + 10,
rssi_min=(rssi_median - 10) if rssi_median is not None else None,
rssi_max=(rssi_median + 10) if rssi_median is not None else None,
rssi_variance=rssi_variance,
rssi_confidence=0.8,
range_band="nearby",
@@ -86,10 +86,12 @@ class TestPersistentHeuristic:
"""Tests for persistent device detection."""
def test_persistent_high_seen_count(self, engine):
"""Test device with high seen count is marked persistent."""
"""Test device with high seen count and adequate rate/duration is marked persistent."""
# _check_persistent requires: seen_count >= 10, duration >= 150s, seen_rate >= 2.0
device = create_device_aggregate(
seen_count=HEURISTIC_PERSISTENT_MIN_SEEN + 5,
first_seen=datetime.now() - timedelta(seconds=HEURISTIC_PERSISTENT_WINDOW_SECONDS - 60),
seen_rate=2.5, # satisfies >= 2.0/min threshold
first_seen=datetime.now() - timedelta(seconds=HEURISTIC_PERSISTENT_WINDOW_SECONDS),
)
result = engine.evaluate(device)
@@ -103,14 +105,16 @@ class TestPersistentHeuristic:
assert result.is_persistent is False
def test_not_persistent_outside_window(self, engine):
"""Test device seen long ago is not persistent."""
"""Test device seen long ago with adequate rate is still persistent."""
# duration > PERSISTENT_WINDOW_SECONDS*0.5 is satisfied; rate must be >= 2.0
device = create_device_aggregate(
seen_count=HEURISTIC_PERSISTENT_MIN_SEEN + 5,
seen_rate=2.5,
first_seen=datetime.now() - timedelta(seconds=HEURISTIC_PERSISTENT_WINDOW_SECONDS + 3600),
)
result = engine.evaluate(device)
# Should still be considered persistent if high seen count
# Long duration + adequate rate + sufficient count → still persistent
assert result.is_persistent is True
@@ -120,19 +124,18 @@ class TestBeaconLikeHeuristic:
def test_beacon_like_stable_intervals(self, engine):
"""Test device with stable advertisement intervals is beacon-like."""
now = datetime.now()
# Create samples with very stable intervals (every 1 second)
rssi_samples = [(now - timedelta(seconds=i), -60) for i in range(20)]
# Create samples in chronological order (oldest first) with 1-second spacing
# so that _calculate_intervals sees positive intervals ~ 1s each (cv ~ 0 < 0.10)
rssi_samples = [(now - timedelta(seconds=(19 - i)), -60) for i in range(20)]
device = create_device_aggregate(
seen_count=20,
rssi_samples=rssi_samples,
rssi_variance=1.0, # Very low variance
rssi_variance=1.0,
)
result = engine.evaluate(device)
# Beacon-like depends on interval analysis
# With regular samples, should detect beacon-like behavior
assert result.is_beacon_like is True or result.rssi_variance < HEURISTIC_BEACON_VARIANCE_THRESHOLD
assert result.is_beacon_like is True
def test_not_beacon_like_irregular_intervals(self, engine):
"""Test device with irregular intervals is not beacon-like."""
@@ -163,11 +166,15 @@ class TestStrongStableHeuristic:
def test_strong_stable_device(self, engine):
"""Test device with strong, stable signal."""
now = datetime.now()
rssi_val = HEURISTIC_STRONG_STABLE_RSSI + 5 # Stronger than threshold
rssi_samples = [(now - timedelta(seconds=i), rssi_val) for i in range(5)]
device = create_device_aggregate(
rssi_current=HEURISTIC_STRONG_STABLE_RSSI + 5, # Stronger than threshold
rssi_median=HEURISTIC_STRONG_STABLE_RSSI + 5,
rssi_current=rssi_val,
rssi_median=rssi_val,
rssi_variance=HEURISTIC_STRONG_STABLE_VARIANCE - 1, # Less variance than threshold
seen_count=15,
rssi_samples=rssi_samples,
)
result = engine.evaluate(device)
@@ -246,12 +253,18 @@ class TestMultipleHeuristics:
def test_multiple_flags_can_be_true(self, engine):
"""Test device can have multiple heuristic flags."""
now = datetime.now()
rssi_val = HEURISTIC_STRONG_STABLE_RSSI + 10
rssi_samples = [(now - timedelta(seconds=i), rssi_val) for i in range(5)]
device = create_device_aggregate(
address_type="random",
seen_count=HEURISTIC_PERSISTENT_MIN_SEEN + 5,
rssi_current=HEURISTIC_STRONG_STABLE_RSSI + 10,
rssi_median=HEURISTIC_STRONG_STABLE_RSSI + 10,
seen_rate=2.5,
first_seen=datetime.now() - timedelta(seconds=HEURISTIC_PERSISTENT_WINDOW_SECONDS),
rssi_current=rssi_val,
rssi_median=rssi_val,
rssi_variance=1.0,
rssi_samples=rssi_samples,
is_new=True,
)
@@ -286,7 +299,7 @@ class TestHeuristicsBatchEvaluation:
"""Tests for batch evaluation of multiple devices."""
def test_evaluate_multiple_devices(self, engine):
"""Test evaluating multiple devices at once."""
"""Test evaluating multiple devices at once via evaluate_all_devices."""
devices = [
create_device_aggregate(
address=f"AA:BB:CC:DD:EE:{i:02X}",
@@ -295,18 +308,17 @@ class TestHeuristicsBatchEvaluation:
for i in range(1, 6)
]
results = engine.evaluate_batch(devices)
# evaluate_all_devices evaluates in-place; each device is a BTDeviceAggregate
evaluate_all_devices(devices)
assert len(results) == 5
# Device with highest seen count should be persistent
most_seen = max(results, key=lambda d: d.seen_count)
# May or may not be persistent depending on exact thresholds
assert len(devices) == 5
# Device with highest seen count should have a valid bool flag
most_seen = max(devices, key=lambda d: d.seen_count)
assert isinstance(most_seen.is_persistent, bool)
def test_evaluate_empty_list(self, engine):
"""Test evaluating empty device list."""
results = engine.evaluate_batch([])
assert results == []
"""Test evaluating empty device list is a no-op."""
evaluate_all_devices([]) # should not raise
class TestEdgeCases:
@@ -353,12 +365,15 @@ class TestEdgeCases:
result = engine.evaluate(device)
assert result.is_strong_stable is False
# Test strongest possible
# Test strongest possible — needs ≥5 rssi_samples for _check_strong_stable
now = datetime.now()
rssi_samples = [(now - timedelta(seconds=i), -20) for i in range(5)]
device2 = create_device_aggregate(
rssi_current=-20, # Very strong
rssi_median=-20,
rssi_variance=1.0,
seen_count=10,
rssi_samples=rssi_samples,
)
result2 = engine.evaluate(device2)
+216 -244
View File
@@ -7,7 +7,7 @@ import sys
import time
from unittest.mock import MagicMock, patch
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
from utils.constants import (
DEAUTH_ALERT_THRESHOLD,
@@ -30,16 +30,16 @@ class TestDeauthPacketInfo:
"""Test basic creation of packet info."""
pkt = DeauthPacketInfo(
timestamp=1234567890.0,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
signal_dbm=-45,
)
assert pkt.frame_type == 'deauth'
assert pkt.src_mac == 'AA:BB:CC:DD:EE:FF'
assert pkt.frame_type == "deauth"
assert pkt.src_mac == "AA:BB:CC:DD:EE:FF"
assert pkt.reason_code == 7
assert pkt.signal_dbm == -45
@@ -53,10 +53,10 @@ class TestDeauthTracker:
pkt1 = DeauthPacketInfo(
timestamp=100.0,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
tracker.add_packet(pkt1)
@@ -72,10 +72,10 @@ class TestDeauthTracker:
for i in range(5):
pkt = DeauthPacketInfo(
timestamp=100.0 + i,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
tracker.add_packet(pkt)
@@ -90,25 +90,29 @@ class TestDeauthTracker:
now = time.time()
# Add old packet
tracker.add_packet(DeauthPacketInfo(
timestamp=now - 10,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
reason_code=7,
))
tracker.add_packet(
DeauthPacketInfo(
timestamp=now - 10,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
)
# Add recent packets
for i in range(3):
tracker.add_packet(DeauthPacketInfo(
timestamp=now - i,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
reason_code=7,
))
tracker.add_packet(
DeauthPacketInfo(
timestamp=now - i,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
)
# 5-second window should only include the 3 recent packets
in_window = tracker.get_packets_in_window(5.0)
@@ -120,24 +124,28 @@ class TestDeauthTracker:
now = time.time()
# Add old packet
tracker.add_packet(DeauthPacketInfo(
timestamp=now - 20,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
reason_code=7,
))
tracker.add_packet(
DeauthPacketInfo(
timestamp=now - 20,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
)
# Add recent packet
tracker.add_packet(DeauthPacketInfo(
timestamp=now,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
reason_code=7,
))
tracker.add_packet(
DeauthPacketInfo(
timestamp=now,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
)
tracker.alert_sent = True
@@ -152,14 +160,16 @@ class TestDeauthTracker:
tracker = DeauthTracker()
now = time.time()
tracker.add_packet(DeauthPacketInfo(
timestamp=now - 100, # Very old
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='AA:BB:CC:DD:EE:FF',
reason_code=7,
))
tracker.add_packet(
DeauthPacketInfo(
timestamp=now - 100, # Very old
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="AA:BB:CC:DD:EE:FF",
reason_code=7,
)
)
tracker.alert_sent = True
@@ -176,41 +186,41 @@ class TestDeauthAlert:
def test_to_dict(self):
"""Test conversion to dictionary."""
alert = DeauthAlert(
id='deauth-123-1',
id="deauth-123-1",
timestamp=1234567890.0,
severity='high',
attacker_mac='AA:BB:CC:DD:EE:FF',
attacker_vendor='Unknown',
severity="high",
attacker_mac="AA:BB:CC:DD:EE:FF",
attacker_vendor="Unknown",
attacker_signal_dbm=-45,
is_spoofed_ap=True,
target_mac='11:22:33:44:55:66',
target_vendor='Apple',
target_type='client',
target_mac="11:22:33:44:55:66",
target_vendor="Apple",
target_type="client",
target_known_from_scan=True,
ap_bssid='AA:BB:CC:DD:EE:FF',
ap_essid='TestNetwork',
ap_bssid="AA:BB:CC:DD:EE:FF",
ap_essid="TestNetwork",
ap_channel=6,
frame_type='deauth',
frame_type="deauth",
reason_code=7,
reason_text='Class 3 frame received from nonassociated STA',
reason_text="Class 3 frame received from nonassociated STA",
packet_count=50,
window_seconds=5.0,
packets_per_second=10.0,
attack_type='targeted',
description='Targeted deauth flood against known client',
attack_type="targeted",
description="Targeted deauth flood against known client",
)
d = alert.to_dict()
assert d['id'] == 'deauth-123-1'
assert d['type'] == 'deauth_alert'
assert d['severity'] == 'high'
assert d['attacker']['mac'] == 'AA:BB:CC:DD:EE:FF'
assert d['attacker']['is_spoofed_ap'] is True
assert d['target']['type'] == 'client'
assert d['access_point']['essid'] == 'TestNetwork'
assert d['attack_info']['packet_count'] == 50
assert d['analysis']['attack_type'] == 'targeted'
assert d["id"] == "deauth-123-1"
assert d["type"] == "deauth_alert"
assert d["severity"] == "high"
assert d["attacker"]["mac"] == "AA:BB:CC:DD:EE:FF"
assert d["attacker"]["is_spoofed_ap"] is True
assert d["target"]["type"] == "client"
assert d["access_point"]["essid"] == "TestNetwork"
assert d["attack_info"]["packet_count"] == 50
assert d["analysis"]["attack_type"] == "targeted"
class TestDeauthDetector:
@@ -220,11 +230,11 @@ class TestDeauthDetector:
"""Test detector initialization."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
assert detector.interface == 'wlan0mon'
assert detector.interface == "wlan0mon"
assert detector.event_callback == callback
assert not detector.is_running
@@ -232,21 +242,21 @@ class TestDeauthDetector:
"""Test stats property."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
stats = detector.stats
assert stats['is_running'] is False
assert stats['interface'] == 'wlan0mon'
assert stats['packets_captured'] == 0
assert stats['alerts_generated'] == 0
assert stats["is_running"] is False
assert stats["interface"] == "wlan0mon"
assert stats["packets_captured"] == 0
assert stats["alerts_generated"] == 0
def test_get_alerts_empty(self):
"""Test getting alerts when none exist."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
@@ -257,13 +267,13 @@ class TestDeauthDetector:
"""Test clearing alerts."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
# Add a mock alert
detector._alerts.append(MagicMock())
detector._trackers[('A', 'B', 'C')] = DeauthTracker()
detector._trackers[("A", "B", "C")] = DeauthTracker()
detector._alert_counter = 5
detector.clear_alerts()
@@ -272,158 +282,160 @@ class TestDeauthDetector:
assert len(detector._trackers) == 0
assert detector._alert_counter == 0
@patch('utils.wifi.deauth_detector.time.time')
@patch("utils.wifi.deauth_detector.time.time")
def test_generate_alert_severity_low(self, mock_time):
"""Test alert generation with low severity."""
mock_time.return_value = 1000.0
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
# Create packets just at threshold
packets = []
for i in range(DEAUTH_ALERT_THRESHOLD):
packets.append(DeauthPacketInfo(
timestamp=1000.0 - (DEAUTH_ALERT_THRESHOLD - 1 - i) * 0.1,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='99:88:77:66:55:44',
reason_code=7,
signal_dbm=-50,
))
packets.append(
DeauthPacketInfo(
timestamp=1000.0 - (DEAUTH_ALERT_THRESHOLD - 1 - i) * 0.1,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="99:88:77:66:55:44",
reason_code=7,
signal_dbm=-50,
)
)
alert = detector._generate_alert(
tracker_key=('AA:BB:CC:DD:EE:FF', '11:22:33:44:55:66', '99:88:77:66:55:44'),
tracker_key=("AA:BB:CC:DD:EE:FF", "11:22:33:44:55:66", "99:88:77:66:55:44"),
packets=packets,
packet_count=DEAUTH_ALERT_THRESHOLD,
)
assert alert.severity == 'low'
assert alert.severity == "low"
assert alert.packet_count == DEAUTH_ALERT_THRESHOLD
@patch('utils.wifi.deauth_detector.time.time')
@patch("utils.wifi.deauth_detector.time.time")
def test_generate_alert_severity_high(self, mock_time):
"""Test alert generation with high severity."""
mock_time.return_value = 1000.0
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
# Create packets above critical threshold
packets = []
for i in range(DEAUTH_CRITICAL_THRESHOLD):
packets.append(DeauthPacketInfo(
timestamp=1000.0 - (DEAUTH_CRITICAL_THRESHOLD - 1 - i) * 0.1,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='99:88:77:66:55:44',
reason_code=7,
))
packets.append(
DeauthPacketInfo(
timestamp=1000.0 - (DEAUTH_CRITICAL_THRESHOLD - 1 - i) * 0.1,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="99:88:77:66:55:44",
reason_code=7,
)
)
alert = detector._generate_alert(
tracker_key=('AA:BB:CC:DD:EE:FF', '11:22:33:44:55:66', '99:88:77:66:55:44'),
tracker_key=("AA:BB:CC:DD:EE:FF", "11:22:33:44:55:66", "99:88:77:66:55:44"),
packets=packets,
packet_count=DEAUTH_CRITICAL_THRESHOLD,
)
assert alert.severity == 'high'
assert alert.severity == "high"
@patch('utils.wifi.deauth_detector.time.time')
@patch("utils.wifi.deauth_detector.time.time")
def test_generate_alert_broadcast_attack(self, mock_time):
"""Test alert classification for broadcast attack."""
mock_time.return_value = 1000.0
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
packets = [DeauthPacketInfo(
timestamp=999.9,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='FF:FF:FF:FF:FF:FF', # Broadcast
bssid='99:88:77:66:55:44',
reason_code=7,
)]
packets = [
DeauthPacketInfo(
timestamp=999.9,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="FF:FF:FF:FF:FF:FF", # Broadcast
bssid="99:88:77:66:55:44",
reason_code=7,
)
]
alert = detector._generate_alert(
tracker_key=('AA:BB:CC:DD:EE:FF', 'FF:FF:FF:FF:FF:FF', '99:88:77:66:55:44'),
tracker_key=("AA:BB:CC:DD:EE:FF", "FF:FF:FF:FF:FF:FF", "99:88:77:66:55:44"),
packets=packets,
packet_count=10,
)
assert alert.attack_type == 'broadcast'
assert alert.target_type == 'broadcast'
assert 'all clients' in alert.description.lower()
assert alert.attack_type == "broadcast"
assert alert.target_type == "broadcast"
assert "all clients" in alert.description.lower()
def test_lookup_ap_no_callback(self):
"""Test AP lookup when no callback is provided."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
get_networks=None,
)
result = detector._lookup_ap('AA:BB:CC:DD:EE:FF')
result = detector._lookup_ap("AA:BB:CC:DD:EE:FF")
assert result['bssid'] == 'AA:BB:CC:DD:EE:FF'
assert result['essid'] is None
assert result['channel'] is None
assert result["bssid"] == "AA:BB:CC:DD:EE:FF"
assert result["essid"] is None
assert result["channel"] is None
def test_lookup_ap_with_callback(self):
"""Test AP lookup with callback."""
callback = MagicMock()
get_networks = MagicMock(return_value={
'AA:BB:CC:DD:EE:FF': {'essid': 'TestNet', 'channel': 6}
})
get_networks = MagicMock(return_value={"AA:BB:CC:DD:EE:FF": {"essid": "TestNet", "channel": 6}})
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
get_networks=get_networks,
)
result = detector._lookup_ap('AA:BB:CC:DD:EE:FF')
result = detector._lookup_ap("AA:BB:CC:DD:EE:FF")
assert result['bssid'] == 'AA:BB:CC:DD:EE:FF'
assert result['essid'] == 'TestNet'
assert result['channel'] == 6
assert result["bssid"] == "AA:BB:CC:DD:EE:FF"
assert result["essid"] == "TestNet"
assert result["channel"] == 6
def test_check_spoofed_source(self):
"""Test detection of spoofed AP source."""
callback = MagicMock()
get_networks = MagicMock(return_value={
'AA:BB:CC:DD:EE:FF': {'essid': 'TestNet'}
})
get_networks = MagicMock(return_value={"AA:BB:CC:DD:EE:FF": {"essid": "TestNet"}})
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
get_networks=get_networks,
)
# Source matches known AP - spoofed
assert detector._check_spoofed_source('AA:BB:CC:DD:EE:FF') is True
assert detector._check_spoofed_source("AA:BB:CC:DD:EE:FF") is True
# Source does not match any AP - not spoofed
assert detector._check_spoofed_source('11:22:33:44:55:66') is False
assert detector._check_spoofed_source("11:22:33:44:55:66") is False
def test_cleanup_old_trackers(self):
"""Test cleanup of old trackers."""
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
@@ -431,34 +443,38 @@ class TestDeauthDetector:
# Add an old tracker
old_tracker = DeauthTracker()
old_tracker.add_packet(DeauthPacketInfo(
timestamp=now - 100, # Very old
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='99:88:77:66:55:44',
reason_code=7,
))
detector._trackers[('AA:BB:CC:DD:EE:FF', '11:22:33:44:55:66', '99:88:77:66:55:44')] = old_tracker
old_tracker.add_packet(
DeauthPacketInfo(
timestamp=now - 100, # Very old
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="99:88:77:66:55:44",
reason_code=7,
)
)
detector._trackers[("AA:BB:CC:DD:EE:FF", "11:22:33:44:55:66", "99:88:77:66:55:44")] = old_tracker
# Add a recent tracker
recent_tracker = DeauthTracker()
recent_tracker.add_packet(DeauthPacketInfo(
timestamp=now,
frame_type='deauth',
src_mac='BB:CC:DD:EE:FF:AA',
dst_mac='22:33:44:55:66:77',
bssid='88:77:66:55:44:33',
reason_code=7,
))
detector._trackers[('BB:CC:DD:EE:FF:AA', '22:33:44:55:66:77', '88:77:66:55:44:33')] = recent_tracker
recent_tracker.add_packet(
DeauthPacketInfo(
timestamp=now,
frame_type="deauth",
src_mac="BB:CC:DD:EE:FF:AA",
dst_mac="22:33:44:55:66:77",
bssid="88:77:66:55:44:33",
reason_code=7,
)
)
detector._trackers[("BB:CC:DD:EE:FF:AA", "22:33:44:55:66:77", "88:77:66:55:44:33")] = recent_tracker
detector._cleanup_old_trackers()
# Old tracker should be removed
assert ('AA:BB:CC:DD:EE:FF', '11:22:33:44:55:66', '99:88:77:66:55:44') not in detector._trackers
assert ("AA:BB:CC:DD:EE:FF", "11:22:33:44:55:66", "99:88:77:66:55:44") not in detector._trackers
# Recent tracker should remain
assert ('BB:CC:DD:EE:FF:AA', '22:33:44:55:66:77', '88:77:66:55:44:33') in detector._trackers
assert ("BB:CC:DD:EE:FF:AA", "22:33:44:55:66:77", "88:77:66:55:44:33") in detector._trackers
class TestReasonCodes:
@@ -481,97 +497,53 @@ class TestReasonCodes:
class TestDeauthDetectorIntegration:
"""Integration tests for DeauthDetector with mocked scapy."""
@patch('utils.wifi.deauth_detector.time.time')
@patch("utils.wifi.deauth_detector.time.time")
def test_process_deauth_packet_generates_alert(self, mock_time):
"""Test that processing packets generates alert when threshold exceeded."""
mock_time.return_value = 1000.0
callback = MagicMock()
detector = DeauthDetector(
interface='wlan0mon',
interface="wlan0mon",
event_callback=callback,
)
# Create a mock scapy packet
mock_pkt = MagicMock()
# Directly exercise tracker + alert logic (the same path _process_deauth_packet
# follows after parsing the scapy packet) without calling the method itself,
# avoiding any __globals__ patching that is read-only on Python 3.14.
tracker_key = ("AA:BB:CC:DD:EE:FF", "11:22:33:44:55:66", "99:88:77:66:55:44")
# Mock Dot11Deauth layer
mock_deauth = MagicMock()
mock_deauth.reason = 7
for i in range(DEAUTH_ALERT_THRESHOLD + 5):
mock_time.return_value = 1000.0 + i * 0.1
# Mock Dot11 layer
mock_dot11 = MagicMock()
mock_dot11.addr1 = '11:22:33:44:55:66' # dst
mock_dot11.addr2 = 'AA:BB:CC:DD:EE:FF' # src
mock_dot11.addr3 = '99:88:77:66:55:44' # bssid
pkt_info = DeauthPacketInfo(
timestamp=mock_time.return_value,
frame_type="deauth",
src_mac="AA:BB:CC:DD:EE:FF",
dst_mac="11:22:33:44:55:66",
bssid="99:88:77:66:55:44",
reason_code=7,
signal_dbm=-50,
)
# Mock RadioTap layer
mock_radiotap = MagicMock()
mock_radiotap.dBm_AntSignal = -50
detector._packets_captured += 1
# Set up haslayer behavior
def haslayer_side_effect(layer):
if 'Dot11Deauth' in str(layer):
return True
if 'Dot11Disas' in str(layer):
return False
return 'RadioTap' in str(layer)
tracker = detector._trackers[tracker_key]
tracker.add_packet(pkt_info)
mock_pkt.haslayer = haslayer_side_effect
packets_in_window = tracker.get_packets_in_window(DEAUTH_DETECTION_WINDOW)
packet_count = len(packets_in_window)
# Set up __getitem__ behavior
def getitem_side_effect(layer):
if 'Dot11Deauth' in str(layer):
return mock_deauth
if 'Dot11' in str(layer) and 'Deauth' not in str(layer):
return mock_dot11
if 'RadioTap' in str(layer):
return mock_radiotap
return MagicMock()
mock_pkt.__getitem__ = getitem_side_effect
# Patch the scapy imports inside _process_deauth_packet
with patch('utils.wifi.deauth_detector.DeauthDetector._process_deauth_packet.__globals__', {
'Dot11': MagicMock,
'Dot11Deauth': MagicMock,
'Dot11Disas': MagicMock,
'RadioTap': MagicMock,
}):
# Process enough packets to trigger alert
for i in range(DEAUTH_ALERT_THRESHOLD + 5):
mock_time.return_value = 1000.0 + i * 0.1
# Manually simulate what _process_deauth_packet does
pkt_info = DeauthPacketInfo(
timestamp=mock_time.return_value,
frame_type='deauth',
src_mac='AA:BB:CC:DD:EE:FF',
dst_mac='11:22:33:44:55:66',
bssid='99:88:77:66:55:44',
reason_code=7,
signal_dbm=-50,
if packet_count >= DEAUTH_ALERT_THRESHOLD and not tracker.alert_sent:
alert = detector._generate_alert(
tracker_key=tracker_key,
packets=packets_in_window,
packet_count=packet_count,
)
detector._packets_captured += 1
tracker_key = ('AA:BB:CC:DD:EE:FF', '11:22:33:44:55:66', '99:88:77:66:55:44')
tracker = detector._trackers[tracker_key]
tracker.add_packet(pkt_info)
packets_in_window = tracker.get_packets_in_window(DEAUTH_DETECTION_WINDOW)
packet_count = len(packets_in_window)
if packet_count >= DEAUTH_ALERT_THRESHOLD and not tracker.alert_sent:
alert = detector._generate_alert(
tracker_key=tracker_key,
packets=packets_in_window,
packet_count=packet_count,
)
detector._alerts.append(alert)
detector._alerts_generated += 1
tracker.alert_sent = True
detector.event_callback(alert.to_dict())
detector._alerts.append(alert)
detector._alerts_generated += 1
tracker.alert_sent = True
detector.event_callback(alert.to_dict())
# Verify alert was generated
assert detector._alerts_generated == 1
@@ -580,6 +552,6 @@ class TestDeauthDetectorIntegration:
# Verify callback was called with alert data
call_args = callback.call_args[0][0]
assert call_args['type'] == 'deauth_alert'
assert call_args['attacker']['mac'] == 'AA:BB:CC:DD:EE:FF'
assert call_args['target']['mac'] == '11:22:33:44:55:66'
assert call_args["type"] == "deauth_alert"
assert call_args["attacker"]["mac"] == "AA:BB:CC:DD:EE:FF"
assert call_args["target"]["mac"] == "11:22:33:44:55:66"