def test_invoke_lambda_function(self): hook = AwsLambdaHook(aws_conn_id='aws_default', function_name="test_function", region_name="us-east-1") hook.get_conn().create_function( FunctionName='test_function', Runtime='python2.7', Role='test-iam-role', Handler='lambda_function.lambda_handler', Code={ 'ZipFile': self.lambda_function(), }, Description='test lambda function', Timeout=3, MemorySize=128, Publish=True, ) payload = {'hello': 'airflow'} response = hook.invoke_lambda(payload=json.dumps(payload)) self.assertEqual(response["StatusCode"], 202)
def test_get_conn_returns_a_boto3_connection(self): hook = AwsLambdaHook(aws_conn_id='aws_default', function_name="test_function", region_name="us-east-1") self.assertIsNotNone(hook.get_conn())