آگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآآگآ 3 y¾œbم@s°ddlZddlZddlZddlZddlmZddlZddlmZm Z m Z ddlmZddl mZddlmZddlmZd d lmZd dlmZGdd „d ƒZGdd„dƒZdS)éN)عpatché)عSleepingBandwidthLimiterعRepositoryCacheعcache_if_remote)ع Repository)عPlaintextKey)عCompressionSpec)عIntegrityErroré)عH)عTestKeyc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) عTestSleepingBandwidthLimitercCs||_||_dS)N)Zexpected_fdZ expected_data)عselfعfdعdata©rْ/usr/lib64/python3.6/remote.pyعexpect_writesz)TestSleepingBandwidthLimiter.expect_writecCst|ƒS)N)عlen)rrrrrrعcheck_writesz(TestSleepingBandwidthLimiter.check_writecCs4|jtd|jƒtdƒ}|jddƒ|jddƒdS)Nعwriteréstest)عsetattrعosrrrr)rعmonkeypatchعitrrrعtest_write_unlimitedsz1TestSleepingBandwidthLimiter.test_write_unlimitedcsً|jtd|jƒ|jtd‡fdd„ƒ|jtddd„ƒd‰tdƒ}|jdd ƒ|jdd ƒ|jdd ƒ|jddƒ|jdd ƒ|jdd ƒˆd7‰|jdd ƒ|jdd ƒˆd7‰|jdd ƒ|jdd ƒˆd7‰|jdd ƒ|jdd ƒdS)NrZ monotoniccsˆS)Nrr)عnowrrع%sz9TestSleepingBandwidthLimiter.test_write..ZsleepcSsdS)Nr)عxrrrr&sédrstests123456s 1234567890é َ1)rrrعtimerrr)rrrr)rrع test_write#s(z'TestSleepingBandwidthLimiter.test_writeN)ع__name__ع __module__ع__qualname__rrrr%rrrrrsrc@s®eZdZejdd„ƒZejdd„ƒZedœdd„Zedœdd „Z edœd d„Z ejdd „ƒZdd„Zejdd„ƒZ ejdd„ƒZejdd„ƒZejdd„ƒZedœdd„ZdS)عTestRepositoryCacheccsntjjt|ƒdƒ|_t|jdddچڈ@}|jtdƒdƒ|jtdƒdƒ|jtdƒtd ƒƒ|VWdQRXdS) Nع repositoryT)Z exclusiveعcreaters1234rs5678ér!) rعpathعjoinعstrZrepository_locationrعputrعbytes)rZtmpdirr*rrrr*IszTestRepositoryCache.repositorycCst|ƒS)N)r)rr*rrrعcacheRszTestRepositoryCache.cache)r2cCsdS)Nr)rr2rrrعtest_simpleVszTestRepositoryCache.test_simplecsF‡fdd„}ˆjtdƒtdƒtdƒgƒ}ˆj}|ˆ_ˆjƒ|ˆ_dS)Ncs dˆ_dS)Nr)Z size_limitr)r2rrعquery_size_limitisz:TestRepositoryCache.test_backoff..query_size_limitrr,r)عget_manyrr4Zbackoff)rr2r4عiteratorZqslr)r2rعtest_backoffhs z TestRepositoryCache.test_backoffc CsDGdd„dƒ}|jtdƒtdƒtdƒgƒ}td|ƒڈWdQRXdS)Nc@s6eZdZdd„Zdd„Zdd„Zdd„Zdd d„Zd S) z4TestRepositoryCache.test_enospc..enospc_opencWsdS)Nr)rعargsrrrع__init__„sz=TestRepositoryCache.test_enospc..enospc_open.__init__cSs|S)Nr)rrrrع __enter__‡sz>TestRepositoryCache.test_enospc..enospc_open.__enter__cSsdS)Nr)rعexc_typeZexc_valZexc_tbrrrع__exit__ٹsz=TestRepositoryCache.test_enospc..enospc_open.__exit__cSsttjdƒ‚dS)NZfoo)عOSErrorعerrnoZENOSPC)rrrrrrچsz:TestRepositoryCache.test_enospc..enospc_open.writeNcSsdS)Nr)rعnrrrعtruncateگsz=TestRepositoryCache.test_enospc..enospc_open.truncate)N)r&r'r(r9r:r<rr@rrrrعenospc_openƒs rArrr,z builtins.open)r5rr)rr2rAr6rrrعtest_enospc‚s zTestRepositoryCache.test_enospccCs,|jddƒtj|tjƒƒ}tdƒj|_|S)NZBORG_PASSPHRASEZtestZnone)Zsetenvrr+r ZMockArgsr Z compressor)rr*rعkeyrrrrCںszTestRepositoryCache.keycCs |j|ƒ}|j||j|ƒƒ|S)N)Zid_hashr0Zencrypt)rrCr*rZid_rrrع_put_encrypted_object¦s z)TestRepositoryCache._put_encrypted_objectcCs|j||dƒS)Ns1234)rD)rrCr*rrrعH1«szTestRepositoryCache.H1cCs|j||dƒS)Ns5678)rD)rrCr*rrrعH2¯szTestRepositoryCache.H2cCs|j||tdƒƒS)Nr!)rDr1)rrCr*rrrعH3³szTestRepositoryCache.H3cCst||ddچS)NT)عdecrypted_cacheZforce_cache)r)rrCr*rrrrH·sz#TestRepositoryCache.decrypted_cache)rHcCs¤t|j|||gƒƒ|j|||gƒ}t|j|ƒdƒڈP}|jdtjƒtj|j ƒdƒdAj ddƒ}|jdtjƒ|j|ƒ|jƒWdQRXt jtƒڈWdQRXdS)Nza+brعlittleréےےےےrJ)عlistr5عopenZkey_filenameعseekعioعSEEK_ENDعintع from_bytesعreadعto_bytesrr@عpytestZraisesr )rrHrErFrGr6rZ corruptedrrrعtest_cache_corruption»s z)TestRepositoryCache.test_cache_corruptionN)r&r'r(rTZfixturer*r2rr3r7rBrCrDrErFrGrHrUrrrrr)Hs r))r>rrNr$Z unittest.mockrrTZremoterrrr*rZ crypto.keyrعcompressr Zhelpersr Z hashindexrrCr rr)rrrrعs6