25 lines
898 B
Python
25 lines
898 B
Python
|
import sys
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
# Mock out MetaTrader5 before importing our module
|
||
|
|
sys.modules['MetaTrader5'] = MagicMock()
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from unittest.mock import patch
|
||
|
|
import pandas as pd
|
||
|
|
from live_trading.multi_bar import TradingApp, log_and_print
|
||
|
|
|
||
|
|
class TestTradingApp(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.app = TradingApp(symbol="EURUSD", lot_size=0.01, magic_number=123456)
|
||
|
|
|
||
|
|
@patch("live_trading.multi_bar.mt5.copy_rates_from_pos")
|
||
|
|
@patch("live_trading.multi_bar.log_and_print")
|
||
|
|
def test_get_data_returns_none(self, mock_log, mock_copy_rates):
|
||
|
|
mock_copy_rates.return_value = None
|
||
|
|
|
||
|
|
# Test what happens when mt5.copy_rates_from_pos returns None
|
||
|
|
result = self.app.get_data("EURUSD", 100, 16408)
|
||
|
|
|
||
|
|
self.assertIsNone(result)
|
||
|
|
mock_log.assert_called_once_with("Could not retrieve data for EURUSD", is_error=True)
|