freqtrade_origin/tests/conftest_trades_usdt.py

389 lines
10 KiB
Python
Raw Normal View History

2021-09-17 08:25:58 +00:00
from datetime import datetime, timedelta, timezone
from freqtrade.persistence.models import Order, Trade
MOCK_TRADE_COUNT = 6
2022-06-10 18:34:17 +00:00
def entry_side(is_short: bool):
return "sell" if is_short else "buy"
def exit_side(is_short: bool):
return "buy" if is_short else "sell"
def direc(is_short: bool):
return "short" if is_short else "long"
def mock_order_usdt_1(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-13 04:46:34 +00:00
'id': f'prod_entry_1_{direc(is_short)}',
'symbol': 'LTC/USDT',
2021-09-17 08:25:58 +00:00
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
2022-06-13 04:46:34 +00:00
'price': 10.0,
'amount': 2.0,
'filled': 2.0,
'remaining': 0.0,
}
def mock_order_usdt_1_exit(is_short: bool):
return {
'id': f'prod_exit_1_{direc(is_short)}',
'symbol': 'LTC/USDT',
'status': 'closed',
'side': exit_side(is_short),
'type': 'limit',
'price': 8.0,
'amount': 2.0,
'filled': 2.0,
2021-09-17 08:25:58 +00:00
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_1(fee, is_short: bool):
2022-06-13 04:46:34 +00:00
"""
Simulate prod entry with open sell order
"""
2021-09-17 08:25:58 +00:00
trade = Trade(
2022-06-13 04:46:34 +00:00
pair='LTC/USDT',
2021-09-17 08:25:58 +00:00
stake_amount=20.0,
2022-06-13 04:46:34 +00:00
amount=2.0,
amount_requested=2.0,
open_date=datetime.now(tz=timezone.utc) - timedelta(days=2, minutes=20),
close_date=datetime.now(tz=timezone.utc) - timedelta(days=2, minutes=5),
2021-09-17 08:25:58 +00:00
fee_open=fee.return_value,
fee_close=fee.return_value,
2022-06-13 04:46:34 +00:00
is_open=False,
open_rate=10.0,
close_rate=8.0,
close_profit=-0.2,
close_profit_abs=-4.09,
2021-09-17 08:25:58 +00:00
exchange='binance',
2022-06-13 04:46:34 +00:00
strategy='SampleStrategy',
open_order_id=f'prod_exit_1_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
timeframe=5,
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-06-13 04:46:34 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_1(is_short), 'LTC/USDT', entry_side(is_short))
trade.orders.append(o)
o = Order.parse_from_ccxt_object(mock_order_usdt_1_exit(is_short),
'LTC/USDT', exit_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-06-10 18:34:17 +00:00
def mock_order_usdt_2(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'1235_{direc(is_short)}',
2022-08-16 08:59:43 +00:00
'symbol': 'NEO/USDT',
2021-09-17 08:25:58 +00:00
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 2.0,
'amount': 100.0,
'filled': 100.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_order_usdt_2_exit(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'12366_{direc(is_short)}',
2022-08-16 08:59:43 +00:00
'symbol': 'NEO/USDT',
2021-09-17 08:25:58 +00:00
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': exit_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 2.05,
'amount': 100.0,
'filled': 100.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_2(fee, is_short: bool):
2021-09-17 08:25:58 +00:00
"""
Closed trade...
"""
trade = Trade(
2022-08-16 08:59:43 +00:00
pair='NEO/USDT',
2021-09-17 08:25:58 +00:00
stake_amount=200.0,
amount=100.0,
amount_requested=100.0,
fee_open=fee.return_value,
fee_close=fee.return_value,
open_rate=2.0,
close_rate=2.05,
close_profit=0.05,
2021-09-17 08:25:58 +00:00
close_profit_abs=3.9875,
exchange='binance',
is_open=False,
2022-06-10 18:34:17 +00:00
open_order_id=f'12366_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
strategy='StrategyTestV2',
timeframe=5,
enter_tag='TEST1',
2022-06-10 18:34:17 +00:00
exit_reason='exit_signal',
2021-09-17 08:25:58 +00:00
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=20),
close_date=datetime.now(tz=timezone.utc) - timedelta(minutes=2),
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-08-16 08:59:43 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_2(is_short), 'NEO/USDT', entry_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(
2022-08-16 08:59:43 +00:00
mock_order_usdt_2_exit(is_short), 'NEO/USDT', exit_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-06-10 18:34:17 +00:00
def mock_order_usdt_3(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'41231a12a_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'XRP/USDT',
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 1.0,
'amount': 30.0,
'filled': 30.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_order_usdt_3_exit(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'41231a666a_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'XRP/USDT',
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': exit_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'stop_loss_limit',
'price': 1.1,
'average': 1.1,
'amount': 30.0,
'filled': 30.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_3(fee, is_short: bool):
2021-09-17 08:25:58 +00:00
"""
Closed trade
"""
trade = Trade(
pair='XRP/USDT',
stake_amount=30.0,
amount=30.0,
amount_requested=30.0,
fee_open=fee.return_value,
fee_close=fee.return_value,
open_rate=1.0,
close_rate=1.1,
close_profit=0.1,
close_profit_abs=2.8425,
2021-09-17 08:25:58 +00:00
exchange='binance',
is_open=False,
strategy='StrategyTestV2',
timeframe=5,
enter_tag='TEST3',
2022-03-24 19:33:47 +00:00
exit_reason='roi',
2021-09-17 08:25:58 +00:00
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=20),
close_date=datetime.now(tz=timezone.utc),
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_3(is_short), 'XRP/USDT', entry_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_3_exit(is_short),
'XRP/USDT', exit_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-06-10 18:34:17 +00:00
def mock_order_usdt_4(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'prod_buy_12345_{direc(is_short)}',
2022-08-16 08:59:43 +00:00
'symbol': 'NEO/USDT',
2021-09-17 08:25:58 +00:00
'status': 'open',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 2.0,
'amount': 10.0,
'filled': 0.0,
'remaining': 30.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_4(fee, is_short: bool):
2021-09-17 08:25:58 +00:00
"""
Simulate prod entry
"""
trade = Trade(
2022-08-16 08:59:43 +00:00
pair='NEO/USDT',
2021-09-17 08:25:58 +00:00
stake_amount=20.0,
amount=10.0,
amount_requested=10.01,
fee_open=fee.return_value,
fee_close=fee.return_value,
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=14),
is_open=True,
open_rate=2.0,
exchange='binance',
2022-06-10 18:34:17 +00:00
open_order_id=f'prod_buy_12345_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
strategy='StrategyTestV2',
timeframe=5,
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-08-16 08:59:43 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_4(is_short), 'NEO/USDT', entry_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-06-10 18:34:17 +00:00
def mock_order_usdt_5(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'prod_buy_3455_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'XRP/USDT',
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 2.0,
'amount': 10.0,
'filled': 10.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_order_usdt_5_stoploss(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'prod_stoploss_3455_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'XRP/USDT',
'status': 'open',
2022-06-10 18:34:17 +00:00
'side': exit_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'stop_loss_limit',
'price': 2.0,
'amount': 10.0,
'filled': 0.0,
'remaining': 30.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_5(fee, is_short: bool):
2021-09-17 08:25:58 +00:00
"""
Simulate prod entry with stoploss
"""
trade = Trade(
pair='XRP/USDT',
stake_amount=20.0,
amount=10.0,
amount_requested=10.01,
fee_open=fee.return_value,
fee_close=fee.return_value,
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=12),
is_open=True,
open_rate=2.0,
exchange='binance',
strategy='SampleStrategy',
2022-06-10 18:34:17 +00:00
stoploss_order_id=f'prod_stoploss_3455_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
timeframe=5,
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_5(is_short), 'XRP/USDT', entry_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_5_stoploss(is_short), 'XRP/USDT', 'stoploss')
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-06-10 18:34:17 +00:00
def mock_order_usdt_6(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'prod_entry_6_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'LTC/USDT',
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 10.0,
'amount': 2.0,
'filled': 2.0,
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_order_usdt_6_exit(is_short: bool):
2021-09-17 08:25:58 +00:00
return {
2022-06-10 18:34:17 +00:00
'id': f'prod_exit_6_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
'symbol': 'LTC/USDT',
'status': 'open',
2022-06-10 18:34:17 +00:00
'side': exit_side(is_short),
2021-09-17 08:25:58 +00:00
'type': 'limit',
'price': 12.0,
'amount': 2.0,
'filled': 0.0,
'remaining': 2.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_6(fee, is_short: bool):
2021-09-17 08:25:58 +00:00
"""
Simulate prod entry with open sell order
"""
trade = Trade(
pair='LTC/USDT',
stake_amount=20.0,
amount=2.0,
amount_requested=2.0,
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=5),
fee_open=fee.return_value,
fee_close=fee.return_value,
is_open=True,
open_rate=10.0,
exchange='binance',
strategy='SampleStrategy',
2022-06-10 18:34:17 +00:00
open_order_id=f'prod_exit_6_{direc(is_short)}',
2021-09-17 08:25:58 +00:00
timeframe=5,
2022-06-10 18:34:17 +00:00
is_short=is_short,
2021-09-17 08:25:58 +00:00
)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_6(is_short), 'LTC/USDT', entry_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
2022-06-10 18:34:17 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_6_exit(is_short),
'LTC/USDT', exit_side(is_short))
2021-09-17 08:25:58 +00:00
trade.orders.append(o)
return trade
2022-02-06 15:02:18 +00:00
2022-06-10 18:34:17 +00:00
def mock_order_usdt_7(is_short: bool):
2022-02-06 15:02:18 +00:00
return {
2022-06-13 04:46:34 +00:00
'id': f'1234_{direc(is_short)}',
'symbol': 'ADA/USDT',
2022-02-06 15:02:18 +00:00
'status': 'closed',
2022-06-10 18:34:17 +00:00
'side': entry_side(is_short),
2022-02-06 15:02:18 +00:00
'type': 'limit',
2022-06-13 04:46:34 +00:00
'price': 2.0,
'amount': 10.0,
'filled': 10.0,
2022-02-06 15:02:18 +00:00
'remaining': 0.0,
}
2022-06-10 18:34:17 +00:00
def mock_trade_usdt_7(fee, is_short: bool):
2022-02-06 15:02:18 +00:00
trade = Trade(
2022-06-13 04:46:34 +00:00
pair='ADA/USDT',
2022-02-06 15:02:18 +00:00
stake_amount=20.0,
2022-06-13 04:46:34 +00:00
amount=10.0,
amount_requested=10.0,
2022-02-06 15:02:18 +00:00
fee_open=fee.return_value,
fee_close=fee.return_value,
2022-06-13 04:46:34 +00:00
is_open=True,
open_date=datetime.now(tz=timezone.utc) - timedelta(minutes=17),
open_rate=2.0,
2022-02-06 15:02:18 +00:00
exchange='binance',
2022-06-13 04:46:34 +00:00
open_order_id=f'1234_{direc(is_short)}',
strategy='StrategyTestV2',
2022-02-06 15:02:18 +00:00
timeframe=5,
2022-06-10 18:34:17 +00:00
is_short=is_short,
2022-02-06 15:02:18 +00:00
)
2022-06-13 04:46:34 +00:00
o = Order.parse_from_ccxt_object(mock_order_usdt_7(is_short), 'ADA/USDT', entry_side(is_short))
2022-02-06 15:02:18 +00:00
trade.orders.append(o)
return trade