def test_make_eb_dir(self):
self.create_config_file()
with patch('os.makedirs') as makedirs_mock:
fileoperations.make_eb_dir('saved_configs')
makedirs_mock.assert_called_once_with(
os.path.join('.elasticbeanstalk', 'saved_configs'))
def test_make_eb_dir__directory_already_exists(self):
self.create_config_file()
os.mkdir(os.path.join('.elasticbeanstalk', 'saved_configs'))
with patch('os.makedirs') as makedirs_mock:
fileoperations.make_eb_dir('saved_configs')
makedirs_mock.assert_not_called()
def test_resolve_config_location_private(self):
cfg_name = 'myfile'
location = 'saved_configs' + os.path.sep + cfg_name + '.cfg.yml'
fileoperations.make_eb_dir('saved_configs')
fileoperations.write_to_eb_data_file(location, self.data)
location = os.path.expanduser(os.getcwd() + os.path.sep +
fileoperations.beanstalk_directory +
location)
result = saved_configs.resolve_config_location(cfg_name)
self.assertEqual(result, location)
def test_update_config_resolve_normal(self, mock_upload):
cfg_name = 'myfile.cfg.yml'
location = 'saved_configs' + os.path.sep + cfg_name
fileoperations.make_eb_dir('saved_configs')
fileoperations.write_to_eb_data_file(location, self.data)
location = os.getcwd() + os.path.sep + \
fileoperations.beanstalk_directory + location
saved_configs.update_config('app', 'myfile')
mock_upload.assert_called_with('app', 'myfile', location)
def test_resolve_config_location_correct_resolve_with_extension(self):
cfg_name = 'myfile.cfg.yml'
location_public = cfg_name
location_private = 'saved_configs' + os.path.sep + cfg_name
fileoperations.make_eb_dir('saved_configs')
fileoperations.write_to_eb_data_file(location_public, self.data)
fileoperations.write_to_eb_data_file(location_private, self.data)
location_public = os.getcwd() + os.path.sep + \
fileoperations.beanstalk_directory + location_public
location_private = os.getcwd() + os.path.sep + \
fileoperations.beanstalk_directory + location_private
result = saved_configs.resolve_config_location(cfg_name)
self.assertEqual(result, location_private)
def download_source_bundle(app_name, env_name):
env = elasticbeanstalk.get_environment(app_name=app_name,
env_name=env_name)
if env.version_label and env.version_label != 'Sample Application':
app_version = elasticbeanstalk.get_application_versions(
app_name,
version_labels=[env.version_label])['ApplicationVersions'][0]
source_bundle = app_version['SourceBundle']
bucket_name = source_bundle['S3Bucket']
key_name = source_bundle['S3Key']
io.echo('Downloading application version...')
data = s3.get_object(bucket_name, key_name)
filename = get_filename(key_name)
else:
# sample app
template = cloudformation.get_template('awseb-' + env.id + '-stack')
try:
url = template['TemplateBody']['Parameters']['AppSource'][
'Default']
except KeyError:
raise NotFoundError('Can not find app source for environment')
utils.get_data_from_url(url)
io.echo('Downloading application version...')
data = utils.get_data_from_url(url, timeout=30)
filename = 'sample.zip'
fileoperations.make_eb_dir('downloads/')
location = fileoperations.get_eb_file_full_location('downloads/' +
filename)
fileoperations.write_to_data_file(location, data)
io.echo('Application version downloaded to:', location)
cwd = os.getcwd()
try:
fileoperations._traverse_to_project_root()
if heuristics.directory_is_empty():
# If we dont have any project code, unzip as current project
io.echo('Unzipping application version as project files.')
fileoperations.unzip_folder(location, os.getcwd())
io.echo('Done.')
finally:
os.chdir(cwd)
def write_to_local_config(cfg_name, data):
fileoperations.make_eb_dir(SAVED_CONFIG_FOLDER_NAME)
file_location = SAVED_CONFIG_FOLDER_NAME + cfg_name + '.cfg.yml'
fileoperations.write_to_eb_data_file(file_location, data)
return fileoperations.get_eb_file_full_location(file_location)
def write_to_local_config(cfg_name, data):
fileoperations.make_eb_dir(SAVED_CONFIG_FOLDER_NAME)
file_location = SAVED_CONFIG_FOLDER_NAME + cfg_name + '.cfg.yml'
fileoperations.write_to_eb_data_file(file_location, data)
return fileoperations.get_eb_file_full_location(file_location)
