fix PEP8

jamielennox · Aug 28, 2020 · 8a40520 · 8a40520
1 parent 305e5cd
commit 8a40520
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 11 deletions.
diff --git a/requests_mock/compat.py b/requests_mock/compat.py
@@ -10,8 +10,6 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 
-import requests
-
 
 class _FakeHTTPMessage(object):
 
@@ -33,10 +31,10 @@ def get_all(self, name, failobj=None):
 
 
 class _FakeHTTPResponse(object):
-    
+
     def __init__(self, headers):
         self._headers = headers
         self.msg = _FakeHTTPMessage(headers)
-    
+
     def isclosed(self):
-        return True
+        return True
diff --git a/requests_mock/response.py b/requests_mock/response.py
@@ -130,13 +130,16 @@ def read(self, *args, **kwargs):
 
 def create_set_cookie_header(response, jar):
     """
-    :param requests.packages.urllib3.response response: the Response that will be updated with the Set-Cookie header
-    :param requests.cookies.RequestsCookieJar jar: a CookieJar containing all cookies to be converted in a Set-Cookie header
+    :param requests.packages.urllib3.response response: the Response that will
+        be updated with the Set-Cookie header
+    :param requests.cookies.RequestsCookieJar jar: a CookieJar containing all
+        cookies to be converted in a Set-Cookie header
     """
     set_cookie_items = []
     for cookie_name, cookie_value in response.cookies.items():
-        set_cookie_items.append(cookie_name + '=' + six.moves.urllib.parse.quote(cookie_value))
-
+        encoded = six.moves.urllib.parse.quote(cookie_value)
+        set_cookie_items.append(cookie_name + '=' + encoded)
+
     set_cookie_header = "; ".join(set_cookie_items)
 
     if set_cookie_header:
@@ -206,7 +209,8 @@ def create_response(request, **kwargs):
     _extract_cookies(request, response, kwargs.get('cookies'))
 
     create_set_cookie_header(response, response.cookies)
-    response.raw._original_response = compat._FakeHTTPResponse(response.headers)
+    response.raw._original_response = \
+        compat._FakeHTTPResponse(response.headers)
 
     return response
 

diff --git a/tests/test_session_cookies.py b/tests/test_session_cookies.py
@@ -13,4 +13,4 @@ def test_cookie(self):
                 r = s.get("mock://test.site")
 
                 self.assertEquals(r.cookies.get("name"), "value")
-                self.assertEquals(s.cookies.get("name"), "value")
+                self.assertEquals(s.cookies.get("name"), "value")