def test_request_timeout_and_backoff(self, mock_get_token, mock_request):
"""
Check whether the request backoffs properly for request() for 5 times in case of Timeout error.
"""
mock_request.side_effect = Timeout
client = RechargeClient("dummy_access_token", "dummy_user_agent", 300)
with self.assertRaises(Timeout):
client.request("GET")
self.assertEquals(mock_request.call_count, 5)
def test_default_value_request_timeout(self, mock_get, mock_request):
"""
Unit tests to ensure that request timeout is set based default value
"""
config = {"access_token": "dummy_at", "user_agent": "dummy_ua"}
client = RechargeClient(**config)
client.request("GET", "dummy_path")
mock_request.assert_called_with(
'GET',
'https://api.rechargeapps.com/dummy_path',
stream=True,
timeout=600,
headers={
'X-Recharge-Access-Token': 'dummy_at',
'Accept': 'application/json',
'User-Agent': 'dummy_ua'
})