Fix test suite for aiohttp>=0.22

aio-libs · Feb 21, 2016 · d5842c4 · d5842c4
1 parent 009db41
commit d5842c4
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/tests/test_get_session.py b/tests/test_get_session.py
@@ -2,6 +2,7 @@
 import unittest
 from unittest import mock
 
+import aiohttp
 from aiohttp.web import Request, HttpVersion
 from aiohttp import CIMultiDict, RawRequestMessage
 
@@ -22,8 +23,14 @@ def tearDown(self):
     def make_request(self, method, path, headers=CIMultiDict(), *,
                      version=HttpVersion(1, 1), closing=False):
         self.app = mock.Mock()
-        message = RawRequestMessage(method, path, version, headers, closing,
-                                    False)
+        if tuple(int(i) for i in aiohttp.__version__.split('.')) < (0, 21):
+            message = RawRequestMessage(method, path, version, headers,
+                                        closing, False)
+        else:
+            message = RawRequestMessage(method, path, version, headers,
+                                        [(k.encode('utf-8'), v.encode('utf-8'))
+                                         for k, v in headers.items()],
+                                        closing, False)
         self.payload = mock.Mock()
         self.transport = mock.Mock()
         self.writer = mock.Mock()