58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
"""Tests for history_sync task registration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
from app.tasks import history_sync
|
|
|
|
|
|
class TestHistorySyncTask:
|
|
async def test_register_schedules_job(self) -> None:
|
|
fake_scheduler = MagicMock()
|
|
class FakeState:
|
|
pass
|
|
|
|
class FakeSettings:
|
|
fail2ban_socket = "/tmp/fake.sock"
|
|
|
|
app = type("FakeApp", (), {})()
|
|
app.state = FakeState()
|
|
app.state.scheduler = fake_scheduler
|
|
app.state.settings = FakeSettings()
|
|
|
|
history_sync.register(app)
|
|
|
|
fake_scheduler.add_job.assert_called_once()
|
|
called_args, called_kwargs = fake_scheduler.add_job.call_args
|
|
assert called_kwargs["id"] == history_sync.JOB_ID
|
|
assert called_kwargs["kwargs"]["settings"] is app.state.settings
|
|
|
|
async def test_backfill_window_is_7_5_days(self) -> None:
|
|
assert history_sync.BACKFILL_WINDOW == 648000
|
|
|
|
async def test_run_sync_delegates_to_history_service(self) -> None:
|
|
fake_app = type("FakeApp", (), {})()
|
|
fake_app.state = type("FakeState", (), {})()
|
|
fake_app.state.settings = type("FakeSettings", (), {})()
|
|
fake_app.state.settings.fail2ban_socket = "/tmp/fake.sock"
|
|
fake_app.state.settings.database_path = "/tmp/fake.db"
|
|
fake_app.state.runtime_settings = None
|
|
|
|
fake_db = AsyncMock()
|
|
fake_db.close = AsyncMock()
|
|
|
|
with patch(
|
|
"app.tasks.db.open_db",
|
|
new_callable=AsyncMock,
|
|
return_value=fake_db,
|
|
), patch(
|
|
"app.tasks.history_sync.history_service.sync_from_fail2ban_db",
|
|
new_callable=AsyncMock,
|
|
return_value=42,
|
|
) as mock_sync:
|
|
await history_sync._run_sync(fake_app)
|
|
|
|
mock_sync.assert_awaited_once_with(fake_db, "/tmp/fake.sock")
|
|
fake_db.close.assert_awaited_once()
|