fix: grandfather pre-cutover stop episodes
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import importlib.util
|
||||
from pathlib import Path
|
||||
|
||||
import sqlalchemy as sa
|
||||
from alembic.migration import MigrationContext
|
||||
from alembic.operations import Operations
|
||||
|
||||
|
||||
def _load_migration_module():
|
||||
path = (
|
||||
Path(__file__).resolve().parents[2]
|
||||
/ "alembic"
|
||||
/ "versions"
|
||||
/ "022_add_paper_trade_reentry_gate_reset.py"
|
||||
)
|
||||
spec = importlib.util.spec_from_file_location("migration_022", path)
|
||||
assert spec is not None and spec.loader is not None
|
||||
module = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(module)
|
||||
return module
|
||||
|
||||
|
||||
def test_upgrade_grandfathers_only_preexisting_initial_stops():
|
||||
migration = _load_migration_module()
|
||||
engine = sa.create_engine("sqlite://")
|
||||
|
||||
with engine.begin() as connection:
|
||||
connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
CREATE TABLE paper_trades (
|
||||
id INTEGER PRIMARY KEY,
|
||||
status VARCHAR NOT NULL,
|
||||
close_reason VARCHAR,
|
||||
closed_at DATETIME
|
||||
)
|
||||
"""
|
||||
)
|
||||
)
|
||||
connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
INSERT INTO paper_trades (id, status, close_reason, closed_at)
|
||||
VALUES
|
||||
(1, 'closed', 'stop', '2026-07-01 12:00:00'),
|
||||
(2, 'closed', 'manual', '2026-07-02 12:00:00'),
|
||||
(3, 'open', NULL, NULL)
|
||||
"""
|
||||
)
|
||||
)
|
||||
|
||||
context = MigrationContext.configure(connection)
|
||||
migration.op = Operations(context)
|
||||
migration.upgrade()
|
||||
|
||||
historical = connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
SELECT closed_at, reentry_gate_failed_at,
|
||||
reentry_gate_requalified_at
|
||||
FROM paper_trades
|
||||
WHERE id = 1
|
||||
"""
|
||||
)
|
||||
).one()
|
||||
assert historical[1] == historical[0]
|
||||
assert historical[2] == historical[0]
|
||||
|
||||
unaffected = connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
SELECT reentry_gate_failed_at, reentry_gate_requalified_at
|
||||
FROM paper_trades
|
||||
WHERE id IN (2, 3)
|
||||
ORDER BY id
|
||||
"""
|
||||
)
|
||||
).all()
|
||||
assert unaffected == [(None, None), (None, None)]
|
||||
|
||||
connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
INSERT INTO paper_trades (id, status, close_reason, closed_at)
|
||||
VALUES (4, 'closed', 'stop', '2026-07-18 12:00:00')
|
||||
"""
|
||||
)
|
||||
)
|
||||
new_stop = connection.execute(
|
||||
sa.text(
|
||||
"""
|
||||
SELECT reentry_gate_failed_at, reentry_gate_requalified_at
|
||||
FROM paper_trades
|
||||
WHERE id = 4
|
||||
"""
|
||||
)
|
||||
).one()
|
||||
assert new_stop == (None, None)
|
||||
@@ -24,6 +24,7 @@ def _stopped_trade(
|
||||
ticker_id: int,
|
||||
*,
|
||||
closed_at: datetime,
|
||||
close_reason: str = "stop",
|
||||
gate_failed_at: datetime | None = None,
|
||||
gate_requalified_at: datetime | None = None,
|
||||
) -> PaperTrade:
|
||||
@@ -39,7 +40,7 @@ def _stopped_trade(
|
||||
opened_at=closed_at - timedelta(days=5),
|
||||
close_price=95.0,
|
||||
closed_at=closed_at,
|
||||
close_reason="stop",
|
||||
close_reason=close_reason,
|
||||
reentry_gate_failed_at=gate_failed_at,
|
||||
reentry_gate_requalified_at=gate_requalified_at,
|
||||
)
|
||||
@@ -126,3 +127,33 @@ async def test_latest_stop_starts_a_new_gate_reset_episode(session):
|
||||
await session.commit()
|
||||
|
||||
assert ticker.id in await get_reentry_gate_locks(session)
|
||||
|
||||
|
||||
async def test_newer_non_stop_exit_supersedes_historical_stop(session):
|
||||
session.add(User(id=1, username="u", password_hash="x", role="user", has_access=True))
|
||||
ticker = Ticker(symbol="LATEREXIT")
|
||||
session.add(ticker)
|
||||
await session.flush()
|
||||
|
||||
stopped_at = datetime.now(timezone.utc) - timedelta(days=20)
|
||||
old_stop = _stopped_trade(ticker.id, closed_at=stopped_at)
|
||||
later_manual_exit = _stopped_trade(
|
||||
ticker.id,
|
||||
closed_at=stopped_at + timedelta(days=10),
|
||||
close_reason="manual",
|
||||
)
|
||||
session.add_all([old_stop, later_manual_exit])
|
||||
await session.commit()
|
||||
|
||||
assert ticker.id not in await get_reentry_gate_locks(session)
|
||||
|
||||
observed_at = datetime.now(timezone.utc)
|
||||
updated = await observe_reentry_gate_transitions(
|
||||
session,
|
||||
evaluated_ticker_ids={ticker.id},
|
||||
qualified_ticker_ids=set(),
|
||||
observed_at=observed_at,
|
||||
)
|
||||
|
||||
assert updated == set()
|
||||
assert old_stop.reentry_gate_failed_at is None
|
||||
|
||||
Reference in New Issue
Block a user