def setUp(self) -> None: self.mock_package_type = mock.Mock() self.mock_client = mock.Mock() self.mock_order = mock.Mock() self.mock_order.status = OrderStatus.PENDING self.order_package = BetfairOrderPackage( self.mock_client, "1.234", [self.mock_order], self.mock_package_type, 0, )
class BetfairOrderPackageTest(unittest.TestCase): def setUp(self) -> None: self.mock_market = mock.Mock() self.mock_package_type = mock.Mock() self.mock_client = mock.Mock() self.mock_order = mock.Mock() self.mock_order.status = OrderStatus.PENDING self.order_package = BetfairOrderPackage( self.mock_client, "1.234", [self.mock_order], self.mock_package_type, self.mock_market, ) def test_init(self): self.assertEqual(self.order_package.EXCHANGE, ExchangeType.BETFAIR) def test_place_instructions(self): self.assertEqual( self.order_package.place_instructions, [self.mock_order.create_place_instruction()], ) def test_cancel_instructions(self): self.assertEqual( self.order_package.cancel_instructions, [self.mock_order.create_cancel_instruction()], ) def test_update_instructions(self): self.assertEqual( self.order_package.update_instructions, [self.mock_order.create_update_instruction()], ) def test_replace_instructions(self): self.assertEqual( self.order_package.replace_instructions, [self.mock_order.create_replace_instruction()], ) def test_order_limit(self): self.assertEqual( self.order_package.order_limit(OrderPackageType.PLACE), 200) self.assertEqual( self.order_package.order_limit(OrderPackageType.CANCEL), 60) self.assertEqual( self.order_package.order_limit(OrderPackageType.UPDATE), 60) self.assertEqual( self.order_package.order_limit(OrderPackageType.REPLACE), 60)