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
|
import pickle
from exchangelib.account import Identity
from exchangelib.credentials import (
Credentials,
OAuth2AuthorizationCodeCredentials,
OAuth2Credentials,
OAuth2LegacyCredentials,
)
from .common import TimedTestCase
class CredentialsTest(TimedTestCase):
def test_hash(self):
# Test that we can use credentials as a dict key
self.assertEqual(hash(Credentials("a", "b")), hash(Credentials("a", "b")))
self.assertNotEqual(hash(Credentials("a", "b")), hash(Credentials("a", "a")))
self.assertNotEqual(hash(Credentials("a", "b")), hash(Credentials("b", "b")))
def test_equality(self):
self.assertEqual(Credentials("a", "b"), Credentials("a", "b"))
self.assertNotEqual(Credentials("a", "b"), Credentials("a", "a"))
self.assertNotEqual(Credentials("a", "b"), Credentials("b", "b"))
def test_type(self):
self.assertEqual(Credentials("a", "b").type, Credentials.UPN)
self.assertEqual(Credentials("a@example.com", "b").type, Credentials.EMAIL)
self.assertEqual(Credentials("a\\n", "b").type, Credentials.DOMAIN)
def test_pickle(self):
# Test that we can pickle, hash, repr, str and compare various credentials types
for o in (
Identity(sid="XXX", upn="YYY", smtp_address="ZZZ", primary_smtp_address="WWW"),
Credentials("XXX", "YYY"),
OAuth2Credentials(client_id="XXX", client_secret="YYY", tenant_id="ZZZZ"),
OAuth2Credentials(
client_id="XXX", client_secret="YYY", tenant_id="ZZZZ", identity=Identity(primary_smtp_address="AAA")
),
OAuth2LegacyCredentials(
client_id="XXX", client_secret="YYY", tenant_id="ZZZZ", username="AAA", password="BBB"
),
OAuth2AuthorizationCodeCredentials(client_id="WWW", client_secret="XXX", authorization_code="YYY"),
OAuth2AuthorizationCodeCredentials(
client_id="WWW", client_secret="XXX", access_token={"access_token": "ZZZ"}
),
OAuth2AuthorizationCodeCredentials(access_token={"access_token": "ZZZ"}),
OAuth2AuthorizationCodeCredentials(
client_id="WWW",
client_secret="XXX",
authorization_code="YYY",
access_token={"access_token": "ZZZ"},
tenant_id="ZZZ",
identity=Identity(primary_smtp_address="AAA"),
),
):
with self.subTest(o=o):
pickled_o = pickle.dumps(o)
unpickled_o = pickle.loads(pickled_o)
self.assertIsInstance(unpickled_o, type(o))
self.assertEqual(o, unpickled_o)
self.assertEqual(hash(o), hash(unpickled_o))
self.assertEqual(repr(o), repr(unpickled_o))
self.assertEqual(str(o), str(unpickled_o))
def test_plain(self):
OAuth2Credentials("XXX", "YYY", "ZZZZ").refresh("XXX") # No-op
def test_oauth_validation(self):
with self.assertRaises(TypeError) as e:
OAuth2AuthorizationCodeCredentials(client_id="WWW", client_secret="XXX", access_token="XXX")
self.assertEqual(
e.exception.args[0],
"'access_token' 'XXX' must be of type <class 'oauthlib.oauth2.rfc6749.tokens.OAuth2Token'>",
)
c = OAuth2AuthorizationCodeCredentials(client_id="WWW", client_secret="XXX", authorization_code="YYY")
self.assertListEqual(c.scope, ["https://outlook.office365.com/.default", "offline_access"])
self.assertDictEqual(c.token_params(), {"include_client_id": True, "code": "YYY"})
c = OAuth2LegacyCredentials(
client_id="XXX", client_secret="YYY", tenant_id="ZZZZ", username="AAA", password="BBB"
)
self.assertListEqual(c.scope, ["https://outlook.office365.com/EWS.AccessAsUser.All"])
self.assertDictEqual(c.token_params(), {"include_client_id": True, "password": "BBB", "username": "AAA"})
c = OAuth2Credentials("XXX", "YYY", "ZZZZ")
self.assertListEqual(c.scope, ["https://outlook.office365.com/.default"])
self.assertDictEqual(c.token_params(), {"include_client_id": True})
c.refresh("XXX") # No-op
with self.assertRaises(TypeError) as e:
c.on_token_auto_refreshed("XXX")
self.assertEqual(
e.exception.args[0],
"'access_token' 'XXX' must be of type <class 'oauthlib.oauth2.rfc6749.tokens.OAuth2Token'>",
)
c.on_token_auto_refreshed(dict(access_token="XXX"))
self.assertIsInstance(c.sig(), int)
|
