def test_build_session_krb_delegation_as_str(self):
winrm_transport = transport.Transport(
endpoint="Endpoint",
server_cert_validation='validate',
username='******',
password='******',
auth_method='kerberos',
kerberos_delegation='True')
self.assertTrue(winrm_transport.kerberos_delegation)
def test_build_session_cert_validate_default(self):
t_default = transport.Transport(
endpoint="https://example.com",
username='******',
password='******',
auth_method='basic',
)
t_default.build_session()
self.assertEqual(True, t_default.session.verify)
def test_build_session_no_password(self):
with self.assertRaises(InvalidCredentialsError) as exc:
transport.Transport(
endpoint="Endpoint",
server_cert_validation='validate',
username='******',
auth_method='basic',
)
self.assertEqual("auth method basic requires a password",
str(exc.exception))
def test_build_session_krb_delegation_as_invalid_str(self):
with self.assertRaises(ValueError) as exc:
transport.Transport(endpoint="Endpoint",
server_cert_validation='validate',
username='******',
password='******',
auth_method='kerberos',
kerberos_delegation='invalid_value')
self.assertEqual("invalid truth value 'invalid_value'",
str(exc.exception))
def test_build_session_cert_validate_default_env(self):
os.environ['REQUESTS_CA_BUNDLE'] = 'path_to_REQUESTS_CA_CERT'
t_default = transport.Transport(
endpoint="https://example.com",
username='******',
password='******',
auth_method='basic',
)
t_default.build_session()
self.assertEqual('path_to_REQUESTS_CA_CERT', t_default.session.verify)
def test_close_session_not_built(self, mock_session):
t_default = transport.Transport(
endpoint="Endpoint",
server_cert_validation='ignore',
username='******',
password='******',
auth_method='basic',
)
t_default.close_session()
self.assertFalse(mock_session.return_value.close.called)
self.assertIsNone(t_default.session)
def test_build_session_server_cert_validation_invalid(self):
with self.assertRaises(WinRMError) as exc:
transport.Transport(
endpoint="Endpoint",
server_cert_validation='invalid_value',
username='******',
password='******',
auth_method='basic',
)
self.assertEqual('invalid server_cert_validation mode: invalid_value',
str(exc.exception))
def test_build_session_invalid_encryption(self):
with self.assertRaises(WinRMError) as exc:
transport.Transport(endpoint="Endpoint",
server_cert_validation='validate',
username='******',
password='******',
auth_method='basic',
message_encryption='invalid_value')
self.assertEqual(
"invalid message_encryption arg: invalid_value. Should be 'auto', 'always', or 'never'",
str(exc.exception))
def test_build_session_cert_override_2(self):
os.environ['CURL_CA_BUNDLE'] = 'path_to_CURL_CA_CERT'
t_default = transport.Transport(
endpoint="https://example.com",
server_cert_validation='validate',
username='******',
password='******',
auth_method='basic',
ca_trust_path='overridepath',
)
t_default.build_session()
self.assertEqual('overridepath', t_default.session.verify)
def test_build_session_invalid_auth(self):
winrm_transport = transport.Transport(
endpoint="Endpoint",
server_cert_validation='validate',
username='******',
password='******',
auth_method='invalid_value',
)
with self.assertRaises(WinRMError) as exc:
winrm_transport.build_session()
self.assertEqual("unsupported auth method: invalid_value",
str(exc.exception))
def test_build_session_proxy_with_env_http(self):
os.environ['HTTP_PROXY'] = 'random_proxy'
t_default = transport.Transport(
endpoint="https://example.com",
server_cert_validation='validate',
username='******',
password='******',
auth_method='basic',
)
t_default.build_session()
self.assertEqual({'http': 'random_proxy'}, t_default.session.proxies)
def test_build_session_proxy_defined(self):
t_default = transport.Transport(endpoint="https://example.com",
server_cert_validation='validate',
username='******',
password='******',
auth_method='basic',
proxy='test_proxy')
t_default.build_session()
self.assertEqual({
'http': 'test_proxy',
'https': 'test_proxy'
}, t_default.session.proxies)
def test_build_session_cert_ignore_2(self):
os.environ['REQUESTS_CA_BUNDLE'] = 'path_to_REQUESTS_CA_CERT'
os.environ['CURL_CA_BUNDLE'] = 'path_to_CURL_CA_CERT'
t_default = transport.Transport(endpoint="https://example.com",
server_cert_validation='ignore',
username='******',
password='******',
auth_method='basic',
ca_trust_path='boguspath')
t_default.build_session()
self.assertIs(False, t_default.session.verify)
