54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import sys, os
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
|
|
|
import pytest
|
|
from app import db
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup_db(tmp_path, monkeypatch):
|
|
test_db = tmp_path / "test.db"
|
|
monkeypatch.setattr(db, "DB_PATH", str(test_db))
|
|
db.init_db()
|
|
yield
|
|
|
|
|
|
def _seed_briefing(drw=1153):
|
|
picks = {
|
|
"core": [{"numbers": [1, 2, 3, 4, 5, 6], "risk_tag": "안정", "reason": "x"}] * 5,
|
|
"bonus": [{"numbers": [7, 8, 9, 10, 11, 12], "risk_tag": "균형", "reason": "x"}] * 5,
|
|
"extended": [{"numbers": [13, 14, 15, 16, 17, 18], "risk_tag": "공격", "reason": "x"}] * 5,
|
|
"pool": [{"numbers": [19, 20, 21, 22, 23, 24], "risk_tag": "안정", "reason": "x"}] * 5,
|
|
}
|
|
db.save_briefing({
|
|
"draw_no": drw, "picks": picks,
|
|
"narrative": {"headline": "h", "summary_3lines": ["a", "b", "c"]},
|
|
"confidence": 70, "model": "test",
|
|
})
|
|
|
|
|
|
def test_bulk_core_inserts_5():
|
|
_seed_briefing()
|
|
r = db.bulk_insert_purchases_from_briefing(1153, "core", 5000)
|
|
assert r["ok"] and r["sets"] == 5
|
|
rows = db.get_purchases(draw_no=1153)
|
|
assert len(rows) == 5
|
|
assert all(row["source_strategy"] == "curator_core" for row in rows)
|
|
|
|
|
|
def test_bulk_full_inserts_20():
|
|
_seed_briefing()
|
|
r = db.bulk_insert_purchases_from_briefing(1153, "full", 20000)
|
|
assert r["ok"] and r["sets"] == 20
|
|
|
|
|
|
def test_bulk_unknown_tier_mode():
|
|
_seed_briefing()
|
|
r = db.bulk_insert_purchases_from_briefing(1153, "garbage", 1000)
|
|
assert r["ok"] is False and "garbage" in r["reason"]
|
|
|
|
|
|
def test_bulk_no_briefing():
|
|
r = db.bulk_insert_purchases_from_briefing(9999, "core", 5000)
|
|
assert r["ok"] is False and "not found" in r["reason"]
|