Daichi Narushima
9356d5dcdf
* Consolidate duplicated export and history streaming helpers. Reduce repeated CLI export plumbing, shared per-symbol SQLite writes, and test mock setup without changing public behavior. Co-authored-by: Cursor <cursoragent@cursor.com> * Bump version from 0.7.0 to 0.7.1. Co-authored-by: Cursor <cursoragent@cursor.com> --------- Co-authored-by: Cursor <cursoragent@cursor.com>
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""Shared pytest fixtures for mt5cli tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
from pytest_mock import MockerFixture # noqa: TC002
|
|
|
|
_DATAFRAME_METHODS = (
|
|
"copy_rates_from_as_df",
|
|
"copy_rates_from_pos_as_df",
|
|
"copy_rates_range_as_df",
|
|
"copy_ticks_from_as_df",
|
|
"copy_ticks_range_as_df",
|
|
"account_info_as_df",
|
|
"terminal_info_as_df",
|
|
"symbols_get_as_df",
|
|
"symbol_info_as_df",
|
|
"orders_get_as_df",
|
|
"positions_get_as_df",
|
|
"history_orders_get_as_df",
|
|
"history_deals_get_as_df",
|
|
"version_as_df",
|
|
"last_error_as_df",
|
|
"symbol_info_tick_as_df",
|
|
"market_book_get_as_df",
|
|
"order_check_as_df",
|
|
"order_send_as_df",
|
|
)
|
|
|
|
|
|
def build_mock_mt5_data_client() -> MagicMock:
|
|
"""Return a MagicMock Mt5DataClient with common DataFrame stubs."""
|
|
client = MagicMock()
|
|
sample_df = pd.DataFrame({"col": [1]})
|
|
for method_name in _DATAFRAME_METHODS:
|
|
getattr(client, method_name).return_value = sample_df
|
|
client.version.return_value = (5, 0, 1)
|
|
client.terminal_info.return_value = {"connected": True, "paths": ["terminal.exe"]}
|
|
client.account_info.return_value = {"login": 123, "limits": {"modes": ["demo"]}}
|
|
client.symbols_total.return_value = 42
|
|
return client
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_client(mocker: MockerFixture) -> MagicMock:
|
|
"""Create and patch a mock Mt5DataClient for CLI and SDK tests."""
|
|
client = build_mock_mt5_data_client()
|
|
mocker.patch("mt5cli.sdk.Mt5DataClient", return_value=client)
|
|
return client
|