"""Test for Notify run failed task."""

from unittest.mock import MagicMock, patch

import pytest

from tasks.accounting_run_calculate.update_run_status_error import status_error


@patch('tasks.accounting_run_calculate.update_run_status_error.'
       'ows.update_accounting_run')
def test_update_run_status_error(
    mock_ows_update_accounting_run: MagicMock,
    mock_accounting_run_calculate_dag_run,
    mock_accounting_run
):
    """Test update accounting run status to 'Error'."""
    with pytest.raises(ValueError):
        status_error(mock_accounting_run_calculate_dag_run)

    mock_ows_update_accounting_run.assert_called_once_with(
        mock_accounting_run['accounting_run_id'],
        accounting_run_status='Error'
    )