­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 yb/ @sdZddlZddlZddlmZmZejddZejddgd d d Zejd d dgdddZ ejddgd ddZ ddZ ddZ ddZ ddZddZdd Zd!d"Zd#d$ZdS)%zM Do benchmarks using pytest-benchmark. Usage: py.test --benchmark-only N) changedircmdccs||jdd|jdd|jdd|jdd|jdt|jd |jd t|jd t|jd V|jd ddS)NZBORG_PASSPHRASEZ123456Z!BORG_CHECK_I_KNOW_WHAT_I_AM_DOINGZYESZ"BORG_DELETE_I_KNOW_WHAT_I_AM_DOINGZ*BORG_UNKNOWN_UNENCRYPTED_REPO_ACCESS_IS_OKyesZ BORG_KEYS_DIRkeysZBORG_CACHE_DIRcacheZ repositoryr)rec)Zsetenvstrjoinremove)requesttmpdirZ monkeypatchr!/usr/lib64/python3.6/benchmark.pyrepo_urls    rnoneZrepokey)paramscCs|dd|j||S)NZinitz --encryption)param)r rrrrrreposrZsessionzerosrandom)Zscoperc csd\}}|jd}|j}|dkr*dd}n|dkr.datarcSs tj|S)N)osurandom)rrrrr,sz&data_type must be 'random' or 'zeros'.wbr)r@B)rr) Zmktempr ValueErrorrangeopenr r writer ) r Ztmpdir_factorycountrpZ data_typerifrrrtestdata"s    r'lz4cCs|d}|dd|j|||S)Nz::testcreatez --compression)r)r rrr'Z archive_urlrrrarchive7sr*cCs.|j|ddd|d|f\}}|dks*tdS)Nr)z --compressionrz::testr)pedanticAssertionError) benchmarkrrr'resultoutrrrtest_create_none>sr0cCs.|j|ddd|d|f\}}|dks*tdS)Nr)z --compressionr(z::testr)r+r,)r-rrr'r.r/rrrtest_create_lz4Csr1c Cs<tt||j|d|f\}}WdQRX|dks8tdS)Nextractr)rr r+r,)r-rr*r r.r/rrr test_extractHsr3cCs$|j|d|f\}}|dks tdS)Ndeleter)r+r,)r-rr*r.r/rrr test_deleteNsr5cCs ||d|\}}|dkstdS)Nlistr)r,)r-rr*r.r/rrr test_listSsr7cCs ||d|\}}|dkstdS)Ninfor)r,)r-rr*r.r/rrr test_infoXsr9cCs.|jdd}||d|\}}|dks*tdS)Nz::rZcheck)splitr,)r-rr*rr.r/rrr test_check]sr;cCs||d\}}|dkstdS)Nhelpr)r,)r-rr.r/rrr test_helpcsr=)__doc__rZpytestZarchiverrrZfixturerrr'r*r0r1r3r5r7r9r;r=rrrrs