def test_add_order_spec(mock_order_spec_class, mock_exchange_class):
exchange = mock_exchange_class.return_value
exchange.options = ExchangeOptions()
exchange.id = "fake_exchange_id"
exchange.name = "coinbase"
exchange.clock = mock.Mock()
exchange.clock.step = 0
wallets = [Wallet(exchange, 10000 * USD), Wallet(exchange, 0 * BTC)]
portfolio = Portfolio(USD, wallets)
# Market order
order = Order(step=0,
exchange_pair=ExchangePair(exchange, USD / BTC),
side=TradeSide.BUY,
trade_type=TradeType.MARKET,
quantity=5000.00 * USD,
portfolio=portfolio,
price=Decimal(7000.00))
order_spec = mock_order_spec_class.return_value
assert len(order._specs) == 0
order.add_order_spec(order_spec)
assert len(order._specs) == 1
assert order_spec in order._specs
def test_add_order_spec(mock_portfolio_class, mock_order_spec_class):
portfolio = mock_portfolio_class.return_value
# Market order
order = Order(side=TradeSide.BUY,
trade_type=TradeType.MARKET,
pair=USD / BTC,
quantity=5000.00 * USD,
portfolio=portfolio,
price=7000.00)
order_spec = mock_order_spec_class.return_value
assert len(order._specs) == 0
order.add_order_spec(order_spec)
assert len(order._specs) == 1
assert order_spec in order._specs
