File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/nonces.cpython-39.opt-1.pyc
a
HZ�h � @ s^ d dl Zd dlZddlmZ ddlmZ ddlmZ ddl m
Z
ddl m Z G dd � d �Z
dS )
� N� )�nonces)� NonceManager)�
bin_to_hex)�get_security_dir��InvalidRPCMethodc @ s� e Zd ZG dd� d�ZG dd� de�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd S ) �TestNonceManagerc @ s@ e Zd ZG dd� d�Ze� Zed�Zee�Zdd� Z dd� Z
dS ) z TestNonceManager.MockRepositoryc @ s e Zd ZdZdS )z)TestNonceManager.MockRepository._Locationz /some/placeN)�__name__�
__module__� __qualname__Zorig� r
r
�;/usr/lib64/python3.9/site-packages/borg/testsuite/nonces.py� _Location s r � c C s | j S �N�� next_free��selfr
r
r �get_free_nonce s z.TestNonceManager.MockRepository.get_free_noncec C s
|| _ d S r r �r Znext_unreservedZ start_noncer
r
r �commit_nonce_reservation s z8TestNonceManager.MockRepository.commit_nonce_reservationN) r
r r r Z _location�bytes�idr �id_strr r r
r
r
r �MockRepository s r c @ s e Zd Zdd� Zdd� ZdS )z"TestNonceManager.MockOldRepositoryc C s t d��d S )N� r r r
r
r r s z1TestNonceManager.MockOldRepository.get_free_noncec C s t �d� d S )NzDcommit_nonce_reservation should never be called on an old repository)�pytestZfailr r
r
r r ! s z;TestNonceManager.MockOldRepository.commit_nonce_reservationN)r
r r r r r
r
r
r �MockOldRepository s r c C s
d | _ d S r )�
repositoryr r
r
r �setUp$ s zTestNonceManager.setUpc C sF t tj�t| jj�d���}|�� W d � S 1 s80 Y d S )N�nonce)�open�os�path�joinr r r �read)r �fdr
r
r � cache_nonce' s z TestNonceManager.cache_noncec C s@ t tj�t| jj�d�d��}W d � n1 s20 Y d S )Nr" �w)r# r$ r% r& r r r )r r" r( r
r
r �set_cache_nonce+ s z TestNonceManager.set_cache_noncec C s4 |� tdd� | �� | _t| jd�}|�dd�}d S �N�NONCE_SPACE_RESERVATIONr � � )�setattrr r r r �ensure_reservation�r Z monkeypatchZmanagerZ
next_noncer
r
r � test_empty_cache_and_old_server/ s
z0TestNonceManager.test_empty_cache_and_old_serverc C s< |� tdd� | �� | _d| j_t| jd�}|�dd�}d S r, �r0 r r r r r r1 r2 r
r
r �test_empty_cache9 s
z!TestNonceManager.test_empty_cachec C sl |� tdd� | �� | _d | j_t| jd�}|�dd�}|�dd�}|�dd�}|�dd �}|�d
d �}d S ) Nr- r r. r/ � �
i i3 � iC �@ r4 r2 r
r
r �test_empty_nonceD s
z!TestNonceManager.test_empty_noncec C sF |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}d S )Nr- r r. �0000000000002000r/ �r0 r r r r r+ r r1 r2 r
r
r �test_sync_nonceh s
z TestNonceManager.test_sync_noncec C sF |� tdd� | �� | _d | j_| �d� t| jd�}|�dd�}d S )Nr- r r; r. r/ r< r2 r
r
r �test_server_just_upgradedv s
z*TestNonceManager.test_server_just_upgradedc C s< |� tdd� | �� | _d| j_t| jd�}|�dd�}d S )Nr- r r. � r/ r4 r2 r
r
r � test_transaction_abort_no_cache� s
z0TestNonceManager.test_transaction_abort_no_cachec C s> |� tdd� | �� | _| �d� t| jd�}|�dd�}d S )Nr- r r; r. r? r/ )r0 r r r r+ r r1 r2 r
r
r �!test_transaction_abort_old_server� s
z2TestNonceManager.test_transaction_abort_old_serverc C sF |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}d S )Nr- r r. Z0000000000001000r? r/ r< r2 r
r
r �&test_transaction_abort_on_other_client� s
z7TestNonceManager.test_transaction_abort_on_other_clientc C sf |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}d| j_|�dd�}|�d d
�}d S ) Nr- r r. r; r/ i @ r6 � i � r< r2 r
r
r �test_interleaved� s
z!TestNonceManager.test_interleavedN)r
r r r r r! r) r+ r3 r5 r: r= r> r@ rA rB rE r
r
r
r r s
$
r )Zos.pathr$ r Zcryptor Z
crypto.noncesr Z
crypto.keyr Zhelpersr Zremoter r r
r
r
r �<module> s