def testValidOrder(self): tc = TradingCenter() accountId = 'accountId' order1 = Order(accountId=accountId, side=Side.BUY, symbol='symbol', price=13.25, share=10) order2 = Order(accountId='unknowAccount', side=Side.BUY, symbol='symbol', price=13.25, share=10) account = self.mock.CreateMock(Account) account.validate(order1).AndReturn(True) account.validate(order1).AndReturn(False) tc._TradingCenter__accounts = {accountId: account} self.mock.ReplayAll() self.assertEquals(False, tc.validateOrder(order2)) # invalid account id self.assertEquals(True, tc.validateOrder(order1)) # True self.assertEquals(False, tc.validateOrder(order1)) # False self.mock.VerifyAll()
def testPlaceOrder_failed(self): tc = TradingCenter() accountId = 'accountId' order1 = Order(accountId = accountId, side = Side.BUY, symbol = 'symbol', price = 13.25, share = 10) account = self.mock.CreateMock(Account) account.validate(order1).AndReturn(False) tc._TradingCenter__accounts = {accountId: account} self.mock.ReplayAll() self.assertEquals(None, tc.placeOrder(order1) ) # True self.mock.VerifyAll()
def testPlaceOrder_failed(self): tc = TradingCenter() accountId = 'accountId' order1 = Order(accountId=accountId, side=Side.BUY, symbol='symbol', price=13.25, share=10) account = self.mock.CreateMock(Account) account.validate(order1).AndReturn(False) tc._TradingCenter__accounts = {accountId: account} self.mock.ReplayAll() self.assertEquals(None, tc.placeOrder(order1)) # True self.mock.VerifyAll()
def testValidOrder(self): tc = TradingCenter() accountId = 'accountId' order1 = Order(accountId = accountId, side = Side.BUY, symbol = 'symbol', price = 13.25, share = 10) order2 = Order(accountId = 'unknowAccount', side = Side.BUY, symbol = 'symbol', price = 13.25, share = 10) account = self.mock.CreateMock(Account) account.validate(order1).AndReturn(True) account.validate(order1).AndReturn(False) tc._TradingCenter__accounts = {accountId: account} self.mock.ReplayAll() self.assertEquals(False, tc.validateOrder(order2) ) # invalid account id self.assertEquals(True, tc.validateOrder(order1) ) # True self.assertEquals(False, tc.validateOrder(order1) ) # False self.mock.VerifyAll()