def test_cancelled_orders_in_spread_calc(self):
book = OrderBook()
limit_buy_order = Order(ID_GENERATOR, 1, 195, "LIMIT", 100)
limit_sell_order = Order(ID_GENERATOR, 2, 200, "LIMIT", -50)
book.add_order(limit_sell_order)
book.add_order(limit_buy_order)
self.assertEqual(book.current_spread(), 5)
limit_sell_order.cancel()
self.assertEqual(book.current_spread(), 0)
def test_change_to_cancel(self):
order = Order(ID_GENERATOR, AGENT_ID, PRICE, "LIMIT", -100)
self.assertEqual(order.order_state, "ACTIVE")
order.cancel()
self.assertEqual(order.order_state, "CANCELLED")
