diff --git a/httpbin/core.py b/httpbin/core.py index 2e8774b..1cc9c9c 100644 --- a/httpbin/core.py +++ b/httpbin/core.py @@ -419,6 +419,7 @@ def drip(): response = Response(generate_bytes(), headers={ "Content-Type": "application/octet-stream", + "Content-Length": str(numbytes), }) response.status_code = code diff --git a/test_httpbin.py b/test_httpbin.py index 8af40d8..b19e999 100755 --- a/test_httpbin.py +++ b/test_httpbin.py @@ -177,11 +177,13 @@ class HttpbinTestCase(unittest.TestCase): def test_drip(self): response = self.app.get('/drip?numbytes=400&duration=2&delay=1') + self.assertEqual(response.content_length, 400) self.assertEqual(len(response.get_data()), 400) self.assertEqual(response.status_code, 200) def test_drip_with_custom_code(self): response = self.app.get('/drip?numbytes=400&duration=2&code=500') + self.assertEqual(response.content_length, 400) self.assertEqual(len(response.get_data()), 400) self.assertEqual(response.status_code, 500)