File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/locking.cpython-39.opt-1.pyc
a
HZ�h�5 � @ s d dl Z d dlZd dlmZmZ d dlmZ d dlZddl m
Z
ddl m Z m
Z
ddlmZmZmZmZmZmZmZmZmZmZmZ dZd Zd
Zd Z e� � d d
� �Z G dd� d�Z e� � dd� �Z G dd� d�Z!G dd� d�Z"e� � dd� �Z#G dd� d�Z$dS )� N)�Thread�Lock)�
format_exc� )� daemonize)�get_process_id�
process_alive) � TimeoutTimer�
ExclusiveLockr �
LockRoster�ADD�REMOVE�SHARED� EXCLUSIVE� LockTimeout� NotLocked� NotMyLock)Zfoo� r )Zbarr r �( g�������?c C s. t � \} }}t�dd�}t| ||�s |S q dS )zBReturn a free PID not used by any process (naturally this is racy)i� i� N)r �randomZrandintr )�host�pid�tid� r �</usr/lib64/python3.9/site-packages/borg/testsuite/locking.py�free_pid s r c @ s e Zd Zdd� Zdd� ZdS )�TestTimeoutTimerc C s" d}t |��� }t�|d � d S )N� �?g �?)r �start�time�sleep)�self�timeout�tr r r � test_timeout s z TestTimeoutTimer.test_timeoutc C s d\}}t ||��� }d S )N)Nr )r r )r! r" r r# r r r �test_notimeout_sleep& s z%TestTimeoutTimer.test_notimeout_sleepN)�__name__�
__module__� __qualname__r$ r% r r r r r s r c C s t | �d��S )N�lock��str�join�Ztmpdirr r r �lockpath/ s r. c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�TestExclusiveLockc C s0 t |dd��}W d � n1 s"0 Y d S )Nr )r" )r
�r! r. r) r r r � test_checks5 s z TestExclusiveLock.test_checksc C sH t |td��� }|�� t |td�� W d � n1 s:0 Y d S �N��id)r
�ID1�acquire�
break_lock�ID2r0 r r r � test_acquire_break_reacquire9 s z.TestExclusiveLock.test_acquire_break_reacquirec C sl t |td��L t�t��" t |tdd��� W d � n1 s@0 Y W d � n1 s^0 Y d S )Nr3 皙�����?�r4 r" )r
r5 �pytest�raisesr r8 r6 �r! r. r r r r$ ? s z TestExclusiveLock.test_timeoutc
C s, t � \}}}}|||f}d}t||d��� } t||d��B t�t�� | �� W d � n1 sd0 Y W d � n1 s�0 Y t�t�� | �� W d � n1 s�0 Y t||d��L t�t��" t||dd��� W d � n1 s�0 Y W d � n1 �s 0 Y d S )N�zfoo.bar.example.netr r r3 r: r; ) r r
r6 r< r= r �releaser r )
r! r. r r r r �our_id�dead_id�cant_know_if_dead_id� dead_lockr r r �test_kill_staleD s
D & z!TestExclusiveLock.test_kill_stalec C s6 t t }}t||d��� }|j}|�||� |j}d S r2 )r5 r8 r
r6 Z unique_name� migrate_lock)r! r. �old_id�new_idr) Zold_unique_nameZnew_unique_namer r r �test_migrate_lockT s
z#TestExclusiveLock.test_migrate_lockc
s� G dd� d�}�fdd��d
� �fdd� }t d� |� }|� }t� �d }d\}}tt�D ],} t|||| ft| ||�|fd �}|�� qX|�� d S ) Nc @ s6 e Zd Zd
dd�Zdd� Zdd� Zdd � Zd
d � Zd S )zBTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounterr c S s t � | _|| _|| _d S �N)�
ThreadingLockr) �count�maxcount)r! rL r r r �__init__e s zKTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.__init__c S s0 | j � | jW d � S 1 s"0 Y d S rJ �r) rL �r! r r r �valuej s zHTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.valuec S s0 | j � | jW d � S 1 s"0 Y d S rJ )r) rM rP r r r �maxvaluen s zKTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.maxvaluec S sR | j �8 | jd7 _| j| jkr*| j| _| jW d � S 1 sD0 Y d S �Nr )r) rL rM rP r r r �incrr s
zGTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.incrc S s> | j �$ | jd8 _| jW d � S 1 s00 Y d S rS rO rP r r r �decry s zGTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.decrN)r )r& r' r( rN rQ rR rT rU r r r r �SynchronizedCounterc s
rV c s0 � � t | � W d � n1 s"0 Y d S rJ )�print)�msg)�
print_lockr r � print_locked~ s z;TestExclusiveLock.test_race_condition.<locals>.print_lockedc s$ �d|| |� f � t |d��� }d}|�� s�|d7 }zvt� | |d dd��N |�� } �d||| f � t�d� |�� } �d ||| f � W d � n1 s�0 Y W q& ty� �d
||f � Y q& |�� }
t � } �d |||
| f � Y q&0 q&�d ||f � |d u�r |�
� d S )
NzIThread %2d: Starting acquire_release_loop(id=%s, timeout=%d); lockpath=%s���r r � )r4 r" r zbThread %2d: Acquired the lock. It's my %d. loop cycle. I am the %d. who has the lock concurrently.g{�G�zt?zjThread %2d: Releasing the lock, finishing my %d. loop cycle. Currently, %d colleagues still have the lock.z9Thread %2d: Got LockTimeout, finishing my %d. loop cycle.zdThread %2d: Exception thrown, finishing my %d. loop cycle. It's the %d. exception seen until now: %sz=Thread %2d: Loop timed out--terminating after %d loop cycles.) r r Z timed_outr
rT r r rU r r r, ) r4 r" � thread_id�lock_owner_counter�exception_counterrY Z last_threadZtimer�cycleZlock_owner_countZexception_count�e)r. rZ r r �acquire_release_loop� s*
4
zCTestExclusiveLock.test_race_condition.<locals>.acquire_release_loop� )Z
differenthosti� )�target�args)N)rW rK �range�RACE_TEST_NUM_THREADSr �RACE_TEST_DURATIONr r, )
r! r. rV rb r^ r_ �threadZhost_idZ
process_idr] r )r. rY rZ r �test_race_conditiona s$ ��
z%TestExclusiveLock.test_race_conditionN) r& r' r( r1 r9 r$ rE rI rj r r r r r/ 4 s
r/ c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestLockc C s8 t |dtd��� }t |dtd��� }|�� |�� d S )NF�� exclusiver4 )r r5 r6 r8 r@ )r! r. Zlock1Zlock2r r r � test_shared� s zTestLock.test_sharedc C s2 t |dtd��}W d � n1 s$0 Y d S �NTrl )r r5 r0 r r r �test_exclusive� s zTestLock.test_exclusivec C s@ t |dd�� }|�� |�� W d � n1 s20 Y d S )NF�rm )r Zupgrader0 r r r � test_upgrade� s zTestLock.test_upgradec C s@ t |dd�� }|�� |�� W d � n1 s20 Y d S )NTrq )r Z downgrader0 r r r �test_downgrade� s zTestLock.test_downgradec C s" t |dtd�}|�� |�� d S ro )r r5 r6 r@ r0 r r r �test_got_exclusive_lock� s z TestLock.test_got_exclusive_lockc C sL t |dtd��� }|�� t |dtd�� W d � n1 s>0 Y d S ro )r r5 r6 r7 r8 r0 r r r �
test_break� s zTestLock.test_breakc C sL t |dtd��N t�t��$ t |dtdd��� W d � n1 sD0 Y W d � n1 sb0 Y t |dtd��N t�t��$ t |dtdd��� W d � n1 s�0 Y W d � n1 s�0 Y t |dtd��P t�t��$ t |dtdd��� W d � n1 �s 0 Y W d � n1 �s>0 Y d S )NFrl Tr: )rm r4 r" )r r5 r< r= r r8 r6 r> r r r r$ � s P P zTestLock.test_timeoutc C s t � \}}}}|||f}d}t||dd��� } | j}
t||d�� W d � n1 sX0 Y t�t�� | �� W d � n1 s�0 Y t||dd��L t�t��" t||dd��� W d � n1 s�0 Y W d � n1 s�0 Y d S )Nr? T�r4 rm r3 r: r; ) r r r6 Z_rosterr< r= r r@ r ) r! r. r r r r rA rB rC rD �rosterr r r rE � s
& zTestLock.test_kill_stalec C sZ t t }}t||dd��� }|�||� |�� t||dd��� }|�||� |�� d S )NTrv F)r5 r8 r r6 rF r@ )r! r. rG rH r) r r r rI � s
zTestLock.test_migrate_lockN) r& r' r( rn rp rr rs rt ru r$ rE rI r r r r rk � s
rk c C s t | �d��S )Nrw r* r- r r r �
rosterpath s rx c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestLockRosterc C s t |�}|�� }|�|� d S rJ )r �loadZsave)r! rx rw �emptyr r r �
test_empty s
zTestLockRoster.test_emptyc C sd t |td�}|�tt� t |td�}|�tt� t |td�}|�tt� t |td�}|�tt� d S r2 )r r5 �modifyr r r8 r
)r! rx �roster1Zroster2r r r �test_modify_get s z TestLockRoster.test_modify_getc C sz t � \}}}}|||f}t||d�}d|_|�tt� d} t|| d�}d|_|�tt� t|�}
|
�tt� t|�} d S )Nr3 Fr? )r r Zkill_stale_locksr} r r ) r! rx r r r r rA rB r~ rC Z
killer_rosterZother_killer_rosterr r r rE ' s
z TestLockRoster.test_kill_stalec C s4 t t }}t||d�}|�tt� |�t||� d S r2 )r5 r8 r r} r r rF )r! rx rG rH rw r r r rI D s
z TestLockRoster.test_migrate_lockN)r& r' r( r| r rE rI r r r r ry s ry )%r r � threadingr r rK � tracebackr r< Zhelpersr �platformr r Zlockingr r
r r r
r r r r r r5 r8 rg rh Zfixturer r r. r/ rk rx ry r r r r �<module> s* 4
y_