"""Reference signing entity logic tests."""

from unittest.mock import patch

from moneyhub.logic import reference_signing_entity as logic
from tests.utils.factories import ReferenceSigningEntityFactory


@patch('moneyhub.logic.reference_signing_entity.models')
def test_get_reference_signing_entity_by_contract_id(mock_models):
    """Test get_by_contract_id method."""
    contract_id = 4444

    entity = ReferenceSigningEntityFactory.build()

    mock_models.ReferenceSigningEntity.get_by_contract_id.return_value = entity

    result = logic.get_reference_signing_entity_by_contract_id(contract_id)

    assert result == entity
    mock_models.ReferenceSigningEntity.get_by_contract_id.assert_called_once_with(
        contract_id)