-
Notifications
You must be signed in to change notification settings - Fork 47
/
test_client.py
85 lines (78 loc) · 3.15 KB
/
test_client.py
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
from typing import Dict, List, Optional, Type
from unittest import mock
import pytest
from up42.http import client, config
SETTINGS = {"some": "settings"}
PROJECT_CREDENTIALS = {"project_id": "some-id", "project_api_key": "some-key"}
ACCOUNT_CREDENTIALS = {"username": "some-user", "password": "some-pass"}
EMPTY_PROJECT_CREDENTIALS = {"project_id": None, "project_api_key": None}
EMPTY_ACCOUNT_CREDENTIALS = {"username": None, "password": None}
TOKEN_URL = "some-token-url"
class TestCreate:
unreachable = mock.MagicMock()
@pytest.mark.parametrize(
"sources, detected_settings",
[
[[EMPTY_PROJECT_CREDENTIALS, ACCOUNT_CREDENTIALS], [None, SETTINGS]],
[[PROJECT_CREDENTIALS, EMPTY_ACCOUNT_CREDENTIALS], [SETTINGS, None]],
],
)
def test_should_create_if_only_one_source_is_given(
self,
sources: List[Optional[Dict]],
detected_settings: List[Optional[Dict]],
):
detect_settings = mock.MagicMock(side_effect=detected_settings)
access_token = "some-token"
retrieve = mock.sentinel.some_object
detect_retriever = mock.MagicMock(return_value=retrieve)
auth = mock.MagicMock()
auth.token.access_token = access_token
create_auth = mock.MagicMock(return_value=auth)
session = mock.MagicMock()
create_session = mock.MagicMock(return_value=session)
result = client.create(
credential_sources=sources,
token_url=TOKEN_URL,
detect_settings=detect_settings,
detect_retriever=detect_retriever,
create_auth=create_auth,
create_session=create_session,
)
assert result.token == access_token
assert result.auth == auth
assert result.session == session
detect_settings.assert_has_calls([mock.call(source) for source in sources])
detect_retriever.assert_called_with(SETTINGS)
create_auth.assert_called_with(retrieve, config.TokenProviderSettings(token_url=TOKEN_URL))
create_session.assert_called_with(auth)
@pytest.mark.parametrize(
"sources, settings, error",
[
[
[PROJECT_CREDENTIALS, ACCOUNT_CREDENTIALS],
SETTINGS,
client.MultipleCredentialsSources,
],
[
[EMPTY_PROJECT_CREDENTIALS, EMPTY_ACCOUNT_CREDENTIALS],
None,
client.MissingCredentials,
],
],
)
def test_fails_to_create_if_multiple_or_no_sources_are_given(
self, sources: List[Optional[Dict]], settings, error: Type[ValueError]
):
detect_settings = mock.MagicMock(return_value=settings)
with pytest.raises(error):
client.create(
credential_sources=sources,
token_url=TOKEN_URL,
detect_settings=detect_settings,
detect_retriever=self.unreachable,
create_auth=self.unreachable,
create_session=self.unreachable,
)
detect_settings.assert_has_calls([mock.call(source) for source in sources])
self.unreachable.assert_not_called()