Mirror/intercept
1
0
Fork 0
mirror of https://github.com/smittix/intercept.git synced 2026-07-08 01:28:13 -07:00
Code Issues Packages Projects Releases Wiki Activity

style: apply ruff-format to entire codebase

Browse Source 
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
2026-07-05 14:48:11 +01:00
parent 82e64104fe
commit 96172ca593
189 changed files with 19883 additions and 19552 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_wifi.py
+60 -51
View File
@@ -5,7 +5,7 @@ from unittest.mock import MagicMock, mock_open, patch
import pytest
from flask import Flask
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 routes.wifi import parse_airodump_csv, wifi_bp
@@ -21,16 +21,19 @@ def mock_app_module(mocker):
mock_app_module.wifi_clients = {}
return mock
@pytest.fixture
def app():
app = Flask(__name__)
app.register_blueprint(wifi_bp)
return app
@pytest.fixture
def client(app):
return app.test_client()
def test_parse_airodump_csv(mocker):
"""Test parsing logic for airodump CSV format."""
csv_content = (
@@ -50,19 +53,22 @@ def test_parse_airodump_csv(mocker):
assert "11:22:33:44:55:66" in clients
assert clients["11:22:33:44:55:66"]["vendor"] == "Apple"
### --- ROUTE TESTS --- ###
def test_get_interfaces(client, mocker):
"""Test the /interfaces endpoint."""
mocker.patch("routes.wifi.detect_wifi_interfaces", return_value=[{'name': 'wlan0', 'type': 'managed'}])
mocker.patch("routes.wifi.detect_wifi_interfaces", return_value=[{"name": "wlan0", "type": "managed"}])
mocker.patch("routes.wifi.check_tool", return_value=True)
response = client.get('/wifi/interfaces')
response = client.get("/wifi/interfaces")
data = response.get_json()
assert response.status_code == 200
assert len(data['interfaces']) == 1
assert data['tools']['airmon'] is True
assert len(data["interfaces"]) == 1
assert data["tools"]["airmon"] is True
def test_toggle_monitor_start_success(client, mocker):
"""Test enabling monitor mode via airmon-ng."""
@@ -72,20 +78,22 @@ def test_toggle_monitor_start_success(client, mocker):
mock_run.return_value = MagicMock(stdout="enabled on [phy0]wlan0mon", stderr="", returncode=0)
with patch("os.path.exists", return_value=True):
response = client.post('/wifi/monitor', json={'action': 'start', 'interface': 'wlan0'})
response = client.post("/wifi/monitor", json={"action": "start", "interface": "wlan0"})
assert response.status_code == 200
assert response.get_json()['status'] == 'success'
assert response.get_json()['monitor_interface'] == 'wlan0mon'
assert response.get_json()["status"] == "success"
assert response.get_json()["monitor_interface"] == "wlan0mon"
def test_start_scan_already_running(client, mock_app_module):
"""Test that we can't start a scan if one is already active."""
mock_app_module.wifi_process = MagicMock()
response = client.post('/wifi/scan/start', json={'interface': 'wlan0mon'})
response = client.post("/wifi/scan/start", json={"interface": "wlan0mon"})
data = response.get_json()
assert data['status'] == 'error'
assert 'already running' in data['message']
assert data["status"] == "error"
assert "already running" in data["message"]
def test_start_scan_execution(client, mock_app_module, mocker):
"""Test the full command construction of airodump-ng."""
@@ -98,28 +106,30 @@ def test_start_scan_execution(client, mock_app_module, mocker):
mock_proc.poll.return_value = None
mock_popen.return_value = mock_proc
payload = {'interface': 'wlan0mon', 'channel': 6, 'band': 'g'}
response = client.post('/wifi/scan/start', json=payload)
payload = {"interface": "wlan0mon", "channel": 6, "band": "g"}
response = client.post("/wifi/scan/start", json=payload)
assert response.status_code == 200
assert response.get_json()['status'] == 'started'
assert response.get_json()["status"] == "started"
args, _ = mock_popen.call_args
cmd = args[0]
assert "-c" in cmd and "6" in cmd
assert "wlan0mon" in cmd
def test_stop_scan(client, mock_app_module):
"""Test terminating the scanning process."""
mock_proc = MagicMock()
mock_app_module.wifi_process = mock_proc
response = client.post('/wifi/scan/stop')
response = client.post("/wifi/scan/stop")
assert response.status_code == 200
assert response.get_json()['status'] == 'stopped'
assert response.get_json()["status"] == "stopped"
mock_proc.terminate.assert_called_once()
def test_send_deauth_success(client, mock_app_module, mocker):
"""Verify deauth command construction and execution."""
mocker.patch("routes.wifi.check_tool", return_value=True)
@@ -127,12 +137,8 @@ def test_send_deauth_success(client, mock_app_module, mocker):
mock_run = mocker.patch("routes.wifi.subprocess.run")
mock_run.return_value = MagicMock(returncode=0)
payload = {
'bssid': 'AA:BB:CC:DD:EE:FF',
'count': 10,
'interface': 'wlan0mon'
}
response = client.post('/wifi/deauth', json=payload)
payload = {"bssid": "AA:BB:CC:DD:EE:FF", "count": 10, "interface": "wlan0mon"}
response = client.post("/wifi/deauth", json=payload)
assert response.status_code == 200
args, _ = mock_run.call_args
@@ -141,21 +147,24 @@ def test_send_deauth_success(client, mock_app_module, mocker):
assert "10" in cmd
assert "AA:BB:CC:DD:EE:FF" in cmd
### --- HANDSHAKE TESTS --- ###
def test_capture_handshake_start(client, mock_app_module, mocker):
"""Test starting airodump-ng for handshake capture."""
mock_app_module.wifi_process = None
mocker.patch("routes.wifi.get_tool_path", return_value="/usr/bin/airodump-ng")
mock_popen = mocker.patch("routes.wifi.subprocess.Popen")
payload = {'bssid': 'AA:BB:CC:DD:EE:FF', 'channel': '6', 'interface': 'wlan0mon'}
response = client.post('/wifi/handshake/capture', json=payload)
payload = {"bssid": "AA:BB:CC:DD:EE:FF", "channel": "6", "interface": "wlan0mon"}
response = client.post("/wifi/handshake/capture", json=payload)
assert response.status_code == 200
assert 'capture_file' in response.get_json()
assert "capture_file" in response.get_json()
assert mock_popen.called
def test_check_handshake_status_found(client, mocker):
"""Verify detection of 'KEY FOUND' in aircrack output."""
mocker.patch("os.path.exists", return_value=True)
@@ -165,24 +174,28 @@ def test_check_handshake_status_found(client, mocker):
mock_run = mocker.patch("routes.wifi.subprocess.run")
mock_run.return_value = MagicMock(stdout="WPA (1 handshake)", stderr="", returncode=0)
payload = {'file': '/tmp/intercept_handshake_test.cap', 'bssid': 'AA:BB:CC:DD:EE:FF'}
response = client.post('/wifi/handshake/status', json=payload)
payload = {"file": "/tmp/intercept_handshake_test.cap", "bssid": "AA:BB:CC:DD:EE:FF"}
response = client.post("/wifi/handshake/status", json=payload)
assert response.get_json()["handshake_found"] is True
assert response.get_json()['handshake_found'] is True
### --- PMKID TESTS --- ###
def test_capture_pmkid_path_traversal_prevention(client):
"""Ensure the status check rejects invalid paths."""
payload = {'file': '/etc/passwd'} # Malicious path
response = client.post('/wifi/pmkid/status', json=payload)
payload = {"file": "/etc/passwd"} # Malicious path
response = client.post("/wifi/pmkid/status", json=payload)
assert response.status_code == 400
assert response.get_json()['status'] == 'error'
assert 'Invalid capture file path' in response.get_json()['message']
assert response.get_json()["status"] == "error"
assert "Invalid capture file path" in response.get_json()["message"]
### --- CRACKING TESTS --- ###
def test_crack_handshake_success(client, mocker):
"""Test successful password extraction using Regex."""
mocker.patch("os.path.exists", return_value=True)
@@ -190,35 +203,31 @@ def test_crack_handshake_success(client, mocker):
mock_run = mocker.patch("routes.wifi.subprocess.run")
# Simulate the actual aircrack-ng success output
mock_run.return_value = MagicMock(
stdout="KEY FOUND! [ secret123 ]",
stderr="",
returncode=0
)
mock_run.return_value = MagicMock(stdout="KEY FOUND! [ secret123 ]", stderr="", returncode=0)
payload = {
'capture_file': '/tmp/intercept_handshake_test.cap',
'wordlist': '/home/user/passwords.txt',
'bssid': 'AA:BB:CC:DD:EE:FF'
"capture_file": "/tmp/intercept_handshake_test.cap",
"wordlist": "/home/user/passwords.txt",
"bssid": "AA:BB:CC:DD:EE:FF",
}
response = client.post('/wifi/handshake/crack', json=payload)
response = client.post("/wifi/handshake/crack", json=payload)
data = response.get_json()
assert data['status'] == 'success'
assert data['password'] == 'secret123'
assert data["status"] == "success"
assert data["password"] == "secret123"
### --- DATA FETCHING TESTS --- ###
def test_get_wifi_networks(client, mock_app_module):
"""Test that the networks endpoint correctly formats internal data."""
mock_app_module.wifi_networks = {
'AA:BB:CC:DD:EE:FF': {'essid': 'Home-WiFi', 'bssid': 'AA:BB:CC:DD:EE:FF'}
}
mock_app_module.wifi_handshakes = ['AA:BB:CC:DD:EE:FF']
mock_app_module.wifi_networks = {"AA:BB:CC:DD:EE:FF": {"essid": "Home-WiFi", "bssid": "AA:BB:CC:DD:EE:FF"}}
mock_app_module.wifi_handshakes = ["AA:BB:CC:DD:EE:FF"]
response = client.get('/wifi/networks')
response = client.get("/wifi/networks")
data = response.get_json()
assert len(data['networks']) == 1
assert data['networks'][0]['essid'] == 'Home-WiFi'
assert 'AA:BB:CC:DD:EE:FF' in data['handshakes']
assert len(data["networks"]) == 1
assert data["networks"][0]["essid"] == "Home-WiFi"
assert "AA:BB:CC:DD:EE:FF" in data["handshakes"]