def test_flatten_all_positions(self):
# Arrange
strategy = TradingStrategy()
strategy.register(
trader_id=self.trader_id,
portfolio=self.portfolio,
msgbus=self.msgbus,
cache=self.cache,
clock=self.clock,
logger=self.logger,
)
# Start strategy and submit orders to open positions
strategy.start()
order1 = strategy.order_factory.market(
USDJPY_SIM.id,
OrderSide.BUY,
Quantity.from_int(100000),
)
order2 = strategy.order_factory.market(
USDJPY_SIM.id,
OrderSide.BUY,
Quantity.from_int(100000),
)
strategy.submit_order(order1)
self.exchange.process(0)
strategy.submit_order(order2)
self.exchange.process(0)
# Act
strategy.flatten_all_positions(USDJPY_SIM.id)
self.exchange.process(0)
# Assert
assert order1.status == OrderStatus.FILLED
assert order2.status == OrderStatus.FILLED
assert strategy.portfolio.is_completely_flat()
