a
}|g @ s d Z ddlZddlZddlZddlZddlmZ ddlZddlmZ ddl m
Z
ddl m Z
z ddlZW n ey~ dZY n0 dZG d d
d
e
jZG d d d e
jZed
kreeejdd eg dS )z!Tests for certbot._internal.lock. N)mock)errors)os)utilFTc @ s$ e Zd ZdZedd Zdd ZdS ) LockDirTestz*Tests for certbot._internal.lock.lock_dir.c O s ddl m} ||i |S )Nr )lock_dir)certbot._internal.lockr )clsargskwargsr r E/usr/lib/python3.9/site-packages/certbot/_internal/tests/lock_test.py_call s zLockDirTest._callc C s8 t | jtj| j| j}tj | jd}t
|| d S )Nz
.certbot.lock) functoolspartial assertRaisesr LockErrorr tempdirr pathjoin test_util
lock_and_call)self
assert_raises lock_pathr r r
test_it s
zLockDirTest.test_itN)__name__
__module__ __qualname____doc__ classmethodr r r r r r
r s
r c s~ e Zd ZdZedd Z fddZdd Zdd Zd
d Z d d
Z
dd Z e
ddd Zdd Zdd Zdd Z ZS ) LockFileTestz*Tests for certbot._internal.lock.LockFile.c O s ddl m} ||i |S )Nr )LockFile)r r" )r r
r r" r r r
r ( s zLockFileTest._callc s t tj| jd| _d S )Nz test.lock)supersetUpr r r r r )r __class__r r
r$ - s
zLockFileTest.setUpc C s0 t j| j| jfd}| | | d S )N)targetr
)multiprocessingZProcessr r startr test_removed)r childr r r
test_acquire_without_deletion1 s z*LockFileTest.test_acquire_without_deletionc C s* t | jtj| j| j}t|| j d S N) r r r r r r r r r )r r r r r
test_contention= s z LockFileTest.test_contentionc C s< | | j}z t|}| || W | n
| 0 d S r- )r r repr_test_repr_commonrelease)r lock_fileZ locked_reprr r r
test_locked_reprB s z LockFileTest.test_locked_reprc C s, | | j}| t|}| || d S r- )r r r1 r/ r0 )r r2 Z
released_reprr r r
test_released_reprK s
z LockFileTest.test_released_reprc C s d S r- r )r r2 Z lock_reprr r r
r0 R s z LockFileTest._test_repr_commonzWRace conditions on lock are specific to the non-blocking file access approach on Linux.c sd ddgddl m fdd}td"}||_ j W d n1 sV0 Y d S )NTFr )statc s& | j kr dr t| | S )z-Wrap os.stat and maybe delete the file first.r )r popr remove)r r Z
should_deleter5 r r
delete_and_stat_ s
z/LockFileTest.test_race..delete_and_stat)certbot._internal.lock.filesystem.os.stat)r r5 r patch side_effectr r )r r9 Z mock_statr r8 r
test_raceV s *zLockFileTest.test_racec C s | | j}| d S r- )r r r1 )r r2 r r r
r* j s zLockFileTest.test_removedc C s t r
d}nd}d}t|Z}t||_z| | j W n$ ty\ } z W Y d }~nd }~0 0 | d W d n1 s|0 Y d S )Nz"certbot._internal.lock.fcntl.lockf%certbot._internal.lock.msvcrt.lockinghi therezIOError not raised
POSIX_MODEr r; OSErrorr< r r Zfail)r Zmocked_functionmsgZ mock_lockerrr r r
$test_unexpected_lockf_or_locking_erro s
z1LockFileTest.test_unexpected_lockf_or_locking_errc C s t r
d}nd}d}t|Z}t||_z| | j W n$ ty\ } z W Y d }~nd }~0 0 | d W d n1 s|0 Y d S )Nr: r> r? zOSError not raisedr@ )r Z
mock_functionrC Zmock_osrD r r r
test_unexpected_os_err~ s
z#LockFileTest.test_unexpected_os_err)r r r r r r r$ r, r. r3 r4 r0 r Zskip_on_windowsr= r* rE rF
__classcell__r r r% r
r! & s
r! __main__ )r r r( sysZunittestr ZpytestZcertbotr Zcertbot.compatr Z
certbot.testsr r Zfcntl ImportErrorrA ZTempDirTestCaser r! r exitmainargv__file__r r r r
s$
i