def __init__(self, file_name):
self.file_name = file_name
#self.s3 = S3()
self.folder = Files.temp_folder('tmp_lambda_')
self.lambda_code = "def run(event, context): return 'hello {0}'.format(event.get('name'))"
self.tmp_file = None
#self.s3_bucket = Globals.lambda_s3_bucket
#self.lambda_name = 'tmp_lambda_dev_test'
#self.s3_key = 'unit_tests/lambdas/{0}.zip'.format(self.lambda_name)
self.create_temp_file()
def __init__(self, lambda_name):
self.lambda_name = lambda_name
self.aws_lambda = Lambda(self.lambda_name)
self.s3_bucket = Globals.lambda_s3_bucket
self.s3_key = f'{Globals.lambda_s3_key_prefix}/{self.lambda_name}.zip'
self.role_arn = Temp_Aws_Roles().for_lambda_invocation()
self.tmp_folder = Files.temp_folder('tmp_lambda_')
(self.aws_lambda.set_s3_bucket(self.s3_bucket).set_s3_key(
self.s3_key).set_role(self.role_arn).set_folder_code(
self.tmp_folder))
def __init__(self, lambda_name):
self.lambda_name = lambda_name
self._lambda = Lambda(self.lambda_name)
self.tmp_s3_bucket = 'gs-lambda-tests'
self.tmp_s3_key = 'unit_tests/lambdas/{0}.zip'.format(self.lambda_name)
self.role_arn = Temp_Aws_Roles().for_lambda_invocation()
self.tmp_folder = Files.temp_folder('tmp_lambda_')
(self._lambda.set_s3_bucket(self.tmp_s3_bucket).set_s3_key(
self.tmp_s3_key).set_role(self.role_arn).set_folder_code(
self.tmp_folder))
def test_temp_folder(self):
assert Files.exists(Files.temp_folder())
assert '/tmp/aa_' in Files.temp_folder('_bb', 'aa_', '/tmp')
def __enter__(self):
if Files.exists(self.zip_file):
if self.target_folder is None:
self.target_folder = Files.temp_folder("unzipped_")
return Files.unzip_file(self.zip_file, self.target_folder)
return None
