o 1ik5@sddlZddlZddlmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZmZejd duZd d ZGdddeejZGdddeejZGdddeZGdddeZ GdddeZ!ej"j#e ddGdddeZ$dS)N)TestCaseoverride_settings) serializersviews)APIRequestFactory countries)settings)Country)CountryFieldMixin)FantasyCountries) MultiCountryPersondrf_spectacularcCsdd|DS)z> Convert Countries into a DRF-OPTIONS formatted dict. cSsg|] \}}||dqS)) display_namevalue).0kvrr`/var/www/html/humari/django-venv/lib/python3.10/site-packages/django_countries/tests/test_drf.py sz%countries_display..rrrrrcountries_displaysrc@eZdZGdddZdS)PersonSerializerc@s&eZdZeZdZddiddidZdS)zPersonSerializer.Metanamecountry other_countryfavourite_countryfantasy_country country_dictT name_only)rrN)__name__ __module__ __qualname__rmodelfields extra_kwargsrrrrMetas  r)Nr#r$r%r)rrrrrrc@r)MultiCountrySerializerc@seZdZeZdZdS)zMultiCountrySerializer.MetarN)r#r$r%r r&r'rrrrr)+sr)Nr*rrrrr,*r+r,c@steZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZdS)TestDRFc Cs2tddd}t|}||jdddddddS)N Chris BeavenNZrr New Zealandrrr assertEqualdataselfperson serializerrrrtest_serialize1s zTestDRF.test_serializec Cs8tddd}t|}||jdddddddddS) Nr.AU)rrr1 Australiacoderr2rr3r6rrrtest_serialize_country_dict?s z#TestDRF.test_serialize_country_dictcCs6tddddd}||||jdddS)NTesterUSNVrrr r5rr assertTrueis_validr4validated_datar7r9rrrtest_deserializeMs  zTestDRF.test_deserializecCs<tddddddd}||||jdddS) Nr@GBAnythingr=rBrCrDrrErIrrrtest_deserialize_country_dictTsz%TestDRF.test_deserialize_country_dictcCsHtddddd}||||jdd||jdd dS) NChrisr2 NeverlandrCrDrr/r rBrErIrrrtest_deserialize_by_name_sz TestDRF.test_deserialize_by_namecCs8tddddd}||||jddgdS)NrNzNo Such ZealandrOrCrDrz("No Such Zealand" is not a valid choice.r assertFalserGr4errorsrIrrrtest_deserialize_invalidks z TestDRF.test_deserialize_invalidcCs,tdd}t|}||jdddgidS)NzNZ,AUrrr;r/r r,r4r5r7mcr9rrrtest_multi_serializexs zTestDRF.test_multi_serializecCs(tdd}t|}||jdgidS)Nr1rrrUrVrrrtest_multi_serialize_empty}s z"TestDRF.test_multi_serialize_emptycCs:tdddgid}||||jdddgdS)Nrr/r;rDr,rFrGr4rHrIrrrtest_multi_deserializeszTestDRF.test_multi_deserializecCs:tdddgid}||||jdddgdS)Nrr2r<rDr/r;rZrIrrrtest_multi_deserialize_by_names  z&TestDRF.test_multi_deserialize_by_namecCsNtdgdid}|||jd}t|tr|d}||dgdS)Nr)r/BADr;rDz"BAD" is not a valid choice.)r,rRrGrS isinstancedictr4)r7r9rSrrrtest_multi_deserialize_invalids   z&TestDRF.test_multi_deserialize_invalidcCsVtdddgid}|||}tjj|jd}||j t dt dgdS)Nrr/r;rD)pk) r,rFrGsaver objectsgetrbr4rr )r7r9savedloadedrrrtest_multi_deserialize_saves z#TestDRF.test_multi_deserialize_savecCs6tdddd}||||jddgdS)NrNr1r0rDrz"" is not a valid choice.rQrIrrrtest_deserialize_blank_invalidsz&TestDRF.test_deserialize_blank_invalidN)r#r$r%r:r?rJrMrPrTrXrYr[r\rarhrirrrrr-0s    r-c@s,eZdZdZejjej ddddZ dS)TestDRFMetadataz> Tests against the DRF OPTIONS API metadata endpoint. zNo i18nreasonc CsGdddtj}dd}|}t}|d}||d}||d}||d}||jd ||tt||tt t d d ||d}||d} | || WddS1sawYdS) Nc@s eZdZdZddZddZdS)z1TestDRFMetadata.test_actions..ExampleViewz Example view.cSsdSNr)r7requestrrrpostsz6TestDRFMetadata.test_actions..ExampleView.postcSstSrm)r)r7rrrget_serializersz@TestDRFMetadata.test_actions..ExampleView.get_serializerN)r#r$r%__doc__rorprrrr ExampleViews rrcSs|jdd|dS)z*Helper method for unpacking response JSON.actionsPOSTchoicesrD)responsekeyrrr_choicessz.TestDRFMetadata.test_actions.._choices/)rnrr fr) LANGUAGE_CODE) rAPIViewas_viewroptionsr4 status_coderrr rassertNotEqual) r7rrrxviewfactoryrnrvcountry_choicesfantasy_choices_enfantasy_choices_frrrr test_actionss         "zTestDRFMetadata.test_actionsN) r#r$r%rqpytestmarkskipifr USE_I18Nrrrrrrjsrjc@s8eZdZdZddZddZddZdd Zd d Zd S) TestDRFSchemaGenerationz@ Tests for OpenAPI schema generation with CountryField. cCsddlm}|dd}|t|d|j}|||d||d}||dd|d ||d |d |d |d ||d d dd ||d d dd ||d d d gdS)z Test that CountryField with country_dict=True provides correct schema annotation. This is used by drf-spectacular for OpenAPI schema generation. r CountryFieldTr!_spectacular_annotationfieldtypeobject propertiesr>rstringrequiredN)"django_countries.serializer_fieldsrrFhasattrrassertIsNotNoneassertInr4r7rr annotationschemarrr#test_country_dict_schema_annotations     z;TestDRFSchemaGeneration.test_country_dict_schema_annotationcCsddlm}|ddd}|t|d|j}|d||d}|||d||t|ddt d d |dD}|d ||d |d t d d |dD}||dddS)za Test that nullable CountryField with country_dict=True includes null in schema. rrT)r! allow_nullrroneOfcs"|] }|ddkr|VqdS)rrNrersrrr  zWTestDRFSchemaGeneration.test_country_dict_schema_annotation_nullable..rr>csr)rnullNrrrrrrrrrN) rrrFrrrrr4lennext)r7rrrr object_schema null_schemarrr,test_country_dict_schema_annotation_nullables      zDTestDRFSchemaGeneration.test_country_dict_schema_annotation_nullablecCs^ddlm}|dd}|t|d|j}|d||d}||||ddd S) z` Test that CountryField with name_only=True provides correct schema annotation. rrTr"rrrrN)rrrFrrrrr4rrrr test_name_only_schema_annotations    z8TestDRFSchemaGeneration.test_name_only_schema_annotationcCspddlm}|ddd}|t|d|j}|d||d}|||d||t|ddd S) z^ Test that nullable CountryField with name_only=True includes null in schema. rrT)r"rrrrrN) rrrFrrrrr4rrrrr)test_name_only_schema_annotation_nullable s     zATestDRFSchemaGeneration.test_name_only_schema_annotation_nullablecCs&ddlm}|}|t|ddS)z Test that standard CountryField doesn't set annotation. This allows the default ChoiceField enum handling to work. rrrN)rrrRr)r7rrrrr%test_standard_field_schema_annotations z=TestDRFSchemaGeneration.test_standard_field_schema_annotationN) r#r$r%rqrrrrrrrrrrs rzdrf-spectacular not installedrkc@s(eZdZdZddZddZddZdS) TestDRFSpectacularIntegrationzO Integration tests with drf-spectacular for OpenAPI schema generation. csddlm}ddlmGfdddtj}|}|jd}||dd}|| d d | d || d |d | d |d | d|dS)zp Test that drf-spectacular generates an object schema (not enum) for country_dict=True. r AutoSchemarceZdZddZdS)zkTestDRFSpectacularIntegration.test_spectacular_country_dict_generates_object_schema..TestSerializerTrNr#r$r%rrrrrTestSerializer4rrrnNrrrr>renum) drf_spectacular.openapirrrr Serializerr'_map_serializer_fieldr4rer assertNotInr7rr auto_schemarrrrr5test_spectacular_country_dict_generates_object_schema+s    zSTestDRFSpectacularIntegration.test_spectacular_country_dict_generates_object_schemacspddlm}ddlmGfdddtj}|}|jd}||dd}|| d d | d |dS) zl Test that drf-spectacular generates a string schema (not enum) for name_only=True. rrrcr)zhTestDRFSpectacularIntegration.test_spectacular_name_only_generates_string_schema..TestSerializerTrNrrrrrrLrrrrnNrrr) rrrrrrr'rr4rerrrrr2test_spectacular_name_only_generates_string_schemaCs   zPTestDRFSpectacularIntegration.test_spectacular_name_only_generates_string_schemacs~ddlm}ddlmGfdddtj}|}|jd}||dd}|d ||d |d |d |d dS) z_ Test that drf-spectacular generates an enum schema for standard CountryField. rrrcseZdZZdS)zkTestDRFSpectacularIntegration.test_spectacular_standard_field_generates_enum_schema..TestSerializerNrrrrrr_s rrrnNrr/rA) rrrrrrr'rrrrrr5test_spectacular_standard_field_generates_enum_schemaWs    zSTestDRFSpectacularIntegration.test_spectacular_standard_field_generates_enum_schemaN)r#r$r%rqrrrrrrrr%s  r)%importlib.util importlibr django.testrrrest_frameworkrrrest_framework.testrdjango_countriesrdjango_countries.confr django_countries.fieldsr django_countries.serializersr 'django_countries.tests.custom_countriesr django_countries.tests.modelsr rutil find_specHAS_DRF_SPECTACULARrModelSerializerrr,r-rjrrrrrrrrs(      u%[