File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/nonces.cpython-36.opt-1.pyc
3
y��b�! � @ 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�ZG dd� d�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � 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.6/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 )N)r )r �next_unreserved� 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 jd� d S )NzDcommit_nonce_reservation should never be called on an old repository)�pytestZfail)r r r r r r
r ! s z;TestNonceManager.MockOldRepository.commit_nonce_reservationN)r r
r r r r r r r
�MockOldRepository s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
z TestNonceManager.MockEncCipherc C s d| _ || _d S )NF)�iv_set�iv)r r r r r
�__init__% s z'TestNonceManager.MockEncCipher.__init__c C s || _ || _d S )N)r r )r �keyr r r r
�reset) s z$TestNonceManager.MockEncCipher.resetc C s. |j ddd�}| j}d| _|j ddd�| _d S )N� �big)� byteorderF)�to_bytesr r )r Z expected_iv�advancer r r r
�expect_iv_and_advance/ s z4TestNonceManager.MockEncCipher.expect_iv_and_advancec C s | j }|jddd�| _d S )Nr" r# )r$ )r r% r )r r& r r r r
�expect_no_reset_and_advance6 s z:TestNonceManager.MockEncCipher.expect_no_reset_and_advanceN)r r
r r r! r' r( r r r r
�
MockEncCipher$ s r) c C s
d | _ d S )N)�
repository)r r r r
�setUp; s zTestNonceManager.setUpc C s0 t tjjt| jj�d�d��
}|j� S Q R X d S )N�nonce�r)�open�os�path�joinr r* r �read)r �fdr r r
� cache_nonce> s z TestNonceManager.cache_noncec C s, t tjjt| jj�d�d��}W d Q R X d S )Nr, �w)r. r/ r0 r1 r r* r )r r, r3 r r r
�set_cache_nonceB s z TestNonceManager.set_cache_noncec C sJ |j tdd� | jd�}| j� | _t| j|d�}|jd� |jdd� d S )N�NONCE_SPACE_RESERVATIONr i � i )�setattrr r) r r* r �ensure_reservationr' )r � monkeypatch�
enc_cipher�managerr r r
� test_empty_cache_and_old_serverF s
z0TestNonceManager.test_empty_cache_and_old_serverc C sR |j tdd� | jd�}| j� | _d| j_t| j|d�}|jd� |jdd� d S )Nr7 r i r8 i ) r9 r r) r r* r r r: r' )r r; r< r= r r r
�test_empty_cacheQ s
z!TestNonceManager.test_empty_cachec C s� |j tdd� | jd�}| j� | _d | j_t| j|d�}|jd� |jdd� |jd� |j d
� |jd� |j d
� |jd� |j d� |jd� |j d� d S )Nr7 r i r8 �
r" �@ i i i i i i3 i i i3 iC i i i3 iC i� )
r9 r r) r r* r r r: r' r( )r r; r< r= r r r
�test_empty_nonce] s
z!TestNonceManager.test_empty_noncec C s\ |j tdd� | jd�}| j� | _d| j_| jd� t| j|d�}|jd� |j dd� d S )Nr7 r i �0000000000002000r8 i )
r9 r r) r r* r r6 r r: r' )r r; r< r= r r r
�test_sync_nonce� s
z TestNonceManager.test_sync_noncec C s\ |j tdd� | jd�}| j� | _d | j_| jd� t| j|d�}|jd� |j dd� d S )Nr7 r i rC r8 i )
r9 r r) r r* r r6 r r: r' )r r; r< r= r r r
�test_server_just_upgraded� s
z*TestNonceManager.test_server_just_upgradedc C sR |j tdd� | jd�}| j� | _d| j_t| j|d�}|jd� |jdd� d S )Nr7 r i i r8 i ) r9 r r) r r* r r r: r' )r r; r< r= r r r
� test_transaction_abort_no_cache� s
z0TestNonceManager.test_transaction_abort_no_cachec C sT |j tdd� | jd�}| j� | _| jd� t| j|d�}|jd� |jdd� d S )Nr7 r i rC i r8 i ) r9 r r) r r* r6 r r: r' )r r; r< r= r r r
�!test_transaction_abort_old_server� s
z2TestNonceManager.test_transaction_abort_old_serverc C s\ |j tdd� | jd�}| j� | _d| j_| jd� t| j|d�}|jd� |j dd� d S )Nr7 r i i Z0000000000001000r8 i )
r9 r r) r r* r r6 r r: r' )r r; r< r= r r r
�&test_transaction_abort_on_other_client� s
z7TestNonceManager.test_transaction_abort_on_other_clientc C s� |j tdd� | jd�}| j� | _d| j_| jd� t| j|d�}|jd� |j dd � d| j_|jd� |j
d � |jd� |j dd � d S )
Nr7 r i rC r8 i @ � � i i i i@ ) r9 r r) r r* r r6 r r: r' r( )r r; r< r= r r r
�test_interleaved� s
z!TestNonceManager.test_interleavedN)r r
r r r r) r+ r4 r6 r> r? rB rD rE rF rG rH rK 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