From a458f50d59ddddfd9de331f45c712fc3600295ec Mon Sep 17 00:00:00 2001 From: Andrey Zelenkov Date: Mon, 30 Jul 2018 16:40:52 +0300 Subject: Tests: fixed header value char tests. Use byte strings to avoid problems with encoding. --- test/unit.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'test/unit.py') diff --git a/test/unit.py b/test/unit.py index c8820280..dfd8f4fc 100644 --- a/test/unit.py +++ b/test/unit.py @@ -250,7 +250,8 @@ class TestUnitHTTP(TestUnit): if '--verbose' in sys.argv: print('>>>', req, sep='\n') - resp = self.recvall(sock) + encoding = 'utf-8' if 'encoding' not in kwargs else kwargs['encoding'] + resp = self.recvall(sock).decode(encoding) if '--verbose' in sys.argv: print('<<<', resp.encode('utf-8'), sep='\n') @@ -289,7 +290,7 @@ class TestUnitHTTP(TestUnit): if not len(part): break - return data.decode() + return data def _resp_to_dict(self, resp): m = re.search('(.*?\x0d\x0a?)\x0d\x0a?(.*)', resp, re.M | re.S) -- cgit