File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/locking.cpython-36.opt-1.pyc
3
�i�% � @ 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� }tj|d � d S )Ng �?g �?)r �start�time�sleep)�self�timeout�tr r r � test_timeout s z TestTimeoutTimer.test_timeoutc C s d\}}t ||�j� }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 | 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 s t |dd��}W d Q R X d S )Nr )r )r )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)r/ g�������?)r/ r )r r0 �pytest�raisesr r3 r1 )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 )r/ T)r/ �kill_stale_locksg�������?)r/ r8 r )r7 r r ) r r r1 r5 r6 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 s6 t t }}t||d�j� }|j}|j||� |j}d S )N)r/ )r0 r3 r r1 Z unique_name� 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' r. r4 r" r> rB 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 s8 t |dtd�j� }t |dtd�j� }|j� |j� d S )NF)� exclusiver/ )r r0 r1 r3 r9 )r r, Zlock1Zlock2r r r � test_shared_ s zTestLock.test_sharedc C s t |dtd��}W d Q R X d S )NT)rD r/ )r r0 )r r, r( r r r �test_exclusivei s zTestLock.test_exclusivec C s, t |dd��}|j� |j� W d Q R X d S )NF)rD )r Zupgrade)r r, r( r r r � test_upgradeo s zTestLock.test_upgradec C s, t |dd��}|j� |j� W d Q R X d S )NT)rD )r Z downgrade)r r, r( r r r �test_downgradew s zTestLock.test_downgradec C s" t |dtd�}|j� |j� d S )NT)rD r/ )r r0 r1 r9 )r r, r( r r r �test_got_exclusive_lock~ s z TestLock.test_got_exclusive_lockc C s8 t |dtd�j� }|j� t |dtd�� W d Q R X d S )NT)rD r/ )r r0 r1 r2 r3 )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)rD r/ Tg�������?)rD r/ r )r r0 r5 r6 r r3 r1 )r r, r r r r" � s ( ( zTestLock.test_timeoutc " C s� t � \}}}}|||f}d }t||dd�j� } | j}
t||dd�� W d Q R X tjt�� | j� W d Q R X t||dd��0 tjt�� t||ddd�j� W d Q R X W d Q R X d S )
N�foo.bar.example.netr r T)r/ rD )r/ r8 g�������?)r/ r8 r )rK r r ) r r r1 Z_rosterr5 r6 r r9 r ) r r, r r r r r: r; r< r= �rosterr r r r> � s
zTestLock.test_kill_stalec C sZ t t }}t||dd�j� }|j||� |j� t||dd�j� }|j||� |j� d S )NT)r/ rD F)r0 r3 r r1 r? r9 )r r, r@ rA r( r r r rB � s
zTestLock.test_migrate_lockN) r% r& r' rE rF rG rH rI rJ r" r> rB r r r r rC ^ s
rC c C s t | jd��S )NrL )r) r* )r+ r r r �
rosterpath� s rM c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestLockRosterc C s t |�}|j� }|j|� d S )N)r �loadZsave)r rM rL �emptyr r r �
test_empty� s
zTestLockRoster.test_emptyc C sd t |td�}|jtt� t |td�}|jtt� t |td�}|jtt� t |td�}|jtt� d S )N)r/ )r r0 �modifyr r
r3 r )r rM �roster1Zroster2r r r �test_modify_get� s z TestLockRoster.test_modify_getc C sv t � \}}}}|||f}t||d�}|jtt� d} t|| d�}|jtt� t|dd�}
|
jtt� t|dd�} d S )N)r/ �foo.bar.example.netr r T)r8 )rU r r )r r rR r r
) r rM r r r r r: r; rS r< Z
killer_rosterZother_killer_rosterr r r r> � s
z TestLockRoster.test_kill_stalec C s4 t t }}t||d�}|jtt� |jt||� d S )N)r/ )r0 r3 r rR r r
r? )r rM r@ rA rL r r r rB � s
z TestLockRoster.test_migrate_lockN)r% r& r' rQ rT r> rB r r r r rN � s rN )r r r )r r r ) r r r5 Zhelpersr �platformr r Zlockingr r r r r
r r r
r r r r0 r3 Zfixturer r r, r- rC rM rN r r r r �<module> s 4 ._