def test_build(self):
builder = Builder('test.zip')
builder._runtime = RUNTIME_NODE_JS
ok_(hasattr(builder.build(), 'read'))
with PyZipFile(builder._zippath, 'r', compression=ZIP_DEFLATED) as zipfile:
ok_('lambda_function.pyc' in zipfile.namelist())
ok_('.lamvery_secret.json' in zipfile.namelist())
def test_archive_file(self):
builder = Builder('test.zip')
builder._archive_file(
self.zipfile, os.path.join(self.pj_root, 'setup.py'))
ok_(isinstance(self.zipfile.getinfo('setup.pyc'), zipfile.ZipInfo))
builder._archive_file(
self.zipfile, os.path.join(self.pj_root, 'README.md'))
ok_(isinstance(self.zipfile.getinfo('README.md'), zipfile.ZipInfo))
def action(self):
archive_name = self._config.get_archive_name()
function_filename = self._config.get_function_filename()
secret = self._config.generate_lambda_secret()
exclude = self._config.get_exclude()
builder = Builder(
filename=archive_name,
function_filename=function_filename,
secret=secret,
single_file=self._single_file,
no_libs=self._no_libs,
exclude=exclude,
runtime=self._config.get_runtime(),
env=self._env,
clean_build=self._config.is_clean_build(),
hooks=self._config.get_build_hooks())
zipfile = builder.build()
with open(archive_name, 'w') as f:
f.write(zipfile.read())
zipfile.close()
self._logger.info('Output archive(zip) to {}'.format(archive_name))
def action(self):
archive_name = self._config.get_archive_name()
function_filename = self._config.get_function_filename()
secret = self._config.generate_lambda_secret()
exclude = self._config.get_exclude()
builder = Builder(
filename=archive_name,
function_filename=function_filename,
secret=secret,
single_file=self._single_file,
no_libs=self._no_libs,
exclude=exclude,
runtime=self._config.get_runtime(),
env=self._env,
clean_build=self._config.is_clean_build(),
hooks=self._config.get_build_hooks())
func_name = self._config.get_function_name()
local_conf = self._config.get_configuration()
zipfile = builder.build()
client = self.get_lambda_client()
remote_conf = client.get_function_conf(func_name)
alias_name = self._set_alias.get_alias_name()
remote_size = client.calculate_capacity()
local_size = builder.get_size()
new_version = None
cur_version = None
vpc_config = self._config.get_vpc_configuration()
if len(remote_conf) == 0:
self._logger.info(
'[Function] Create new function "{}"'.format(func_name))
self._print_diff(
prefix='[Function]',
remote=remote_conf, local=local_conf, keys=CONF_DIFF_KEYS)
self._print_diff(
prefix='[Function-VPC]',
remote=remote_conf.get('VpcConfig', {}), local=vpc_config, keys=VPC_DIFF_KEYS)
if len(remote_conf) > 0:
if self._enable_versioning():
cur_version = client.get_alias(
func_name, alias_name).get('FunctionVersion')
else:
local_size -= remote_conf['CodeSize']
client.update_function_conf(local_conf)
self._print_capacity(remote=remote_size, local=local_size)
new_version = client.update_function_code(
zipfile, local_conf, self._enable_versioning())
else:
if self._enable_versioning():
local_size *= 2
self._print_capacity(
remote=remote_size, local=local_size)
new_version = client.create_function(
zipfile, local_conf, self._enable_versioning())
zipfile.close()
if new_version is not None:
self._logger.info(
'[Function] Deployed version: {}'.format(new_version))
if cur_version is not None:
self._logger.info(
'[Function] Previous version: {}'.format(cur_version))
self._set_alias._alias = previous_alias(alias_name)
self._set_alias._version = cur_version
self._set_alias.action()
if alias_name is not None:
self._set_alias._alias = alias_name
self._set_alias._version = new_version
self._set_alias.action()
def test_is_exclude_file(self):
builder = Builder('test.zip')
eq_(builder.is_exclude_file('test.zip'), True)
eq_(builder.is_exclude_file('foo.txt'), False)
builder.is_exclude = Mock(return_value=True)
eq_(builder.is_exclude_file('foo.txt'), True)
def test_is_exclude(self):
builder = Builder('test.zip', exclude=['^\.lamvery\.yml$'])
eq_(builder.is_exclude('foo.txt'), False)
eq_(builder.is_exclude('.lamvery.yml'), True)
def test_archive_single_file(self):
self._single_file = True
builder = Builder('test.zip', single_file=True)
builder._archive_file(
self.zipfile, os.path.join(self.pj_root, 'setup.py'))
ok_(isinstance(self.zipfile.getinfo('setup.py'), zipfile.ZipInfo))
def test_archive_dist(self):
builder = Builder('test.zip')
builder._archive_dist(self.zipfile, 'lamvery.js')
ok_(isinstance(self.zipfile.getinfo('lamvery.js'), zipfile.ZipInfo))
def test_archive_dir(self):
builder = Builder('test.zip')
builder._archive_dir(self.zipfile, self.pj_root)
ok_(isinstance(self.zipfile.getinfo('setup.pyc'), zipfile.ZipInfo))
def test_generate_json(self):
builder = Builder('test.zip')
builder._generate_json(
JSON_FILE_NAME, {'foo': 2, 'bar': 3})
data = json.load(open(JSON_FILE_NAME, 'r'))
eq_(data.get('foo'), 2)
def test_build_with_single_file(self):
builder = Builder('test.zip', function_filename='lambda_function.py', single_file=True)
builder.build()
with PyZipFile(builder._zippath, 'r', compression=ZIP_DEFLATED) as zipfile:
ok_('lambda_function.py' in zipfile.namelist())
ok_(not ('.lamvery_secret.json' in zipfile.namelist()))
def test_get_size(self):
builder = Builder('test.zip')
builder.build()
ok_(isinstance(builder.get_size(), int))
def test_get_paths(self):
builder = Builder('test.zip')
paths = builder._get_paths()
ok_(os.path.join(self.pj_root, 'lamvery') in paths)
builder = Builder('test.zip', no_libs=True)
paths = builder._get_paths()
ok_(os.path.join(self.pj_root, 'lamvery') in paths)
ok_(os.path.join(self.pj_root, 'lambda_function.py') in paths)
ok_(os.path.join(self.pj_root, 'lambda_function.pyc') in paths)
ok_(os.path.join(self.pj_root, '.lamvery.yml') in paths)
builder = Builder('test.zip', function_filename='test.py', single_file=True)
paths = builder._get_paths()
ok_(os.path.join(self.pj_root, 'test.py') in paths)
ok_(not os.path.join(self.pj_root, 'lambda_function.pyc') in paths)
ok_(not os.path.join(self.pj_root, '.lamvery.yml') in paths)
builder = Builder('test.zip', clean_build=True)
paths = builder._get_paths()
ok_(os.path.join(self.pj_root, 'lamvery') not in paths)
del os.environ['VIRTUAL_ENV']
builder = Builder('test.zip')
paths = builder._get_paths()
ok_(os.path.join(self.pj_root, 'lamvery') in paths)
def test_is_source_file(self):
builder = Builder('test.zip')
eq_(builder.is_source_file('foo.py'), True)
eq_(builder.is_source_file('foo.pyc'), True)
eq_(builder.is_source_file('foo.php'), False)