def test_amazon_sns_request_response(): expected_task = Task( "TestTask", Resource="arn:aws:states:::sns:publish", Parameters=Parameters( TopicArn="arn:aws:sns:us-east-1:123456789012:accretion-notify", Message=JsonPath("$.foo.bar")), ) expected = expected_task.to_dict() test_task = AmazonSns( "TestTask", TopicArn="arn:aws:sns:us-east-1:123456789012:accretion-notify", Message=JsonPath("$.foo.bar")) test = test_task.to_dict() assert test == expected
def test_amazon_sns_callback(): expected_task = Task( "TestTask", Resource="arn:aws:states:::sns:publish.waitForTaskToken", Parameters=Parameters( TopicArn="arn:aws:sns:us-east-1:123456789012:accretion-notify", Message=JsonPath("$.foo.bar")), ) expected = expected_task.to_dict() test_task = AmazonSns( "TestTask", Pattern=IntegrationPattern.WAIT_FOR_CALLBACK, TopicArn="arn:aws:sns:us-east-1:123456789012:accretion-notify", Message=JsonPath("$.foo.bar"), ) test = test_task.to_dict() assert test == expected
def test_aws_lambda_request_response(): expected_task = Task( "TestTask", Resource="arn:aws:states:::lambda:invoke", Parameters=Parameters( FunctionName= "arn:aws:lambda:us-east-1:123456789012:function:ship-val", Payload=Parameters(test="value")), ) expected = expected_task.to_dict() test_task = AwsLambda( "TestTask", FunctionName="arn:aws:lambda:us-east-1:123456789012:function:ship-val", Payload=Parameters(test="value"), ) test = test_task.to_dict() assert test == expected
def test_aws_lambda_callback(): expected_task = Task( "TestTask", Resource="arn:aws:states:::lambda:invoke.waitForTaskToken", Parameters=Parameters( FunctionName= "arn:aws:lambda:us-east-1:123456789012:function:ship-val", Payload=Parameters(test="value")), ) expected = expected_task.to_dict() test_task = AwsLambda( "TestTask", Pattern=IntegrationPattern.WAIT_FOR_CALLBACK, FunctionName="arn:aws:lambda:us-east-1:123456789012:function:ship-val", Payload=Parameters(test="value"), ) test = test_task.to_dict() assert test == expected