refactor: move profit test next to each other
This commit is contained in:
Matthias
@@ -967,6 +967,92 @@ async def test_telegram_profit_handle(
|
|||||||
assert "*Trading volume:* `126 USDT`" in msg_mock.call_args_list[-1][0][0]
|
assert "*Trading volume:* `126 USDT`" in msg_mock.call_args_list[-1][0][0]
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_telegram_profit_long_short_handle(
|
||||||
|
default_conf_usdt, update, ticker_usdt, fee, mocker
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
Test the /profit_long and /profit_short commands to ensure the output content
|
||||||
|
is consistent with /profit, covering both no trades and trades present cases.
|
||||||
|
"""
|
||||||
|
|
||||||
|
mocker.patch("freqtrade.rpc.rpc.CryptoToFiatConverter._find_price", return_value=1.1)
|
||||||
|
mocker.patch.multiple(EXMS, fetch_ticker=ticker_usdt, get_fee=fee)
|
||||||
|
telegram, freqtradebot, msg_mock = get_telegram_testobject(mocker, default_conf_usdt)
|
||||||
|
|
||||||
|
# When there are no trades
|
||||||
|
await telegram._profit_long(update=update, context=MagicMock())
|
||||||
|
assert msg_mock.call_count == 1
|
||||||
|
assert "No long trades yet." in msg_mock.call_args_list[0][0][0]
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
# Test support with "/profit long"
|
||||||
|
context = MagicMock()
|
||||||
|
context.args = ["long"]
|
||||||
|
await telegram._profit(update=update, context=context)
|
||||||
|
assert msg_mock.call_count == 1
|
||||||
|
assert "No long trades yet." in msg_mock.call_args_list[0][0][0]
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
await telegram._profit_short(update=update, context=MagicMock())
|
||||||
|
assert msg_mock.call_count == 1
|
||||||
|
assert "No short trades yet." in msg_mock.call_args_list[0][0][0]
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
# Test support with "/profit short"
|
||||||
|
context = MagicMock()
|
||||||
|
context.args = ["short"]
|
||||||
|
await telegram._profit(update=update, context=context)
|
||||||
|
assert msg_mock.call_count == 1
|
||||||
|
assert "No short trades yet." in msg_mock.call_args_list[0][0][0]
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
# When there are trades
|
||||||
|
create_mock_trades_usdt(fee)
|
||||||
|
|
||||||
|
# Keep only long trades
|
||||||
|
for t in Trade.get_trades_proxy():
|
||||||
|
t.is_short = False
|
||||||
|
Trade.commit()
|
||||||
|
await telegram._profit_long(update=update, context=MagicMock())
|
||||||
|
msg = msg_mock.call_args_list[0][0][0]
|
||||||
|
assert "*ROI: Closed long trades*" in msg
|
||||||
|
assert "*ROI: All long trades" in msg
|
||||||
|
assert "*Total Trade Count:*" in msg
|
||||||
|
assert "*Winrate:*" in msg
|
||||||
|
assert "*Expectancy (Ratio):*" in msg
|
||||||
|
assert "*Best Performing:*" in msg
|
||||||
|
assert "*Profit factor:*" in msg
|
||||||
|
assert "*Max Drawdown:*" in msg
|
||||||
|
assert "*Current Drawdown:*" in msg
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
# Keep only short trades
|
||||||
|
for t in Trade.get_trades_proxy():
|
||||||
|
t.is_short = True
|
||||||
|
Trade.commit()
|
||||||
|
await telegram._profit_short(update=update, context=MagicMock())
|
||||||
|
msg = msg_mock.call_args_list[0][0][0]
|
||||||
|
assert "*ROI: Closed short trades*" in msg
|
||||||
|
assert "*ROI: All short trades" in msg
|
||||||
|
assert "*Total Trade Count:*" in msg
|
||||||
|
assert "*Winrate:*" in msg
|
||||||
|
assert "*Expectancy (Ratio):*" in msg
|
||||||
|
assert "*Best Performing:*" in msg
|
||||||
|
assert "*Profit factor:*" in msg
|
||||||
|
assert "*Max Drawdown:*" in msg
|
||||||
|
assert "*Current Drawdown:*" in msg
|
||||||
|
msg_mock.reset_mock()
|
||||||
|
|
||||||
|
# Test parameter passing
|
||||||
|
context = MagicMock()
|
||||||
|
context.args = ["2"]
|
||||||
|
await telegram._profit_long(update=update, context=context)
|
||||||
|
assert msg_mock.call_count == 1
|
||||||
|
await telegram._profit_short(update=update, context=context)
|
||||||
|
assert msg_mock.call_count == 2
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("is_short", [True, False])
|
@pytest.mark.parametrize("is_short", [True, False])
|
||||||
async def test_telegram_stats(default_conf, update, ticker, fee, mocker, is_short) -> None:
|
async def test_telegram_stats(default_conf, update, ticker, fee, mocker, is_short) -> None:
|
||||||
mocker.patch("freqtrade.rpc.rpc.CryptoToFiatConverter._find_price", return_value=15000.0)
|
mocker.patch("freqtrade.rpc.rpc.CryptoToFiatConverter._find_price", return_value=15000.0)
|
||||||
@@ -2982,93 +3068,3 @@ async def test__tg_info(default_conf_usdt, mocker, update):
|
|||||||
content = context.bot.send_message.call_args[1]["text"]
|
content = context.bot.send_message.call_args[1]["text"]
|
||||||
assert "Freqtrade Bot Info:\n" in content
|
assert "Freqtrade Bot Info:\n" in content
|
||||||
assert '"chat_id": "1235"' in content
|
assert '"chat_id": "1235"' in content
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
|
||||||
async def test_telegram_profit_long_short_handle(
|
|
||||||
default_conf_usdt,
|
|
||||||
update,
|
|
||||||
ticker_usdt,
|
|
||||||
fee,
|
|
||||||
mocker,
|
|
||||||
):
|
|
||||||
"""
|
|
||||||
Test the /profit_long and /profit_short commands to ensure the output content
|
|
||||||
is consistent with /profit, covering both no trades and trades present cases.
|
|
||||||
"""
|
|
||||||
|
|
||||||
mocker.patch("freqtrade.rpc.rpc.CryptoToFiatConverter._find_price", return_value=1.1)
|
|
||||||
mocker.patch.multiple(EXMS, fetch_ticker=ticker_usdt, get_fee=fee)
|
|
||||||
telegram, freqtradebot, msg_mock = get_telegram_testobject(mocker, default_conf_usdt)
|
|
||||||
|
|
||||||
# When there are no trades
|
|
||||||
await telegram._profit_long(update=update, context=MagicMock())
|
|
||||||
assert msg_mock.call_count == 1
|
|
||||||
assert "No long trades yet." in msg_mock.call_args_list[0][0][0]
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
# Test support with "/profit long"
|
|
||||||
context = MagicMock()
|
|
||||||
context.args = ["long"]
|
|
||||||
await telegram._profit(update=update, context=context)
|
|
||||||
assert msg_mock.call_count == 1
|
|
||||||
assert "No long trades yet." in msg_mock.call_args_list[0][0][0]
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
await telegram._profit_short(update=update, context=MagicMock())
|
|
||||||
assert msg_mock.call_count == 1
|
|
||||||
assert "No short trades yet." in msg_mock.call_args_list[0][0][0]
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
# Test support with "/profit short"
|
|
||||||
context = MagicMock()
|
|
||||||
context.args = ["short"]
|
|
||||||
await telegram._profit(update=update, context=context)
|
|
||||||
assert msg_mock.call_count == 1
|
|
||||||
assert "No short trades yet." in msg_mock.call_args_list[0][0][0]
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
# When there are trades
|
|
||||||
create_mock_trades_usdt(fee)
|
|
||||||
|
|
||||||
# Keep only long trades
|
|
||||||
for t in Trade.get_trades_proxy():
|
|
||||||
t.is_short = False
|
|
||||||
Trade.commit()
|
|
||||||
await telegram._profit_long(update=update, context=MagicMock())
|
|
||||||
msg = msg_mock.call_args_list[0][0][0]
|
|
||||||
assert "*ROI: Closed long trades*" in msg
|
|
||||||
assert "*ROI: All long trades" in msg
|
|
||||||
assert "*Total Trade Count:*" in msg
|
|
||||||
assert "*Winrate:*" in msg
|
|
||||||
assert "*Expectancy (Ratio):*" in msg
|
|
||||||
assert "*Best Performing:*" in msg
|
|
||||||
assert "*Profit factor:*" in msg
|
|
||||||
assert "*Max Drawdown:*" in msg
|
|
||||||
assert "*Current Drawdown:*" in msg
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
# Keep only short trades
|
|
||||||
for t in Trade.get_trades_proxy():
|
|
||||||
t.is_short = True
|
|
||||||
Trade.commit()
|
|
||||||
await telegram._profit_short(update=update, context=MagicMock())
|
|
||||||
msg = msg_mock.call_args_list[0][0][0]
|
|
||||||
assert "*ROI: Closed short trades*" in msg
|
|
||||||
assert "*ROI: All short trades" in msg
|
|
||||||
assert "*Total Trade Count:*" in msg
|
|
||||||
assert "*Winrate:*" in msg
|
|
||||||
assert "*Expectancy (Ratio):*" in msg
|
|
||||||
assert "*Best Performing:*" in msg
|
|
||||||
assert "*Profit factor:*" in msg
|
|
||||||
assert "*Max Drawdown:*" in msg
|
|
||||||
assert "*Current Drawdown:*" in msg
|
|
||||||
msg_mock.reset_mock()
|
|
||||||
|
|
||||||
# Test parameter passing
|
|
||||||
context = MagicMock()
|
|
||||||
context.args = ["2"]
|
|
||||||
await telegram._profit_long(update=update, context=context)
|
|
||||||
assert msg_mock.call_count == 1
|
|
||||||
await telegram._profit_short(update=update, context=context)
|
|
||||||
assert msg_mock.call_count == 2
|
|
||||||
|
|||||||
Reference in New Issue
Block a user