def test_status_code_error(self):
# raise
with self.assertRaises(exceptions.BetfairError):
raise exceptions.StatusCodeError("test")
# pickle
error = exceptions.StatusCodeError("test")
self.assertEqual(
str(pickle.loads(pickle.dumps(error))),
str(error),
)
def test_poll_market_catalogue_status_error(self, mock_events):
mock_context = mock.Mock()
mock_flumine = mock.Mock()
mock_client = mock.Mock()
mock_flumine.clients.get_betfair_default.return_value = mock_client
mock_market_one = mock.Mock(market_id="1.234",
update_market_catalogue=True,
closed=False)
mock_flumine.markets.markets = {
"1.234": mock_market_one,
}
mock_client.betting_client.betting.list_market_catalogue.side_effect = (
exceptions.StatusCodeError("503"))
worker.poll_market_catalogue(mock_context, mock_flumine)
mock_client.betting_client.betting.list_market_catalogue.assert_called_with(
filter={"marketIds": ["1.234"]},
market_projection=[
"COMPETITION",
"EVENT",
"EVENT_TYPE",
"RUNNER_DESCRIPTION",
"RUNNER_METADATA",
"MARKET_START_TIME",
"MARKET_DESCRIPTION",
],
max_results=25,
)
mock_flumine.handler_queue.put.assert_not_called()
def test__get_cleared_market_status_error(self, mock_events, mock_config):
mock_flumine = mock.Mock()
mock_betting_client = mock.Mock()
mock_betting_client.betting.list_cleared_orders.side_effect = (
exceptions.StatusCodeError("503"))
self.assertFalse(
worker._get_cleared_market(mock_flumine, mock_betting_client,
"1.23"))
mock_betting_client.betting.list_cleared_orders.assert_called_with(
bet_status="SETTLED",
market_ids=["1.23"],
group_by="MARKET",
customer_strategy_refs=[mock_config.customer_strategy_ref],
)