def setup(self):
self.mock_conn = Mock()
self.mock_client_config = Mock()
type(self.mock_client_config).region_name = 'us-east-1'
type(self.mock_conn)._client_config = self.mock_client_config
self.cls = TrustedAdvisor({}, {})
self.cls.conn = self.mock_conn
def test_boto_kwargs(self):
mock_svc = Mock(spec_set=_AwsService)
mock_svc.get_limits.return_value = {}
boto_args = dict(region_name='myregion',
aws_access_key_id='myaccesskey',
aws_secret_access_key='mysecretkey',
aws_session_token='mytoken')
cls = TrustedAdvisor(
{'foo': mock_svc},
boto_args,
ta_refresh_mode=123,
ta_refresh_timeout=456
)
assert cls.conn is None
cls_boto_args = cls._boto3_connection_kwargs
assert cls_boto_args.get('region_name') == 'us-east-1'
assert cls_boto_args.get('aws_access_key_id') == 'myaccesskey'
assert cls_boto_args.get('aws_secret_access_key') == 'mysecretkey'
assert cls_boto_args.get('aws_session_token') == 'mytoken'
assert cls.ta_region == 'myregion'
assert cls.all_services == {'foo': mock_svc}
assert cls.limits_updated is False
assert cls.refresh_mode == 123
assert cls.refresh_timeout == 456
def test_init_sts(self):
cls = TrustedAdvisor(account_id='aid', account_role='role', region='r')
assert cls.conn is None
assert cls.account_id == 'aid'
assert cls.account_role == 'role'
assert cls.region == 'us-east-1'
assert cls.ta_region == 'r'
assert cls.external_id is None
def test_init(self):
cls = TrustedAdvisor()
assert cls.conn is None
assert cls.account_id is None
assert cls.account_role is None
assert cls.region == 'us-east-1'
assert cls.ta_region is None
assert cls.external_id is None
def test_simple(self):
cls = TrustedAdvisor({}, {})
assert cls.conn is None
assert cls._boto3_connection_kwargs == {'region_name': 'us-east-1'}
assert cls.all_services == {}
assert cls.limits_updated is False
assert cls.refresh_mode is None
assert cls.refresh_timeout is None
def test_connect(self):
cls = TrustedAdvisor()
mock_conn = Mock(spec_set=SupportConnection, name='mock_conn')
with patch('awslimitchecker.trustedadvisor.boto.connect_support'
'',
autospec=True) as mock_connect:
mock_connect.return_value = mock_conn
cls.connect()
assert cls.conn == mock_conn
assert mock_connect.mock_calls == [call()]
def test_connect_again(self):
cls = TrustedAdvisor()
mock_original_conn = Mock(spec_set=SupportConnection)
cls.conn = mock_original_conn
mock_conn = Mock(spec_set=SupportConnection)
with patch('awslimitchecker.trustedadvisor.boto.connect_support'
'') as mock_connect:
mock_connect.return_value = mock_conn
cls.connect()
assert cls.conn == mock_original_conn
assert mock_connect.mock_calls == []
def setup(self):
self.mock_conn = Mock(spec_set=SupportConnection)
type(self.mock_conn).region = RegionInfo(name='us-east-1')
self.cls = TrustedAdvisor()
self.cls.conn = self.mock_conn
self.mock_svc1 = Mock(spec_set=_AwsService)
self.mock_svc2 = Mock(spec_set=_AwsService)
self.services = {
'SvcFoo': self.mock_svc1,
'SvcBar': self.mock_svc2,
}
def test_init(self):
cls = TrustedAdvisor({})
assert cls.conn is None
assert cls.account_id is None
assert cls.account_role is None
assert cls.region == 'us-east-1'
assert cls.ta_region is None
assert cls.external_id is None
assert cls.mfa_serial_number is None
assert cls.mfa_token is None
assert cls.all_services == {}
assert cls.limits_updated is False
def test_init_sts_external_id(self):
cls = TrustedAdvisor(account_id='aid',
account_role='role',
region='r',
external_id='myeid')
assert cls.conn is None
assert cls.account_id == 'aid'
assert cls.account_role == 'role'
assert cls.region == 'us-east-1'
assert cls.ta_region == 'r'
assert cls.external_id == 'myeid'
assert cls.mfa_serial_number is None
assert cls.mfa_token is None
def setup(self):
self.mock_conn = Mock()
self.mock_client_config = Mock()
type(self.mock_client_config).region_name = 'us-east-1'
type(self.mock_conn)._client_config = self.mock_client_config
self.cls = TrustedAdvisor()
self.cls.conn = self.mock_conn
self.mock_svc1 = Mock(spec_set=_AwsService)
self.mock_svc2 = Mock(spec_set=_AwsService)
self.services = {
'SvcFoo': self.mock_svc1,
'SvcBar': self.mock_svc2,
}
def test_connect_region(self):
cls = TrustedAdvisor(account_id='foo', account_role='bar', region='re')
mock_conn = Mock(spec_set=SupportConnection, name='mock_conn')
mock_conn_via = Mock(spec_set=SupportConnection, name='mock_conn')
with patch('awslimitchecker.trustedadvisor.TrustedAdvisor.connect_via'
'') as mock_connect_via:
mock_connect_via.return_value = mock_conn_via
with patch(
'awslimitchecker.trustedadvisor.boto.connect_support'
'',
autospec=True) as mock_connect:
mock_connect.return_value = mock_conn
cls.connect()
assert cls.conn == mock_conn_via
assert mock_connect.mock_calls == []
assert mock_connect_via.mock_calls == [call(connect_to_region)]
def test_init_sts(self):
mock_svc = Mock(spec_set=_AwsService)
mock_svc.get_limits.return_value = {}
cls = TrustedAdvisor({'foo': mock_svc},
account_id='aid',
account_role='role',
region='r')
assert cls.conn is None
assert cls.account_id == 'aid'
assert cls.account_role == 'role'
assert cls.region == 'us-east-1'
assert cls.ta_region == 'r'
assert cls.external_id is None
assert cls.mfa_serial_number is None
assert cls.mfa_token is None
assert cls.all_services == {'foo': mock_svc}
assert cls.limits_updated is False
