Big refactoring
This commit is contained in:
@@ -228,23 +228,23 @@ async def test_scan_ticker_full_flow_quality_selection_and_persistence(
|
||||
)
|
||||
|
||||
# -- Assert: database persistence --
|
||||
# Old dummy setup should be gone, only the 2 new setups should exist
|
||||
# History is preserved: old setup remains, 2 new setups are appended
|
||||
db_result = await scan_session.execute(
|
||||
select(TradeSetup).where(TradeSetup.ticker_id == ticker.id)
|
||||
)
|
||||
persisted = list(db_result.scalars().all())
|
||||
assert len(persisted) == 2, (
|
||||
f"Expected 2 persisted setups (old deleted), got {len(persisted)}"
|
||||
assert len(persisted) == 3, (
|
||||
f"Expected 3 persisted setups (1 old + 2 new), got {len(persisted)}"
|
||||
)
|
||||
|
||||
persisted_directions = sorted(s.direction for s in persisted)
|
||||
assert persisted_directions == ["long", "short"], (
|
||||
f"Expected ['long', 'short'] persisted, got {persisted_directions}"
|
||||
assert persisted_directions == ["long", "long", "short"], (
|
||||
f"Expected ['long', 'long', 'short'] persisted, got {persisted_directions}"
|
||||
)
|
||||
|
||||
# Verify persisted records match returned setups
|
||||
persisted_long = [s for s in persisted if s.direction == "long"][0]
|
||||
persisted_short = [s for s in persisted if s.direction == "short"][0]
|
||||
# Verify latest persisted records match returned setups
|
||||
persisted_long = max((s for s in persisted if s.direction == "long"), key=lambda s: s.id)
|
||||
persisted_short = max((s for s in persisted if s.direction == "short"), key=lambda s: s.id)
|
||||
|
||||
assert persisted_long.target == long_setup.target
|
||||
assert persisted_long.rr_ratio == long_setup.rr_ratio
|
||||
|
||||
Reference in New Issue
Block a user