diff options
-rw-r--r-- | tests/test_common.py | 174 |
1 files changed, 89 insertions, 85 deletions
diff --git a/tests/test_common.py b/tests/test_common.py index 5cf54bf..7f0e35b 100644 --- a/tests/test_common.py +++ b/tests/test_common.py @@ -1,5 +1,4 @@ # -*- coding: utf-8 -*- - import oauthlib from oauthlib.common import ( CaseInsensitiveDict, Request, add_params_to_uri, extract_params, @@ -7,40 +6,43 @@ from oauthlib.common import ( urldecode, ) -from .unittest import TestCase +from tests.unittest import TestCase -PARAMS_DICT = {"foo": "bar", "baz": "123"} -PARAMS_TWOTUPLE = [("foo", "bar"), ("baz", "123")] -PARAMS_FORMENCODED = "foo=bar&baz=123" -URI = "http://www.someuri.com" +PARAMS_DICT = {'foo': 'bar', 'baz': '123', } +PARAMS_TWOTUPLE = [('foo', 'bar'), ('baz', '123')] +PARAMS_FORMENCODED = 'foo=bar&baz=123' +URI = 'http://www.someuri.com' class EncodingTest(TestCase): + def test_urldecode(self): - self.assertCountEqual(urldecode(""), []) - self.assertCountEqual(urldecode("="), [("", "")]) - self.assertCountEqual(urldecode("%20"), [(" ", "")]) - self.assertCountEqual(urldecode("+"), [(" ", "")]) - self.assertCountEqual(urldecode("c2"), [("c2", "")]) - self.assertCountEqual(urldecode("c2="), [("c2", "")]) - self.assertCountEqual(urldecode("foo=bar"), [("foo", "bar")]) - self.assertCountEqual(urldecode("foo_%20~=.bar-"), [("foo_ ~", ".bar-")]) - self.assertCountEqual(urldecode("foo=1,2,3"), [("foo", "1,2,3")]) - self.assertCountEqual(urldecode("foo=(1,2,3)"), [("foo", "(1,2,3)")]) - self.assertCountEqual(urldecode("foo=bar.*"), [("foo", "bar.*")]) - self.assertCountEqual(urldecode("foo=bar@spam"), [("foo", "bar@spam")]) - self.assertCountEqual(urldecode("foo=bar/baz"), [("foo", "bar/baz")]) - self.assertCountEqual(urldecode("foo=bar?baz"), [("foo", "bar?baz")]) - self.assertCountEqual(urldecode("foo=bar's"), [("foo", "bar's")]) - self.assertCountEqual(urldecode("foo=$"), [("foo", "$")]) - self.assertRaises(ValueError, urldecode, "foo bar") - self.assertRaises(ValueError, urldecode, "%R") - self.assertRaises(ValueError, urldecode, "%RA") - self.assertRaises(ValueError, urldecode, "%AR") - self.assertRaises(ValueError, urldecode, "%RR") + self.assertCountEqual(urldecode(''), []) + self.assertCountEqual(urldecode('='), [('', '')]) + self.assertCountEqual(urldecode('%20'), [(' ', '')]) + self.assertCountEqual(urldecode('+'), [(' ', '')]) + self.assertCountEqual(urldecode('c2'), [('c2', '')]) + self.assertCountEqual(urldecode('c2='), [('c2', '')]) + self.assertCountEqual(urldecode('foo=bar'), [('foo', 'bar')]) + self.assertCountEqual(urldecode('foo_%20~=.bar-'), + [('foo_ ~', '.bar-')]) + self.assertCountEqual(urldecode('foo=1,2,3'), [('foo', '1,2,3')]) + self.assertCountEqual(urldecode('foo=(1,2,3)'), [('foo', '(1,2,3)')]) + self.assertCountEqual(urldecode('foo=bar.*'), [('foo', 'bar.*')]) + self.assertCountEqual(urldecode('foo=bar@spam'), [('foo', 'bar@spam')]) + self.assertCountEqual(urldecode('foo=bar/baz'), [('foo', 'bar/baz')]) + self.assertCountEqual(urldecode('foo=bar?baz'), [('foo', 'bar?baz')]) + self.assertCountEqual(urldecode('foo=bar\'s'), [('foo', 'bar\'s')]) + self.assertCountEqual(urldecode('foo=$'), [('foo', '$')]) + self.assertRaises(ValueError, urldecode, 'foo bar') + self.assertRaises(ValueError, urldecode, '%R') + self.assertRaises(ValueError, urldecode, '%RA') + self.assertRaises(ValueError, urldecode, '%AR') + self.assertRaises(ValueError, urldecode, '%RR') class ParameterTest(TestCase): + def test_extract_params_dict(self): self.assertCountEqual(extract_params(PARAMS_DICT), PARAMS_TWOTUPLE) @@ -48,28 +50,30 @@ class ParameterTest(TestCase): self.assertCountEqual(extract_params(PARAMS_TWOTUPLE), PARAMS_TWOTUPLE) def test_extract_params_formencoded(self): - self.assertCountEqual(extract_params(PARAMS_FORMENCODED), PARAMS_TWOTUPLE) + self.assertCountEqual(extract_params(PARAMS_FORMENCODED), + PARAMS_TWOTUPLE) def test_extract_params_blank_string(self): - self.assertCountEqual(extract_params(""), []) + self.assertCountEqual(extract_params(''), []) def test_extract_params_empty_list(self): self.assertCountEqual(extract_params([]), []) def test_extract_non_formencoded_string(self): - self.assertIsNone(extract_params("not a formencoded string")) + self.assertIsNone(extract_params('not a formencoded string')) def test_extract_invalid(self): self.assertIsNone(extract_params(object())) - self.assertIsNone(extract_params([("")])) + self.assertIsNone(extract_params([('')])) def test_add_params_to_uri(self): - correct = "{}?{}".format(URI, PARAMS_FORMENCODED) + correct = '{}?{}'.format(URI, PARAMS_FORMENCODED) self.assertURLEqual(add_params_to_uri(URI, PARAMS_DICT), correct) self.assertURLEqual(add_params_to_uri(URI, PARAMS_TWOTUPLE), correct) class GeneratorTest(TestCase): + def test_generate_timestamp(self): timestamp = generate_timestamp() self.assertIsInstance(timestamp, str) @@ -108,18 +112,21 @@ class GeneratorTest(TestCase): class RequestTest(TestCase): + def test_non_unicode_params(self): r = Request( - b"http://a.b/path?query", - http_method=b"GET", - body=b"you=shall+pass", - headers={b"a": b"b"}, + b'http://a.b/path?query', + http_method=b'GET', + body=b'you=shall+pass', + headers={ + b'a': b'b', + } ) - self.assertEqual(r.uri, "http://a.b/path?query") - self.assertEqual(r.http_method, "GET") - self.assertEqual(r.body, "you=shall+pass") - self.assertEqual(r.decoded_body, [("you", "shall pass")]) - self.assertEqual(r.headers, {"a": "b"}) + self.assertEqual(r.uri, 'http://a.b/path?query') + self.assertEqual(r.http_method, 'GET') + self.assertEqual(r.body, 'you=shall+pass') + self.assertEqual(r.decoded_body, [('you', 'shall pass')]) + self.assertEqual(r.headers, {'a': 'b'}) def test_none_body(self): r = Request(URI) @@ -134,11 +141,11 @@ class RequestTest(TestCase): self.assertEqual(r.decoded_body, []) def test_empty_string_body(self): - r = Request(URI, body="") + r = Request(URI, body='') self.assertEqual(r.decoded_body, []) def test_non_formencoded_string_body(self): - body = "foo bar" + body = 'foo bar' r = Request(URI, body=body) self.assertIsNone(r.decoded_body) @@ -156,58 +163,54 @@ class RequestTest(TestCase): self.assertCountEqual(r.decoded_body, PARAMS_TWOTUPLE) def test_getattr_existing_attribute(self): - r = Request(URI, body="foo bar") - self.assertEqual("foo bar", getattr(r, "body")) + r = Request(URI, body='foo bar') + self.assertEqual('foo bar', getattr(r, 'body')) def test_getattr_return_default(self): - r = Request(URI, body="") - actual_value = getattr(r, "does_not_exist", "foo bar") - self.assertEqual("foo bar", actual_value) + r = Request(URI, body='') + actual_value = getattr(r, 'does_not_exist', 'foo bar') + self.assertEqual('foo bar', actual_value) def test_getattr_raise_attribute_error(self): - r = Request(URI, body="foo bar") + r = Request(URI, body='foo bar') with self.assertRaises(AttributeError): - getattr(r, "does_not_exist") + getattr(r, 'does_not_exist') def test_sanitizing_authorization_header(self): - r = Request( - URI, - headers={ - "Accept": "application/json", - "Authorization": "Basic Zm9vOmJhcg==", - }, - ) - self.assertNotIn("Zm9vOmJhcg==", repr(r)) - self.assertIn("<SANITIZED>", repr(r)) + r = Request(URI, headers={'Accept': 'application/json', + 'Authorization': 'Basic Zm9vOmJhcg=='} + ) + self.assertNotIn('Zm9vOmJhcg==', repr(r)) + self.assertIn('<SANITIZED>', repr(r)) # Double-check we didn't modify the underlying object: - self.assertEqual(r.headers["Authorization"], "Basic Zm9vOmJhcg==") + self.assertEqual(r.headers['Authorization'], 'Basic Zm9vOmJhcg==') def test_token_body(self): - payload = "client_id=foo&refresh_token=bar" + payload = 'client_id=foo&refresh_token=bar' r = Request(URI, body=payload) - self.assertNotIn("bar", repr(r)) - self.assertIn("<SANITIZED>", repr(r)) + self.assertNotIn('bar', repr(r)) + self.assertIn('<SANITIZED>', repr(r)) - payload = "refresh_token=bar&client_id=foo" + payload = 'refresh_token=bar&client_id=foo' r = Request(URI, body=payload) - self.assertNotIn("bar", repr(r)) - self.assertIn("<SANITIZED>", repr(r)) + self.assertNotIn('bar', repr(r)) + self.assertIn('<SANITIZED>', repr(r)) def test_password_body(self): - payload = "username=foo&password=bar" + payload = 'username=foo&password=bar' r = Request(URI, body=payload) - self.assertNotIn("bar", repr(r)) - self.assertIn("<SANITIZED>", repr(r)) + self.assertNotIn('bar', repr(r)) + self.assertIn('<SANITIZED>', repr(r)) - payload = "password=bar&username=foo" + payload = 'password=bar&username=foo' r = Request(URI, body=payload) - self.assertNotIn("bar", repr(r)) - self.assertIn("<SANITIZED>", repr(r)) + self.assertNotIn('bar', repr(r)) + self.assertIn('<SANITIZED>', repr(r)) def test_headers_params(self): - r = Request(URI, headers={"token": "foobar"}, body="token=banana") - self.assertEqual(r.headers["token"], "foobar") - self.assertEqual(r.token, "banana") + r = Request(URI, headers={'token': 'foobar'}, body='token=banana') + self.assertEqual(r.headers['token'], 'foobar') + self.assertEqual(r.token, 'banana') def test_sanitized_request_non_debug_mode(self): """make sure requests are sanitized when in non debug mode. @@ -216,24 +219,25 @@ class RequestTest(TestCase): """ try: oauthlib.set_debug(False) - r = Request(URI, headers={"token": "foobar"}, body="token=banana") - self.assertNotIn("token", repr(r)) - self.assertIn("SANITIZED", repr(r)) + r = Request(URI, headers={'token': 'foobar'}, body='token=banana') + self.assertNotIn('token', repr(r)) + self.assertIn('SANITIZED', repr(r)) finally: # set flag back for other tests oauthlib.set_debug(True) class CaseInsensitiveDictTest(TestCase): + def test_basic(self): cid = CaseInsensitiveDict({}) - cid["a"] = "b" - cid["c"] = "d" - del cid["c"] - self.assertEqual(cid["A"], "b") - self.assertEqual(cid["a"], "b") + cid['a'] = 'b' + cid['c'] = 'd' + del cid['c'] + self.assertEqual(cid['A'], 'b') + self.assertEqual(cid['a'], 'b') def test_update(self): cid = CaseInsensitiveDict({}) - cid.update({"KeY": "value"}) - self.assertEqual(cid["kEy"], "value") + cid.update({'KeY': 'value'}) + self.assertEqual(cid['kEy'], 'value') |