"""Functional tests for ReferenceSigningEntity."""

from tests.utils.factories import ReferenceSigningEntityFactory


def test_get_reference_signing_entity_by_id(fixture_client):
    """Test to get ReferenceSigningEntity by ID."""
    reference_signing_entity = ReferenceSigningEntityFactory.create(
        company_code='4914'
    )

    reference_signing_entity_id = reference_signing_entity.reference_signing_entity_id

    res = fixture_client.get(
        f'/reference-signing-entity/{reference_signing_entity_id}'
    )

    assert res.status_code == 200
    assert res.json == {
        'reference_signing_entity_id':
            reference_signing_entity.reference_signing_entity_id,
        'reference_payment_entity_id':
            reference_signing_entity.reference_payment_entity_id,
        'reference_sap_profit_center_id':
            reference_signing_entity.reference_sap_profit_center_id,
        'company_code': reference_signing_entity.company_code,
        'legal_name': reference_signing_entity.legal_name,
        'vat_number': reference_signing_entity.vat_number,
        'company_registration_number':
            reference_signing_entity.company_registration_number,
        'address': reference_signing_entity.address
    }


def test_get_reference_signing_entity_list(fixture_client):
    """Test to get a list of reference_signing_entity."""
    company_codes = ['4915', '4916', '4917']
    [
        ReferenceSigningEntityFactory.create(
            company_code=company_code
        ) for company_code in ['4915', '4916', '4917']
    ]

    res = fixture_client.get('/reference-signing-entities/')

    assert res.status_code == 200

    items = res.json.get('items')
    assert len(company_codes) == res.json.get('total_count')
    assert items[0]['company_code'] == company_codes[0]
    assert items[1]['company_code'] == company_codes[1]
    assert items[2]['company_code'] == company_codes[2]