CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-03 18:43:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

ruff format: rpc tests

Browse Source
This commit is contained in:
Matthias
2024-05-12 15:44:42 +02:00
parent 8c7d80b78e
commit 1cbd49fd4e
7 changed files with 3700 additions and 3248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

318
tests/rpc/test_rpc_emc.py
View File

@@ -1,6 +1,7 @@
"""
Unit test file for rpc/external_message_consumer.py
"""
import asyncio
import logging
from datetime import datetime, timezone
@@ -21,19 +22,16 @@ _TEST_WS_PORT = 9989
@pytest.fixture
def patched_emc(default_conf, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": "null",
"port": 9891,
"ws_token": _TEST_WS_TOKEN
}
]
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{"name": "default", "host": "null", "port": 9891, "ws_token": _TEST_WS_TOKEN}
],
}
}
})
)
dataprovider = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dataprovider)
@@ -81,7 +79,7 @@ def test_emc_init(patched_emc):
# Parametrize this?
def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
test_producer = {"name": "test", "url": "ws://test", "ws_token": "test"}
producer_name = test_producer['name']
producer_name = test_producer["name"]
invalid_msg = r"Invalid message .+"
caplog.set_level(logging.DEBUG)
@@ -92,7 +90,8 @@ def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
assert log_has(f"Received message of type `whitelist` from `{producer_name}`", caplog)
assert log_has(
f"Consumed message from `{producer_name}` of type `RPCMessageType.WHITELIST`", caplog)
f"Consumed message from `{producer_name}` of type `RPCMessageType.WHITELIST`", caplog
)
# Test handle analyzed_df single candle message
df_message = {
@@ -100,8 +99,8 @@ def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
"data": {
"key": ("BTC/USDT", "5m", "spot"),
"df": ohlcv_history,
"la": datetime.now(timezone.utc)
}
"la": datetime.now(timezone.utc),
},
}
patched_emc.handle_producer_message(test_producer, df_message)
@@ -124,11 +123,7 @@ def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
malformed_message = {
"type": "analyzed_df",
"data": {
"key": "BTC/USDT",
"df": ohlcv_history,
"la": datetime.now(timezone.utc)
}
"data": {"key": "BTC/USDT", "df": ohlcv_history, "la": datetime.now(timezone.utc)},
}
patched_emc.handle_producer_message(test_producer, malformed_message)
@@ -138,13 +133,13 @@ def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
# Empty dataframe
malformed_message = {
"type": "analyzed_df",
"data": {
"key": ("BTC/USDT", "5m", "spot"),
"df": ohlcv_history.loc[ohlcv_history['open'] < 0],
"la": datetime.now(timezone.utc)
}
}
"type": "analyzed_df",
"data": {
"key": ("BTC/USDT", "5m", "spot"),
"df": ohlcv_history.loc[ohlcv_history["open"] < 0],
"la": datetime.now(timezone.utc),
},
}
patched_emc.handle_producer_message(test_producer, malformed_message)
assert log_has(f"Received message of type `analyzed_df` from `{producer_name}`", caplog)
@@ -166,29 +161,32 @@ def test_emc_handle_producer_message(patched_emc, caplog, ohlcv_history):
async def test_emc_create_connection_success(default_conf, caplog, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60,
}
}
})
)
mocker.patch('freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start',
MagicMock())
mocker.patch(
"freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start", MagicMock()
)
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)
test_producer = default_conf['external_message_consumer']['producers'][0]
test_producer = default_conf["external_message_consumer"]["producers"][0]
lock = asyncio.Lock()
emc._running = True
@@ -205,27 +203,27 @@ async def test_emc_create_connection_success(default_conf, caplog, mocker):
emc.shutdown()
@pytest.mark.parametrize('host,port', [
(_TEST_WS_HOST, -1),
("10000.1241..2121/", _TEST_WS_PORT),
])
@pytest.mark.parametrize(
"host,port",
[
(_TEST_WS_HOST, -1),
("10000.1241..2121/", _TEST_WS_PORT),
],
)
async def test_emc_create_connection_invalid_url(default_conf, caplog, mocker, host, port):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": host,
"port": port,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{"name": "default", "host": host, "port": port, "ws_token": _TEST_WS_TOKEN}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60,
}
}
})
)
dp = DataProvider(default_conf, None, None, None)
# Handle start explicitly to avoid messing with threading in tests
@@ -242,25 +240,27 @@ async def test_emc_create_connection_invalid_url(default_conf, caplog, mocker, h
async def test_emc_create_connection_error(default_conf, caplog, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 60,
"ping_timeout": 60,
"sleep_timeout": 60,
}
}
})
)
# Test unexpected error
mocker.patch('websockets.connect', side_effect=RuntimeError)
mocker.patch("websockets.connect", side_effect=RuntimeError)
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)
@@ -275,28 +275,31 @@ async def test_emc_create_connection_error(default_conf, caplog, mocker):
async def test_emc_receive_messages_valid(default_conf, caplog, mocker):
caplog.set_level(logging.DEBUG)
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 1,
"ping_timeout": 60,
"sleep_time": 60
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 1,
"ping_timeout": 60,
"sleep_time": 60,
}
}
})
)
mocker.patch('freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start',
MagicMock())
mocker.patch(
"freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start", MagicMock()
)
lock = asyncio.Lock()
test_producer = default_conf['external_message_consumer']['producers'][0]
test_producer = default_conf["external_message_consumer"]["producers"][0]
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)
@@ -319,28 +322,31 @@ async def test_emc_receive_messages_valid(default_conf, caplog, mocker):
async def test_emc_receive_messages_invalid(default_conf, caplog, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 1,
"ping_timeout": 60,
"sleep_time": 60
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 1,
"ping_timeout": 60,
"sleep_time": 60,
}
}
})
)
mocker.patch('freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start',
MagicMock())
mocker.patch(
"freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start", MagicMock()
)
lock = asyncio.Lock()
test_producer = default_conf['external_message_consumer']['producers'][0]
test_producer = default_conf["external_message_consumer"]["producers"][0]
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)
@@ -363,28 +369,31 @@ async def test_emc_receive_messages_invalid(default_conf, caplog, mocker):
async def test_emc_receive_messages_timeout(default_conf, caplog, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 0.1,
"ping_timeout": 1,
"sleep_time": 1
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 0.1,
"ping_timeout": 1,
"sleep_time": 1,
}
}
})
)
mocker.patch('freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start',
MagicMock())
mocker.patch(
"freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start", MagicMock()
)
lock = asyncio.Lock()
test_producer = default_conf['external_message_consumer']['producers'][0]
test_producer = default_conf["external_message_consumer"]["producers"][0]
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)
@@ -411,28 +420,31 @@ async def test_emc_receive_messages_timeout(default_conf, caplog, mocker):
async def test_emc_receive_messages_handle_error(default_conf, caplog, mocker):
default_conf.update({
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN
}
],
"wait_timeout": 1,
"ping_timeout": 1,
"sleep_time": 1
default_conf.update(
{
"external_message_consumer": {
"enabled": True,
"producers": [
{
"name": "default",
"host": _TEST_WS_HOST,
"port": _TEST_WS_PORT,
"ws_token": _TEST_WS_TOKEN,
}
],
"wait_timeout": 1,
"ping_timeout": 1,
"sleep_time": 1,
}
}
})
)
mocker.patch('freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start',
MagicMock())
mocker.patch(
"freqtrade.rpc.external_message_consumer.ExternalMessageConsumer.start", MagicMock()
)
lock = asyncio.Lock()
test_producer = default_conf['external_message_consumer']['producers'][0]
test_producer = default_conf["external_message_consumer"]["producers"][0]
dp = DataProvider(default_conf, None, None, None)
emc = ExternalMessageConsumer(default_conf, dp)