async def test_cancel_open_orders_multi_symbol(self):
config, exchange_manager, trader_inst = await self.init_default()
orders_manager = exchange_manager.exchange_personal_data.orders_manager
# Test buy order
market_buy = BuyMarketOrder(trader_inst)
market_buy.update(order_type=TraderOrderType.BUY_MARKET,
symbol="BTC/USDC",
current_price=70,
quantity=10,
price=70)
# Test buy order
limit_sell = SellLimitOrder(trader_inst)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="NANO/USDT",
current_price=70,
quantity=10,
price=70)
# Test sell order
market_sell = SellMarketOrder(trader_inst)
market_sell.update(order_type=TraderOrderType.SELL_MARKET,
symbol=self.DEFAULT_SYMBOL,
current_price=70,
quantity=10,
price=70)
# Test buy order
limit_buy = BuyLimitOrder(trader_inst)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol=self.DEFAULT_SYMBOL,
current_price=70,
quantity=10,
price=70)
await trader_inst.create_order(market_buy)
await trader_inst.create_order(market_sell)
await trader_inst.create_order(limit_buy)
await trader_inst.create_order(limit_sell)
assert market_buy in orders_manager.get_open_orders()
assert market_sell in orders_manager.get_open_orders()
assert limit_buy in orders_manager.get_open_orders()
assert limit_sell in orders_manager.get_open_orders()
await trader_inst.cancel_open_orders(self.DEFAULT_SYMBOL)
assert market_buy in orders_manager.get_open_orders()
assert market_sell not in orders_manager.get_open_orders()
assert limit_buy not in orders_manager.get_open_orders()
assert limit_sell in orders_manager.get_open_orders()
await self.stop(exchange_manager)
async def test_notify_order_close_with_linked_orders(self):
config, exchange_manager, trader_inst = await self.init_default()
orders_manager = exchange_manager.exchange_personal_data.orders_manager
# Test buy order
market_buy = BuyMarketOrder(trader_inst)
market_buy.update(order_type=TraderOrderType.BUY_MARKET,
symbol="BTC/USDC",
current_price=70,
quantity=10,
price=70)
# Test buy order
limit_sell = SellLimitOrder(trader_inst)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="NANO/USDT",
current_price=70,
quantity=10,
price=70)
# Test stop loss order
stop_loss = StopLossOrder(trader_inst)
stop_loss.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=60,
quantity=10,
price=60)
stop_loss.linked_orders = [limit_sell]
limit_sell.linked_orders = [stop_loss]
await trader_inst.create_order(market_buy)
await trader_inst.create_order(stop_loss)
await trader_inst.create_order(limit_sell)
await trader_inst.notify_order_close(limit_sell)
assert market_buy in orders_manager.get_open_orders()
assert stop_loss not in orders_manager.get_open_orders()
assert limit_sell not in orders_manager.get_open_orders()
await self.stop(exchange_manager)
async def test_update_portfolio_with_multiple_filled_orders(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# Test buy order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=90,
quantity=4,
price=90)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=60,
quantity=2,
price=60)
# Test buy order
limit_buy_2 = BuyLimitOrder(trader)
limit_buy_2.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=4,
price=50)
# Test buy order
limit_buy_3 = BuyLimitOrder(trader)
limit_buy_3.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=46,
quantity=2,
price=46)
# Test buy order
limit_buy_4 = BuyLimitOrder(trader)
limit_buy_4.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=41,
quantity=1.78,
price=41)
# Test buy order
limit_buy_5 = BuyLimitOrder(trader)
limit_buy_5.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=0.2122427,
quantity=3.72448,
price=0.2122427)
# Test buy order
limit_buy_6 = BuyLimitOrder(trader)
limit_buy_6.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=430,
quantity=1.05,
price=430)
# Test sell order
limit_sell_2 = SellLimitOrder(trader)
limit_sell_2.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=10,
quantity=2,
price=10)
# Test stop loss order
stop_loss_2 = StopLossOrder(trader)
stop_loss_2.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=10,
quantity=2,
price=10)
# Test sell order
limit_sell_3 = SellLimitOrder(trader)
limit_sell_3.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=20,
quantity=1,
price=20)
# Test stop loss order
stop_loss_3 = StopLossOrder(trader)
stop_loss_3.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=20,
quantity=1,
price=20)
# Test sell order
limit_sell_4 = SellLimitOrder(trader)
limit_sell_4.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=0.2,
price=50)
# Test stop loss order
stop_loss_4 = StopLossOrder(trader)
stop_loss_4.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=45,
quantity=0.2,
price=45)
# Test sell order
limit_sell_5 = SellLimitOrder(trader)
limit_sell_5.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=11,
quantity=0.7,
price=11)
# Test stop loss order
stop_loss_5 = StopLossOrder(trader)
stop_loss_5.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=9,
quantity=0.7,
price=9)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_2, True)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_3, True)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_4, True)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_5, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_2, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_3, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_4, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_5, True)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_2, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_3, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_4, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_5, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_6, True)
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE), 1) == 2.1
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE), 7) == 62.7295063
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# cancels
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_3, False)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_5, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_2, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_3, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_5, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_4, False)
# filling
await fill_limit_or_stop_order(stop_loss_2, 9, 11)
await fill_limit_or_stop_order(limit_sell, 89, 91)
await fill_limit_or_stop_order(limit_sell_3, 19, 21)
await fill_limit_or_stop_order(limit_buy_2, 49, 51)
await fill_limit_or_stop_order(limit_sell_5, 9, 12)
await fill_limit_or_stop_order(stop_loss_4, 44, 46)
await fill_limit_or_stop_order(limit_buy_4, 40, 42)
await fill_limit_or_stop_order(limit_buy_5, 0.2122426, 0.2122428)
await fill_limit_or_stop_order(limit_buy_6, 429, 431)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, stop_loss_2)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_buy_4)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_sell)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_sell_3)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_buy_2)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_sell_5)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, stop_loss_4)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_buy_5)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_buy_6)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 12.65448
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 692.22
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 12.65448
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL), 7) == 691.4295063
await self.stop_default(exchange_manager)
async def test_reset_portfolio_available(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# Test buy order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=90,
quantity=4,
price=90)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
portfolio_manager.portfolio.reset_portfolio_available()
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# Test sell order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=90,
quantity=4,
price=90)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="VEN/BTC",
current_price=0.5,
quantity=4,
price=0.5)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
# Test buy order
btc_limit_buy = BuyLimitOrder(trader)
btc_limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=10,
quantity=50,
price=10)
portfolio_manager.portfolio.update_portfolio_available(
btc_limit_buy, True)
# Test buy order
btc_limit_buy2 = BuyLimitOrder(trader)
btc_limit_buy2.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=10,
quantity=50,
price=10)
portfolio_manager.portfolio.update_portfolio_available(
btc_limit_buy2, True)
# reset equivalent of the ven buy order
portfolio_manager.portfolio.reset_portfolio_available("BTC", 4 * 0.5)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 6
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 0
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# reset equivalent of the btc buy orders 1 and 2
portfolio_manager.portfolio.reset_portfolio_available("USDT")
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 6
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
await self.stop_default(exchange_manager)
async def test_update_portfolio_with_some_filled_orders(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# Test buy order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=90,
quantity=4,
price=90)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=60,
quantity=2,
price=60)
# Test buy order
limit_buy_2 = BuyLimitOrder(trader)
limit_buy_2.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=4,
price=50)
# Test sell order
limit_sell_2 = SellLimitOrder(trader)
limit_sell_2.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=10,
quantity=2,
price=10)
# Test stop loss order
stop_loss_2 = StopLossOrder(trader)
stop_loss_2.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=10,
quantity=2,
price=10)
# Test sell order
limit_sell_3 = SellLimitOrder(trader)
limit_sell_3.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=20,
quantity=1,
price=20)
# Test stop loss order
stop_loss_3 = StopLossOrder(trader)
stop_loss_3.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=20,
quantity=1,
price=20)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_2, True)
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_3, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_2, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_3, True)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy_2, True)
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE), 1) == 3
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 680
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# cancels
portfolio_manager.portfolio.update_portfolio_available(
stop_loss_3, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell_2, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy, False)
# filling
await fill_limit_or_stop_order(stop_loss_2, 9, 11)
await fill_limit_or_stop_order(limit_sell, 89, 91)
await fill_limit_or_stop_order(limit_sell_3, 19, 21)
await fill_limit_or_stop_order(limit_buy_2, 49, 51)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, stop_loss_2)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_sell)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_sell_3)
await self._update_portfolio_from_order_without_fees(
portfolio_manager, limit_buy_2)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 7
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1200
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 7
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1200
await self.stop_default(exchange_manager)
async def test_update_portfolio_with_stop_loss_orders(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# Test buy order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=90,
quantity=4,
price=90)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=4,
price=50)
# Test stop loss order
stop_loss = StopLossOrder(trader)
stop_loss.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=60,
quantity=4,
price=60)
await fill_limit_or_stop_order(stop_loss, 59, 61)
portfolio_manager.portfolio.update_portfolio_available(stop_loss, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE), 1) == 6
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 800
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# cancel limits
portfolio_manager.portfolio.update_portfolio_available(
limit_buy, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, False)
# filling stop loss
# typical stop loss behavior --> update available before update portfolio
await self._update_portfolio_from_order_without_fees(
portfolio_manager, stop_loss)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 6
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1240
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 6
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1240
await self.stop_default(exchange_manager)
async def test_update_portfolio_with_cancelled_orders(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# force fees => shouldn't do anything
exchange_manager.exchange.config[CONFIG_SIMULATOR][
CONFIG_SIMULATOR_FEES] = {
CONFIG_SIMULATOR_FEES_MAKER: 0.05,
CONFIG_SIMULATOR_FEES_TAKER: 0.1
}
# Test buy order
market_sell = SellMarketOrder(trader)
market_sell.update(order_type=TraderOrderType.SELL_MARKET,
symbol="BTC/USDT",
current_price=80,
quantity=4.1,
price=80)
# Test sell order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=10,
quantity=4.2,
price=10)
# Test stop loss order
stop_loss = StopLossOrder(trader)
stop_loss.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=80,
quantity=3.6,
price=80)
portfolio_manager.portfolio.update_portfolio_available(stop_loss, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=4,
price=50)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
portfolio_manager.portfolio.update_portfolio_available(
market_sell, True)
assert round(
portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE), 1) == 1.7
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 800
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# with no filled orders
portfolio_manager.portfolio.update_portfolio_available(
stop_loss, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_buy, False)
portfolio_manager.portfolio.update_portfolio_available(
market_sell, False)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
await self.stop_default(exchange_manager)
async def test_update_portfolio_with_filled_orders(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# force fees => should have consequences
exchange_manager.exchange.config[CONFIG_SIMULATOR][
CONFIG_SIMULATOR_FEES] = {
CONFIG_SIMULATOR_FEES_MAKER: 0.05,
CONFIG_SIMULATOR_FEES_TAKER: 0.1
}
# Test buy order
market_sell = SellMarketOrder(trader)
market_sell.update(order_type=TraderOrderType.SELL_MARKET,
symbol="BTC/USDT",
current_price=70,
quantity=3,
price=70)
await fill_market_order(market_sell, 70)
# Test sell order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=100,
quantity=4.2,
price=100)
# Test stop loss order
stop_loss = StopLossOrder(trader)
stop_loss.update(order_type=TraderOrderType.STOP_LOSS,
symbol="BTC/USDT",
current_price=80,
quantity=4.2,
price=80)
limit_sell.add_linked_order(stop_loss)
stop_loss.add_linked_order(limit_sell)
# Test buy order
limit_buy = BuyLimitOrder(trader)
limit_buy.update(order_type=TraderOrderType.BUY_LIMIT,
symbol="BTC/USDT",
current_price=50,
quantity=2,
price=50)
await fill_limit_or_stop_order(limit_buy, 49, 51)
# update portfolio with creations
portfolio_manager.portfolio.update_portfolio_available(
market_sell, True)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
portfolio_manager.portfolio.update_portfolio_available(stop_loss, True)
portfolio_manager.portfolio.update_portfolio_available(limit_buy, True)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 2.8
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 900
# when cancelling limit sell, market sell and stop orders
portfolio_manager.portfolio.update_portfolio_available(
stop_loss, False)
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, False)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 7
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 900
# when filling limit buy
await self._update_portfolio_from_order_with_fees(
portfolio_manager, limit_buy)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 8.998
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 900
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 11.998
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 900
# when filling market sell
await self._update_portfolio_from_order_with_fees(
portfolio_manager, market_sell)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 8.998
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1109.79
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 8.998
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1109.79
await self.stop_default(exchange_manager)
async def test_update_portfolio_available(self):
_, exchange_manager, portfolio_manager, trader = await self.init_default(
)
# Test buy order
market_buy = BuyMarketOrder(trader)
market_buy.update(order_type=TraderOrderType.BUY_MARKET,
symbol="BTC/USDT",
current_price=70,
quantity=10,
price=70)
# test buy order creation
portfolio_manager.portfolio.update_portfolio_available(
market_buy, True)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 300
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# test buy order canceled --> return to init state and the update_portfolio will sync TOTAL with AVAILABLE
portfolio_manager.portfolio.update_portfolio_available(
market_buy, False)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# Test sell order
limit_sell = SellLimitOrder(trader)
limit_sell.update(order_type=TraderOrderType.SELL_LIMIT,
symbol="BTC/USDT",
current_price=60,
quantity=8,
price=60)
# test sell order creation
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, True)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 2
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
# test sell order canceled --> return to init state and the update_portfolio will sync TOTAL with AVAILABLE
portfolio_manager.portfolio.update_portfolio_available(
limit_sell, False)
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_AVAILABLE) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_AVAILABLE) == 1000
assert portfolio_manager.portfolio.get_currency_portfolio(
"BTC", PORTFOLIO_TOTAL) == 10
assert portfolio_manager.portfolio.get_currency_portfolio(
"USDT", PORTFOLIO_TOTAL) == 1000
await self.stop_default(exchange_manager)
