genichinandClaude Opus 4.8
0b7db66463
데몬이 시스템 UTC 날짜로 슬롯을 1회만 preallocate 해 trading_date 가 KST 캘린더와 어긋나고 재시작 없이는 다음날 슬롯이 안 생기던 문제(issue #14) 정정. 거버넌스: 전부 change-req (신규 REQ 없음, dcy-014 OQ-3; 스케줄러는 REQ-030 패턴 준용). - REQ-005: trading_date 를 설정 타임존(기본 Asia/Seoul) 기준 계산(timestamp 는 UTC 유지). DailyPreallocationScheduler 신규 — TZ 일자당 1회 멱등 fire, 재시작 없이 다음날 슬롯 자동 생성. setup() 배선 + trex-preallocation 스레드 - REQ-048: TOML trading_timezone/daily_preallocation_time 키 + from_config 전파 - REQ-009: runtime-flows/data-model 정합 + instance.example.toml OQ-1 기본 00:00(TZ 자정, TOML 설정 가능) / OQ-2 매 캘린더일 생성(거래일 캘린더 후속). 멱등 preallocate 로 중복 tick·다중 인스턴스 중복 생성 0. 검증: 단위 302 passed/32 skipped, 인프라 게이트(PG+RabbitMQ) 334 passed, ruff clean. 신규 test_preallocation_scheduler(고정 clock) + test_instance_config 확장. BAT-018 release-candidate. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
90 lines
3.4 KiB
Python
90 lines
3.4 KiB
Python
"""Per-instance TOML config loader tests (REQ-048 / BAT-013)."""
|
|
from __future__ import annotations
|
|
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from trex_core.runtime.config import DaemonConfig, DaemonInstanceConfig
|
|
|
|
_REPO = Path(__file__).resolve().parents[2]
|
|
|
|
|
|
def _write_toml(tmp_path, body: str) -> str:
|
|
p = tmp_path / "inst.toml"
|
|
p.write_text(body)
|
|
return str(p)
|
|
|
|
|
|
def test_from_toml_parses(tmp_path):
|
|
"""AC-1: from_toml reads instance_key + portfolio_hub_token."""
|
|
path = _write_toml(tmp_path, 'instance_key = "inst-1"\nportfolio_hub_token = "tok"\n')
|
|
cfg = DaemonInstanceConfig.from_toml(path)
|
|
assert cfg.instance_key == "inst-1"
|
|
assert cfg.portfolio_hub_token == "tok"
|
|
|
|
|
|
def test_from_toml_missing_instance_key_raises(tmp_path):
|
|
"""AC-4: missing required instance_key → error (no silent default)."""
|
|
path = _write_toml(tmp_path, 'portfolio_hub_token = "tok"\n')
|
|
with pytest.raises(RuntimeError):
|
|
DaemonInstanceConfig.from_toml(path)
|
|
|
|
|
|
def test_from_config_injects_token(tmp_path, monkeypatch):
|
|
"""AC-3 / S-5: TOML token is injected into PortfolioHubConfig when base_url is set,
|
|
even with TREX_PORTFOLIO_HUB_TOKEN absent from the environment."""
|
|
monkeypatch.setenv("TREX_POSTGRESQL_DSN", "postgresql://x/y")
|
|
monkeypatch.setenv("TREX_PORTFOLIO_HUB_BASE_URL", "http://hub:18000")
|
|
monkeypatch.delenv("TREX_PORTFOLIO_HUB_TOKEN", raising=False)
|
|
path = _write_toml(tmp_path, 'instance_key = "inst-9"\nportfolio_hub_token = "toml-tok"\n')
|
|
cfg = DaemonConfig.from_config(path)
|
|
assert cfg.instance_key == "inst-9"
|
|
assert cfg.portfolio_hub is not None
|
|
assert cfg.portfolio_hub.token == "toml-tok"
|
|
|
|
|
|
def test_default_instance_key_when_from_env(monkeypatch):
|
|
monkeypatch.setenv("TREX_POSTGRESQL_DSN", "postgresql://x/y")
|
|
assert DaemonConfig.from_env().instance_key == "default"
|
|
|
|
|
|
def test_trading_timezone_defaults_to_kst(tmp_path, monkeypatch):
|
|
"""dcy-014 / REQ-048: trading TZ + daily trigger default to KST / midnight."""
|
|
monkeypatch.setenv("TREX_POSTGRESQL_DSN", "postgresql://x/y")
|
|
inst = DaemonInstanceConfig.from_toml(_write_toml(tmp_path, 'instance_key = "i1"\n'))
|
|
assert inst.trading_timezone == "Asia/Seoul"
|
|
assert inst.daily_preallocation_time == "00:00"
|
|
assert DaemonConfig.from_env().trading_timezone == "Asia/Seoul"
|
|
|
|
|
|
def test_trading_timezone_from_toml_propagates(tmp_path, monkeypatch):
|
|
"""dcy-014 / REQ-048: TOML trading_timezone + trigger flow into DaemonConfig."""
|
|
monkeypatch.setenv("TREX_POSTGRESQL_DSN", "postgresql://x/y")
|
|
path = _write_toml(
|
|
tmp_path,
|
|
'instance_key = "i2"\ntrading_timezone = "UTC"\ndaily_preallocation_time = "07:30"\n',
|
|
)
|
|
cfg = DaemonConfig.from_config(path)
|
|
assert cfg.trading_timezone == "UTC"
|
|
assert cfg.daily_preallocation_time == "07:30"
|
|
|
|
|
|
def test_config_required_cli_exits_nonzero():
|
|
"""AC-2: daemon entry without --config exits non-zero (argparse required)."""
|
|
proc = subprocess.run(
|
|
[sys.executable, "-m", "trex_core.runtime.daemon"],
|
|
capture_output=True, text=True, timeout=30, cwd=str(_REPO),
|
|
)
|
|
assert proc.returncode != 0
|
|
assert "config" in (proc.stderr + proc.stdout).lower()
|
|
|
|
|
|
def test_gitignore_excludes_instance_toml():
|
|
"""AC-5: .gitignore excludes instance TOML but keeps the example."""
|
|
gi = (_REPO / ".gitignore").read_text()
|
|
assert "/config/*.toml" in gi
|
|
assert "!/config/instance.example.toml" in gi
|