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
|
import json
import os
import shutil
import tempfile
from unittest.mock import patch
import sys
import msal
import pytest
from msal_extensions import *
from .http_client import MinimalResponse
@pytest.fixture
def temp_location():
test_folder = tempfile.mkdtemp(prefix="test_token_cache_roundtrip")
yield os.path.join(test_folder, 'token_cache.bin')
shutil.rmtree(test_folder, ignore_errors=True)
def _test_token_cache_roundtrip(persistence):
desired_scopes = ['https://graph.microsoft.com/.default']
apps = [ # Multiple apps sharing same persistence
msal.ConfidentialClientApplication(
"fake_client_id", client_credential="fake_client_secret",
token_cache=PersistedTokenCache(persistence)) for i in range(2)]
with patch.object(apps[0].http_client, "post", return_value=MinimalResponse(
status_code=200, text=json.dumps({
"token_type": "Bearer",
"access_token": "app token",
"expires_in": 3600,
}))) as mocked_post:
token1 = apps[0].acquire_token_for_client(scopes=desired_scopes)
assert token1["token_source"] == "identity_provider", "Initial token should come from IdP"
token2 = apps[1].acquire_token_for_client(scopes=desired_scopes) # Hit token cache in MSAL 1.23+
assert token2["token_source"] == "cache", "App2 should hit cache written by app1"
assert token1['access_token'] == token2['access_token'], "Cache should hit"
def test_token_cache_roundtrip_with_persistence_builder(temp_location):
_test_token_cache_roundtrip(build_encrypted_persistence(temp_location))
def test_token_cache_roundtrip_with_file_persistence(temp_location):
_test_token_cache_roundtrip(FilePersistence(temp_location))
def test_file_not_found_error_is_not_raised():
persistence = FilePersistence('non_existing_file')
cache = PersistedTokenCache(persistence)
# An exception raised here will fail the test case as it is supposed to be a NO-OP
cache.find('')
|
