a012635dff
Remove 7 test files fully covered by e2e tests (admin, account, models, API routes, integration MFA/OIDC, notifications). Trim 5 more files to keep only edge cases not reachable via e2e. Fix conftest to replace wiregui.db engine/session at import time so all code uses the test database. Use session-scoped tables with per-test savepoint isolation to prevent data leaking between tests.
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
"""Tests for services — WireGuard event error handling and rule events."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
from wiregui.models.device import Device
|
|
from wiregui.models.rule import Rule
|
|
from wiregui.services.events import on_device_created, on_device_deleted, on_device_updated, on_rule_created
|
|
|
|
|
|
def _make_device(**kwargs) -> Device:
|
|
defaults = dict(
|
|
name="test",
|
|
public_key="pk-test",
|
|
preshared_key="psk-test",
|
|
ipv4="10.3.2.5",
|
|
ipv6="fd00::3:2:5",
|
|
user_id="00000000-0000-0000-0000-000000000000",
|
|
)
|
|
defaults.update(kwargs)
|
|
return Device(**defaults)
|
|
|
|
|
|
# --- Events (WG disabled) ---
|
|
|
|
|
|
@patch("wiregui.services.events.get_settings")
|
|
@patch("wiregui.services.events.wireguard")
|
|
async def test_events_skip_when_wg_disabled(mock_wg, mock_settings):
|
|
mock_settings.return_value.wg_enabled = False
|
|
mock_wg.add_peer = AsyncMock()
|
|
mock_wg.remove_peer = AsyncMock()
|
|
|
|
device = _make_device()
|
|
await on_device_created(device)
|
|
await on_device_deleted(device)
|
|
await on_device_updated(device)
|
|
|
|
mock_wg.add_peer.assert_not_awaited()
|
|
mock_wg.remove_peer.assert_not_awaited()
|
|
|
|
|
|
# --- Events (WG error handling) ---
|
|
|
|
|
|
@patch("wiregui.services.events.get_settings")
|
|
@patch("wiregui.services.events.firewall")
|
|
@patch("wiregui.services.events.wireguard")
|
|
async def test_on_device_created_handles_wg_error(mock_wg, mock_fw, mock_settings):
|
|
mock_settings.return_value.wg_enabled = True
|
|
mock_wg.add_peer = AsyncMock(side_effect=RuntimeError("wg failed"))
|
|
mock_fw.add_device_jump_rule = AsyncMock()
|
|
|
|
device = _make_device()
|
|
# Should not raise — error is logged
|
|
await on_device_created(device)
|
|
|
|
|
|
# --- Rule events ---
|
|
|
|
|
|
@patch("wiregui.services.events.get_settings")
|
|
@patch("wiregui.services.events.firewall")
|
|
async def test_on_rule_created_calls_apply_rule(mock_fw, mock_settings):
|
|
mock_settings.return_value.wg_enabled = True
|
|
mock_fw.apply_rule = AsyncMock()
|
|
|
|
rule = Rule(
|
|
action="accept",
|
|
destination="10.0.0.0/8",
|
|
port_type="tcp",
|
|
port_range="80",
|
|
user_id="00000000-0000-0000-0000-000000000000",
|
|
)
|
|
await on_rule_created(rule)
|
|
|
|
mock_fw.apply_rule.assert_awaited_once_with(
|
|
"00000000-0000-0000-0000-000000000000", "10.0.0.0/8", "accept", "tcp", "80",
|
|
)
|