"""Test for invoke_adjustment_file_validation_lambda task."""

from unittest.mock import patch


from hooks.lambda_invoke.invocation_response import LambdaInvocationResponse
from tasks.adjustment_file_upload.invoke_adjustment_file_validation_lambda import \
    invoke_adjustment_file_validation_lambda_task


@patch('tasks.adjustment_file_upload.invoke_adjustment_file_validation_lambda.'
       'OrchLambdaHook')
def test_invoke_adjustment_file_validation_lambda(
    mock_orchlambdahook,
    mock_adjustment_file_upload_dag_run,
    lambda_response_success
):
    """Test invoke_adjustment_file_validation_lambda task."""
    mock_lambda_response = LambdaInvocationResponse(lambda_response_success)
    mock_orchlambdahook.return_value.invoke_lambda.return_value = mock_lambda_response

    invoke_adjustment_file_validation_lambda_task(mock_adjustment_file_upload_dag_run)

    mock_orchlambdahook.assert_called_once()
    mock_orchlambdahook.return_value.invoke_lambda.assert_called_once()