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
|
# pylint: disable=missing-docstring
import pytest
from ibm_cloud_sdk_core.authenticators import ContainerAuthenticator, Authenticator
def test_container_authenticator():
authenticator = ContainerAuthenticator(iam_profile_name='iam-user-123')
assert authenticator is not None
assert authenticator.authentication_type() == Authenticator.AUTHTYPE_CONTAINER
assert authenticator.token_manager.cr_token_filename is None
assert authenticator.token_manager.iam_profile_name == 'iam-user-123'
assert authenticator.token_manager.iam_profile_id is None
assert authenticator.token_manager.client_id is None
assert authenticator.token_manager.client_secret is None
assert authenticator.token_manager.disable_ssl_verification is False
assert authenticator.token_manager.headers is None
assert authenticator.token_manager.proxies is None
assert authenticator.token_manager.scope is None
authenticator.set_cr_token_filename('path/to/token')
assert authenticator.token_manager.cr_token_filename == 'path/to/token'
# Set the IAM profile to None to trigger a validation which will fail,
# because both of the profile and ID are None.
with pytest.raises(ValueError) as err:
authenticator.set_iam_profile_name(None)
assert str(err.value) == 'At least one of iam_profile_name or iam_profile_id must be specified.'
authenticator.set_iam_profile_id('iam-id-123')
assert authenticator.token_manager.iam_profile_id == 'iam-id-123'
authenticator.set_iam_profile_name('iam-user-123')
assert authenticator.token_manager.iam_profile_name == 'iam-user-123'
authenticator.set_client_id_and_secret('tom', 'jerry')
assert authenticator.token_manager.client_id == 'tom'
assert authenticator.token_manager.client_secret == 'jerry'
authenticator.set_scope('scope1 scope2 scope3')
assert authenticator.token_manager.scope == 'scope1 scope2 scope3'
with pytest.raises(TypeError) as err:
authenticator.set_headers('dummy')
assert str(err.value) == 'headers must be a dictionary'
authenticator.set_headers({'dummy': 'headers'})
assert authenticator.token_manager.headers == {'dummy': 'headers'}
with pytest.raises(TypeError) as err:
authenticator.set_proxies('dummy')
assert str(err.value) == 'proxies must be a dictionary'
authenticator.set_proxies({'dummy': 'proxies'})
assert authenticator.token_manager.proxies == {'dummy': 'proxies'}
def test_disable_ssl_verification():
authenticator = ContainerAuthenticator(iam_profile_name='iam-user-123', disable_ssl_verification=True)
assert authenticator.token_manager.disable_ssl_verification is True
authenticator.set_disable_ssl_verification(False)
assert authenticator.token_manager.disable_ssl_verification is False
def test_invalid_disable_ssl_verification_type():
with pytest.raises(TypeError) as err:
authenticator = ContainerAuthenticator(iam_profile_name='iam-user-123', disable_ssl_verification='True')
assert str(err.value) == 'disable_ssl_verification must be a bool'
authenticator = ContainerAuthenticator(iam_profile_name='iam-user-123')
assert authenticator.token_manager.disable_ssl_verification is False
with pytest.raises(TypeError) as err:
authenticator.set_disable_ssl_verification('True')
assert str(err.value) == 'status must be a bool'
def test_container_authenticator_with_scope():
authenticator = ContainerAuthenticator(iam_profile_name='iam-user-123', scope='scope1 scope2')
assert authenticator is not None
assert authenticator.token_manager.scope == 'scope1 scope2'
def test_authenticator_validate_failed():
with pytest.raises(ValueError) as err:
ContainerAuthenticator(None)
assert str(err.value) == 'At least one of iam_profile_name or iam_profile_id must be specified.'
with pytest.raises(ValueError) as err:
ContainerAuthenticator(iam_profile_name='iam-user-123', client_id='my_client_id')
assert str(err.value) == 'Both client_id and client_secret should be initialized.'
with pytest.raises(ValueError) as err:
ContainerAuthenticator(iam_profile_name='iam-user-123', client_secret='my_client_secret')
assert str(err.value) == 'Both client_id and client_secret should be initialized.'
|
