From 4a025164a681781e9aff993aa0aa0aff042dbb62 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 21 Jun 2017 13:19:31 +0800
Subject: [PATCH] apiserver: add get_response_with_headers

---
 test/api_server.py     |  9 +++++++++
 test/test_apiserver.py | 10 ++++++++++
 2 files changed, 19 insertions(+)

diff --git a/test/api_server.py b/test/api_server.py
index 6ff7589d..6c01e2ec 100644
--- a/test/api_server.py
+++ b/test/api_server.py
@@ -27,6 +27,15 @@ def index():
 def get_response_with_status_code(status_code):
     return "Status Code: %d" % status_code, status_code
 
+@app.route('/response_headers', methods=['POST'])
+def get_response_with_headers():
+    headers_dict = request.get_json()
+    content = "Response headers: %s" % json.dumps(headers_dict)
+    response = make_response(content)
+    for header_key, header_value in headers_dict.items():
+        response.headers[header_key] = header_value
+    return response
+
 @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 02d6b083..09ef5650 100644
--- a/test/test_apiserver.py
+++ b/test/test_apiserver.py
@@ -113,3 +113,13 @@ class TestApiServer(ApiServerUnittest):
         url = "%s/status_code/%d" % (self.host, status_code)
         resp = self.api_client.get(url)
         self.assertEqual(status_code, resp.status_code)
+
+    def test_get_response_with_headers(self):
+        headers = {
+            'abc': 123,
+            'def': 456
+        }
+        url = "%s/response_headers" % self.host
+        resp = self.api_client.post(url, json=headers)
+        self.assertIn('abc', resp.headers)
+        self.assertIn('123', resp.headers['abc'])