def test_existing_profiles_are_preserved(self):
write_config_file(self.TEST_FILE,
'[test]',
'output = none',
'region = us-west-2',
'aws_access_key_id = TEST_KEY',
'aws_secret_access_key = TEST_ACCESS',
'aws_security_token = TEST_TOKEN',
'aws_session_token = TEST_TOKEN')
AwsCredentialsFile(self.TEST_FILE).add_profile(
'dev', 'un-west-5', Struct({'access_key': 'ACCESS_KEY',
'secret_key': 'SECRET_KEY',
'security_token': 'SESSION_TOKEN',
'session_token': 'SESSION_TOKEN',
'expiration': 'TEST_EXPIRATION'}))
six.assertCountEqual(self, read_config_file(self.TEST_FILE),
['[test]',
'region = us-west-2',
'aws_access_key_id = TEST_KEY',
'aws_secret_access_key = TEST_ACCESS',
'output = none',
'aws_security_token = TEST_TOKEN',
'aws_session_token = TEST_TOKEN',
'',
'[dev]',
'output = json',
'region = un-west-5',
'aws_access_key_id = ACCESS_KEY',
'aws_secret_access_key = SECRET_KEY',
'aws_security_token = SESSION_TOKEN',
'aws_session_token = SESSION_TOKEN',
''])
def test_existing_profiles_are_preserved(self):
write_config_file(self.TEST_FILE, '[test]', 'output = none',
'region = us-west-2', 'aws_access_key_id = TEST_KEY',
'aws_secret_access_key = TEST_ACCESS',
'aws_security_token = TEST_TOKEN',
'aws_session_token = TEST_TOKEN')
AwsCredentialsFile(self.TEST_FILE).add_profile(
'dev', 'un-west-5',
Struct({
'access_key': 'ACCESS_KEY',
'secret_key': 'SECRET_KEY',
'security_token': 'SESSION_TOKEN',
'session_token': 'SESSION_TOKEN',
'expiration': 'TEST_EXPIRATION'
}))
six.assertCountEqual(self, read_config_file(self.TEST_FILE), [
'[test]', 'region = us-west-2', 'aws_access_key_id = TEST_KEY',
'aws_secret_access_key = TEST_ACCESS', 'output = none',
'aws_security_token = TEST_TOKEN',
'aws_session_token = TEST_TOKEN', '', '[dev]', 'output = json',
'region = un-west-5', 'aws_access_key_id = ACCESS_KEY',
'aws_secret_access_key = SECRET_KEY',
'aws_security_token = SESSION_TOKEN',
'aws_session_token = SESSION_TOKEN', ''
])
def test_profile_is_updated(self):
write_config_file(self.TEST_FILE,
'[dev]',
'output = none',
'region = us-west-2',
'aws_access_key_id = OLD',
'aws_secret_access_key = REDUNDANT',
'aws_session_token = EXPIRED')
AwsCredentialsFile(self.TEST_FILE).add_profile(
'dev', 'un-west-5', Struct({'access_key': 'ACCESS_KEY',
'secret_key': 'SECRET_KEY',
'session_token': 'SESSION_TOKEN',
'expiration': 'TEST_EXPIRATION'}), True)
self.assertItemsEqual(read_config_file(self.TEST_FILE),
['[dev]',
'region = un-west-5',
'aws_access_key_id = ACCESS_KEY',
'aws_secret_access_key = SECRET_KEY',
'output = json',
'aws_security_token = SESSION_TOKEN',
'aws_session_token = SESSION_TOKEN',
''])
