← Back
Editing: lock_test.cpython-38.pyc
U �]~ � @ s� d Z ddlZddlZddlZz ddlZW n ek r@ dZY nX dZddlZddlm Z ddl m Z ddl m Z G dd � d ej�ZG d d � d ej�Zed kr�e�� dS ) zTests for certbot.lock.� NFT)�errors)�os)�utilc @ s$ e Zd ZdZedd� �Zdd� ZdS )� LockDirTestz Tests for certbot.lock.lock_dir.c O s ddl m} |||�S )Nr )�lock_dir)� certbot.lockr )�cls�args�kwargsr � r �9/usr/lib/python3/dist-packages/certbot/tests/lock_test.py�_call s zLockDirTest._callc C s8 t �| jtj| j| j�}tj� | jd�}t � ||� d S )Nz .certbot.lock) � functools�partial� assertRaisesr � LockErrorr �tempdirr �path�join� 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� fdd�Zdd� Zdd � Zd d � Z d d � Z dd� Z e � d�dd� �Zdd� Zdd� Zdd� Z� ZS )� LockFileTestz Tests for certbot.lock.LockFile.c O s ddl m} |||�S )Nr )�LockFile)r r! )r r r r! r r r r # s zLockFileTest._callc s$ t t| ��� tj�| jd�| _d S )Nz test.lock)�superr �setUpr r r r r )r �� __class__r r r# ( s zLockFileTest.setUpc C sR t j| j| jfd�}|�� |�� | �|jd� | �t j � | j�� | � � d S )N)�targetr r ) �multiprocessingZProcessr r �startr Z assertEqualZexitcode� assertTruer r �exists� test_removed)r Zchildr r r � test_acquire_without_deletion, 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_contention8 s �z LockFileTest.test_contentionc C sB | � | j�}z&t|�}| �||� | �d|k� W 5 |�� X d S )NZacquired�r r �release�repr�_test_repr_commonr) )r � lock_fileZ locked_reprr r r �test_locked_repr= s z LockFileTest.test_locked_reprc C s: | � | j�}|�� t|�}| �||� | �d|k� d S )NZreleasedr/ )r r3 Z released_reprr r r �test_released_reprF s z LockFileTest.test_released_reprc C s& | � |jj|k� | � | j|k� d S r- )r) r% r r )r r3 Z lock_reprr r r r2 M s z LockFileTest._test_repr_commonzWRace conditions on lock are specific to the non-blocking file access approach on Linux.c sZ ddg�ddl m� � ��fdd�}t�d��}||_� �� j� W 5 Q R X � ��� d S )NTFr )�statc s& | � j kr ��d�r 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_statZ s z/LockFileTest.test_race.<locals>.delete_and_stat� certbot.lock.filesystem.os.stat)r r6 �mock�patch� side_effectr r � assertFalse)r r: Z mock_statr r9 r � test_raceQ s zLockFileTest.test_racec C s, | � | j�}|�� | �tj�| j�� d S r- )r r r0 r? r r r* )r r3 r r r r+ e s zLockFileTest.test_removedc C s� t r d}nd}d}t�|��`}t|�|_z| �| j� W n4 tk rl } z| �|t|�k� W 5 d }~X Y n X | � d� W 5 Q R X d S )Nzcertbot.lock.fcntl.lockf�certbot.lock.msvcrt.locking�hi therezIOError not raised) � POSIX_MODEr< r= �IOErrorr> r r r) �str�fail)r Zmocked_function�msgZ mock_lock�errr r r �$test_unexpected_lockf_or_locking_errj s $z1LockFileTest.test_unexpected_lockf_or_locking_errc C s� t r d}nd}d}t�|��`}t|�|_z| �| j� W n4 tk rl } z| �|t|�k� W 5 d }~X Y n X | � d� W 5 Q R X d S )Nr; rA rB zOSError not raised) rC r< r= �OSErrorr> r r r) rE rF )r Z mock_functionrG Zmock_osrH r r r �test_unexpected_os_erry s $z#LockFileTest.test_unexpected_os_err)r r r r r r r# r, r. r4 r5 r2 r Zskip_on_windowsr@ r+ rI rK � __classcell__r r r$ r r ! s � r �__main__)r r r' ZunittestZfcntl� ImportErrorrC r< Zcertbotr Zcertbot.compatr Z certbot.testsr r ZTempDirTestCaser r r �mainr r r r �<module> s i
Save File
Cancel