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
|
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
from unittest import TestCase
from unittest.mock import MagicMock
from azure.communication.administration._shared.user_credential import CommunicationUserCredential
from azure.communication.administration._shared.utils import create_access_token
class TestCommunicationUserCredential(TestCase):
sample_token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9."+\
"eyJleHAiOjMyNTAzNjgwMDAwfQ.9i7FNNHHJT8cOzo-yrAUJyBSfJ-tPPk2emcHavOEpWc"
sample_token_expiry = 32503680000
expired_token = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9."+\
"eyJleHAiOjEwMH0.1h_scYkNp-G98-O4cW6KvfJZwiz54uJMyeDACE4nypg"
def test_communicationusercredential_decodes_token(self):
credential = CommunicationUserCredential(self.sample_token)
access_token = credential.get_token()
self.assertEqual(access_token.token, self.sample_token)
def test_communicationusercredential_throws_if_invalid_token(self):
self.assertRaises(ValueError, lambda: CommunicationUserCredential("foo.bar.tar"))
def test_communicationusercredential_throws_if_nonstring_token(self):
self.assertRaises(TypeError, lambda: CommunicationUserCredential(454))
def test_communicationusercredential_static_token_returns_expired_token(self):
credential = CommunicationUserCredential(self.expired_token)
self.assertEqual(credential.get_token().token, self.expired_token)
def test_communicationusercredential_token_expired_refresh_called(self):
refresher = MagicMock(return_value=self.sample_token)
access_token = CommunicationUserCredential(
self.expired_token,
token_refresher=refresher).get_token()
refresher.assert_called_once()
self.assertEqual(access_token, self.sample_token)
def test_communicationusercredential_token_expired_refresh_called_asnecessary(self):
refresher = MagicMock(return_value=create_access_token(self.expired_token))
credential = CommunicationUserCredential(
self.expired_token,
token_refresher=refresher)
credential.get_token()
access_token = credential.get_token()
self.assertEqual(refresher.call_count, 2)
self.assertEqual(access_token.token, self.expired_token)
|
