minor PEP 8 fixes, compatibility with python 2.7

pull/26/head
Jean-Hugues de Raigniac 2021-04-26 12:45:36 +04:00
parent 606fd8f891
commit 55c7f4138a
1 changed files with 8 additions and 2 deletions

View File

@ -1,8 +1,8 @@
#!/usr/bin/env python
from __future__ import absolute_import, division, print_function, unicode_literals
from __future__ import absolute_import, division, print_function
import os, sys, unittest, json, logging, base64
import os, sys, unittest, logging, base64
import requests
from requests.adapters import HTTPAdapter
@ -13,8 +13,10 @@ from requests_http_signature import HTTPSignatureAuth, HTTPSignatureHeaderAuth,
hmac_secret = b"monorail_cat"
passphrase = b"passw0rd"
class TestAdapter(HTTPAdapter):
def __init__(self, testcase):
super(TestAdapter, self).__init__()
self.testcase = testcase
def send(self, request, *args, **kwargs):
@ -33,12 +35,16 @@ class TestAdapter(HTTPAdapter):
response.url = request.url
return response
class DigestlessSignatureAuth(HTTPSignatureAuth):
def add_digest(self, request):
pass
class TestRequestsHTTPSignature(unittest.TestCase):
def setUp(self):
if not hasattr(unittest.TestCase, 'assertRaisesRegex'):
setattr(unittest.TestCase, 'assertRaisesRegex', unittest.TestCase.assertRaisesRegexp)
logging.basicConfig(level="DEBUG")
self.session = requests.Session()
self.session.mount("http://", TestAdapter(self))