test: add full integration test for simulation

test_full_simulation_workflow_generates_signals_and_trades:
- Creates klines with clear price movements
- Uses very low threshold (0.1%) to ensure signals generated
- Verifies signals are NOT empty
- Verifies trade_log is NOT empty
- Verifies BUY signals are generated
- Verifies results contain signals and trade_log

This test ensures the simulation engine is working correctly.
This commit is contained in:
shokollm
2026-04-12 05:00:46 +00:00
parent 1505bc9913
commit 984656c83c

View File

@@ -293,6 +293,93 @@ class TestSimulateEngine:
# Last close should be the last candle's close
assert engine.last_close == 104.0
# ==================== Integration Tests ====================
@pytest.mark.asyncio
async def test_full_simulation_workflow_generates_signals_and_trades(self):
"""
Full integration test: provides klines with clear price movements
and verifies signals and trade_log are populated.
This test ensures the simulation is working by:
1. Creating klines with obvious price movements (drops > 0.1%)
2. Using a very low threshold (0.1%)
3. Verifying signals are generated
4. Verifying trade_log is populated
5. Verifying we have buy/sell actions
"""
# Create klines with clear price drops and rises
klines = [
{"time": 1000, "open": 100, "high": 101, "low": 99, "close": 100, "volume": 1000}, # Flat
{"time": 2000, "open": 100, "high": 101, "low": 99.9, "close": 99.95, "volume": 1000}, # 0.05% drop
{"time": 3000, "open": 99.95, "high": 100, "low": 99.5, "close": 99.5, "volume": 1000}, # 0.45% drop
{"time": 4000, "open": 99.5, "high": 100, "low": 99, "close": 99.2, "volume": 1000}, # 0.30% drop
{"time": 5000, "open": 99.2, "high": 100, "low": 98, "close": 98.5, "volume": 1000}, # 0.71% drop
{"time": 6000, "open": 98.5, "high": 99, "low": 98, "close": 98.8, "volume": 1000}, # 0.30% rise
{"time": 7000, "open": 98.8, "high": 99, "low": 98, "close": 98.3, "volume": 1000}, # 0.51% drop
{"time": 8000, "open": 98.3, "high": 99, "low": 97, "close": 97.5, "volume": 1000}, # 0.81% drop
{"time": 9000, "open": 97.5, "high": 98, "low": 96, "close": 96.5, "volume": 1000}, # 1.03% drop
]
# Use very low threshold to ensure signals are generated
config_override = {
"max_candles": 100,
"strategy_config": {
"conditions": [
{"type": "price_drop", "threshold": 0.1, "token": "TEST", "token_address": "0x1234"}
],
"actions": [
{"type": "buy", "amount_percent": 10}
],
"risk_management": {
"stop_loss_percent": 5,
"take_profit_percent": 5
}
}
}
engine = create_engine(config_override=config_override, klines_data=klines)
engine.running = True
engine.run_id = "integration-test"
results = await engine.run()
# Verify results
print(f"\n=== Integration Test Results ===")
print(f"Status: {engine.status}")
print(f"Candles processed: {results.get('candles_processed')}")
print(f"Signals count: {len(engine.signals)}")
print(f"Trade log count: {len(engine.trade_log)}")
# ASSERTIONS - These should NEVER fail if simulation is working
assert engine.status == "completed", "Simulation should complete successfully"
assert results.get("candles_processed") == len(klines), f"Should process all {len(klines)} candles"
# Critical: signals should NOT be empty
assert len(engine.signals) > 0, "SIGNALS SHOULD NOT BE EMPTY! Simulation is not generating signals."
print(f"Signals: {[s['signal_type'] for s in engine.signals]}")
# Critical: trade_log should NOT be empty
assert len(engine.trade_log) > 0, "TRADE_LOG SHOULD NOT BE EMPTY! No activity logged."
print(f"Trade log: {[t['action'] for t in engine.trade_log]}")
# Should have at least one BUY signal
buy_signals = [s for s in engine.signals if s['signal_type'] == 'buy']
assert len(buy_signals) > 0, "Should have at least one BUY signal"
print(f"Buy signals: {len(buy_signals)}")
# Verify trade_log has BUY action
buy_trades = [t for t in engine.trade_log if t['action'] == 'buy']
assert len(buy_trades) > 0, "Trade log should contain BUY actions"
# Verify results contain the data
assert "signals" in results, "Results should contain signals"
assert "trade_log" in results, "Results should contain trade_log"
print("\n=== Integration Test PASSED ===")
print(f"Simulation working correctly!")
print(f"Generated {len(engine.signals)} signals and {len(engine.trade_log)} trade log entries")
if __name__ == "__main__":
pytest.main([__file__, "-v"])