o iM@sdZddlZddlmZddlZddlZddlmZddlmZ ddlm Z Gdd d ej Z e j ZGd d d ej Zej ZGd d d ej Zej ZGdddej ZGdddej ZGdddej ZejddZejdddgddZejdddgddZejdddgddZejdddgdd Zejdddgd!d"Zd#d$Zejdddgd%d&Zejdddgd'd(Z ejdddgd)d*Z!ejdddgd+d,Z"ejdddgd-d.Z#ejd/gd0d1d2Z$ejd/gd0d3d4Z%ejd/gd0d5d6Z&ejd/gd0d7d8Z'ejd/gd0d9d:Z(ejd/gd0d;d<Z)ejdddgd=d>Z*ejdddgd?d@Z+ejdddgdAdBZ,dS)Czrrr)test_t_schema_for_list_of_pydantic_schemas   rCcCs0t|t}t|tjsJ|jdjsJdS)z,Tests t_schema when null fields are present.rN)rr,rr.rr=r(nullable)rr?rrrtest_t_schema_for_null_fieldss rEcCsZdddddiddigddd d }|D]\}}t|}t|||ks*Jqd S) zSTests handle_null_fields() doesn't change anything when no null fields are present.Namestring)r)r*r*integerr nullzTotal Area Sq Mi)anyOfdefaultr))rrN)rBcopydeepcopyrhandle_null_fields)test_properties_schema schema_beforerrr,test_schema_with_no_null_fields_is_unchangeds  rScCsTt|jt}tjtjdddtjdddddd d gd d d gd }||ks(JdS)NSTRINGrFr*r)INTEGERr Populationr*rKr)rrOBJECTrrrr(r*r+r)property_ordering)rr, _api_clientrrr=rr?expected_schemarrrtest_schema_with_default_values( r`cCsft|t}tjtjdddtjtjddtjddgddd d d d gd d d gd}||ks1JdS)NrTrFrUrV)r*NUMBERzRestaurants Per Capita)any_ofr))rrrZrrrr[)rr,rrr=r^rrrtest_schema_with_any_ofs&    rccCs|jrPddddgdddddd dd d d d d d dddgddgddddd dd d dd dddgddgdgddid}t|}t||||ksRJdSdS)z]When a dict schema is passed to process_schema, the only change should be camel-casing anyOf.rZz Fruit Basketz-A structured representation of a fruit basketfruitARRAYz*An ordered list of the fruit in the basketzA piece of fruitApplezDescribes an applerTzAlways 'apple')r* descriptionz$The color of the apple (e.g., 'red'))r*colorr*rh)r)rgr*r(propertyOrderingr+OrangezDescribes an orangezAlways 'orange'z'The size of the orange (e.g., 'medium'))r*sizerk)rgrJ)r*rgrB)r*r)rgr+r(N)r!rLrMrprocess_schema)r dict_schemarRrrr0test_complex_dict_schema_with_anyof_is_unchangedsZ   9  =rncCs2ddd}ddgd}t||||ksJdS)z||ksJd S) z7The `order_properties` setting should apply to 'items'.rerZr*rTr|r)r*rBr}r~rrNrLrMrrlrrrr:test_process_schema_order_properties_propagates_into_itemss,   rcCsddddiddiddgd}t|}ddddiddiddd gd gd}tj|||d |r:||ks8Jd S||ks@Jd S) z=The `order_properties` setting should apply to 'prefixItems'.rerZr*rTr|r)r* prefixItemsr}r~rrNrrrrrAtest_process_schema_order_properties_propagates_into_prefix_itemss0  rcCsddddiddiddddidd}t|}ddddiddidddgd ddidd d gd }tj|||d |rE||ksCJd S||ksKJd S)z||kss               B    * % ) , , )