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' })