Codex Release Bot
78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
"""Scheduled nightly agent install."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from server.application.services import agent_install_schedule as sched
|
|
from server.background.schedule_runner import _cron_match
|
|
|
|
|
|
def test_cron_matches_beijing_2340():
|
|
# 2026-06-17 23:40 Beijing = 2026-06-17 15:40 UTC
|
|
at = datetime(2026, 6, 17, 15, 40, tzinfo=timezone.utc)
|
|
assert _cron_match("40 23 * * *", at) is True
|
|
|
|
|
|
def test_cron_not_match_2340_at_other_minute():
|
|
at = datetime(2026, 6, 17, 15, 41, tzinfo=timezone.utc)
|
|
assert _cron_match("40 23 * * *", at) is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_scheduled_skips_without_servers(monkeypatch):
|
|
monkeypatch.setattr(sched.settings, "AGENT_INSTALL_SCHEDULE_ENABLED", "true")
|
|
monkeypatch.setattr(sched.settings, "API_BASE_URL", "https://api.example.com")
|
|
monkeypatch.setattr(sched, "list_no_agent_server_ids", AsyncMock(return_value=[]))
|
|
monkeypatch.setattr(sched, "_already_ran_today", AsyncMock(return_value=False))
|
|
monkeypatch.setattr(sched, "_install_agent_batch_running", AsyncMock(return_value=False))
|
|
mark = AsyncMock()
|
|
monkeypatch.setattr(sched, "_mark_ran_today", mark)
|
|
|
|
result = await sched.run_scheduled_agent_install()
|
|
|
|
assert result["action"] == "skipped"
|
|
assert result["reason"] == "no_servers_without_agent"
|
|
mark.assert_awaited_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_scheduled_starts_install_batch(monkeypatch):
|
|
monkeypatch.setattr(sched.settings, "AGENT_INSTALL_SCHEDULE_ENABLED", "true")
|
|
monkeypatch.setattr(sched.settings, "API_BASE_URL", "https://api.example.com")
|
|
monkeypatch.setattr(sched, "list_no_agent_server_ids", AsyncMock(return_value=[10, 11]))
|
|
monkeypatch.setattr(sched, "_already_ran_today", AsyncMock(return_value=False))
|
|
monkeypatch.setattr(sched, "_install_agent_batch_running", AsyncMock(return_value=False))
|
|
monkeypatch.setattr(sched, "_mark_ran_today", AsyncMock())
|
|
|
|
start = AsyncMock(return_value={"job_id": 42, "label": "批量安装 Agent（定时）"})
|
|
monkeypatch.setattr(
|
|
"server.application.services.server_batch_service.start_batch_job",
|
|
start,
|
|
)
|
|
|
|
result = await sched.run_scheduled_agent_install()
|
|
|
|
assert result["action"] == "started"
|
|
assert result["job_id"] == 42
|
|
start.assert_awaited_once_with(
|
|
op="install-agent",
|
|
server_ids=[10, 11],
|
|
operator="system",
|
|
params={"scheduled": True, "cron": "40 23 * * *"},
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_scheduled_skips_without_api_base_url(monkeypatch):
|
|
monkeypatch.setattr(sched.settings, "AGENT_INSTALL_SCHEDULE_ENABLED", "true")
|
|
monkeypatch.setattr(sched.settings, "API_BASE_URL", "")
|
|
monkeypatch.setattr(sched, "_already_ran_today", AsyncMock(return_value=False))
|
|
|
|
result = await sched.run_scheduled_agent_install()
|
|
|
|
assert result == {"action": "skipped", "reason": "api_base_url_unconfigured"}
|