def test_move_changed_order(self): """Test that Supervisor will amend order, not cancel it.""" order1 = Order(order_type='Limit', qty=228, price=1000, side='Buy') order1_copy = Order(order_type='Limit', qty=228, price=1000, side='Buy') self.supervisor.add_order(order1) self.exchange_mock.get_open_orders_ws.return_value = [order1_copy] order1.move(to=1001) self.supervisor.cancel_needless_orders() self.exchange_mock.bulk_cancel_orders.assert_not_called() self.exchange_mock.move_order.assert_called_once_with(order=order1)
def test_move_bad_order(self): # Nothing to move: no price or stop_px order = Order(order_type='Stop', qty=228, side='Buy') with self.assertRaises(RuntimeError): order.move(to=1234)
def test_move_limit_order(self): order = Order(order_type='Limit', qty=228, side='Buy', price=1000) order.move(to=1001) self.assertEqual(1001, order.price)
def test_move_stop_order(self): order = Order(order_type='Stop', qty=228, side='Buy', stop_px=1000) order.move(to=1001) self.assertEqual(1001, order.stop_px)