trade_executor - 종목당 4% 상한을 기존 보유분 포함 누적으로 (sizing_core.remaining_position_room) - 추천 경로는 스윙 소유 행만 매수·매도 (external/manual/daytrading/us_swing skip, 대칭), 고아 DT 행 warning - 일일 매수한도는 스윙 소유 매수만 집계, 집계 실패 시 fail-closed - 부분매도 반복 차단에서만 DT 매도 제외 (재매수 쿨다운은 유지) - 체결 확인: 부분체결 잔량 자동 재주문 제거(가드 전부 우회·취소/거부 시 발동), PARTIAL→COMPLETED+체결수량, 초과체결 캡 korea_investment_api - get_order_execution 이 존재하지 않는 /trading/inquire-ccnl 을 호출하던 결함 → inquire-daily-ccld (TTTC0081R 우선, 구 TR 폴백, 페이지네이션, ODNO 매칭, EXCG_ID_DVSN_CD=KRX), 실패 로그 warning daytrading - 시간손절은 loss_tracker 중립 (42일 밴이 스윙까지 막던 문제) - 실현손익을 수수료·거래세 차감 순손익으로 (trade_cost_core), 체결가 백그라운드 보정 (fill_reconciler) - legacy _record_trade strategy_type=daytrading 명시 db_worker - update_trade_history_by_order_id 를 당일(KST)+최신 1행으로 한정 (일별 재사용 ODNO 로 과거 행 덮어쓰기 방지) alembic 0013: trade_journals strategy_name='unknown' 을 trade_history.strategy_type 으로 보정 (백업 테이블 포함) CLAUDE.md: 변경 동작 반영, MT 예약 리비전 0013→0014 998 passed Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
214 lines
10 KiB
Python
214 lines
10 KiB
Python
"""유닛 테스트 계층 공용 안전망 — 실제 KIS 호출·라이브 DB를 구조적으로 차단한다.
|
|
|
|
## 왜 필요한가
|
|
파일별 방어(각 테스트가 자기 목을 다는 방식)는 두 번 실패했다.
|
|
|
|
1. `test_daytrading_stale_position_guard.py` — 리팩터링으로 `close_stale_positions()`가
|
|
`_execute_exit` → `_execute_exit_stale` 로 바뀌자 목이 빗나가 **실제 KIS 매도 주문이
|
|
2회 전송**됐다.
|
|
2. `test_daytrading_orb_watchlist_and_capital_split.py` — `get_balance`만 목킹하고
|
|
`get_buyable_cash()`는 빠뜨려 `_get_access_token → _request_new_token` 으로 내려갔고,
|
|
`korea_investment_api._handle_rate_limit()`의 "토큰 발급 1분 제한: 69.8초 대기"
|
|
(`asyncio.sleep`)에 걸려 테스트 하나가 140초를 소비했다. 소켓을 막아도 예외가
|
|
재시도 로직에 삼켜진 뒤 잠들기 때문에 느려지는 것은 그대로다.
|
|
|
|
프로젝트 메모리에는 멀티테넌트 1단계 작업 중 라이브 경로 호출로 **모의투자 1주 오발주**
|
|
사고가 기록돼 있다. 규약은 "격리 Postgres(`_test` 접미사) + KIS mock"인데 유닛 계층에는
|
|
`conftest.py`가 없어 강제 수단이 전혀 없었다.
|
|
|
|
## 설계 원칙
|
|
- **그럴듯한 값을 반환하지 않는다.** 모든 차단 대상은 호출되는 즉시 예외를 던진다.
|
|
조용히 성공하는 스텁은 목킹 누락을 감추고 테스트를 거짓 통과시킨다.
|
|
- 예외는 `Exception`이 아니라 **`BaseException` 파생**(`KISCallBlocked`)이다.
|
|
프로덕션 코드 곳곳의 `except Exception:` 재시도/폴백 블록이 삼켜버리면 차단이 무력화되고
|
|
(위 사례 2처럼) 오히려 느려지기만 한다. `BaseException`이면 통과해서 테스트를 실패시킨다.
|
|
- **차단은 클래스 메서드에 적용**한다(`KoreaInvestmentAPI.<method>`). 인스턴스가 아니라
|
|
클래스를 막아야 `from ... import korea_investment_api`로 싱글톤 참조를 미리 캡처한
|
|
모듈(예: `daytrading/entry_service`, `api/endpoints/portfolio`)이나 멀티테넌트에서 새로
|
|
만드는 인스턴스까지 전부 덮인다. 개별 테스트의 `patch.object(instance, ...)`는 인스턴스
|
|
속성으로 클래스 속성을 가리므로 **테스트 쪽 목이 항상 이긴다**(autouse 픽스처가 먼저
|
|
적용되고 테스트 내 patch가 나중에 적용되는 순서).
|
|
- **시세·잔고 계열도 차단**한다. 전역 기본 스텁을 주면 기존 테스트의 기대값이 조용히
|
|
바뀌므로, 값을 주는 대신 막고 각 테스트가 명시적으로 덮게 한다.
|
|
- `tests/integration/`은 제외한다(격리 컨테이너 Postgres + 자체 목 규약, 별도 conftest).
|
|
|
|
## 사용법
|
|
cd backend
|
|
DATABASE_URL="postgresql+asyncpg://u:p@127.0.0.1:5432/autostock_test" \
|
|
python -m pytest tests/ --ignore=tests/integration
|
|
|
|
`--allow-network` 를 주면 KIS 차단을 끈다(실 연동 점검 전용, 기본 꺼짐).
|
|
"""
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# 차단 예외 — `except Exception:` 에 삼켜지지 않도록 BaseException 파생
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
class KISCallBlocked(BaseException):
|
|
"""테스트 중 실제 KIS/주문 경로가 호출됨 — 해당 테스트의 목킹이 누락됐다는 뜻."""
|
|
|
|
|
|
def _blocked(dotted_name: str):
|
|
"""호출 즉시 실패하는 async 스텁 — 값을 반환하지 않는다(거짓 통과 방지)."""
|
|
|
|
async def _boom(*args, **kwargs):
|
|
raise KISCallBlocked(
|
|
f"[tests/conftest.py] 실호출 차단: {dotted_name}() 가 목킹되지 않은 채 호출됐습니다. "
|
|
f"해당 테스트에서 patch.object(...)로 명시적으로 덮으십시오. "
|
|
f"(실 연동 점검이라면 tests/integration/ 으로 옮기거나 --allow-network 사용)"
|
|
)
|
|
|
|
_boom.__name__ = dotted_name.rsplit(".", 1)[-1]
|
|
_boom.__qualname__ = dotted_name
|
|
return _boom
|
|
|
|
|
|
# 차단 대상 — 추측이 아니라 코드에서 정의 위치를 확인한 목록 (2026-09-27 기준).
|
|
#
|
|
# app/services/korea_investment_api.py
|
|
# 174 class KoreaInvestmentAPI
|
|
# 410 async def _request_new_token
|
|
# 461 async def _get_access_token
|
|
# 632 async def get_current_price
|
|
# 652 async def get_daily_chart
|
|
# 801 async def get_overseas_price
|
|
# 822 async def get_minute_chart
|
|
# 904 async def get_minute_chart_full_day
|
|
# 1001 async def get_overseas_daily_chart
|
|
# 1044 async def place_overseas_order
|
|
# 1087 async def get_balance
|
|
# 1164 async def get_buyable_cash
|
|
# 1202 async def get_overseas_balance
|
|
# 1234 async def place_order
|
|
# app/services/trade_executor.py
|
|
# 260 async def safe_place_order
|
|
# 334 async def safe_place_order_v2
|
|
_KIS_CLASS_METHODS = (
|
|
# ── 주문 계열 (실발주 위험) ──
|
|
"place_order",
|
|
"place_overseas_order",
|
|
# ── 토큰 계열 (rate-limit sleep 70초의 근원) ──
|
|
"_request_new_token",
|
|
"_get_access_token",
|
|
# ── 시세·잔고 계열 ──
|
|
"get_current_price",
|
|
"get_daily_chart",
|
|
"get_overseas_price",
|
|
"get_overseas_daily_chart",
|
|
"get_minute_chart",
|
|
"get_minute_chart_full_day",
|
|
"get_minute_chart_full_day_v2", # 진입 판정 경로(전일 워밍업 포함) — 2026-09-29
|
|
"get_balance",
|
|
"get_buyable_cash",
|
|
"get_overseas_balance",
|
|
# ── 체결 조회 계열 (2026-10-01: DT 체결가 확인이 백그라운드로 호출) ──
|
|
"get_todays_orders",
|
|
"get_order_execution",
|
|
)
|
|
|
|
_TRADE_EXECUTOR_FUNCS = (
|
|
"safe_place_order",
|
|
"safe_place_order_v2",
|
|
)
|
|
|
|
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
# pytest 훅
|
|
# ─────────────────────────────────────────────────────────────────────────────
|
|
def pytest_addoption(parser):
|
|
parser.addoption(
|
|
"--allow-network",
|
|
action="store_true",
|
|
default=False,
|
|
help="KIS 실호출 차단을 해제한다 (실 연동 점검 전용 — 라이브 계좌 주문 위험).",
|
|
)
|
|
|
|
|
|
def pytest_configure(config):
|
|
"""라이브 DB 가드 — DB 이름이 `_test`로 끝나지 않으면 세션 자체를 중단.
|
|
|
|
`tests/integration/conftest.py` 도 동일한 가드를 import 시점에 수행한다. 여기서는
|
|
유닛 계층까지 포함해 세션 전체에 적용하되, 판정 기준(`settings.DATABASE_URL`의 DB 이름)이
|
|
같으므로 integration 단독 실행의 기존 동작은 바뀌지 않는다. env만 보던 방식은
|
|
env 미설정 시 `settings.DATABASE_URL` 기본값(라이브 가능성)으로 조용히 새는 결함이 있어
|
|
`settings`를 기준으로 본다.
|
|
"""
|
|
from app.core.config import settings
|
|
|
|
db_url = settings.DATABASE_URL or os.environ.get("DATABASE_URL") or ""
|
|
db_name = db_url.rsplit("/", 1)[-1].split("?", 1)[0]
|
|
if not db_name.endswith("_test"):
|
|
pytest.exit(
|
|
"settings.DATABASE_URL의 DB 이름이 '_test'로 끝나지 않습니다"
|
|
f"(현재: {db_name!r}). 테스트는 반드시 격리 DB(예: autostock_test)를 대상으로 "
|
|
"실행해야 합니다. 예:\n"
|
|
' DATABASE_URL="postgresql+asyncpg://u:p@127.0.0.1:5432/autostock_test" '
|
|
"python -m pytest tests/ --ignore=tests/integration",
|
|
returncode=1,
|
|
)
|
|
|
|
|
|
def _is_integration(request) -> bool:
|
|
"""`tests/integration/` 하위이거나 `integration` 마커가 붙은 테스트인가."""
|
|
try:
|
|
path = str(request.node.fspath)
|
|
except Exception: # pragma: no cover - fspath 미지원 노드 방어
|
|
path = ""
|
|
if f"{os.sep}integration{os.sep}" in path:
|
|
return True
|
|
return request.node.get_closest_marker("integration") is not None
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _block_live_kis(request, monkeypatch):
|
|
"""모든 유닛 테스트에 자동 적용 — KIS 주문/토큰/시세/잔고 경로를 호출 즉시 실패로 치환.
|
|
|
|
적용 순서상 이 픽스처가 먼저 클래스 속성을 덮고, 테스트 본문의
|
|
`patch.object(instance, ...)` / `patch("...module.attr", ...)` 가 나중에 적용되므로
|
|
개별 테스트의 명시적 목이 항상 우선한다.
|
|
"""
|
|
if request.config.getoption("--allow-network"):
|
|
return
|
|
if _is_integration(request):
|
|
return
|
|
|
|
from app.services import korea_investment_api as kis_mod
|
|
|
|
for name in _KIS_CLASS_METHODS:
|
|
if not hasattr(kis_mod.KoreaInvestmentAPI, name):
|
|
raise AssertionError(
|
|
f"tests/conftest.py 차단 목록이 코드와 어긋납니다: "
|
|
f"KoreaInvestmentAPI.{name} 가 존재하지 않습니다(이름 변경/삭제?)."
|
|
)
|
|
monkeypatch.setattr(
|
|
kis_mod.KoreaInvestmentAPI,
|
|
name,
|
|
_blocked(f"korea_investment_api.{name}"),
|
|
raising=True,
|
|
)
|
|
|
|
from app.services import trade_executor as te_mod
|
|
|
|
for name in _TRADE_EXECUTOR_FUNCS:
|
|
if not hasattr(te_mod, name):
|
|
raise AssertionError(
|
|
f"tests/conftest.py 차단 목록이 코드와 어긋납니다: "
|
|
f"trade_executor.{name} 가 존재하지 않습니다(이름 변경/삭제?)."
|
|
)
|
|
monkeypatch.setattr(
|
|
te_mod, name, _blocked(f"trade_executor.{name}"), raising=True
|
|
)
|
|
|
|
# [2026-10-01] DT 체결가 확인(daytrading/fill_reconciler)은 주문 직후 백그라운드 태스크를
|
|
# 띄운다. 유닛 테스트에서는 기본 OFF — 테스트 종료 뒤 남은 태스크가 차단된 KIS 를 치거나
|
|
# "Task was destroyed but it is pending" 노이즈를 내지 않게 한다. 검증하는 테스트만 켠다.
|
|
from app.core.config import settings as _settings
|
|
monkeypatch.setattr(_settings, "DAY_TRADE_FILL_CHECK_ENABLED", False, raising=False)
|