File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/nonces.cpython-39.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 || j ksJ �|| _ 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 sN t tj�t| jj�d�d�� }|�|�s,J �W d � n1 s@0 Y d S )Nr" �w)r# r$ r% r& r r r �write)r r" r( r
r
r �set_cache_nonce+ s z TestNonceManager.set_cache_noncec C sP |� tdd� | �� | _t| jd�}|�dd�}|dks<J �| �� dksLJ �d S �N�NONCE_SPACE_RESERVATIONr � � �0000000000002033)�setattrr r r r �ensure_reservationr) �r Z monkeypatchZmanagerZ
next_noncer
r
r � test_empty_cache_and_old_server/ s
z0TestNonceManager.test_empty_cache_and_old_serverc C sX |� tdd� | �� | _d| j_t| jd�}|�dd�}|dksDJ �| �� dksTJ �d S r- �r2 r r r r r r3 r) r4 r
r
r �test_empty_cache9 s
z!TestNonceManager.test_empty_cachec C sP |� tdd� | �� | _d | j_t| jd�}|�dd�}|dksDJ �| �� dksTJ �| jjdksdJ �|�dd�}|dks|J �| �� dks�J �| jjdks�J �|�d d�}|d ks�J �| �� dks�J �| jjdks�J �|�dd
�}|dks�J �| �� d ks�J �| jjd k�sJ �|�d
d�}|d k�s(J �| �� dk�s:J �| jjdk�sLJ �d S )Nr. r r/ r0 r1 �3 � �
i � Z0000000000002063ic iC �@ Z00000000000020c3i� r6 r4 r
r
r �test_empty_nonceD s0
z!TestNonceManager.test_empty_noncec C sr |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}|dksNJ �| �� dks^J �| jjdksnJ �d S )Nr. r r/ �0000000000002000r0 r1 r8 � r2 r r r r r, r r3 r) r4 r
r
r �test_sync_nonceh s
z TestNonceManager.test_sync_noncec C sr |� tdd� | �� | _d | j_| �d� t| jd�}|�dd�}|dksNJ �| �� dks^J �| jjdksnJ �d S )Nr. r r> r/ r0 r1 r8 r? r4 r
r
r �test_server_just_upgradedv s
z*TestNonceManager.test_server_just_upgradedc C sh |� tdd� | �� | _d| j_t| jd�}|�dd�}|dksDJ �| �� dksTJ �| jjdksdJ �d S )Nr. r r/ � r0 r1 r8 r6 r4 r
r
r � test_transaction_abort_no_cache� s
z0TestNonceManager.test_transaction_abort_no_cachec C sZ |� tdd� | �� | _| �d� t| jd�}|�dd�}|dksFJ �| �� dksVJ �d S )Nr. r r> r/ rB r0 r1 )r2 r r r r, r r3 r) r4 r
r
r �!test_transaction_abort_old_server� s
z2TestNonceManager.test_transaction_abort_old_serverc C sr |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}|dksNJ �| �� dks^J �| jjdksnJ �d S ) Nr. r r/ Z0000000000001000rB r0 r1 r8 r? r4 r
r
r �&test_transaction_abort_on_other_client� s
z7TestNonceManager.test_transaction_abort_on_other_clientc C s� |� tdd� | �� | _d| j_| �d� t| jd�}|�dd�}|dksNJ �| �� dks^J �| jjdksnJ �d| j_|�d d
�}|d ks�J �| �� dks�J �| jjdks�J �|�d d �}|dks�J �| �� d
ks�J �| jjdks�J �d S )Nr. r r/ r> r0 r1 r8 i @ r9 � i � Z0000000000004035i5@ r? r4 r
r
r �test_interleaved� s$
z!TestNonceManager.test_interleavedN)r
r r r r r! r) r, r5 r7 r= r@ rA rC rD rE rH 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