diff --git a/test/api_server.py b/test/api_server.py index c5c04187..6ff7589d 100644 --- a/test/api_server.py +++ b/test/api_server.py @@ -23,6 +23,10 @@ users_dict = {} def index(): return "Hello World!" +@app.route('/status_code/') +def get_response_with_status_code(status_code): + return "Status Code: %d" % status_code, status_code + @app.route('/api/users') def get_users(): users_list = [user for uid, user in users_dict.items()] diff --git a/test/test_apiserver.py b/test/test_apiserver.py index 73ae3bc3..02d6b083 100644 --- a/test/test_apiserver.py +++ b/test/test_apiserver.py @@ -1,4 +1,5 @@ import requests +import random from .base import ApiServerUnittest class TestApiServer(ApiServerUnittest): @@ -106,3 +107,9 @@ class TestApiServer(ApiServerUnittest): resp = self.delete_user(1000) self.assertEqual(200, resp.status_code) self.assertEqual(resp.json()['success'], True) + + def test_get_response_with_status_code(self): + status_code = random.randint(200, 511) + url = "%s/status_code/%d" % (self.host, status_code) + resp = self.api_client.get(url) + self.assertEqual(status_code, resp.status_code)