@@ -349,22 +349,40 @@ def test_retry_fail_for_unsupported_state():
349
349
c1 .add_catch (Catch (error_equals = ["States.NoChoiceMatched" ], next_step = Fail ("ChoiceFailed" )))
350
350
351
351
352
- def test_result_path_none ():
353
- task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' , result_path = None )
352
+ def test_paths_none ():
353
+ task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' ,
354
+ result_path = None ,
355
+ input_path = None ,
356
+ output_path = None )
354
357
assert 'ResultPath' in task_state .to_dict ()
355
358
assert task_state .to_dict ()['ResultPath' ] is None
356
359
360
+ assert 'InputPath' in task_state .to_dict ()
361
+ assert task_state .to_dict ()['InputPath' ] is None
357
362
358
- def test_result_path_none_converted_to_null ():
359
- task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' , result_path = None )
363
+ assert 'OutputPath' in task_state .to_dict ()
364
+ assert task_state .to_dict ()['OutputPath' ] is None
365
+
366
+
367
+ def test_paths_none_converted_to_null ():
368
+ task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' ,
369
+ result_path = None ,
370
+ input_path = None ,
371
+ output_path = None )
360
372
assert '"ResultPath": null' in task_state .to_json ()
373
+ assert '"InputPath": null' in task_state .to_json ()
374
+ assert '"OutputPath": null' in task_state .to_json ()
361
375
362
376
363
- def test_default_result_path_not_included ():
377
+ def test_default_paths_not_included ():
364
378
task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' )
365
379
assert 'ResultPath' not in task_state .to_dict ()
380
+ assert 'InputPath' not in task_state .to_dict ()
381
+ assert 'OutputPath' not in task_state .to_dict ()
366
382
367
383
368
- def test_default_result_path_not_converted_to_null ():
384
+ def test_default_paths_not_converted_to_null ():
369
385
task_state = Task ('Task' , resource = 'arn:aws:lambda:us-east-1:1234567890:function:StartLambda' )
370
386
assert '"ResultPath": null' not in task_state .to_json ()
387
+ assert '"InputPath": null' not in task_state .to_json ()
388
+ assert '"OutputPath": null' not in task_state .to_json ()
0 commit comments