o !2g$ @sUddlZddlZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z ddlZddlZddlmZddlmZd d lmZmZmZmZe d Zd d gZe eed<d ddedededdfddZeGdddZej !deej !dgdej j!dddgddgddededdfddZ"ej !deej !dgdej j!dddgddgddededdfd d!Z#ej !deej !dd"d#gej j!dddgddgddededdfd$d%Z$dS)&N) dataclass)datetimetimezone)Path)AnyListLiteralUnion)CachewMarshall)Json) gc_controlprofile running_on_citimer)cachewcattrslegacyrrImpls)impl test_namecountrreturnc sJ|dkr tr td|dkrtd}|j}|j}nP|dkr1ddlm}|} | j }| j }n:|dkrgd d l m } d d l m} | d d lmfd d} j} j| }fdd}nt|ttjdt|d)td|dt||d}Wdn1swYWdn1swYddt|D}t|d1td|dt|D] }|||||<qWdn1swYWdn1swYddt|D}t|d4td|dt|D] }t||||<qWdn 1s'wYWdn 1s7wYtd}|j rLt!"|j|j#t|dMtd|d2t$%|}|&d |'d!d"d|DWdn 1swY|(Wdn 1swYWdn 1swYd#dt|D}t|d$Qtd%|d6t$%|}d }|&d&D] \}|||<|d'7}qWdn 1swY|(Wdn 1swYWdn 1swY|jd(}t|d)Ftd*|d+|)d+}|D] }|*|d,q5Wdn 1sKwYWdn 1s[wYWdn 1skwYd-dt|D}t|d.Ntd/|d3d }|)d0}|D]}|+d,}|||<|d'7}qWdn 1swYWdn 1swYWdn 1swY|dd|d1d|dd|d1dksJd2dt|D}t|d34td4|dt|D] }t,||||<qWdn 1s(wYWdn 1s8wYd5dt|D}t|d63td7|dt|D] }|||||<q\Wdn 1sswYWdn 1swY|dd|d1d|dd|d1dksJdS)8Ndz1test too heavy for CI, only meant to run manuallyr)Type_rr )NTBinderrr) Converter)configure_tagged_union)get_argscsfdd}|S)Nc sZ|}|dur dS|D]}z ||}W|Sty"Yq wtd|d|)NzCould not cast z to ) structure Exception ValueError)datatype_argstres converterrM/home/garg/my-data/venv/lib/python3.10/site-packages/cachew/tests/marshall.py union_hookIs zAdo_test..union_structure_hook_factory..union_hookr*)_r,r(r*r+union_structure_hook_factoryHsz-do_test..union_structure_hook_factorycs |SNr*)x)Type struct_funcr*r+js zdo_test..)filez :baselinezbuilding z objects of type )rcSg|]}dqSr/r*.0r-r*r*r+ szdo_test..z :serializezserializing cSr5r/r*r6r*r*r+r8xr9z :json_dumpzjson dump z/tmp/cachew_test/db.sqlitez :sqlite_dumpzsqlite dump zCREATE TABLE data (value BLOB)z#INSERT INTO data (value) VALUES (?)cSsg|]}|fqSr*r*)r7sr*r*r+r8scSr5r/r*r6r*r*r+r8r9z :sqlite_loadzsqlite load zSELECT value FROM datar z cache.jsonlz :jsonl_dumpzjsonl dump wb cSr5r/r*r6r*r*r+r8r9z :jsonl_loadzjsonl load rbicSr5r/r*r6r*r*r+r8r9z :json_loadzjson load cSr5r/r*r6r*r*r+r8r9z :deserializezdeserializing )-rpytestskipr dumploadrrmaketo_rowfrom_rowrrcattrs.strategiesrtypingr_unstructure_funcdispatch_structure_func RuntimeErrorprintsysstderrrrlistrangeorjsondumpsrparentexistsshutilrmtreemkdirsqlite3connectexecute executemanycloseopenwriterstriploads)rr1factoryrrmarshallto_json from_jsonrbinderrrr. unstruct_funcobjectsjsonsistrsdbconnstrs2valuecachefwr:strs3frljsons2objects2r*)r1r)rr2r+do_test#s           $$ $    $   $    $ $   6$ $ :ruc@eZdZUeed<eed<dS)NamefirstlastN)__name__ __module__ __qualname__str__annotations__r*r*r*r+rw  rw)c@Bi@KLgc_onTFgc_off)idscCsB|dkr tddtfdd}t|jjtttf|||ddS)Nrz1TODO need to adjust the handling of Union types..rcSsPg}t|D]}|ddkr|t|q|td|d|dq|S)Nr rzfirst zlast )rxry)rOappendr}rw)rrfrhr*r*r+r`s    z)test_union_str_dataclass..factoryrr1r`rr) r>r?intrunodenamer r}rwrrrrequestr`r*r*r+test_union_str_dataclasss " rcsF|dkr tdddldtffdd }t|jjt|||ddS)Nrz*TODO support datetime with pytz for cattrsrrc3svdtjdg}td}td}|||}t|D]}|||}||t|}|j|dVq!dS)Nz Europe/BerlinzAmerica/New_Yorkz1990-01-01T00:00:00z2030-01-01T00:00:00)tzinfo)rutcr fromisoformatrOlenreplace)rtzsstartendsteprhdttzpytzr*r+r`s     ztest_datetimes..factoryr)r>r?rrrurrrrr*rr+test_datetimess  rrrcsVtGdddtGfddddtffdd }t|jj|||ddS) Nc@rv)z"test_nested_dataclass..UUUxxyyNrzr{r|rr~r*r*r*r+UUUrrcs&eZdZUeed<ed<eed<dS)z"test_nested_dataclass..TE2rmuuuvalue2Nrr*)rr*r+TE2s  rrc3s,t|D]}|||d|dVqdS)N)rr)rmrr)rO)rrhrrr*r+r`s z&test_nested_dataclass..factoryr)rrrurrrr*rr+test_nested_dataclasss r)%rTrWrL dataclassesrrrpathlibrrFrrrr rPr>marshall.cachewr marshall.commonr utilsrrrrImplrr~r}rrurwmark parametrizerrrr*r*r*r+s@