"""Test for complete run task method."""

from unittest.mock import MagicMock
from unittest.mock import patch

from lib.constants import ACCOUNTING_RUN_STATUSES
from tasks.accounting_run_calculate.update_run_status_complete import \
    status_complete


@patch('tasks.accounting_run_calculate.update_run_status_complete.helpers')
@patch('tasks.accounting_run_calculate.update_run_status_complete.ows')
def test_update_run_status_complete(
    mock_ows,
    mock_helpers,
    mock_accounting_run_calculate_dag_run
):
    """Test updating accounting_run_status."""
    accounting_run_id = 123
    mock_helpers.get_event_from_params.return_value = MagicMock(
        target_id=accounting_run_id
    )

    status_complete(mock_accounting_run_calculate_dag_run)

    mock_helpers.get_event_from_params.assert_called_once_with(
        mock_accounting_run_calculate_dag_run
    )
    mock_ows.update_accounting_run.assert_called_once_with(
        accounting_run_id,
        accounting_run_status=ACCOUNTING_RUN_STATUSES.COMPLETE
    )