fix byte streaming endpoint in python3

This commit is contained in:
Kevin McCarthy
2014-06-08 17:42:43 -10:00
parent 16d6fea9b8
commit 7e2214ce14
2 changed files with 6 additions and 1 deletions
+1 -1
View File
@@ -481,7 +481,7 @@ def stream_random_bytes(n):
chunks = []
for i in xrange(n):
chunks.append(chr(random.randint(0, 255)))
chunks.append(os.urandom(1))
if len(chunks) == chunk_size:
yield(bytes().join(chunks))
chunks = []
+5
View File
@@ -134,6 +134,11 @@ class HttpbinTestCase(unittest.TestCase):
self.assertEqual(len(response.get_data()), 1024)
self.assertEqual(response.status_code, 200)
def test_stream_bytes(self):
response = self.app.get('/stream-bytes/1024')
self.assertEqual(len(response.get_data()), 1024)
self.assertEqual(response.status_code, 200)
if __name__ == '__main__':
unittest.main()