Send an explicit Content-Length header with /drip/

It's useful for testing tools which track the progress of a request to know
the content length ahead of time.

Test included. I've verified that the new tests do fail without this change,
response.content_length is None if an explicit Content-Length isn't set.
This commit is contained in:
Peter Burns
2014-09-18 16:03:19 -07:00
parent 128bc1b986
commit 5021b420ef
2 changed files with 3 additions and 0 deletions
+1
View File
@@ -419,6 +419,7 @@ def drip():
response = Response(generate_bytes(), headers={
"Content-Type": "application/octet-stream",
"Content-Length": str(numbytes),
})
response.status_code = code
+2
View File
@@ -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)