163 lines
3.7 KiB
Python
163 lines
3.7 KiB
Python
import pytest
|
|
|
|
from freqtrade.persistence.models import Order, Trade
|
|
|
|
|
|
def mock_trade_1(fee):
|
|
trade = Trade(
|
|
pair='ETH/BTC',
|
|
stake_amount=0.001,
|
|
amount=123.0,
|
|
amount_requested=123.0,
|
|
fee_open=fee.return_value,
|
|
fee_close=fee.return_value,
|
|
open_rate=0.123,
|
|
exchange='bittrex',
|
|
open_order_id='dry_run_buy_12345',
|
|
strategy='DefaultStrategy',
|
|
)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': '1234',
|
|
'symbol': 'ETH/BTC',
|
|
'status': 'closed',
|
|
'side': 'buy',
|
|
'price': 0.123,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'buy')
|
|
trade.orders.append(o)
|
|
return trade
|
|
|
|
|
|
def mock_trade_2(fee):
|
|
"""
|
|
Closed trade...
|
|
"""
|
|
trade = Trade(
|
|
pair='ETC/BTC',
|
|
stake_amount=0.001,
|
|
amount=123.0,
|
|
amount_requested=123.0,
|
|
fee_open=fee.return_value,
|
|
fee_close=fee.return_value,
|
|
open_rate=0.123,
|
|
close_rate=0.128,
|
|
close_profit=0.005,
|
|
exchange='bittrex',
|
|
is_open=False,
|
|
open_order_id='dry_run_sell_12345',
|
|
strategy='DefaultStrategy',
|
|
)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': '1235',
|
|
'symbol': 'ETC/BTC',
|
|
'status': 'closed',
|
|
'side': 'buy',
|
|
'price': 0.123,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'buy')
|
|
trade.orders.append(o)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': '12366',
|
|
'symbol': 'ETC/BTC',
|
|
'status': 'closed',
|
|
'side': 'sell',
|
|
'price': 0.128,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'sell')
|
|
trade.orders.append(o)
|
|
return trade
|
|
|
|
|
|
def mock_trade_3(fee):
|
|
"""
|
|
Closed trade
|
|
"""
|
|
trade = Trade(
|
|
pair='XRP/BTC',
|
|
stake_amount=0.001,
|
|
amount=123.0,
|
|
fee_open=fee.return_value,
|
|
fee_close=fee.return_value,
|
|
open_rate=0.05,
|
|
close_rate=0.06,
|
|
close_profit=0.01,
|
|
exchange='bittrex',
|
|
is_open=False,
|
|
)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': '41231a12a',
|
|
'symbol': 'XRP/BTC',
|
|
'status': 'closed',
|
|
'side': 'buy',
|
|
'price': 0.05,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'buy')
|
|
trade.orders.append(o)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': '41231a666a',
|
|
'symbol': 'XRP/BTC',
|
|
'status': 'closed',
|
|
'side': 'stop_loss',
|
|
'price': 0.06,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'sell')
|
|
trade.orders.append(o)
|
|
return trade
|
|
|
|
|
|
def mock_trade_4(fee):
|
|
"""
|
|
Simulate prod entry
|
|
"""
|
|
trade = Trade(
|
|
pair='ETC/BTC',
|
|
stake_amount=0.001,
|
|
amount=123.0,
|
|
amount_requested=124.0,
|
|
fee_open=fee.return_value,
|
|
fee_close=fee.return_value,
|
|
open_rate=0.123,
|
|
exchange='bittrex',
|
|
open_order_id='prod_buy_12345',
|
|
strategy='DefaultStrategy',
|
|
)
|
|
o = Order.parse_from_ccxt_object({
|
|
'id': 'prod_buy_12345',
|
|
'symbol': 'ETC/BTC',
|
|
'status': 'open',
|
|
'side': 'buy',
|
|
'price': 0.123,
|
|
'amount': 123.0,
|
|
'filled': 123.0,
|
|
'remaining': 0.0,
|
|
}, 'ETH/BTC', 'buy')
|
|
trade.orders.append(o)
|
|
|
|
|
|
def create_mock_trades(fee):
|
|
"""
|
|
Create some fake trades ...
|
|
"""
|
|
# Simulate dry_run entries
|
|
trade = mock_trade_1(fee)
|
|
Trade.session.add(trade)
|
|
|
|
trade = mock_trade_2(fee)
|
|
Trade.session.add(trade)
|
|
|
|
trade = mock_trade_3(fee)
|
|
Trade.session.add(trade)
|
|
|
|
trade = mock_trade_4(fee)
|
|
Trade.session.add(trade)
|