From f3e6726098220701dc2193c440852d04508cf972 Mon Sep 17 00:00:00 2001
From: Max Romanov <max.romanov@nginx.com>
Date: Thu, 12 Mar 2020 17:54:15 +0300
Subject: Tests: added Python input readline and iterator tests.

---
 test/python/input_iter/wsgi.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

(limited to 'test/python/input_iter/wsgi.py')

diff --git a/test/python/input_iter/wsgi.py b/test/python/input_iter/wsgi.py
index d3bf437f..04a6a06c 100644
--- a/test/python/input_iter/wsgi.py
+++ b/test/python/input_iter/wsgi.py
@@ -1,5 +1,16 @@
 def application(environ, start_response):
-    body = bytes(environ['wsgi.input'].__iter__())
+    body = []
+    content_length = 0
 
-    start_response('200', [('Content-Length', str(len(body)))])
-    return [body]
+    for l in environ['wsgi.input'].__iter__():
+        body.append(l)
+        content_length += len(l)
+
+    start_response(
+        '200',
+        [
+            ('Content-Length', str(content_length)),
+            ('X-Lines-Count', str(len(body))),
+        ],
+    )
+    return body
-- 
cgit