Tests: added read_buffer_size option in http().

1 files changed, 8 insertions, 1 deletions

diff --git a/test/unit/http.py b/test/unit/http.py
index 01611221..82a6bd6a 100644
--- a/test/unit/http.py
+++ b/test/unit/http.py
@@ -12,6 +12,11 @@ class TestHTTP(TestUnit):
         port = 7080 if 'port' not in kwargs else kwargs['port']
         url = '/' if 'url' not in kwargs else kwargs['url']
         http = 'HTTP/1.0' if 'http_10' in kwargs else 'HTTP/1.1'
+        read_buffer_size = (
+            4096
+            if 'read_buffer_size' not in kwargs
+            else kwargs['read_buffer_size']
+        )
 
         headers = (
             {'Host': 'localhost', 'Connection': 'close'}
@@ -94,7 +99,9 @@ class TestHTTP(TestUnit):
             read_timeout = (
                 30 if 'read_timeout' not in kwargs else kwargs['read_timeout']
             )
-            resp = self.recvall(sock, read_timeout=read_timeout).decode(enc)
+            resp = self.recvall(
+                sock, read_timeout=read_timeout, buff_size=read_buffer_size
+            ).decode(enc)
 
         if TestUnit.detailed:
             print('<<<')