"""Functional tests for reference signing entity endpoints."""

from tests.functional.conftest import insert_mock_data


def _insert_mock_data():
    insert_mock_data({
        'reference_payment_entity': {'reference_payment_entity_id': 1},
        'reference_sap_profit_center': {'reference_sap_profit_center_id': 1, 'company_code': '321'},
        'reference_signing_entity': {
            'reference_signing_entity_id': 1,
            'reference_payment_entity_id': 1,
            'reference_sap_profit_center_id': 1,
            'company_code': '321',
            'tax_entity_company_code': '321',
            'legal_name': 'Company',
            'vat_number': '321',
            'company_registration_number': '12341234',
            'address': None,
        },
        'contract': {
            'contract_id': 1001,
            'reference_signing_entity_id': 1,
        },
    })


def test_get_reference_signing_entity_by_contract_id(fixture_client):
    """Test getting a reference signing entity by contract."""
    _insert_mock_data()

    contract_id = 1001

    result = fixture_client.get(
        f'/reference-signing-entity/contract/{contract_id}')

    assert result.status_code == 200
    assert result.json() == {
        'reference_signing_entity_id': 1,
        'company_code': '321',
        'tax_entity_company_code': '321',
        'legal_name': 'Company',
        'vat_number': '321',
        'company_registration_number': '12341234',
        'address': None,
    }


def test_get_reference_signing_entity_by_contract_id_no_data(fixture_client):
    """Test getting a reference signing entity by contract that doesn't exist."""
    _insert_mock_data()

    contract_id = 9999

    result = fixture_client.get(
        f'/reference-signing-entity/contract/{contract_id}')

    assert result.status_code == 200
    assert result.json() is None