[Phase 0 Reset]
- 清除舊版 app/、alembic/versions/、雜亂測試腳本
- 新 requirements.txt (移除 caldav/redis/keycloak-lib,加入 apscheduler/croniter/docker/paramiko/ping3/dnspython)
[Phase 1 資料庫]
- 9 張資料表 SQLAlchemy Models:tenants / accounts / schedules / schedule_logs /
tenant_schedule_results / account_schedule_results / servers / server_status_logs / system_status_logs
- Alembic migration 001_create_all_tables (已套用到 10.1.0.20:5433/virtual_mis)
- seed.py:schedules 初始 3 筆 / servers 初始 4 筆
[Phase 2 CRUD API]
- GET/POST/PUT/DELETE: /api/v1/tenants / accounts / servers / schedules
- /api/v1/system-status
- 帳號編碼自動產生 (prefix + seq_no 4碼左補0)
- 燈號 (lights) 從最新排程結果取得
[Phase 3 Watchdog]
- APScheduler interval 3分鐘,原子 UPDATE status=Going 防重複執行
- 手動觸發 API: POST /api/v1/schedules/{id}/run
[Phase 4 Service Clients]
- KeycloakClient:vmis-admin realm,REST API (不用 python-keycloak)
- MailClient:Docker Mailserver @ 10.1.0.254:8080,含 MX DNS 驗證
- DockerClient:docker-py 本機 + paramiko SSH 遠端 compose
- NextcloudClient:OCS API user/quota
- SystemChecker:功能驗證 (traefik routers>0 / keycloak token / SMTP EHLO / DB SELECT 1 / ping)
[TDD]
- 37 tests / 37 passed (2.11s)
- SQLite in-memory + StaticPool,無需外部 DB
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
"""TDD: Schedule API tests"""
|
|
from datetime import datetime
|
|
|
|
|
|
def _seed_schedules(db):
|
|
"""Insert initial schedule records for tests"""
|
|
from app.models.schedule import Schedule
|
|
from croniter import croniter
|
|
schedules = [
|
|
Schedule(id=1, name="租戶檢查", cron_timer="0 */3 * * * *",
|
|
status="Waiting", next_run_at=croniter("0 */3 * * * *", datetime.utcnow()).get_next(datetime)),
|
|
Schedule(id=2, name="帳號檢查", cron_timer="0 */3 * * * *",
|
|
status="Waiting", next_run_at=croniter("0 */3 * * * *", datetime.utcnow()).get_next(datetime)),
|
|
Schedule(id=3, name="系統狀態", cron_timer="0 0 8 * * *",
|
|
status="Waiting", next_run_at=croniter("0 0 8 * * *", datetime.utcnow()).get_next(datetime)),
|
|
]
|
|
for s in schedules:
|
|
db.add(s)
|
|
db.commit()
|
|
|
|
|
|
def test_list_schedules(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.get("/api/v1/schedules")
|
|
assert resp.status_code == 200
|
|
schedules = resp.json()
|
|
assert len(schedules) == 3
|
|
names = [s["name"] for s in schedules]
|
|
assert "租戶檢查" in names
|
|
assert "帳號檢查" in names
|
|
assert "系統狀態" in names
|
|
|
|
|
|
def test_get_schedule(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.get("/api/v1/schedules/1")
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["id"] == 1
|
|
assert data["name"] == "租戶檢查"
|
|
assert data["status"] == "Waiting"
|
|
assert data["cron_timer"] == "0 */3 * * * *"
|
|
assert data["next_run_at"] is not None
|
|
|
|
|
|
def test_get_schedule_not_found(client, db):
|
|
resp = client.get("/api/v1/schedules/99")
|
|
assert resp.status_code == 404
|
|
|
|
|
|
def test_update_schedule_cron(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.put("/api/v1/schedules/1", json={"cron_timer": "0 */5 * * * *"})
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["cron_timer"] == "0 */5 * * * *"
|
|
assert data["next_run_at"] is not None
|
|
|
|
|
|
def test_update_schedule_invalid_cron(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.put("/api/v1/schedules/1", json={"cron_timer": "invalid cron"})
|
|
assert resp.status_code == 422
|
|
|
|
|
|
def test_manual_run_schedule(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.post("/api/v1/schedules/1/run")
|
|
assert resp.status_code == 202
|
|
data = resp.json()
|
|
assert data["schedule_id"] == 1
|
|
|
|
|
|
def test_manual_run_already_going(client, db):
|
|
_seed_schedules(db)
|
|
from app.models.schedule import Schedule
|
|
s = db.get(Schedule, 1)
|
|
s.status = "Going"
|
|
db.commit()
|
|
resp = client.post("/api/v1/schedules/1/run")
|
|
assert resp.status_code == 409
|
|
|
|
|
|
def test_schedule_logs_empty(client, db):
|
|
_seed_schedules(db)
|
|
resp = client.get("/api/v1/schedules/1/logs")
|
|
assert resp.status_code == 200
|
|
assert resp.json() == []
|