92554d4089
generate_keypair() used synchronous subprocess.run() which blocked the NiceGUI event loop during wg genkey/pubkey calls. This caused WebSocket disconnects, page reloads, and the config dialog never appearing after device creation. Switched to asyncio.create_subprocess_exec so the event loop stays responsive while waiting for the wg CLI.
54 lines
2 KiB
Python
54 lines
2 KiB
Python
"""End-to-end tests for device management UI using NiceGUI's User fixture."""
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
from nicegui.testing import User
|
|
|
|
from wiregui.models.user import User as UserModel
|
|
from tests.e2e.conftest import TEST_EMAIL, TEST_PASSWORD
|
|
|
|
# Fake WG keys for testing (valid base64, 32 bytes)
|
|
FAKE_PRIVATE_KEY = "YFake0PrivateKey00000000000000000000000000w="
|
|
FAKE_PUBLIC_KEY = "ZFake0PublicKey000000000000000000000000000w="
|
|
|
|
|
|
async def _login(user: User):
|
|
"""Helper to log in via the UI."""
|
|
await user.open("/login")
|
|
user.find("Email").type(TEST_EMAIL)
|
|
user.find("Password").type(TEST_PASSWORD)
|
|
user.find("Sign in").click()
|
|
await user.should_see("My Devices")
|
|
|
|
|
|
@pytest.mark.parametrize("user", [{"storage": {}}], indirect=True)
|
|
async def test_add_device_via_ui(user: User, test_user: UserModel):
|
|
"""Test the full flow: login → devices → add device → see it in table."""
|
|
with patch("wiregui.pages.devices.generate_keypair", new_callable=AsyncMock, return_value=(FAKE_PRIVATE_KEY, FAKE_PUBLIC_KEY)), \
|
|
patch("wiregui.pages.devices.generate_preshared_key", return_value="cHJlc2hhcmVkMDAwMDAwMDAwMDAwMDAwMDAwMDAwMDA="):
|
|
|
|
await _login(user)
|
|
|
|
# Open create dialog
|
|
user.find("Add Device").click()
|
|
await user.should_see("New Device")
|
|
|
|
# Fill device name and submit
|
|
user.find("Device Name").type("Test Laptop")
|
|
user.find("Create").click()
|
|
|
|
# Should see config dialog with the device config
|
|
await user.should_see("Test Laptop")
|
|
|
|
|
|
@pytest.mark.parametrize("user", [{"storage": {}}], indirect=True)
|
|
async def test_add_device_requires_name(user: User, test_user: UserModel):
|
|
"""Test that creating a device without a name shows an error."""
|
|
await _login(user)
|
|
|
|
# Open create dialog and submit without name
|
|
user.find("Add Device").click()
|
|
await user.should_see("New Device")
|
|
user.find("Create").click()
|
|
await user.should_see("Device name is required")
|