"""Unit tests for contract_party schemas."""

from abacus_contract.constants.constants import CONTRACT_PARTY_TARGET_TYPES
from abacus_contract.schemas.contract_party import ContractPartyDetailSchema
from abacus_contract.schemas.contract_party import ContractPartyPostSchema
from tests.utils.factories import ContractFactory
from tests.utils.factories import ContractPartyFactory


def test_contract_party_detail_schema():
    """Test ContractPartyDetailSchema."""
    mock_contract = ContractFactory.create()
    mock_contract_party = ContractPartyFactory.create(contract=mock_contract)
    res = ContractPartyDetailSchema().dump(mock_contract_party)

    assert res == {
        'contract_party_id': mock_contract_party.contract_party_id,
        'contract_id': mock_contract.contract_id,
        'target_id': mock_contract_party.target_id,
        'target_type': mock_contract_party.target_type
    }


def test_contract_party_post_schema():
    """Test ContractPartyPostSchema."""
    mock_contract = ContractFactory.create()
    mock_contract_party_post_body = {
        'contract_id': mock_contract.contract_id,
        'target_id': '1ec7c1bf-2318-4052-9406-a3e35a620bd3',
        'target_type': CONTRACT_PARTY_TARGET_TYPES.CONTRIBUTOR
    }
    res = ContractPartyPostSchema().dump(mock_contract_party_post_body)

    assert res == mock_contract_party_post_body