def test_invoke_lambda_function(self, session): z = Zappa(session) z.credentials_arn = 'arn:aws:iam::724336686645:role/ZappaLambdaExecution' function_name = 'django-helloworld-unicode' payload = '{"event": "hello"}' response = z.invoke_lambda_function(function_name, payload)
def test_rollback_lambda_function_version_docker(self, session): z = Zappa(session) z.credentials_arn = "arn:aws:iam::724336686645:role/ZappaLambdaExecution" function_name = "django-helloworld-unicode" with self.assertRaises(NotImplementedError): z.rollback_lambda_function_version(function_name)
def test_cognito_trigger_existing_UnusedAccountValidityDays(self, session): z = Zappa(session) z.update_cognito( "Zappa-Trigger-Test", "us-east-1_9jUv74DH8", {"PreSignUp": "test.tasks.pre_signup"}, "arn:aws:lambda:us-east-1:12345:function:Zappa-Trigger-Test", )
def test_rollback_lambda_function_version(self, session): z = Zappa(session) z.credentials_arn = 'arn:aws:iam::724336686645:role/ZappaLambdaExecution' function_name = 'django-helloworld-unicode' too_many_versions = z.rollback_lambda_function_version(function_name, 99999) self.assertFalse(too_many_versions) function_arn = z.rollback_lambda_function_version(function_name, 1)
def test_upload_remove_s3(self, session): bucket_name = 'test_zappa_upload_s3' z = Zappa(session) zip_path = z.create_lambda_zip(minify=False) res = z.upload_to_s3(zip_path, bucket_name) self.assertTrue(res) s3 = session.resource('s3') # will throw ClientError with 404 if bucket doesn't exist s3.meta.client.head_bucket(Bucket=bucket_name) # will throw ClientError with 404 if object doesn't exist s3.meta.client.head_object( Bucket=bucket_name, Key=zip_path, ) res = z.remove_from_s3(zip_path, bucket_name) self.assertTrue(res) fail = z.upload_to_s3('/tmp/this_isnt_real', bucket_name) self.assertFalse(fail) #Will graciouly handle quirky S3 behavior on 'us-east-1' region name' z.aws_region = 'us-east-1' res = z.upload_to_s3(zip_path, bucket_name) os.remove(zip_path) self.assertTrue(res)
def test_copy_on_s3(self, session): bucket_name = 'test_zappa_upload_s3' z = Zappa(session) zip_path = z.create_lambda_zip(minify=False) res = z.upload_to_s3(zip_path, bucket_name) self.assertTrue(res) s3 = session.resource('s3') # will throw ClientError with 404 if bucket doesn't exist s3.meta.client.head_bucket(Bucket=bucket_name) # will throw ClientError with 404 if object doesn't exist s3.meta.client.head_object( Bucket=bucket_name, Key=zip_path, ) zp = 'copy_' + zip_path res = z.copy_on_s3(zip_path, zp, bucket_name) os.remove(zip_path) self.assertTrue(res)
def test_create_lambda_function_s3(self, session): bucket_name = 'lmbda' zip_path = 'Spheres-dev-1454694878.zip' z = Zappa(session) z.aws_region = 'us-east-1' z.load_credentials(session) z.credentials_arn = 'arn:aws:iam::12345:role/ZappaLambdaExecution' arn = z.create_lambda_function(bucket=bucket_name, s3_key=zip_path, function_name='test_lmbda_function55', handler='runme.lambda_handler') arn = z.update_lambda_function( bucket=bucket_name, s3_key=zip_path, function_name='test_lmbda_function55', )
def test_create_lambda_function_docker(self, session): bucket_name = "lmbda" docker_image_uri = "docker_image_uri" z = Zappa(session) z.aws_region = "us-east-1" z.load_credentials(session) z.credentials_arn = "arn:aws:iam::12345:role/ZappaLambdaExecution" arn = z.create_lambda_function( bucket=bucket_name, docker_image_uri=docker_image_uri, function_name="test_lmbda_function55", ) arn = z.update_lambda_function( bucket=bucket_name, docker_image_uri=docker_image_uri, function_name="test_lmbda_function55", )
def test_create_lambda_function_s3(self, session): bucket_name = "lmbda" zip_path = "Spheres-dev-1454694878.zip" z = Zappa(session) z.aws_region = "us-east-1" z.load_credentials(session) z.credentials_arn = "arn:aws:iam::12345:role/ZappaLambdaExecution" arn = z.create_lambda_function( bucket=bucket_name, s3_key=zip_path, function_name="test_lmbda_function55", handler="runme.lambda_handler", ) arn = z.update_lambda_function( bucket=bucket_name, s3_key=zip_path, function_name="test_lmbda_function55", )
def test_create_lambda_function_s3(self, session): bucket_name = 'lmbda' zip_path = 'Spheres-dev-1454694878.zip' z = Zappa(session) z.aws_region = 'us-east-1' z.load_credentials(session) z.credentials_arn = 'arn:aws:iam::12345:role/ZappaLambdaExecution' arn = z.create_lambda_function( bucket=bucket_name, s3_key=zip_path, function_name='test_lmbda_function55', handler='runme.lambda_handler' ) arn = z.update_lambda_function( bucket=bucket_name, s3_key=zip_path, function_name='test_lmbda_function55', )
def test_create_iam_roles(self, session): z = Zappa(session) arn, updated = z.create_iam_roles() self.assertEqual(arn, "arn:aws:iam::123:role/{}".format(z.role_name))
def test_cognito_trigger_existing(self, session): z = Zappa(session) z.update_cognito( 'Zappa-Trigger-Test', 'us-east-1_9jUv74DH8', {'PreSignUp': 'test.tasks.pre_signup'}, 'arn:aws:lambda:us-east-1:12345:function:Zappa-Trigger-Test')
def test_get_api_url(self, session): z = Zappa(session) z.credentials_arn = 'arn:aws:iam::724336686645:role/ZappaLambdaExecution' url = z.get_api_url('Spheres-demonstration', 'demonstration')
def test_fetch_logs(self, session): z = Zappa(session) z.credentials_arn = 'arn:aws:iam::12345:role/ZappaLambdaExecution' events = z.fetch_logs('Spheres-demonstration') self.assertTrue(events is not None)
def test_cognito_trigger_existing(self, session): z = Zappa(session) z.update_cognito('Zappa-Trigger-Test', 'us-east-1_9jUv74DH8', {'PreSignUp': 'test.tasks.pre_signup'}, 'arn:aws:lambda:us-east-1:12345:function:Zappa-Trigger-Test')
def test_is_lambda_function_ready(self, session): z = Zappa(session) z.credentials_arn = "arn:aws:iam::724336686645:role/ZappaLambdaExecution" function_name = "django-helloworld-unicode" z.is_lambda_function_ready(function_name)