"""Tests for pad_zip in the PO email processor."""

import pytest


@pytest.mark.parametrize(
    ("raw", "expected"),
    [
        # Missing values pass through untouched
        (None, None),
        ("", ""),
        # Leading zeros restored (Coupa drops them: "MA 2149" -> "02149")
        ("2149", "02149"),
        ("7001", "07001"),
        ("501", "00501"),
        # Already-valid 5-digit zips are unchanged
        ("85704", "85704"),
        ("02149", "02149"),
        # Zip+4: leading zeros restored on the base, suffix preserved
        ("2149-1234", "02149-1234"),
        ("85704-1234", "85704-1234"),
        # Surrounding whitespace is handled
        (" 2149 ", "02149"),
        # Non-numeric values are left alone
        ("ABC12", "ABC12"),
    ],
)
def test_pad_zip(po_handler, raw, expected):
    assert po_handler.pad_zip(raw) == expected