o 1i'@s<dZddlmZddlmZddlmZGdddeZdS)zATests for displaying CountryField(multiple=True) in Django admin.)display_for_field)TestCase)modelsc@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)TestMultipleCountryAdminDisplayz=Test that multiple country fields display correctly in admin.cCsTtjjjdd}tjjd}t|j|d}||t | |d| d|dS)z Test that multiple country fields work in list_display. Regression test for #311: TypeError "unhashable type: 'list'" when using CountryField(multiple=True) in list_display. zUS,CA,MX countriesr- United StatesN) r MultiCountryobjectscreate_meta get_fieldrrassertIsInstancestrassertNotEqualassertInselfmultifieldresultrj/var/www/html/humari/django-venv/lib/python3.10/site-packages/django_countries/tests/test_admin_display.py(test_display_for_field_with_list_display s   zHTestMultipleCountryAdminDisplay.test_display_for_field_with_list_displaycCs`tjjjdd}tjjd}t|j|d}||d| |t | d|| d|dS)z Test that multiple country fields display correctly when readonly. Regression test for #463: CountryField shows up as "-" in admin when readonly with multiple=True. US,CArrr r CanadaN) rr r r rrrrrrrrrrrrtest_display_for_field_readonly s   z?TestMultipleCountryAdminDisplay.test_display_for_field_readonlycCs@tjjjdd}tjjd}t|j|d}||ddgdS)z9Test that empty multiple country fields display properly.rrrr N) rr r r rrrrrrrrr&test_display_for_field_empty_countries5szFTestMultipleCountryAdminDisplay.test_display_for_field_empty_countriescCshtjjjdd}tjjd}||}|t|dt |}| d|| d|| |t dS)z Test that value_from_object returns a MultipleCountriesDescriptor. This is what Django admin uses when accessing field values. The descriptor should be list-like but provide proper string display. NZ,AUrrz New Zealand AustraliaN) rr r r rrvalue_from_object assertEquallenrrr)rrrvalue str_valuerrr5test_value_from_object_returns_displayable_descriptorAs   zUTestMultipleCountryAdminDisplay.test_value_from_object_returns_displayable_descriptorcCstjjjdd}dd|jD}||gd|jd}||jd|t|jd|t |jt |j}| d |d S) z7Test that MultipleCountriesDescriptor acts like a list.zNZ,AU,CArcSsg|]}|jqSr)code).0countryrrr ^szdTestMultipleCountryAdminDisplay.test_multiple_countries_descriptor_list_behavior..)AUCANZrr-CountryN) rr r r rr$r)r% assertTrueboolreprr)rr country_codes first_countryrepr_strrrr0test_multiple_countries_descriptor_list_behaviorYs  zPTestMultipleCountryAdminDisplay.test_multiple_countries_descriptor_list_behaviorcCsHtjjjdd}|t|jd|t|j|t |jddS)z1Test MultipleCountriesDescriptor with empty list.rrrN) rr r r r$rr assertFalser3r%)rrrrr1test_multiple_countries_descriptor_empty_behavioroszQTestMultipleCountryAdminDisplay.test_multiple_countries_descriptor_empty_behaviorcCsTtjjjdd}tjjjdd}tjjjdd}||j|j||j|jdS)z6Test MultipleCountriesDescriptor equality comparisons.r rrN)rr r r r$rr)rmulti1multi2multi3rrr+test_multiple_countries_descriptor_equality|s zKTestMultipleCountryAdminDisplay.test_multiple_countries_descriptor_equalitycCsVddlm}|jjddd}|jd}|j}||t|j |d}| |dd S) zFTest that single (non-multiple) fields still use flatchoices normally.r)PersonTestUS)namer+r+r zUnited States of AmericaN) django_countries.tests.modelsr?r r rr flatchoicesassertIsNotNonerr+r$)rr?personrrDrrrr%test_single_country_field_flatchoicess   zETestMultipleCountryAdminDisplay.test_single_country_field_flatchoicesN) __name__ __module__ __qualname____doc__rrrr(r8r:r>rGrrrrr s  rN)rKdjango.contrib.admin.utilsr django.testrdjango_countries.testsrrrrrrs