folder stricture reorganize
This commit is contained in:
Dobromir Popov
222
tests/test_mexc_balance_orders.py
Normal file
222
tests/test_mexc_balance_orders.py
Normal file
@ -0,0 +1,222 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script for MEXC balance retrieval and $1 order execution
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import logging
|
||||
from pathlib import Path
|
||||
|
||||
# Add project root to path
|
||||
sys.path.insert(0, os.path.abspath('.'))
|
||||
|
||||
from core.trading_executor import TradingExecutor
|
||||
from core.data_provider import DataProvider
|
||||
|
||||
# Setup logging
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
def test_mexc_balance():
|
||||
"""Test MEXC balance retrieval"""
|
||||
print("="*60)
|
||||
print("TESTING MEXC BALANCE RETRIEVAL")
|
||||
print("="*60)
|
||||
|
||||
try:
|
||||
# Initialize trading executor
|
||||
executor = TradingExecutor()
|
||||
|
||||
# Check if trading is enabled
|
||||
print(f"Trading enabled: {executor.trading_enabled}")
|
||||
print(f"Dry run mode: {executor.dry_run}")
|
||||
|
||||
if not executor.trading_enabled:
|
||||
print("❌ Trading not enabled - check config.yaml and API keys")
|
||||
return False
|
||||
|
||||
# Test balance retrieval
|
||||
print("\n📊 Retrieving account balance...")
|
||||
balances = executor.get_account_balance()
|
||||
|
||||
if not balances:
|
||||
print("❌ No balances retrieved - check API connectivity")
|
||||
return False
|
||||
|
||||
print(f"✅ Retrieved balances for {len(balances)} assets:")
|
||||
for asset, balance_info in balances.items():
|
||||
free = balance_info['free']
|
||||
locked = balance_info['locked']
|
||||
total = balance_info['total']
|
||||
print(f" {asset}: Free: {free:.6f}, Locked: {locked:.6f}, Total: {total:.6f}")
|
||||
|
||||
# Check USDT balance specifically
|
||||
if 'USDT' in balances:
|
||||
usdt_free = balances['USDT']['free']
|
||||
print(f"\n💰 USDT available for trading: ${usdt_free:.2f}")
|
||||
|
||||
if usdt_free >= 2.0: # Need at least $2 for testing
|
||||
print("✅ Sufficient USDT balance for $1 order testing")
|
||||
return True
|
||||
else:
|
||||
print(f"⚠️ Insufficient USDT balance for testing (need $2+, have ${usdt_free:.2f})")
|
||||
return False
|
||||
else:
|
||||
print("❌ No USDT balance found")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error testing MEXC balance: {e}")
|
||||
return False
|
||||
|
||||
def test_mexc_order_execution():
|
||||
"""Test $1 order execution (dry run)"""
|
||||
print("\n" + "="*60)
|
||||
print("TESTING $1 ORDER EXECUTION (DRY RUN)")
|
||||
print("="*60)
|
||||
|
||||
try:
|
||||
# Initialize components
|
||||
executor = TradingExecutor()
|
||||
data_provider = DataProvider()
|
||||
|
||||
if not executor.trading_enabled:
|
||||
print("❌ Trading not enabled - cannot test order execution")
|
||||
return False
|
||||
|
||||
# Test symbol
|
||||
symbol = "ETH/USDT"
|
||||
|
||||
# Get current price
|
||||
print(f"\n📈 Getting current price for {symbol}...")
|
||||
ticker_data = data_provider.get_historical_data(symbol, '1m', limit=1, refresh=True)
|
||||
|
||||
if ticker_data is None or ticker_data.empty:
|
||||
print(f"❌ Could not get price data for {symbol}")
|
||||
return False
|
||||
|
||||
current_price = float(ticker_data['close'].iloc[-1])
|
||||
print(f"✅ Current {symbol} price: ${current_price:.2f}")
|
||||
|
||||
# Calculate order size for $1
|
||||
usd_amount = 1.0
|
||||
crypto_amount = usd_amount / current_price
|
||||
print(f"💱 $1 USD = {crypto_amount:.6f} ETH")
|
||||
|
||||
# Test buy signal execution
|
||||
print(f"\n🛒 Testing BUY signal execution...")
|
||||
buy_success = executor.execute_signal(
|
||||
symbol=symbol,
|
||||
action='BUY',
|
||||
confidence=0.75,
|
||||
current_price=current_price
|
||||
)
|
||||
|
||||
if buy_success:
|
||||
print("✅ BUY signal executed successfully")
|
||||
|
||||
# Check position
|
||||
positions = executor.get_positions()
|
||||
if symbol in positions:
|
||||
position = positions[symbol]
|
||||
print(f"📍 Position opened: {position.quantity:.6f} {symbol} @ ${position.entry_price:.2f}")
|
||||
|
||||
# Test sell signal execution
|
||||
print(f"\n💰 Testing SELL signal execution...")
|
||||
sell_success = executor.execute_signal(
|
||||
symbol=symbol,
|
||||
action='SELL',
|
||||
confidence=0.80,
|
||||
current_price=current_price * 1.001 # Simulate small price increase
|
||||
)
|
||||
|
||||
if sell_success:
|
||||
print("✅ SELL signal executed successfully")
|
||||
|
||||
# Check trade history
|
||||
trades = executor.get_trade_history()
|
||||
if trades:
|
||||
last_trade = trades[-1]
|
||||
print(f"📊 Trade completed: P&L = ${last_trade.pnl:.4f}")
|
||||
|
||||
return True
|
||||
else:
|
||||
print("❌ SELL signal failed")
|
||||
return False
|
||||
else:
|
||||
print("❌ No position found after BUY signal")
|
||||
return False
|
||||
else:
|
||||
print("❌ BUY signal failed")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error testing order execution: {e}")
|
||||
return False
|
||||
|
||||
def test_dashboard_balance_integration():
|
||||
"""Test dashboard balance integration"""
|
||||
print("\n" + "="*60)
|
||||
print("TESTING DASHBOARD BALANCE INTEGRATION")
|
||||
print("="*60)
|
||||
|
||||
try:
|
||||
from web.dashboard import TradingDashboard
|
||||
|
||||
# Create dashboard with trading executor
|
||||
executor = TradingExecutor()
|
||||
dashboard = TradingDashboard(trading_executor=executor)
|
||||
|
||||
print(f"Dashboard starting balance: ${dashboard.starting_balance:.2f}")
|
||||
|
||||
if dashboard.starting_balance > 0:
|
||||
print("✅ Dashboard successfully retrieved starting balance")
|
||||
return True
|
||||
else:
|
||||
print("⚠️ Dashboard using default balance (MEXC not connected)")
|
||||
return False
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error testing dashboard integration: {e}")
|
||||
return False
|
||||
|
||||
def main():
|
||||
"""Run all tests"""
|
||||
print("🚀 MEXC INTEGRATION TESTING")
|
||||
print("Testing balance retrieval and $1 order execution")
|
||||
|
||||
# Test 1: Balance retrieval
|
||||
balance_test = test_mexc_balance()
|
||||
|
||||
# Test 2: Order execution (only if balance test passes)
|
||||
if balance_test:
|
||||
order_test = test_mexc_order_execution()
|
||||
else:
|
||||
print("\n⏭️ Skipping order execution test (balance test failed)")
|
||||
order_test = False
|
||||
|
||||
# Test 3: Dashboard integration
|
||||
dashboard_test = test_dashboard_balance_integration()
|
||||
|
||||
# Summary
|
||||
print("\n" + "="*60)
|
||||
print("TEST SUMMARY")
|
||||
print("="*60)
|
||||
print(f"Balance Retrieval: {'✅ PASS' if balance_test else '❌ FAIL'}")
|
||||
print(f"Order Execution: {'✅ PASS' if order_test else '❌ FAIL'}")
|
||||
print(f"Dashboard Integration: {'✅ PASS' if dashboard_test else '❌ FAIL'}")
|
||||
|
||||
if balance_test and order_test and dashboard_test:
|
||||
print("\n🎉 ALL TESTS PASSED - Ready for live $1 testing!")
|
||||
return True
|
||||
else:
|
||||
print("\n⚠️ Some tests failed - check configuration and API keys")
|
||||
return False
|
||||
|
||||
if __name__ == "__main__":
|
||||
success = main()
|
||||
sys.exit(0 if success else 1)
|
||||
Reference in New Issue
Block a user