def test_s3_client_created_lazily(self):
session = Mock()
command = CloudTrailValidateLogs(session)
parsed_globals = Mock(region=None, verify_ssl=None, endpoint_url=None)
command.setup_services(None, parsed_globals)
create_client_calls = session.create_client.call_args_list
self.assertEqual(create_client_calls, [call("cloudtrail", verify=None, region_name=None)])
def test_s3_client_created_lazily(self):
session = Mock()
command = CloudTrailValidateLogs(session)
parsed_globals = Mock(region=None, verify_ssl=None, endpoint_url=None)
command.setup_services(parsed_globals)
create_client_calls = session.create_client.call_args_list
self.assertEqual(create_client_calls,
[call('cloudtrail', verify=None, region_name=None)])
def test_endpoint_url_is_used_for_cloudtrail(self):
endpoint_url = "https://mycloudtrail.aws.amazon.com/"
session = Mock()
command = CloudTrailValidateLogs(session)
parsed_globals = Mock(region="foo", verify_ssl=None, endpoint_url=endpoint_url)
command.setup_services(None, parsed_globals)
create_client_calls = session.create_client.call_args_list
self.assertEqual(
create_client_calls, [call("cloudtrail", verify=None, region_name="foo", endpoint_url=endpoint_url)]
)
def test_endpoint_url_is_used_for_cloudtrail(self):
endpoint_url = 'https://mycloudtrail.aws.amazon.com/'
session = Mock()
command = CloudTrailValidateLogs(session)
parsed_globals = Mock(region='foo',
verify_ssl=None,
endpoint_url=endpoint_url)
command.setup_services(parsed_globals)
create_client_calls = session.create_client.call_args_list
self.assertEqual(create_client_calls, [
call('cloudtrail',
verify=None,
region_name='foo',
endpoint_url=endpoint_url)
])
def test_endpoint_url_is_used_for_cloudtrail(self):
endpoint_url = 'https://mycloudtrail.aws.amazon.com/'
session = mock.Mock()
command = CloudTrailValidateLogs(session)
parsed_globals = mock.Mock(region='foo',
verify_ssl=None,
endpoint_url=endpoint_url)
command.setup_services(parsed_globals)
create_client_calls = session.create_client.call_args_list
self.assertEqual(
create_client_calls,
[
mock.call('organizations', verify=None, region_name='foo'),
# Here we should inject the endpoint_url only for cloudtrail.
mock.call('cloudtrail',
verify=None,
region_name='foo',
endpoint_url=endpoint_url)
])
