"""Update Contract State logic tests."""

from unittest.mock import patch

from abacus_state.constants.constants import CONTRACT_ACTION_NAMES
from abacus_state.logic import contract_state as logic

from tests.utils.factories import AbacusStateFactory


@patch('abacus_state.logic.contract_state.AbacusState')
def test_update_contract_state(mock_model, request_engine):
    """Test update_contract_state method."""
    abacus_state = AbacusStateFactory.create(
        action_name=CONTRACT_ACTION_NAMES.LEGACY_SYNC,
        action_status='init',
        parent_table_name='contract',
        parent_table_id=1,
    )

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

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