"""Tests for address converter."""

from src import convert_address, batch_convert, ConversionStatus, MappingType


class TestMergedWard:
    """Test wards that were merged into new wards."""

    def test_phuc_xa_merged_to_hong_ha(self):
        result = convert_address("Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội")
        assert result.status == ConversionStatus.SUCCESS
        assert result.mapping_type == MappingType.MERGED
        assert result.new.ward == "Phường Hồng Hà"
        assert result.new.province == "Thành phố Hà Nội"

    def test_an_binh_can_tho(self):
        result = convert_address("Phường An Bình, Quận Ninh Kiều, Thành phố Cần Thơ")
        assert result.status == ConversionStatus.SUCCESS
        assert result.new.province == "Thành phố Cần Thơ"


class TestUnchangedWard:
    """Test wards that remain unchanged."""

    def test_tan_loc_can_tho(self):
        result = convert_address("Phường Tân Lộc, Quận Thốt Nốt, Thành phố Cần Thơ")
        assert result.status == ConversionStatus.SUCCESS
        assert result.mapping_type == MappingType.UNCHANGED
        assert result.new.ward == "Phường Tân Lộc"


class TestRenamedWard:
    """Test wards that were renamed."""

    def test_long_hoa_renamed(self):
        result = convert_address("Phường Long Hòa, Quận Bình Thủy, Thành phố Cần Thơ")
        assert result.status == ConversionStatus.SUCCESS
        assert result.mapping_type == MappingType.RENAMED
        assert result.new.ward == "Phường Long Tuyền"


class TestDividedWard:
    """Test wards that were split into multiple new wards."""

    def test_divided_selects_default(self):
        result = convert_address("Xã Tân Thạnh, Huyện Thới Lai, Thành phố Cần Thơ")
        assert result.status == ConversionStatus.SUCCESS
        assert result.mapping_type == MappingType.DIVIDED


class TestAbbreviations:
    """Test address abbreviation expansion."""

    def test_p_q_tp(self):
        result = convert_address("P. Phúc Xá, Q. Ba Đình, TP. Hà Nội")
        assert result.status == ConversionStatus.SUCCESS
        assert result.new.ward == "Phường Hồng Hà"
        assert result.new.province == "Thành phố Hà Nội"

    def test_tp_shorthand(self):
        result = convert_address("P.Phúc Xá, Q.Ba Đình, TP.Hà Nội")
        assert result.status == ConversionStatus.SUCCESS
        assert result.new.province == "Thành phố Hà Nội"


class TestPartialAddress:
    """Test partial addresses (province-only, no ward)."""

    def test_province_only(self):
        result = convert_address("Thành phố Hà Nội")
        assert result.status == ConversionStatus.PARTIAL
        assert result.new.province == "Thành phố Hà Nội"

    def test_unknown_province(self):
        result = convert_address("Tỉnh Không Tồn Tại")
        assert result.status == ConversionStatus.NOT_FOUND


class TestWithStreet:
    """Test addresses that include street information."""

    def test_street_preserved(self):
        result = convert_address("123 Phố Hàng Bông, Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội")
        assert result.status == ConversionStatus.SUCCESS
        assert "123" in result.converted
        assert "Phường Hồng Hà" in result.converted


class TestBatchConvert:
    """Test batch conversion."""

    def test_batch(self):
        addresses = [
            "Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội",
            "Phường Tân Lộc, Quận Thốt Nốt, Thành phố Cần Thơ",
            "Tỉnh Không Tồn Tại",
        ]
        results = batch_convert(addresses)
        assert len(results) == 3
        assert results[0].status == ConversionStatus.SUCCESS
        assert results[1].status == ConversionStatus.SUCCESS
        assert results[2].status == ConversionStatus.NOT_FOUND


class TestProvinceMapping:
    """Test province-level conversions (63 -> 34)."""

    def test_ha_noi_stays(self):
        result = convert_address("Thành phố Hà Nội")
        assert result.new.province == "Thành phố Hà Nội"

    def test_merged_province(self):
        # Hà Giang merged with another province
        result = convert_address("Tỉnh Hà Giang")
        assert result.status == ConversionStatus.PARTIAL
        assert result.new.province  # should have a new province name