"""Test for notify success task."""

from unittest.mock import MagicMock, patch

from tasks.apply_pending_adjustments.notify_success import notify_success_task


@patch('tasks.apply_pending_adjustments.notify_failure.ows.update_abacus_state')
@patch('tasks.apply_pending_adjustments.notify_failure.helpers.get_abacus_state')
def test_notify_success_task(
    mock_helpers_get_abacus_state: MagicMock,
    mock_ows_update_abacus_state: MagicMock,
    mock_apply_pending_adjustments_event,
    mock_apply_pending_adjustments_run,
):
    """Test for updating action_status to 'complete'."""
    mock_abacus_state = {
        'abacus_state_id': 11,
        'action_name': 'apply_file',
    }
    mock_helpers_get_abacus_state.return_value = mock_abacus_state
    mock_params = {'action_name': 'apply_file'}
    res = notify_success_task(mock_apply_pending_adjustments_run, **mock_params)
    assert res is None
    mock_helpers_get_abacus_state.assert_called_once_with(
        'apply_file',
        mock_apply_pending_adjustments_event['target_id'],
    )
    mock_ows_update_abacus_state.assert_called_once_with(
        mock_abacus_state['abacus_state_id'],
        body={
            'action_status': 'complete',
        }
    )