Tests: "no_recv" option introduced.

2018-09-04 02:30:27 +03:00
parent 6bebbe3e8a
commit 9f5349d347
1 changed files with 5 additions and 2 deletions
--- a/test/unit.py
+++ b/test/unit.py
@@ -250,8 +250,11 @@ class TestUnitHTTP(TestUnit):
        if '--verbose' in sys.argv:
            print('>>>', req, sep='\n')

-        encoding = 'utf-8' if 'encoding' not in kwargs else kwargs['encoding']
-        resp = self.recvall(sock).decode(encoding)
+        resp = ''
+
+        if 'no_recv' not in kwargs:
+           enc = 'utf-8' if 'encoding' not in kwargs else kwargs['encoding']
+           resp = self.recvall(sock).decode(enc)

        if '--verbose' in sys.argv:
            print('<<<', resp.encode('utf-8'), sep='\n')