U ?h@sdZddlZddlmZddlmZddlZz$ddlZeej j ddddZ Wne k rldZdZ YnXdd 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_pathrI/var/www/html/venv/lib/python3.8/site-packages/pooch/tests/test_hashes.pydata_dir_mirror:src Cstdd}|zt||jddt|jdd}|}W5QRX|tksRt t |did}| |jt |d}| d}||kst t|W5t|jXd S) zGCheck that the registry builder creates the right file names and hashesFdelete recursiveutf-8encodingsome bogus URLpathbase_urlregistry tiny-data.txtN)rcloseosremovenamer openreadREGISTRYAssertionErrorr load_registrystrfetchr )routfilefoutr$puptruefnamerrrtest_make_registryCs       r6c Cstdd}|zt||jddt|jdd}|}W5QRX|tksRt t |did}| |jt |d | d kst t| d t |d d }|| d kst t| d W5t|jXd S) z7Check that the registry builder works in recursive modeFrTrrrr r!r%subdirzsubdir/tiny-data.txtN)rr&r'r(r)r r*r+REGISTRY_RECURSIVEr-rr.r/r0r )rr1r2r$r3r4rrrtest_make_registry_recursiveXs   r9c Cs8tt}tdddW5QRXdt|jks4tdS)z!Test an invalid hashing algorithmZ somethingZblah)r5algz'blah'N)pytestraises ValueErrorr r/valuer-)excrrr test_file_hash_invalid_algorithmns r@zalg,expected_hash)ZidscCsf|dr6tdkrtd|dkr6tdkr6tdtjtd}t |t ||}||ksbt dS)z2Test the hash calculation using hashlib and xxhashxxhNrequires xxhashrrrrequires xxhash > 2.0r%) startswithxxhashr;skipXXHASH_MAJOR_VERSIONr'r"joinrr r r-)r: expected_hashr5Z returned_hashrrrtest_file_hashus    rKcCs|dr6tdkrtd|dkr6tdkr6tdtjtd}t ||d|}t ||sht |d }t ||rt dS) z*Make sure the hash checking function worksrANrBrCrrDr%: :blablablabla) rErFr;rGrHr'r"rIrr r r-)r:rJr5 known_hashrrrtest_hash_matchess    rOc Cstjtd}t||d|}t||dds6t|d}tt }t||dddW5QRXdt |j ksxttt }t||dddW5QRX|t |j kstdS) zBMake sure the hash checking function raises an exception if strictr%rLTstrictrM NeverlandrQsourceN) r'r"rIrr r r-r;r<r=r/r>)r:rJr5rNZbad_hasherrorrrrtest_hash_matches_stricts   rVcCsDtjtd}t|ddsttddds.tt|ddds@tdS)zHThe hash checking function should always returns True if known_hash=Noner%N)rN)r5rNT)rNrQ)r'r"rIrr r-)r5rrrtest_hash_matches_nonesrXc Cs~tjtd}t||d|}t||dds:tt t }t||dddddW5QRXdt |j ksztdS) z:Hash matching should be independent of upper or lower caser%rLTrPNrRrS) r'r"rIrr upperr r-r;r<r=r/r>)r:rJr5rNrUrrrtest_hash_matches_uppercases "r[)0__doc__r'pathlibrtempfilerr;rFintVERSIONsplitrH ImportErrorcorerhashesr r r utilsr r r/__file__parentrr,r8ZTINY_DATA_HASHES_HASHLIBZTINY_DATA_HASHES_XXHcopyZTINY_DATA_HASHESupdateZfixturerr6r9r@markZ parametrizelistitemskeysrKrOrVrXr[rrrrsx