"""Reference Signing Entity model tests."""

from moneyhub.models.reference_signing_entity import ReferenceSigningEntity
from tests.unit.conftest import insert_mock_data


def test_reference_signing_entity():
    """Test to get reference signing entity."""
    contract_id = 4343

    insert_mock_data({
        'reference_payment_entity': {'reference_payment_entity_id': 1},
        'reference_sap_profit_center': {
            'reference_sap_profit_center_id': 1,
            'profit_center': 'UK1234',
            'company_code': '4914',
            'business_group': 'ORC'
        },
        'reference_signing_entity': {
            'reference_signing_entity_id': 1,
            'reference_payment_entity_id': 1,
            'reference_sap_profit_center_id': 1,
            'company_code': '4914',
            'tax_entity_company_code': '4914',
            'legal_name': 'Company',
            'vat_number': '12341234',
            'company_registration_number': '12341234',
            'address': None,
        },
        'contract': [
            {
                'contract_id': contract_id,
                'reference_signing_entity_id': 1,
            },
            {
                'contract_id': 999,
                'reference_signing_entity_id': 1,
            },
        ],

    })

    res = ReferenceSigningEntity.get_by_contract_id(contract_id)
    assert res.reference_signing_entity_id == 1
    assert res.company_code == '4914'