"""Step Functions module unit tests.""" from unittest.mock import patch import step_functions @patch('step_functions.client') def test_start_execution(boto3_client_mock): """Test start_execution function.""" test_execution_input = 'test_input' test_execution_name = 'test_name' test_state_machine_arn = 'test_arn' step_functions.start_execution( test_execution_input, test_execution_name, test_state_machine_arn) boto3_client_mock.start_execution.assert_called_once_with( input=test_execution_input, name=test_execution_name, stateMachineArn=test_state_machine_arn )