Merge pull request #24 from sergeyKogan/master

Added the ability to provide custom response reason
internap · Jun 19, 2020 · 9af761d · 9af761d
2 parents 900f5ab + d89d8c2
commit 9af761d
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/mockserver_friendly/__init__.py b/mockserver_friendly/__init__.py
@@ -51,9 +51,10 @@ def request(method=None, path=None, querystring=None, body=None, headers=None, c
     )
 
 
-def response(code=None, body=None, headers=None, cookies=None, delay=None):
+def response(code=None, body=None, headers=None, cookies=None, delay=None, reason=None):
     return _non_null_options_to_dict(
         _Option("statusCode", code),
+        _Option("reasonPhrase", reason),
         _Option("body", body),
         _Option("headers", headers, formatter=_to_named_values_list),
         _Option("delay", delay, formatter=_to_delay),

diff --git a/test/test_basic_responses.py b/test/test_basic_responses.py
@@ -15,6 +15,16 @@ def test_code_response(self):
         result = requests.get(MOCK_SERVER_URL)
         self.assertEqual(result.status_code, 418)
 
+    def test_reason_phrase_response(self):
+        reason_phrase = "Custom reasonPhrase"
+        self.client.stub(
+            request(),
+            response(reason=reason_phrase)
+        )
+
+        result = requests.get(MOCK_SERVER_URL)
+        self.assertEqual(result.reason, reason_phrase)
+
     def test_body_response(self):
         self.client.stub(
             request(),