"""Test for invoke_ledger_accounting_run_balance_lambda task."""

from unittest.mock import patch

import pytest

from hooks.lambda_invoke.invocation_response import LambdaInvocationResponse
from tasks.accounting_run_calculate.invoke_ledger_accounting_run_balance_lambda import \
    invoke_ledger_accounting_run_balance_lambda_task


@patch('tasks.accounting_run_calculate.invoke_ledger_accounting_run_balance_lambda'
       '.OrchLambdaHook')
def test_invoke_ledger_accounting_run_balance_lambda_task(
    mock_orchlambdahook,
    mock_accounting_run_calculate_dag_run,
    lambda_response_error
):
    """Test for invoke_ledger_accounting_run_balance_lambda task."""
    mock_lambda_response = LambdaInvocationResponse(lambda_response_error)
    mock_orchlambdahook.return_value.invoke_lambda.return_value = mock_lambda_response
    with pytest.raises(Exception):
        invoke_ledger_accounting_run_balance_lambda_task(
            mock_accounting_run_calculate_dag_run
        )