diff --git a/mockserver_friendly/__init__.py b/mockserver_friendly/__init__.py index 403470b..ed61a23 100644 --- 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 index 097675a..11cb947 100644 --- 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(),