"""Update Accounting Run State logic tests."""

from unittest.mock import patch

from abacus_state.constants.constants import ACCOUNTING_RUN_ACTION_NAMES
from abacus_state.logic import accounting_run_state as logic

from tests.utils.factories import AbacusStateFactory


@patch('abacus_state.logic.accounting_run_state.AbacusState')
def test_update_accounting_run_state(mock_model, request_engine):
    """Test update_accounting_run_state method."""
    abacus_state = AbacusStateFactory.create(
        action_name=ACCOUNTING_RUN_ACTION_NAMES.COMMIT_ROYALTIES,
        action_status='init',
        parent_table_name='accounting_run',
        parent_table_id=1,
    )

    mock_put_request = {'action_status': 'running', 'message': 'testing'}

    res = logic.update_accounting_run_state(abacus_state, **mock_put_request)
    assert res.status == 200
    mock_model.commit_changes.assert_called_once()