mirror of
https://github.com/jkingsman/Remote-Terminal-for-MeshCore.git
synced 2026-08-07 01:03:34 +02:00
Reorganize for great victory and move to blob for payload hasg
This commit is contained in:
@@ -5,8 +5,11 @@ import shutil
|
||||
import tempfile
|
||||
from pathlib import Path
|
||||
|
||||
import httpx
|
||||
import pytest
|
||||
|
||||
from app.database import Database
|
||||
|
||||
# Use an isolated file-backed SQLite DB for tests that import app.main/TestClient.
|
||||
# This must be set before app.config/app.database are imported, otherwise the global
|
||||
# Database instance will bind to the default runtime DB (data/meshcore.db).
|
||||
@@ -20,3 +23,52 @@ def cleanup_test_db_dir():
|
||||
"""Clean up temporary pytest DB directory after the test session."""
|
||||
yield
|
||||
shutil.rmtree(_TEST_DB_DIR, ignore_errors=True)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def test_db():
|
||||
"""Create an in-memory test database with schema + migrations."""
|
||||
from app.repository import channels, contacts, messages, raw_packets, settings
|
||||
|
||||
db = Database(":memory:")
|
||||
await db.connect()
|
||||
|
||||
submodules = [contacts, channels, messages, raw_packets, settings]
|
||||
originals = [(mod, mod.db) for mod in submodules]
|
||||
|
||||
for mod in submodules:
|
||||
mod.db = db
|
||||
|
||||
# Also patch the db reference used by the packets router for VACUUM
|
||||
import app.routers.packets as packets_module
|
||||
|
||||
original_packets_db = packets_module.db
|
||||
packets_module.db = db
|
||||
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
for mod, original in originals:
|
||||
mod.db = original
|
||||
packets_module.db = original_packets_db
|
||||
await db.disconnect()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def client():
|
||||
"""Create an httpx AsyncClient for testing the app."""
|
||||
from app.main import app
|
||||
|
||||
transport = httpx.ASGITransport(app=app)
|
||||
return httpx.AsyncClient(transport=transport, base_url="http://test")
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def captured_broadcasts():
|
||||
"""Capture WebSocket broadcasts for verification."""
|
||||
broadcasts = []
|
||||
|
||||
def mock_broadcast(event_type: str, data: dict):
|
||||
broadcasts.append({"type": event_type, "data": data})
|
||||
|
||||
return broadcasts, mock_broadcast
|
||||
|
||||
Reference in New Issue
Block a user