File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/locking.cpython-39.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 |��� }|�� r J �t�|d � |�� s6J �d S )N� �?g �?)r �start� timed_out�time�sleep)�self�timeout�tr r r � test_timeout s
z TestTimeoutTimer.test_timeoutc C sf d\}}t ||��� }|�� r"J �t�� |jd| ks<J �|�� rHJ �t�� |jd| ksbJ �d S )N)Nr r r )r r Ztimed_out_or_sleepr Z
start_time)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 sD t |dd��$}|�� r |�� s"J �W d � n1 s60 Y d S )Nr )r# )r
Z is_locked�by_me�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�ID2r2 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 )Nr5 皙�����?�r6 r# )r
r7 �pytest�raisesr r: r8 �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 r5 r< r= ) r r
r8 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 s� t t }}|d |d ks J �t||d��� }|j|ks<J �|j}|�� sNJ �|�||� |j|kshJ �|j}|�� szJ �||ks�J �d S �Nr r5 )r7 r: r
r8 r6 Z unique_namer1 � 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
ks�J d ��|�� d ks�J d
|�� ��|� � d
ks�J d|� � ��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" rO 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 rM �r* rO �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 rM )r* rP rS 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* rO rP rS r r r �incrr s
zGTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.incrc S s> | j �$ | jd8 _| jW d � S 1 s00 Y d S rV rR rS r r r �decry s zGTestExclusiveLock.test_race_condition.<locals>.SynchronizedCounter.decrN)r )r' r( r) rQ rT rU rW rX r r r r �SynchronizedCounterc s
rY c s0 � � t | � W d � n1 s"0 Y d S rM )�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 � )r6 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 r r
rW r r! rX r r r- ) r6 r# � thread_id�lock_owner_counter�exception_counterr\ Z last_threadZtimer�cycleZlock_owner_countZexception_count�e)r/ r] r r �acquire_release_loop� s*
4
zCTestExclusiveLock.test_race_condition.<locals>.acquire_release_loop� )Z
differenthosti� )�target�argsr z3Never gained the lock? Something went wrong here...r zKMaximal number of concurrent lock holders was %d. So exclusivity is broken.zFExclusiveLock threw %d exceptions due to unclean concurrency handling.)N)
rZ rN �range�RACE_TEST_NUM_THREADSr �RACE_TEST_DURATIONr r- rU rT )
r" r/ rY re ra rb �threadZhost_idZ
process_idr` r )r/ r\ r] r �test_race_conditiona s( ��
z%TestExclusiveLock.test_race_conditionN) r' r( r) r3 r; r% rG rL rm r r r r r0 4 s
r0 c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestLockc C s� t |dtd��� }t |dtd��� }t|j�t��dks<J �t|j�t��dksTJ �|j� tt�rfJ �|j� t�svJ �|�
� |�
� d S )NF�� exclusiver6 r r ) r r7 r8 r: �len�_roster�getr r �emptyrB )r" r/ Zlock1Zlock2r r r � test_shared� s zTestLock.test_sharedc C st t |dtd��R}t|j�t��dks(J �t|j�t��dks@J �|j�tt�rRJ �W d � n1 sf0 Y d S )NTro r r )r r7 rq rr rs r r rt r2 r r r �test_exclusive� s zTestLock.test_exclusivec C s� t |dd��b}|�� |�� t|j�t��dks6J �t|j�t��dksNJ �|j�tt�r`J �W d � n1 st0 Y d S )NF�rp r r )r Zupgraderq rr rs r r rt r2 r r r � test_upgrade� s zTestLock.test_upgradec C sp t |dd��P}|�� |�� t|j�t��dks6J �t|j�t��dksNJ �W d � n1 sb0 Y d S )NTrw r r )r Z downgraderq rr rs r r r2 r r r �test_downgrade� s
zTestLock.test_downgradec C sF t |dtd�}|�� rJ �|�� |�� s.J �|�� |�� rBJ �d S )NTro )r r7 Zgot_exclusive_lockr8 rB r2 r r r �test_got_exclusive_lock� s z TestLock.test_got_exclusive_lockc C s| t |dtd��� }|�� t|j�t��dks2J �t|j�t��dksJJ �t |dt d�� W d � n1 sn0 Y d S )NTro r )
r r7 r8 r9 rq rr rs r r r: r2 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 )NFro Tr< )rp r6 r# )r r7 r> r? r r: r8 r@ r r r r% � s P P zTestLock.test_timeoutc C sV t � \}}}}|||f}d}t||dd��� } | j}
t||d��8 |
�t�t� ksXJ �|
�t�|hkslJ �W d � n1 s�0 Y |
�t�t� ks�J �|
�t�t� ks�J �t� t
�� | � � W d � n1 s�0 Y t||dd��N t� t ��" t||dd��� W d � n1 �s(0 Y W d � n1 �sH0 Y d S )NrA T�r6 rp r5 r< r= )
r r r8 rr rs r �setr r> r? r rB r ) r" r/ r r r r rC rD rE rF �rosterr r r rG � s
2 & zTestLock.test_kill_stalec C s� t t }}|d |d ks J �t||dd��� }|j|ks>J �|�||� |j|ksXJ �|�� t||dd��� }|j|ks�J �|�||� |j|ks�J �|�� d S )Nr Tr| F)r7 r: r r8 r6 rI rB )r" r/ rJ rK r* r r r rL � s
zTestLock.test_migrate_lockN) r' r( r) ru rv rx ry rz r{ r% rG rL r r r r rn � s
rn c C s t | �d��S )Nr~ r+ r. r r r �
rosterpath s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestLockRosterc C s* t |�}|�� }|�|� |i ks&J �d S rM )r �loadZsave)r" r r~ rt r r r �
test_empty s
zTestLockRoster.test_emptyc C s� t |td�}|�t�t� ks J �|�tt� |�t�thks@J �t |td�}|�tt� |�t�tthksnJ �t |td�}|�tt� |�t�thks�J �t |td�}|�tt� |�t�t� ks�J �d S r4 ) r r7 rs r r} �modifyr r: r
)r" r �roster1Zroster2r r r �test_modify_get s z TestLockRoster.test_modify_getc C s t � \}}}}|||f}t||d�}d|_|�t�t� ks@J �|�tt� |�t�|hks`J �d} t|| d�}d|_|�t�|hks�J �|�tt� |�t�|| hks�J �t|�}
|
�t�| hks�J �|
�tt� |
�t�|| hks�J �t|�} | �t�|| hk�s
J �d S )Nr5 FrA )r r Zkill_stale_locksrs r r} r� r ) r" r r r r r rC rD r� rE Z
killer_rosterZother_killer_rosterr r r rG ' s&
z TestLockRoster.test_kill_stalec C s� t t }}|d |d ks J �t||d�}|j|ks8J �|�tt� |�t�|hksXJ �|�t||� |j|kstJ �|�t�|hks�J �d S rH ) r7 r: r r6 r� r r rs rI )r" r rJ rK r~ r r r rL D s
z TestLockRoster.test_migrate_lockN)r' r( r) r� r� rG rL r r r r r� s r� )%r r � threadingr r rN � tracebackr r> Zhelpersr �platformr r Zlockingr r
r r r
r r r r r r7 r: rj rk Zfixturer r r/ r0 rn r r� r r r r �<module> s* 4
y_