def test_can_generate_package(self):
generator = mock.Mock(spec=packager.LambdaDeploymentPackager)
generator.create_deployment_package.return_value = 'package.zip'
package = models.DeploymentPackage(models.Placeholder.BUILD_STAGE)
config = Config.create()
p = DeploymentPackager(generator)
p.handle(config, package)
assert package.filename == 'package.zip'
def test_package_not_generated_if_filename_populated(self):
generator = mock.Mock(spec=packager.LambdaDeploymentPackager)
generator.create_deployment_package.return_value = 'NEWPACKAGE.zip'
package = models.DeploymentPackage(filename='original-name.zip')
config = Config.create()
p = DeploymentPackager(generator)
p.handle(config, package)
assert package.filename == 'original-name.zip'
assert not generator.create_deployment_package.called