o i@sdZddlZddlmZddlmZddlZzddlZeej j ddddZ Wn e y5dZdZ Ynwdd l mZdd lmZmZmZdd lmZmZeeejd d ZdZdZddddZddddddZeZe eej!ddZ"ddZ#ddZ$dd Z%ej&j'd!e(e)e(e*d"d#d$Z+ej&j'd!e(e)e(e*d"d%d&Z,ej&j'd!e(e)e(e*d"d'd(Z-d)d*Z.ej&j'd!e(e)e(e*d"d+d,Z/dS)-z3 Test the hash calculation and checking functions. N)Path)NamedTemporaryFile.)maxsplit)Pooch) make_registry file_hash hash_matches)check_tiny_datamirror_directorydatastorezOtiny-data.txt baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2d zsubdir/tiny-data.txt baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2d tiny-data.txt baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2d Z(c03148994acd89317915ea2f2d080d6dd127aa09Z@baee0894dba14b12085eacb204284b97e362f4f3e5a5807693cc90ef415c1b2dZ 70e2afd3fd7e336ae478b1e740a5f08e)sha1sha256md5Zf843815fe57948faZ98d6f1a2Z 0267d220db258fffb0c567c0ecd1b689Z811e3f2a12aec53f)xxh64xxh32Zxxh128Zxxh3_128Zxxh3_64cCs tt|S)z Mirror the test data folder on a temporary directory. Needed to avoid permission errors when pooch is installed on a non-writable path. )r DATA_DIR)Ztmp_pathrf/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/pooch/tests/test_hashes.pydata_dir_mirror:s rc Cstdd}|zTt||jddt|jdd }|}Wdn1s(wY|tks3Jt|did}||jt |d }| d }||ksQJt |Wt |jdSt |jw) zGCheck that the registry builder creates the right file names and hashesFdelete recursiveutf-8encodingNsome bogus URLpathbase_urlregistry tiny-data.txt)rcloser nameopenreadREGISTRYr load_registrystrfetchr osremove)routfilefoutr$puptruefnamerrrtest_make_registryCs        r5c Cstdd}|zkt||jddt|jdd }|}Wdn1s(wY|tks3Jt|did }||jt |d | d ksMJt | d t |d d }|| d kseJt | d Wt |jdSt |jw) z7Check that the registry builder works in recursive modeFrTrrrNr r!r%subdirzsubdir/tiny-data.txt)rr&r r'r(r)REGISTRY_RECURSIVErr+r,r-r r.r/)rr0r1r$r2r3rrrtest_make_registry_recursiveXs    r8cCsLtt}tdddWdn1swYdt|jvs$JdS)z!Test an invalid hashing algorithmZ somethingZblah)r4algNz'blah')pytestraises ValueErrorr r,value)excrrr test_file_hash_invalid_algorithmns r?zalg,expected_hash)ZidscCsf|drtdurtd|dvrtdkrtdtjtd}t |t ||}||ks1JdS)z2Test the hash calculation using hashlib and xxhashxxhNrequires xxhashrrrrequires xxhash > 2.0r%) startswithxxhashr:skipXXHASH_MAJOR_VERSIONr.r"joinrr r )r9 expected_hashr4Z returned_hashrrrtest_file_hashus    rJcCs|drtdurtd|dvrtdkrtdtjtd}t ||d|}t ||s4J|d }t ||r@JdS) z*Make sure the hash checking function worksr@NrArBrrCr%: :blablablabla) rDrEr:rFrGr.r"rHrr r )r9rIr4 known_hashrrrtest_hash_matchess    rNcCstjtd}t||d|}t||ddsJ|d}tt}t||dddWdn1s8wYdt |j vsFJtt}t||dddWdn1s^wY|t |j vslJdS) zBMake sure the hash checking function raises an exception if strictr%rKTstrictrL NeverlandrPsourceN) r.r"rHrr r r:r;r<r,r=)r9rIr4rMZbad_hasherrorrrrtest_hash_matches_stricts   rUcCsDtjtd}t|ddsJtdddsJt|ddds JdS)zHThe hash checking function should always returns True if known_hash=Noner%N)rM)r4rMT)rMrP)r.r"rHrr )r4rrrtest_hash_matches_nonesrWcCstjtd}t||d|}t||ddsJtt }t||dddddWdn1s9wYdt |j vsGJdS) z:Hash matching should be independent of upper or lower caser%rKTrONrQrR) r.r"rHrr upperr r:r;r<r,r=)r9rIr4rMrTrrrtest_hash_matches_uppercases rZ)0__doc__r.pathlibrtempfilerr:rEintVERSIONsplitrG ImportErrorcorerhashesr r r utilsr r r,__file__parentrr*r7ZTINY_DATA_HASHES_HASHLIBZTINY_DATA_HASHES_XXHcopyZTINY_DATA_HASHESupdateZfixturerr5r8r?markZ parametrizelistitemskeysrJrNrUrWrZrrrrs|