"""Unit Tests for vendor_dms_master_restriction schema."""

from abacus_legacy_sync.schemas.vendor_dms_master_restriction import (
    VendorDmsMasterRestrictionDetailSchema,
)


def test_vendor_dms_master_restriction_schema():
    """Test vendor_dms_master_restriction detail serialization."""
    mock_data = {
        'restriction_id': 1,
        'customer_master_master_id': 1,
        'distribution_type_id': 2,
        'vendor_contract_id': 11,
    }

    result = VendorDmsMasterRestrictionDetailSchema().dump(mock_data)
    assert result == mock_data