1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
from __future__ import absolute_import, unicode_literals
import datetime
import os
from oauthlib.common import PY3
from oauthlib.oauth2.rfc6749.utils import (escape, generate_age, host_from_uri,
is_secure_transport, list_to_scope,
params_from_uri, scope_to_list)
from ...unittest import TestCase
class ScopeObject:
"""
Fixture for testing list_to_scope()/scope_to_list() with objects other
than regular strings.
"""
def __init__(self, scope):
self.scope = scope
if PY3:
def __str__(self):
return self.scope
else:
def __unicode__(self):
return self.scope
class UtilsTests(TestCase):
def test_escape(self):
"""Assert that we are only escaping unicode"""
self.assertRaises(ValueError, escape, b"I am a string type. Not a unicode type.")
self.assertEqual(escape("I am a unicode type."), "I%20am%20a%20unicode%20type.")
def test_host_from_uri(self):
"""Test if hosts and ports are properly extracted from URIs.
This should be done according to the MAC Authentication spec.
Defaults ports should be provided when none is present in the URI.
"""
self.assertEqual(host_from_uri('http://a.b-c.com:8080'), ('a.b-c.com', '8080'))
self.assertEqual(host_from_uri('https://a.b.com:8080'), ('a.b.com', '8080'))
self.assertEqual(host_from_uri('http://www.example.com'), ('www.example.com', '80'))
self.assertEqual(host_from_uri('https://www.example.com'), ('www.example.com', '443'))
def test_is_secure_transport(self):
"""Test check secure uri."""
if 'OAUTHLIB_INSECURE_TRANSPORT' in os.environ:
del os.environ['OAUTHLIB_INSECURE_TRANSPORT']
self.assertTrue(is_secure_transport('https://example.com'))
self.assertFalse(is_secure_transport('http://example.com'))
os.environ['OAUTHLIB_INSECURE_TRANSPORT'] = '1'
self.assertTrue(is_secure_transport('http://example.com'))
del os.environ['OAUTHLIB_INSECURE_TRANSPORT']
def test_params_from_uri(self):
self.assertEqual(params_from_uri('http://i.b/?foo=bar&g&scope=a+d'),
{'foo': 'bar', 'g': '', 'scope': ['a', 'd']})
def test_generate_age(self):
issue_time = datetime.datetime.now() - datetime.timedelta(
days=3, minutes=1, seconds=4)
self.assertGreater(float(generate_age(issue_time)), 259263.0)
def test_list_to_scope(self):
expected = 'foo bar baz'
string_list = ['foo', 'bar', 'baz']
self.assertEqual(list_to_scope(string_list), expected)
string_tuple = ('foo', 'bar', 'baz')
self.assertEqual(list_to_scope(string_tuple), expected)
obj_list = [ScopeObject('foo'), ScopeObject('bar'), ScopeObject('baz')]
self.assertEqual(list_to_scope(obj_list), expected)
set_list = set(string_list)
set_scope = list_to_scope(set_list)
assert len(set_scope.split(' ')) == 3
for x in string_list:
assert x in set_scope
self.assertRaises(ValueError, list_to_scope, object())
def test_scope_to_list(self):
expected = ['foo', 'bar', 'baz']
string_scopes = 'foo bar baz '
self.assertEqual(scope_to_list(string_scopes), expected)
string_list_scopes = ['foo', 'bar', 'baz']
self.assertEqual(scope_to_list(string_list_scopes), expected)
tuple_list_scopes = ('foo', 'bar', 'baz')
self.assertEqual(scope_to_list(tuple_list_scopes), expected)
obj_list_scopes = [ScopeObject('foo'), ScopeObject('bar'), ScopeObject('baz')]
self.assertEqual(scope_to_list(obj_list_scopes), expected)
set_list_scopes = set(string_list_scopes)
set_list = scope_to_list(set_list_scopes)
self.assertEqual(sorted(set_list), sorted(string_list_scopes))
|