def test_configure_from_config_file(self):
# this is not a known configuration so this is ignored.
config_file_vars = {
'foo': 'bar',
CDP_ACCESS_KEY_ID_KEY_NAME: 'key_id',
CDP_PRIVATE_KEY_KEY_NAME: 'mysecretkey'
}
context = FakeContext(
all_variables={'config_file': '/config/location'},
config_file_vars=config_file_vars)
context.context_var_map = {'region': ('region', "CDP_REGION")}
context.full_config = {
'profiles': {
'default': {
'region': 'CDP_REGION'
}
}
}
stream = six.StringIO()
self.configure_list = ConfigureListCommand(stream)
self.configure_list(context, args=[], parsed_globals=None)
rendered = stream.getvalue()
self.assertRegexpMatches(rendered,
'profile\\s+<not set>\\s+None\\s+None')
self.assertRegexpMatches(
rendered, 'cdp_access_key_id\\s+\\*+y_id\\s+config-file')
self.assertRegexpMatches(rendered,
'cdp_private_key\\s+\\*+tkey\\s+config-file')
def test_configure_from_env(self):
env_vars = {'profile': 'myprofilename'}
context = FakeContext(
all_variables={'config_file': '/config/location'},
environment_vars=env_vars)
context.context_var_map = {'profile': (None, "PROFILE_ENV_VAR")}
context.full_config = {
'profiles': {
'default': {
'region': 'CDP_REGION'
}
}
}
stream = six.StringIO()
self.configure_list = ConfigureListCommand(stream)
self.configure_list(context, args=[], parsed_globals=None)
rendered = stream.getvalue()
self.assertRegexpMatches(
rendered, 'profile\\s+myprofilename\\s+env\\s+PROFILE_ENV_VAR')
def test_configure_from_multiple_sources(self):
# Here the profile is from an env var, the
# region is from the config file, and the credentials
# are from an iam-role.
env_vars = {'profile': 'myprofilename'}
credentials = mock.Mock()
credentials.access_key_id = 'access_key'
credentials.private_key = 'private_key'
credentials.method = 'foobar'
context = FakeContext(
all_variables={'config_file': '/config/location'},
environment_vars=env_vars,
config_file_vars={},
credentials=credentials)
context.context_var_map = {
'region': ('region', 'CDP_REGION'),
'profile': ('profile', 'CDP_DEFAULT_PROFILE')
}
context.full_config = {
'profiles': {
'default': {
'region': 'CDP_REGION'
}
}
}
stream = six.StringIO()
self.configure_list = ConfigureListCommand(stream)
self.configure_list(context, args=[], parsed_globals=None)
rendered = stream.getvalue()
# The profile came from an env var.
self.assertRegexpMatches(
rendered, 'profile\\s+myprofilename\\s+env\\s+CDP_DEFAULT_PROFILE')
# The credentials came from 'foobar'. Note how we're
# also checking that the access_key/private_key are masked
# with '*' chars except for the last 4 chars.
self.assertRegexpMatches(rendered,
r'cdp_access_key_id\s+\*+_key\s+foobar')
self.assertRegexpMatches(rendered,
r'cdp_private_key\s+\*+_key\s+foobar')