x/deb-python-tempest-lib
Code Issues Proposed changes

Add RateLimitExceeded test case

Browse Source 
The rest_client module raises RateLimitExceeded exception, but
there was not any tests for the exception. This patch adds it.

Change-Id: I933ebe2ed52750088e3941b308b669f088d7f8ed
This commit is contained in:
Ken'ichi Ohmichi
2015-01-19 04:46:39 +00:00
parent 84dee6b781
commit 9018af62c4
1 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
tempest_lib/tests/test_rest_client.py
View File

@@ -274,10 +274,15 @@ class TestRestClientParseRespJSON(BaseRestClientTestClass):
class TestRestClientErrorCheckerJSON(base.TestCase): class TestRestClientErrorCheckerJSON(base.TestCase):
c_type = "application/json" c_type = "application/json"
def set_data(self, r_code, enc=None, r_body=None): def set_data(self, r_code, enc=None, r_body=None, absolute_limit=True):
if enc is None: if enc is None:
enc = self.c_type enc = self.c_type
resp_dict = {'status': r_code, 'content-type': enc} resp_dict = {'status': r_code, 'content-type': enc}
resp_body = {'resp_body': 'fake_resp_body'}
if absolute_limit is False:
resp_dict.update({'retry-after': 120})
resp_body.update({'overLimit': {'message': 'fake_message'}})
resp = httplib2.Response(resp_dict) resp = httplib2.Response(resp_dict)
data = { data = {
"method": "fake_method", "method": "fake_method",
@@ -285,7 +290,7 @@ class TestRestClientErrorCheckerJSON(base.TestCase):
"headers": "fake_headers", "headers": "fake_headers",
"body": "fake_body", "body": "fake_body",
"resp": resp, "resp": resp,
"resp_body": '{"resp_body": "fake_resp_body"}', "resp_body": json.dumps(resp_body)
} }
if r_body is not None: if r_body is not None:
data.update({"resp_body": r_body}) data.update({"resp_body": r_body})
@@ -329,6 +334,11 @@ class TestRestClientErrorCheckerJSON(base.TestCase):
self.rest_client._error_checker, self.rest_client._error_checker,
**self.set_data("413")) **self.set_data("413"))
def test_response_413_without_absolute_limit(self):
self.assertRaises(exceptions.RateLimitExceeded,
self.rest_client._error_checker,
**self.set_data("413", absolute_limit=False))
def test_response_415(self): def test_response_415(self):
self.assertRaises(exceptions.InvalidContentType, self.assertRaises(exceptions.InvalidContentType,
self.rest_client._error_checker, self.rest_client._error_checker,
@@ -381,6 +391,11 @@ class TestRestClientErrorCheckerTEXT(TestRestClientErrorCheckerJSON):
self.rest_client._error_checker, self.rest_client._error_checker,
**self.set_data("405", enc="fake_enc")) **self.set_data("405", enc="fake_enc"))
def test_response_413_without_absolute_limit(self):
# Skip this test because rest_client cannot get overLimit message
# from text body.
pass
class TestRestClientUtils(BaseRestClientTestClass): class TestRestClientUtils(BaseRestClientTestClass):