## 재기동 내구성 - scheduler: 복원 직후 reset_if_new_day()가 복원분을 전부 폐기하던 버그 (begin_restore()로 날짜 경계를 선확정. 복원 성공 로그는 뜨는데 실제로는 지워졌다) - 일일 안전장치(daily_pnl/loss_streak_today/closed_today) Redis 영속화 날짜를 키와 페이로드 양쪽에 실어 이중 게이트, 복원은 보수적 병합 (손익 min/연패 max/청산목록 합집합 — 복원이 안전장치를 약화시킬 수 없다) - db_worker 종료 drain이 insert_trade_history만 커밋하고 포지션 추적 행은 DLQ로 버리던 비대칭 제거 + DLQ 재처리 경로 신설(쓰기 전용이었다) - 장중 기동 시 ORB 소급 수집(backfill), 전일 캔들 혼입 차단 - swing/exit_service "보유 종목 없음" debug→info, 10회 연속 warning ## DT 진입 복구 - 워밍업 회귀 복구: 1b9257c(2026-09-28)가 조회 계층에 당일 필터를 넣어 5분봉 RSI(14)/BB(20) 워밍업이 죽고 VWAP/RSI_BB가 10:15/10:40 이전 발화 불가가 됐다. 필터를 삭제하지 않고 용도별로 분리 (get_minute_chart_full_day_v2 신설, 기존 메서드·3개 호출부 동작 무변경) - 거래량 게이트 비교 대상 교정: 형성 중 봉 → 완성봉 (5분봉·ORB 1분봉 양쪽) 임계값(1.3/1.5)은 불변, 고친 것은 비교 대상뿐 - 세션 경계 차단: _last_session_candles()로 거래량 비교가 전일 봉을 잡아 개장 직후 게이트가 자동 통과하는 것을 방지 - 스캔 선조회(동시성 4)·종목 타임아웃·시간예산·라운드로빈 + 잘린 종목 수 로그 - watchlist 30분 갱신 시 ORB 보유 종목 우선 유지 ## 유입 잔고 격리 - sync_portfolio()가 시스템 매수 근거 없는 KIS 잔고를 자기 스윙 포지션으로 편입해 자동 손절로 처분하던 문제 → strategy_type='external' 격리 (승격만 허용, 강등 없음. 기존 행은 수동 SQL 필요) - 격리분 손절선 이탈은 집행 없이 warning + summary 노출 - 수동 청산 경로에 record_trade_result 누락 보강, 전략 간 당일 쿨다운 ## 4% 상한 - market_data 표시 경로 클램프 누락 + confidence 배율 후 재클램프 - 회귀 방지 테스트(AST로 가드가 try 밖에 있는지 검증) ## 프론트 - 총자산 보정 가드(3%·종목수·가격결손·국내분 한정), 원인 미규명 명시 - cumulative_realized_pl이 /portfolio/combined에 없어 항상 0이던 문제 테스트 579 → 677 passed / 0 failed (소켓 차단 하 통과, 실호출 없음) 전략 임계값 변경 0건 ⚠️ 워밍업 복구의 효과 크기는 미확정 — KIS 분봉 API에 날짜 파라미터가 없어 미래 ref_time에 전일 데이터를 주는지가 미검증 전제다. 배포 후 "[분봉] 워밍업 5분봉 N봉 (당일 a + 전일 b)" 로그의 b로 확정할 것. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
205 lines
9.8 KiB
Python
205 lines
9.8 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",
|
|
)
|
|
|
|
_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
|
|
)
|