def setUp(self):
self.mock_betting_client = mock.Mock(lightweight=False)
self.base_client = BaseClient(self.mock_betting_client,
1024,
100,
0.02,
True,
username="******")
class BaseClientTest(unittest.TestCase):
def setUp(self):
self.mock_betting_client = mock.Mock()
self.base_client = BaseClient(self.mock_betting_client, 1024, 100, 0.02, True)
def test_init(self):
self.assertEqual(self.base_client.betting_client, self.mock_betting_client)
self.assertEqual(self.base_client.transaction_limit, 1024)
self.assertEqual(self.base_client.capital_base, 100)
self.assertEqual(self.base_client.commission_base, 0.02)
self.assertTrue(self.base_client.interactive_login)
self.assertIsNone(self.base_client.account_details)
self.assertIsNone(self.base_client.account_funds)
self.assertEqual(self.base_client.commission_paid, 0)
self.assertEqual(self.base_client.chargeable_transaction_count, 0)
self.assertIsNone(self.base_client.execution)
self.assertEqual(self.base_client.trading_controls, [])
self.assertTrue(self.base_client.order_stream)
self.assertTrue(self.base_client.best_price_execution)
def test_login(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.login()
def test_keep_alive(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.keep_alive()
def test_logout(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.logout()
def test_update_account_details(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.update_account_details()
def test_add_execution(self):
mock_flumine = mock.Mock()
self.base_client.EXCHANGE = ExchangeType.SIMULATED
self.base_client.add_execution(mock_flumine)
self.assertEqual(self.base_client.execution, mock_flumine.simulated_execution)
self.base_client.EXCHANGE = ExchangeType.BETFAIR
self.base_client.add_execution(mock_flumine)
self.assertEqual(self.base_client.execution, mock_flumine.betfair_execution)
def test_min_bet_size(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bet_size
def test_min_bet_payout(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bet_payout
def test_min_bsp_liability(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bsp_liability
def test_info(self):
self.assertTrue(self.base_client.info)
def setUp(self):
self.mock_betting_client = mock.Mock()
self.base_client = BaseClient(self.mock_betting_client, 1024, 100,
0.02, True)
def test_init_assert(self):
with self.assertRaises(AssertionError):
BaseClient(mock.Mock(lightweight=True), 1024, 100, 0.02, True)
class BaseClientTest(unittest.TestCase):
def setUp(self):
self.mock_betting_client = mock.Mock(lightweight=False)
self.base_client = BaseClient(self.mock_betting_client,
1024,
100,
0.02,
True,
username="******")
def test_init(self):
self.assertEqual(self.base_client.betting_client,
self.mock_betting_client)
self.assertEqual(self.base_client.transaction_limit, 1024)
self.assertEqual(self.base_client.capital_base, 100)
self.assertEqual(self.base_client.commission_base, 0.02)
self.assertTrue(self.base_client.interactive_login)
self.assertEqual(self.base_client._username, "test")
self.assertIsNone(self.base_client.account_details)
self.assertIsNone(self.base_client.account_funds)
self.assertEqual(self.base_client.commission_paid, 0)
self.assertIsNone(self.base_client.execution)
self.assertEqual(self.base_client.trading_controls, [])
self.assertTrue(self.base_client.order_stream)
self.assertTrue(self.base_client.best_price_execution)
self.assertTrue(self.base_client.min_bet_validation)
self.assertFalse(self.base_client.paper_trade)
def test_init_assert(self):
with self.assertRaises(AssertionError):
BaseClient(mock.Mock(lightweight=True), 1024, 100, 0.02, True)
def test_login(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.login()
def test_keep_alive(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.keep_alive()
def test_logout(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.logout()
def test_update_account_details(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.update_account_details()
def test_add_execution(self):
mock_flumine = mock.Mock()
self.base_client.EXCHANGE = ExchangeType.SIMULATED
self.base_client.add_execution(mock_flumine)
self.assertEqual(self.base_client.execution,
mock_flumine.simulated_execution)
self.base_client.EXCHANGE = ExchangeType.BETFAIR
self.base_client.add_execution(mock_flumine)
self.assertEqual(self.base_client.execution,
mock_flumine.betfair_execution)
def test_add_execution_paper(self):
self.base_client.paper_trade = True
self.base_client.EXCHANGE = ExchangeType.BETFAIR
mock_flumine = mock.Mock()
self.base_client.add_execution(mock_flumine)
self.assertEqual(self.base_client.execution,
mock_flumine.simulated_execution)
def test_add_transaction(self):
mock_trading_control = mock.Mock()
self.base_client.trading_controls.append(mock_trading_control)
self.base_client.add_transaction(123, True)
mock_trading_control.add_transaction.assert_called_with(123, True)
def test_current_transaction_count_total(self):
self.assertIsNone(self.base_client.current_transaction_count_total)
mock_trading_control = mock.Mock(NAME="MAX_TRANSACTION_COUNT",
current_transaction_count_total=123)
self.base_client.trading_controls.append(mock_trading_control)
self.assertEqual(self.base_client.current_transaction_count_total, 123)
def test_transaction_count_total(self):
self.assertIsNone(self.base_client.transaction_count_total)
mock_trading_control = mock.Mock(NAME="MAX_TRANSACTION_COUNT",
transaction_count_total=123)
self.base_client.trading_controls.append(mock_trading_control)
self.assertEqual(self.base_client.transaction_count_total, 123)
def test_min_bet_size(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bet_size
def test_min_bet_payout(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bet_payout
def test_min_bsp_liability(self):
with self.assertRaises(NotImplementedError):
assert self.base_client.min_bsp_liability
def test_username(self):
self.assertEqual(self.base_client.username,
self.base_client.betting_client.username)
self.base_client.betting_client = None
self.assertEqual(self.base_client.username, self.base_client._username)
def test_info(self):
self.assertTrue(self.base_client.info)