gahusb
c5303151c0
- LottoAgent.sync_evolver_activity(): lotto-lab evolver status polling → agent_office.db task+log 미러링 - UTC 날짜 기준 멱등 guard (get_tasks_by_agent_date_kind 활용) - 일요일(dow=6) → 5 clamp (lotto-lab trials는 0~5) - 월요일 6-trial 완성 시 evolver_generate task 추가 생성 - scheduler.py: lotto_evolver_activity_sync cron 09:30 등록 - tests: creates_apply_task / idempotent / no_picks_no_task 3종 Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
122 lines
3.5 KiB
Python
122 lines
3.5 KiB
Python
# agent-office/tests/test_sync_evolver_activity.py
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
import gc
|
|
from datetime import datetime, timezone, timedelta
|
|
|
|
_fd, _TMP = tempfile.mkstemp(suffix=".db")
|
|
os.close(_fd)
|
|
os.unlink(_TMP)
|
|
os.environ["AGENT_OFFICE_DB_PATH"] = _TMP
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import pytest
|
|
from app import db
|
|
db.DB_PATH = _TMP
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fresh_db():
|
|
gc.collect()
|
|
if os.path.exists(_TMP):
|
|
os.remove(_TMP)
|
|
db.init_db()
|
|
yield
|
|
gc.collect()
|
|
if os.path.exists(_TMP):
|
|
try:
|
|
os.remove(_TMP)
|
|
except PermissionError:
|
|
pass
|
|
|
|
|
|
def _today_dow_clamped():
|
|
"""오늘의 weekday() (일요일=6은 5로 clamp)."""
|
|
KST = timezone(timedelta(hours=9))
|
|
dow = datetime.now(KST).weekday()
|
|
return 5 if dow == 6 else dow
|
|
|
|
|
|
def _fake_status_with_picks(dow_with_picks):
|
|
async def fake():
|
|
return {
|
|
"week_start": "2026-05-18",
|
|
"current_base": [0.2] * 5,
|
|
"trials": [
|
|
{
|
|
"id": 100 + i,
|
|
"day_of_week": i,
|
|
"weight": [0.2] * 5,
|
|
"source": "perturb",
|
|
"picks": ([
|
|
{"id": j, "numbers": [1,2,3,4,5,6], "meta_score": 0.5}
|
|
for j in range(5)
|
|
] if i == dow_with_picks else []),
|
|
}
|
|
for i in range(6)
|
|
],
|
|
}
|
|
return fake
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_evolver_activity_creates_apply_task(monkeypatch):
|
|
"""오늘 trial에 picks가 있으면 evolver_apply task 1개 생성."""
|
|
from app.agents.lotto import LottoAgent
|
|
from app import service_proxy
|
|
|
|
dow = _today_dow_clamped()
|
|
monkeypatch.setattr(service_proxy, "lotto_evolver_status", _fake_status_with_picks(dow))
|
|
|
|
agent = LottoAgent()
|
|
await agent.sync_evolver_activity()
|
|
|
|
apply_tasks = db.get_agent_tasks("lotto", task_type="evolver_apply", days=1)
|
|
assert len(apply_tasks) == 1
|
|
assert apply_tasks[0]["result_data"]["n_picks"] == 5
|
|
assert apply_tasks[0]["input_data"]["day_of_week"] == dow
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_evolver_activity_idempotent(monkeypatch):
|
|
"""같은 날 두 번 호출해도 task는 1개만 (멱등)."""
|
|
from app.agents.lotto import LottoAgent
|
|
from app import service_proxy
|
|
|
|
dow = _today_dow_clamped()
|
|
monkeypatch.setattr(service_proxy, "lotto_evolver_status", _fake_status_with_picks(dow))
|
|
|
|
agent = LottoAgent()
|
|
await agent.sync_evolver_activity()
|
|
await agent.sync_evolver_activity()
|
|
|
|
apply_tasks = db.get_agent_tasks("lotto", task_type="evolver_apply", days=1)
|
|
assert len(apply_tasks) == 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_evolver_activity_no_picks_no_task(monkeypatch):
|
|
"""오늘 trial에 picks가 없으면 task 생성하지 않음."""
|
|
from app.agents.lotto import LottoAgent
|
|
from app import service_proxy
|
|
|
|
async def fake_status():
|
|
return {
|
|
"week_start": "2026-05-18",
|
|
"current_base": [0.2] * 5,
|
|
"trials": [
|
|
{"id": 100 + i, "day_of_week": i, "weight": [0.2]*5,
|
|
"source": "perturb", "picks": []}
|
|
for i in range(6)
|
|
],
|
|
}
|
|
monkeypatch.setattr(service_proxy, "lotto_evolver_status", fake_status)
|
|
|
|
agent = LottoAgent()
|
|
await agent.sync_evolver_activity()
|
|
|
|
apply_tasks = db.get_agent_tasks("lotto", task_type="evolver_apply", days=1)
|
|
assert len(apply_tasks) == 0
|