def testCancelAllOpenOrders(self):
security=Stock(symbol='symbol', description='a stock', ISIN='US123456789', exchange=self.exchange)
order1=BuyOrder(account=self.account, security=security, price=13.2, share=10)
order2=BuyOrder(account=self.account, security=security, price=13.25, share=10)
self.session.add(order1)
self.session.add(order2)
self.session.commit()
tc=TradingCenter(self.session, self.pubsub)
tc.cancel_orders()
self.assertEquals([], tc.open_orders(security))
def testCancelOrder(self):
stock = Stock(symbol='symbol',
description='a stock',
ISIN='US123456789',
exchange=self.exchange)
order1 = BuyOrder(account=self.account,
security=stock,
price=13.2,
share=10)
order2 = BuyOrder(account=self.account,
security=stock,
price=13.25,
share=10)
self.session.add(order1)
self.session.add(order2)
self.session.commit()
tc = TradingCenter(self.session, self.pubsub)
order1.cancel()
self.assertEquals([order2], tc.open_orders(stock))
# cancel open orders
self.assertEquals([order2], tc.cancel_orders())
self.assertEquals(CancelOrderStage, type(order1.current_stage))
self.assertEquals([], tc.open_orders(stock))
def testCancelAllOpenOrders(self):
security = Stock(symbol='symbol',
description='a stock',
ISIN='US123456789',
exchange=self.exchange)
order1 = BuyOrder(account=self.account,
security=security,
price=13.2,
share=10)
order2 = BuyOrder(account=self.account,
security=security,
price=13.25,
share=10)
self.session.add(order1)
self.session.add(order2)
self.session.commit()
tc = TradingCenter(self.session, self.pubsub)
tc.cancel_orders()
self.assertEquals([], tc.open_orders(security))
def testCancelOrder(self):
stock=Stock(symbol='symbol', description='a stock', ISIN='US123456789', exchange=self.exchange)
order1=BuyOrder(account=self.account, security=stock, price=13.2, share=10)
order2=BuyOrder(account=self.account, security=stock, price=13.25, share=10)
self.session.add(order1)
self.session.add(order2)
self.session.commit()
tc=TradingCenter(self.session, self.pubsub)
order1.cancel()
self.assertEquals([order2], tc.open_orders(stock))
# cancel open orders
self.assertEquals([order2], tc.cancel_orders())
self.assertEquals(CancelOrderStage, type(order1.current_stage))
self.assertEquals([], tc.open_orders(stock))
