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 sP t j| j| jfd}| | |jdks2J tj | jsDJ |
d S )N)targetr
r ) multiprocessingZProcessr r startr Zexitcoder r exists 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 sH | | j}z,t|}| || d|v s.J W | n
| 0 d S )NZacquired)r r repr_test_repr_commonrelease)r lock_fileZ locked_reprr r r
test_locked_reprB s z LockFileTest.test_locked_reprc C s8 | | j}| t|}| || d|v s4J d S )NZreleased)r r r2 r0 r1 )r r3 Z
released_reprr r r
test_released_reprK s
z LockFileTest.test_released_reprc C s" |j j|v sJ | j|v s J d S r. )r& r r )r r3 Z lock_reprr r r
r1 R s z LockFileTest._test_repr_commonzWRace conditions on lock are specific to the non-blocking file access approach on Linux.c st ddgddl m fdd}td"}||_ j W d n1 sV0 Y tdkspJ 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_deleter6 r r
delete_and_stat_ s
z/LockFileTest.test_race..delete_and_stat)certbot._internal.lock.filesystem.os.stat)r r6 r patch side_effectr r len)r r: Z mock_statr r9 r
test_raceV s *zLockFileTest.test_racec C s* | | j}| tj| jr&J d S r. )r r r2 r r r* )r r3 r r r
r+ j s zLockFileTest.test_removedc C s t r
d}nd}d}t|j}t||_z| | j W n4 tyl } z |t|v sXJ W Y d }~nd }~0 0 | d W d n1 s0 Y d S )Nz"certbot._internal.lock.fcntl.lockf%certbot._internal.lock.msvcrt.lockinghi therezIOError not raised
POSIX_MODEr r< OSErrorr= r r strZfail)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|j}t||_z| | j W n4 tyl } z |t|v sXJ W Y d }~nd }~0 0 | d W d n1 s0 Y d S )Nr; r@ rA zOSError not raisedrB )r Z
mock_functionrF Zmock_osrG r r r
test_unexpected_os_err~ s
&z#LockFileTest.test_unexpected_os_err)r r r r r r r$ r- r/ r4 r5 r1 r Zskip_on_windowsr? r+ rH rI
__classcell__r r r% r
r! & s
r! __main__ )r r r( sysZunittestr ZpytestZcertbotr Zcertbot.compatr Z
certbot.testsr r Zfcntl ImportErrorrC ZTempDirTestCaser r! r exitmainargv__file__r r r r
s$
i