­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ 3 i%@sddlZddlZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZdZdZejd d ZGd d d Zejd dZGdddZGdddZejddZGdddZdS)N) daemonize)get_process_id process_alive) TimeoutTimer ExclusiveLockLock LockRosterADDREMOVESHARED EXCLUSIVE LockTimeout NotLocked NotMyLockfoobarcCs2t\}}}x tjdd}t|||s|SqWdS)zBReturn a free PID not used by any process (naturally this is racy)iiN)rrandomZrandintr)hostpidtidr/usr/lib64/python3.6/locking.pyfree_pids    rc@seZdZddZddZdS)TestTimeoutTimercCs<d}t|j}|j sttj|d|js8tdS)Ng?g?)rstartZ timed_outAssertionErrortimesleep)selftimeouttrrr test_timeouts  zTestTimeoutTimer.test_timeoutcCsjd\}}t||j}|j s$ttj|jd|ks>t|j sLttj|jd|ksftdS)N?rr)Nr$)rrZtimed_out_or_sleeprrZ start_time)r r!rr"rrrtest_notimeout_sleep"s z%TestTimeoutTimer.test_notimeout_sleepN)__name__ __module__ __qualname__r#r%rrrrrsrcCst|jdS)Nlock)strjoin)tmpdirrrrlockpath+sr-c@s4eZdZddZddZddZddZd d Zd S) TestExclusiveLockc Cs0t|dd}|jr|js"tWdQRXdS)Nr)r!)rZ is_lockedby_mer)r r-r)rrr test_checks1szTestExclusiveLock.test_checksc Cs4t|tdj}|jt|tdWdQRXdS)N)id)rID1acquire break_lockID2)r r-r)rrrtest_acquire_break_reacquire5sz.TestExclusiveLock.test_acquire_break_reacquirecCsDt|td.tjtt|tddjWdQRXWdQRXdS)N)r1g?)r1r!)rr2pytestraisesrr5r3)r r-rrrr#;s zTestExclusiveLock.test_timeoutc )Cst\}}}}|||f}d }t||dj} t||dd$tjt| jWdQRXWdQRXtjt| jWdQRXt||d0tjtt||dddjWdQRXWdQRXdS) Nfoo.bar.example.netrr)r1T)r1kill_stale_locksg?)r1r:r!)r9rr) rrr3r7r8rreleaserr) r r-rrrrour_iddead_idcant_know_if_dead_id dead_lockrrrtest_kill_stale@s    z!TestExclusiveLock.test_kill_stalecCstt}}|d|dkstt||dj}|j|ksr?rosterrrrr@s   zTestLock.test_kill_stalecCstt}}|d|dkstt||ddj}|j|ks>t|j|||j|ksXt|jt||ddj}|j|kst|j|||j|kst|jdS)NrT)r1rFF)r2r5rrr3r1rAr;)r r-rBrCr)rrrrDs   zTestLock.test_migrate_lockN) r&r'r(rKrLrMrNrOrPr#r@rDrrrrrE^s  rEcCst|jdS)NrS)r*r+)r,rrr rosterpathsrTc@s,eZdZddZddZddZddZd S) TestLockRostercCs*t|}|j}|j||iks&tdS)N)r loadZsaver)r rTrSrJrrr test_emptys zTestLockRoster.test_emptycCst|td}|jttks t|jtt|jtthks@tt|td}|jtt|jttthksntt|td}|jtt |jtthkstt|td}|jtt |jttkstdS)N)r1) r r2rIr rRrmodifyr r5r )r rTroster1Zroster2rrrtest_modify_gets        zTestLockRoster.test_modify_getc Cs t\}}}}|||f}t||d}|jttks:t|jtt|jt|hksZtd} t|| d}|jt|hks~t|jtt|jt|| hkstt|dd} | jt| hkst| jtt| jt|| hkstt|dd} | jt|| hkstdS)N)r1foo.bar.example.netrrT)r:)r[rr)rr rIr rRrrXr ) r rTrrrrr<r=rYr>Z killer_rosterZother_killer_rosterrrrr@s"        zTestLockRoster.test_kill_stalecCstt}}|d|dkstt||d}|j|ks8t|jtt|jt|hksXt|j t|||j|kstt|jt|hkstdS)Nr)r1) r2r5rr r1rXr r rIrA)r rTrBrCrSrrrrDs   z TestLockRoster.test_migrate_lockN)r&r'r(rWrZr@rDrrrrrUsrU)rrr)rrr)rrr7ZhelpersrplatformrrZlockingrrrr r r r r rrrr2r5Zfixturerrr-r.rErTrUrrrrs 4 ._