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
|
# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import pytest
from azure.core.credentials import AccessToken
from azure.core.exceptions import ClientAuthenticationError, HttpResponseError
from unittest.mock import MagicMock
from azure.communication.identity._shared.token_exchange import (
TokenExchangeClient
)
class DummyPipelineResponse:
def __init__(self, status_code=200, token="tok", expires_on="2999-12-31T23:59:59Z", content=None):
self.http_response = MagicMock()
self.http_response.status_code = status_code
if content is not None:
self.http_response.text.return_value = content
else:
self.http_response.text.return_value = (
f'{{"accessToken": {{"token": "{token}", "expiresOn": "{expires_on}"}}}}'
if status_code == 200 else '{}'
)
class TestTokenExchangeClient:
def test_exchange_entra_token_success(self):
client = TokenExchangeClient("https://endpoint", MagicMock())
dummy_response = DummyPipelineResponse()
client._pipeline = MagicMock()
client._pipeline.run.return_value = dummy_response
token = client.exchange_entra_token()
assert isinstance(token, AccessToken)
assert token.token == "tok"
def test_exchange_entra_token_failure(self):
client = TokenExchangeClient("https://endpoint", MagicMock())
dummy_response = DummyPipelineResponse(status_code=400)
client._pipeline = MagicMock()
client._pipeline.run.return_value = dummy_response
with pytest.raises(HttpResponseError):
client.exchange_entra_token()
def test_parse_access_token_from_response_invalid_json(self):
client = TokenExchangeClient("https://endpoint", MagicMock())
dummy_response = DummyPipelineResponse(content="not json")
with pytest.raises(ValueError):
client._parse_access_token_from_response(dummy_response)
def test_parse_access_token_from_response_missing_fields(self):
client = TokenExchangeClient("https://endpoint", MagicMock())
dummy_response = DummyPipelineResponse(content='{"invalid": "data"}')
with pytest.raises(ValueError):
client._parse_access_token_from_response(dummy_response)
|
