Hyungi Ahn
99672292d3
프로덕션 컨테이너는 /app 을 cwd 로 실행하고 import 는 `from api...`, `from core...`, `from workers...` 처럼 무접두 스타일을 사용한다. PR-A 내부 import 가 `from app.policy...`, `from app.ai.envelope` 로 되어 있어서 컨테이너에서 ModuleNotFoundError 발생. 변경: - app/policy/*.py: `from app.policy.X` → `from policy.X` - app/services/prompt_versions.py: lazy import 도 `from policy.prompt_render` - app/ai/envelope.py: 영향 없음 (내부 import 없음) - tests/policy/*.py: 모두 `from policy.X` / `from ai.envelope` 로 통일 - tests/policy/conftest.py: 로컬 pytest 용 sys.path.insert(app/) 추가 (MacBook 에서 repo-root 기준 실행 시 app/ 를 package root 로 취급) CI: pytest tests/policy/ -q → 98 passed (로컬, 동일 결과) 프로덕션: docker exec fastapi python -c "from policy.loader import load_policy" → OK Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
88 lines
2.4 KiB
Python
88 lines
2.4 KiB
Python
"""InMemoryShadowLogger 동작 + Protocol 계약."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from policy.routing import RoutingDecision, decide_routing
|
|
from policy.shadow import InMemoryShadowLogger, ShadowLogger
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_decision(policy) -> RoutingDecision:
|
|
return decide_routing(
|
|
subject_domain="safety_reference",
|
|
content_chars=1000,
|
|
self_declared_high_impact=False,
|
|
confidence=0.95,
|
|
policy=policy,
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_inmem_logger_records(sample_decision):
|
|
logger = InMemoryShadowLogger()
|
|
await logger.record_would_route(
|
|
doc_id="doc-001",
|
|
decision=sample_decision,
|
|
actual_model_used="4B",
|
|
prompt_version="v1-abc",
|
|
policy_version="hash-1234",
|
|
)
|
|
assert logger.count() == 1
|
|
rec = logger.records[0]
|
|
assert rec.doc_id == "doc-001"
|
|
assert rec.decision == sample_decision
|
|
assert rec.actual_model_used == "4B"
|
|
assert rec.prompt_version == "v1-abc"
|
|
assert rec.policy_version == "hash-1234"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_inmem_logger_multiple(sample_decision):
|
|
logger = InMemoryShadowLogger()
|
|
for i in range(5):
|
|
await logger.record_would_route(
|
|
doc_id=f"doc-{i}",
|
|
decision=sample_decision,
|
|
actual_model_used="4B",
|
|
prompt_version="v1",
|
|
policy_version="h",
|
|
)
|
|
assert logger.count() == 5
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_inmem_logger_clear(sample_decision):
|
|
logger = InMemoryShadowLogger()
|
|
await logger.record_would_route(
|
|
doc_id="doc-1",
|
|
decision=sample_decision,
|
|
actual_model_used="4B",
|
|
prompt_version="v1",
|
|
policy_version="h",
|
|
)
|
|
logger.clear()
|
|
assert logger.count() == 0
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_inmem_logger_extra_payload(sample_decision):
|
|
logger = InMemoryShadowLogger()
|
|
await logger.record_would_route(
|
|
doc_id="doc-1",
|
|
decision=sample_decision,
|
|
actual_model_used="4B",
|
|
prompt_version="v1",
|
|
policy_version="h",
|
|
extra={"latency_ms": 120, "note": "test"},
|
|
)
|
|
rec = logger.records[0]
|
|
assert rec.extra == {"latency_ms": 120, "note": "test"}
|
|
|
|
|
|
def test_inmem_logger_satisfies_protocol():
|
|
"""InMemoryShadowLogger 가 ShadowLogger Protocol 을 만족."""
|
|
logger = InMemoryShadowLogger()
|
|
assert isinstance(logger, ShadowLogger)
|