| Current File : //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/locking.cpython-36.pyc |
3
y��b�% � @ s� d dl Z d dlZd dlZddlmZ ddlmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZmZmZmZ dZdZej� d d
� �ZG d d � d �Zej� d
d� �ZG dd� d�ZG dd� d�Zej� dd� �Z G dd� d�Z dS )� N� )� daemonize)�get_process_id�
process_alive) � TimeoutTimer�
ExclusiveLock�Lock�
LockRoster�ADD�REMOVE�SHARED� EXCLUSIVE� LockTimeout� NotLocked� NotMyLock�foo� �barc C s2 t � \} }}x tjdd�}t| ||�s|S qW 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.6/locking.py�free_pid s
r c @ s e Zd Zdd� Zdd� ZdS )�TestTimeoutTimerc C s< d}t |�j� }|j� s t�tj|d � |j� s8t�d S )Ng �?g �?)r �startZ timed_out�AssertionError�time�sleep)�self�timeout�tr r r � test_timeout s
z TestTimeoutTimer.test_timeoutc C sj d\}}t ||�j� }|j� s$t�tj� |jd| ks>t�|j� sLt�tj� |jd| ksft�d S )N� �?r r )Nr$ )r r Ztimed_out_or_sleepr r 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 | jd��S )N�lock)�str�join)�tmpdirr r r �lockpath+ s r- c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestExclusiveLockc C s0 t |dd��}|j� r |j� s"t�W d Q R X d S )Nr )r! )r Z is_locked�by_mer )r r- r) r r r � test_checks1 s z TestExclusiveLock.test_checksc C s4 t |td�j� }|j� t |td�� W d Q R X d S )N)�id)r �ID1�acquire�
break_lock�ID2)r r- r) r r r � test_acquire_break_reacquire5 s z.TestExclusiveLock.test_acquire_break_reacquirec C sD t |td��. tjt�� t |tdd�j� W d Q R X W d Q R X d S )N)r1 g�������?)r1 r! )r r2 �pytest�raisesr r5 r3 )r r- r r r r# ; s z TestExclusiveLock.test_timeoutc
) C s� t � \}}}}|||f}d }t||d�j� } t||dd��$ tjt�� | j� W d Q R X W d Q R X tjt�� | j� W d Q R X t||d��0 tjt�� t||ddd�j� W d Q R X W d Q R X d S )
N�foo.bar.example.netr r )r1 T)r1 �kill_stale_locksg�������?)r1 r: r! )r9 r r ) r r r3 r7 r8 r �releaser r )
r r- r r r r �our_id�dead_id�cant_know_if_dead_id� dead_lockr r r �test_kill_stale@ s
z!TestExclusiveLock.test_kill_stalec C s� t t }}|d |d ks t�t||d�j� }|j|ks<t�|j}|j� sNt�|j||� |j|ksht�|j}|j� szt�||ks�t�d S )Nr )r1 ) r2 r5 r r r3 r1 Z unique_namer/ � migrate_lock)r r- �old_id�new_idr) Zold_unique_nameZnew_unique_namer r r �test_migrate_lockP s
z#TestExclusiveLock.test_migrate_lockN)r& r' r( r0 r6 r# r@ rD r r r r r. 0 s
r. 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�j� }t |dtd�j� }t|jjt��dks<t�t|jjt ��dksTt�|jj
tt � sht�|jj
t �sxt�|j � |j � d S )NF)� exclusiver1 r r ) r r2 r3 r5 �len�_roster�getr r r
�emptyr; )r r- Zlock1Zlock2r r r � test_shared_ s zTestLock.test_sharedc
C sb t |dtd��J}t|jjt��dks(t�t|jjt��dks@t�|jjtt� sTt�W d Q R X d S )NT)rF r1 r r ) r r2 rG rH rI r r r
rJ )r r- r) r r r �test_exclusivei s zTestLock.test_exclusivec
C sp t |dd��Z}|j� |j� t|jjt��dks6t�t|jjt��dksNt�|jjtt� sbt�W d Q R X d S )NF)rF r r ) r ZupgraderG rH rI r r r
rJ )r r- r) r r r � test_upgradeo s zTestLock.test_upgradec
C s\ t |dd��F}|j� |j� t|jjt��dks6t�t|jjt��dksNt�W d Q R X d S )NT)rF r r )r Z downgraderG rH rI r r r
)r r- r) r r r �test_downgradew s
zTestLock.test_downgradec C sJ t |dtd�}|j� s t�|j� |j� s0t�|j� |j� sFt�d S )NT)rF r1 )r r2 Zgot_exclusive_lockr r3 r; )r r- r) r r r �test_got_exclusive_lock~ s z TestLock.test_got_exclusive_lockc C sh t |dtd�j� }|j� t|jjt��dks2t�t|jjt ��dksJt�t |dt
d�� W d Q R X d S )NT)rF r1 r ) r r2 r3 r4 rG rH rI r r r
r5 )r r- r) r r r �
test_break� s zTestLock.test_breakc 0 C s� t |dtd��0 tjt�� t |dtdd�j� W d Q R X W d Q R X t |dtd��0 tjt�� t |dtdd�j� W d Q R X W d Q R X t |dtd��0 tjt�� t |dtdd�j� W d Q R X W d Q R X d S )NF)rF r1 Tg�������?)rF r1 r! )r r2 r7 r8 r r5 r3 )r r- r r r r# � s ( ( zTestLock.test_timeoutc " C s t � \}}}}|||f}d }t||dd�j� } | j}
t||dd��. |
jt�t� ksZt�|
jt�|hksnt�W d Q R X |
jt�t� ks�t�|
jt�t� ks�t�t j
t �� | j � W d Q R X t||dd��0 t j
t
�� t||ddd�j� W d Q R X W d Q R X d S )
N�foo.bar.example.netr r T)r1 rF )r1 r: g�������?)r1 r: r! )rQ r r )r r r3 rH rI r
�setr r r7 r8 �KeyErrorr; r ) r r- r r r r r< r= r> r? �rosterr r r r@ � s
zTestLock.test_kill_stalec C s� t t }}|d |d ks t�t||dd�j� }|j|ks>t�|j||� |j|ksXt�|j� t||dd�j� }|j|ks�t�|j||� |j|ks�t�|j� d S )Nr T)r1 rF F)r2 r5 r r r3 r1 rA r; )r r- rB rC r) r r r rD � s
zTestLock.test_migrate_lockN) r& r' r( rK rL rM rN rO rP r# r@ rD r r r r rE ^ s
rE c C s t | jd��S )NrT )r* r+ )r, r r r �
rosterpath� s rU c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestLockRosterc C s* t |�}|j� }|j|� |i ks&t�d S )N)r �loadZsaver )r rU rT rJ r r r �
test_empty� s
zTestLockRoster.test_emptyc C s� t |td�}|jt�t� ks t�|jtt� |jt�thks@t�t |td�}|jtt� |jt�tthksnt�t |td�}|jtt � |jt�thks�t�t |td�}|jtt � |jt�t� ks�t�d S )N)r1 )
r r2 rI r rR r �modifyr
r5 r )r rU �roster1Zroster2r r r �test_modify_get� s z TestLockRoster.test_modify_getc C s
t � \}}}}|||f}t||d�}|jt�t� ks:t�|jtt� |jt�|hksZt�d} t|| d�}|jt�|hks~t�|jtt� |jt�|| hks�t�t|dd�}
|
jt�| hks�t�|
jtt� |
jt�|| hks�t�t|dd�} | jt�|| hk�st�d S )N)r1 �foo.bar.example.netr r T)r: )r\ r r )r r rI r rR r rY r
) r rU r r r r r< r= rZ r> Z
killer_rosterZother_killer_rosterr r r r@ � s"
z TestLockRoster.test_kill_stalec C s� t t }}|d |d ks t�t||d�}|j|ks8t�|jtt� |jt�|hksXt�|j t||� |j|kstt�|jt�|hks�t�d S )Nr )r1 )
r2 r5 r r r1 rY r r
rI rA )r rU rB rC rT r r r rD � s
z TestLockRoster.test_migrate_lockN)r& r' r( rX r[ r@ rD r r r r rV � s rV )r r r )r r r ) r r r7 Zhelpersr �platformr r Zlockingr r r r r
r r r
r r r r2 r5 Zfixturer r r- r. rE rU rV r r r r �<module> s 4 ._