fix byte streaming endpoint in python3

postmanlabs · Jun 9, 2014 · 7e2214c · 7e2214c
1 parent 16d6fea
commit 7e2214c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/httpbin/core.py b/httpbin/core.py
@@ -481,7 +481,7 @@ def generate_bytes():
         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 = []

diff --git a/test_httpbin.py b/test_httpbin.py
@@ -134,6 +134,11 @@ def test_get_bytes(self):
         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()