From c21646a01f0a145a7f1d656b5da453c787206a00 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 21 Jun 2017 11:43:21 +0800
Subject: [PATCH] apiserver: add get_response_with_status_code

---
 test/api_server.py     | 4 ++++
 test/test_apiserver.py | 7 +++++++
 2 files changed, 11 insertions(+)

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/<int: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)