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], )