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)