ruff format: rpc tests

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

View File

@@ -26,47 +26,43 @@ def get_webhook_dict() -> dict:
"value2": "limit {limit:8f}",
"value3": "{stake_amount:8f} {stake_currency}",
"value4": "leverage {leverage:.1f}",
"value5": "direction {direction}"
"value5": "direction {direction}",
},
"webhookentrycancel": {
"value1": "Cancelling Open Buy Order for {pair}",
"value2": "limit {limit:8f}",
"value3": "{stake_amount:8f} {stake_currency}",
"value4": "leverage {leverage:.1f}",
"value5": "direction {direction}"
"value5": "direction {direction}",
},
"webhookentryfill": {
"value1": "Buy Order for {pair} filled",
"value2": "at {open_rate:8f}",
"value3": "{stake_amount:8f} {stake_currency}",
"value4": "leverage {leverage:.1f}",
"value5": "direction {direction}"
"value5": "direction {direction}",
},
"webhookexit": {
"value1": "Selling {pair}",
"value2": "limit {limit:8f}",
"value3": "profit: {profit_amount:8f} {stake_currency} ({profit_ratio})"
"value3": "profit: {profit_amount:8f} {stake_currency} ({profit_ratio})",
},
"webhookexitcancel": {
"value1": "Cancelling Open Sell Order for {pair}",
"value2": "limit {limit:8f}",
"value3": "profit: {profit_amount:8f} {stake_currency} ({profit_ratio})"
"value3": "profit: {profit_amount:8f} {stake_currency} ({profit_ratio})",
},
"webhookexitfill": {
"value1": "Sell Order for {pair} filled",
"value2": "at {close_rate:8f}",
"value3": ""
"value3": "",
},
"webhookstatus": {
"value1": "Status: {status}",
"value2": "",
"value3": ""
}
"webhookstatus": {"value1": "Status: {status}", "value2": "", "value3": ""},
}
def test__init__(mocker, default_conf):
default_conf['webhook'] = {'enabled': True, 'url': "https://DEADBEEF.com"}
default_conf["webhook"] = {"enabled": True, "url": "https://DEADBEEF.com"}
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
assert webhook._config == default_conf
@@ -80,255 +76,290 @@ def test_send_msg_webhook(default_conf, mocker):
msg_mock = MagicMock()
mocker.patch("freqtrade.rpc.webhook.Webhook._send_msg", msg_mock)
msg = {
'type': RPCMessageType.ENTRY,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 1.0,
'direction': 'Long',
'limit': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 1.0,
"direction": "Long",
"limit": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["entry"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["entry"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["entry"]["value3"].format(**msg))
assert (msg_mock.call_args[0][0]["value4"] ==
default_conf["webhook"]["entry"]["value4"].format(**msg))
assert (msg_mock.call_args[0][0]["value5"] ==
default_conf["webhook"]["entry"]["value5"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["entry"]["value1"].format(
**msg
)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["entry"]["value2"].format(
**msg
)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["entry"]["value3"].format(
**msg
)
assert msg_mock.call_args[0][0]["value4"] == default_conf["webhook"]["entry"]["value4"].format(
**msg
)
assert msg_mock.call_args[0][0]["value5"] == default_conf["webhook"]["entry"]["value5"].format(
**msg
)
# Test short
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.ENTRY,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 2.0,
'direction': 'Short',
'limit': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 2.0,
"direction": "Short",
"limit": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["entry"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["entry"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["entry"]["value3"].format(**msg))
assert (msg_mock.call_args[0][0]["value4"] ==
default_conf["webhook"]["entry"]["value4"].format(**msg))
assert (msg_mock.call_args[0][0]["value5"] ==
default_conf["webhook"]["entry"]["value5"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["entry"]["value1"].format(
**msg
)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["entry"]["value2"].format(
**msg
)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["entry"]["value3"].format(
**msg
)
assert msg_mock.call_args[0][0]["value4"] == default_conf["webhook"]["entry"]["value4"].format(
**msg
)
assert msg_mock.call_args[0][0]["value5"] == default_conf["webhook"]["entry"]["value5"].format(
**msg
)
# Test buy cancel
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.ENTRY_CANCEL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 1.0,
'direction': 'Long',
'limit': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY_CANCEL,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 1.0,
"direction": "Long",
"limit": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookentrycancel"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookentrycancel"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookentrycancel"]["value3"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookentrycancel"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookentrycancel"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookentrycancel"][
"value3"
].format(**msg)
# Test short cancel
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.ENTRY_CANCEL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 2.0,
'direction': 'Short',
'limit': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY_CANCEL,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 2.0,
"direction": "Short",
"limit": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookentrycancel"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookentrycancel"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookentrycancel"]["value3"].format(**msg))
assert (msg_mock.call_args[0][0]["value4"] ==
default_conf["webhook"]["webhookentrycancel"]["value4"].format(**msg))
assert (msg_mock.call_args[0][0]["value5"] ==
default_conf["webhook"]["webhookentrycancel"]["value5"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookentrycancel"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookentrycancel"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookentrycancel"][
"value3"
].format(**msg)
assert msg_mock.call_args[0][0]["value4"] == default_conf["webhook"]["webhookentrycancel"][
"value4"
].format(**msg)
assert msg_mock.call_args[0][0]["value5"] == default_conf["webhook"]["webhookentrycancel"][
"value5"
].format(**msg)
# Test buy fill
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.ENTRY_FILL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 1.0,
'direction': 'Long',
'open_rate': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY_FILL,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 1.0,
"direction": "Long",
"open_rate": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookentryfill"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookentryfill"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookentryfill"]["value3"].format(**msg))
assert (msg_mock.call_args[0][0]["value4"] ==
default_conf["webhook"]["webhookentrycancel"]["value4"].format(**msg))
assert (msg_mock.call_args[0][0]["value5"] ==
default_conf["webhook"]["webhookentrycancel"]["value5"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookentryfill"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookentryfill"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookentryfill"][
"value3"
].format(**msg)
assert msg_mock.call_args[0][0]["value4"] == default_conf["webhook"]["webhookentrycancel"][
"value4"
].format(**msg)
assert msg_mock.call_args[0][0]["value5"] == default_conf["webhook"]["webhookentrycancel"][
"value5"
].format(**msg)
# Test short fill
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.ENTRY_FILL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'leverage': 2.0,
'direction': 'Short',
'open_rate': 0.005,
'stake_amount': 0.8,
'stake_amount_fiat': 500,
'stake_currency': 'BTC',
'fiat_currency': 'EUR'
"type": RPCMessageType.ENTRY_FILL,
"exchange": "Binance",
"pair": "ETH/BTC",
"leverage": 2.0,
"direction": "Short",
"open_rate": 0.005,
"stake_amount": 0.8,
"stake_amount_fiat": 500,
"stake_currency": "BTC",
"fiat_currency": "EUR",
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookentryfill"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookentryfill"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookentryfill"]["value3"].format(**msg))
assert (msg_mock.call_args[0][0]["value4"] ==
default_conf["webhook"]["webhookentrycancel"]["value4"].format(**msg))
assert (msg_mock.call_args[0][0]["value5"] ==
default_conf["webhook"]["webhookentrycancel"]["value5"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookentryfill"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookentryfill"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookentryfill"][
"value3"
].format(**msg)
assert msg_mock.call_args[0][0]["value4"] == default_conf["webhook"]["webhookentrycancel"][
"value4"
].format(**msg)
assert msg_mock.call_args[0][0]["value5"] == default_conf["webhook"]["webhookentrycancel"][
"value5"
].format(**msg)
# Test sell
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.EXIT,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'gain': "profit",
'limit': 0.005,
'amount': 0.8,
'order_type': 'limit',
'open_rate': 0.004,
'current_rate': 0.005,
'profit_amount': 0.001,
'profit_ratio': 0.20,
'stake_currency': 'BTC',
'sell_reason': ExitType.STOP_LOSS.value
"type": RPCMessageType.EXIT,
"exchange": "Binance",
"pair": "ETH/BTC",
"gain": "profit",
"limit": 0.005,
"amount": 0.8,
"order_type": "limit",
"open_rate": 0.004,
"current_rate": 0.005,
"profit_amount": 0.001,
"profit_ratio": 0.20,
"stake_currency": "BTC",
"sell_reason": ExitType.STOP_LOSS.value,
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookexit"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookexit"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookexit"]["value3"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookexit"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookexit"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookexit"][
"value3"
].format(**msg)
# Test sell cancel
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.EXIT_CANCEL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'gain': "profit",
'limit': 0.005,
'amount': 0.8,
'order_type': 'limit',
'open_rate': 0.004,
'current_rate': 0.005,
'profit_amount': 0.001,
'profit_ratio': 0.20,
'stake_currency': 'BTC',
'sell_reason': ExitType.STOP_LOSS.value
"type": RPCMessageType.EXIT_CANCEL,
"exchange": "Binance",
"pair": "ETH/BTC",
"gain": "profit",
"limit": 0.005,
"amount": 0.8,
"order_type": "limit",
"open_rate": 0.004,
"current_rate": 0.005,
"profit_amount": 0.001,
"profit_ratio": 0.20,
"stake_currency": "BTC",
"sell_reason": ExitType.STOP_LOSS.value,
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookexitcancel"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookexitcancel"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookexitcancel"]["value3"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookexitcancel"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookexitcancel"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookexitcancel"][
"value3"
].format(**msg)
# Test Sell fill
msg_mock.reset_mock()
msg = {
'type': RPCMessageType.EXIT_FILL,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'gain': "profit",
'close_rate': 0.005,
'amount': 0.8,
'order_type': 'limit',
'open_rate': 0.004,
'current_rate': 0.005,
'profit_amount': 0.001,
'profit_ratio': 0.20,
'stake_currency': 'BTC',
'sell_reason': ExitType.STOP_LOSS.value
"type": RPCMessageType.EXIT_FILL,
"exchange": "Binance",
"pair": "ETH/BTC",
"gain": "profit",
"close_rate": 0.005,
"amount": 0.8,
"order_type": "limit",
"open_rate": 0.004,
"current_rate": 0.005,
"profit_amount": 0.001,
"profit_ratio": 0.20,
"stake_currency": "BTC",
"sell_reason": ExitType.STOP_LOSS.value,
}
webhook.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookexitfill"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookexitfill"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookexitfill"]["value3"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookexitfill"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookexitfill"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookexitfill"][
"value3"
].format(**msg)
for msgtype in [RPCMessageType.STATUS,
RPCMessageType.WARNING,
RPCMessageType.STARTUP]:
for msgtype in [RPCMessageType.STATUS, RPCMessageType.WARNING, RPCMessageType.STARTUP]:
# Test notification
msg = {
'type': msgtype,
'status': 'Unfilled sell order for BTC cancelled due to timeout'
}
msg = {"type": msgtype, "status": "Unfilled sell order for BTC cancelled due to timeout"}
msg_mock = MagicMock()
mocker.patch("freqtrade.rpc.webhook.Webhook._send_msg", msg_mock)
webhook.send_msg(msg)
assert msg_mock.call_count == 1
assert (msg_mock.call_args[0][0]["value1"] ==
default_conf["webhook"]["webhookstatus"]["value1"].format(**msg))
assert (msg_mock.call_args[0][0]["value2"] ==
default_conf["webhook"]["webhookstatus"]["value2"].format(**msg))
assert (msg_mock.call_args[0][0]["value3"] ==
default_conf["webhook"]["webhookstatus"]["value3"].format(**msg))
assert msg_mock.call_args[0][0]["value1"] == default_conf["webhook"]["webhookstatus"][
"value1"
].format(**msg)
assert msg_mock.call_args[0][0]["value2"] == default_conf["webhook"]["webhookstatus"][
"value2"
].format(**msg)
assert msg_mock.call_args[0][0]["value3"] == default_conf["webhook"]["webhookstatus"][
"value3"
].format(**msg)
def test_exception_send_msg(default_conf, mocker, caplog):
@@ -338,9 +369,8 @@ def test_exception_send_msg(default_conf, mocker, caplog):
del default_conf["webhook"]["webhookentry"]
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
webhook.send_msg({'type': RPCMessageType.ENTRY})
assert log_has(f"Message type '{RPCMessageType.ENTRY}' not configured for webhooks",
caplog)
webhook.send_msg({"type": RPCMessageType.ENTRY})
assert log_has(f"Message type '{RPCMessageType.ENTRY}' not configured for webhooks", caplog)
default_conf["webhook"] = get_webhook_dict()
default_conf["webhook"]["strategy_msg"] = {"value1": "{DEADBEEF:8f}"}
@@ -348,112 +378,103 @@ def test_exception_send_msg(default_conf, mocker, caplog):
mocker.patch("freqtrade.rpc.webhook.Webhook._send_msg", msg_mock)
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
msg = {
'type': RPCMessageType.STRATEGY_MSG,
'msg': 'hello world',
"type": RPCMessageType.STRATEGY_MSG,
"msg": "hello world",
}
webhook.send_msg(msg)
assert log_has("Problem calling Webhook. Please check your webhook configuration. "
"Exception: 'DEADBEEF'", caplog)
assert log_has(
"Problem calling Webhook. Please check your webhook configuration. "
"Exception: 'DEADBEEF'",
caplog,
)
# Test no failure for not implemented but known messagetypes
for e in RPCMessageType:
msg = {
'type': e,
'status': 'whatever'
}
msg = {"type": e, "status": "whatever"}
webhook.send_msg(msg)
# Test no failure for not implemented but known messagetypes
for e in RPCMessageType:
msg = {
'type': e,
'status': 'whatever'
}
msg = {"type": e, "status": "whatever"}
webhook.send_msg(msg)
def test__send_msg(default_conf, mocker, caplog):
default_conf["webhook"] = get_webhook_dict()
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
msg = {'value1': 'DEADBEEF',
'value2': 'ALIVEBEEF',
'value3': 'FREQTRADE'}
msg = {"value1": "DEADBEEF", "value2": "ALIVEBEEF", "value3": "FREQTRADE"}
post = MagicMock()
mocker.patch("freqtrade.rpc.webhook.post", post)
webhook._send_msg(msg)
assert post.call_count == 1
assert post.call_args[1] == {'data': msg, 'timeout': 10}
assert post.call_args[0] == (default_conf['webhook']['url'], )
assert post.call_args[1] == {"data": msg, "timeout": 10}
assert post.call_args[0] == (default_conf["webhook"]["url"],)
post = MagicMock(side_effect=RequestException)
mocker.patch("freqtrade.rpc.webhook.post", post)
webhook._send_msg(msg)
assert log_has('Could not call webhook url. Exception: ', caplog)
assert log_has("Could not call webhook url. Exception: ", caplog)
def test__send_msg_with_json_format(default_conf, mocker, caplog):
default_conf["webhook"] = get_webhook_dict()
default_conf["webhook"]["format"] = "json"
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
msg = {'text': 'Hello'}
msg = {"text": "Hello"}
post = MagicMock()
mocker.patch("freqtrade.rpc.webhook.post", post)
webhook._send_msg(msg)
assert post.call_args[1] == {'json': msg, 'timeout': 10}
assert post.call_args[1] == {"json": msg, "timeout": 10}
def test__send_msg_with_raw_format(default_conf, mocker, caplog):
default_conf["webhook"] = get_webhook_dict()
default_conf["webhook"]["format"] = "raw"
webhook = Webhook(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
msg = {'data': 'Hello'}
msg = {"data": "Hello"}
post = MagicMock()
mocker.patch("freqtrade.rpc.webhook.post", post)
webhook._send_msg(msg)
assert post.call_args[1] == {
'data': msg['data'],
'headers': {'Content-Type': 'text/plain'},
'timeout': 10
"data": msg["data"],
"headers": {"Content-Type": "text/plain"},
"timeout": 10,
}
def test_send_msg_discord(default_conf, mocker):
default_conf["discord"] = {
'enabled': True,
'webhook_url': "https://webhookurl..."
}
default_conf["discord"] = {"enabled": True, "webhook_url": "https://webhookurl..."}
msg_mock = MagicMock()
mocker.patch("freqtrade.rpc.webhook.Webhook._send_msg", msg_mock)
discord = Discord(RPC(get_patched_freqtradebot(mocker, default_conf)), default_conf)
msg = {
'type': RPCMessageType.EXIT_FILL,
'trade_id': 1,
'exchange': 'Binance',
'pair': 'ETH/BTC',
'direction': 'Long',
'gain': "profit",
'close_rate': 0.005,
'amount': 0.8,
'order_type': 'limit',
'open_date': datetime.now() - timedelta(days=1),
'close_date': datetime.now(),
'open_rate': 0.004,
'current_rate': 0.005,
'profit_amount': 0.001,
'profit_ratio': 0.20,
'stake_currency': 'BTC',
'enter_tag': 'enter_tagggg',
'exit_reason': ExitType.STOP_LOSS.value,
"type": RPCMessageType.EXIT_FILL,
"trade_id": 1,
"exchange": "Binance",
"pair": "ETH/BTC",
"direction": "Long",
"gain": "profit",
"close_rate": 0.005,
"amount": 0.8,
"order_type": "limit",
"open_date": datetime.now() - timedelta(days=1),
"close_date": datetime.now(),
"open_rate": 0.004,
"current_rate": 0.005,
"profit_amount": 0.001,
"profit_ratio": 0.20,
"stake_currency": "BTC",
"enter_tag": "enter_tagggg",
"exit_reason": ExitType.STOP_LOSS.value,
}
discord.send_msg(msg=msg)
assert msg_mock.call_count == 1
assert 'embeds' in msg_mock.call_args_list[0][0][0]
assert 'title' in msg_mock.call_args_list[0][0][0]['embeds'][0]
assert 'color' in msg_mock.call_args_list[0][0][0]['embeds'][0]
assert 'fields' in msg_mock.call_args_list[0][0][0]['embeds'][0]
assert "embeds" in msg_mock.call_args_list[0][0][0]
assert "title" in msg_mock.call_args_list[0][0][0]["embeds"][0]
assert "color" in msg_mock.call_args_list[0][0][0]["embeds"][0]
assert "fields" in msg_mock.call_args_list[0][0][0]["embeds"][0]