| Current File : //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/nonces.cpython-36.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 || j kst�|| _ d S )N)r �AssertionError)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 ks t �|dk st �|| _|| _d S )NF)r r r )r �keyr r r r
�reset) s z$TestNonceManager.MockEncCipher.resetc C s: |j ddd�}| j}||ks t�d| _|j ddd�| _d S )N� �big)� byteorderF)�to_bytesr r r )r Z expected_iv�advancer r r r
�expect_iv_and_advance/ s
z4TestNonceManager.MockEncCipher.expect_iv_and_advancec C s& | j }|dkst�|jddd�| _d S )NFr# r$ )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��}|j|�s,t�W d Q R X d S )Nr- �w) r/ r0 r1 r2 r r+ r �writer )r r- r4 r r r
�set_cache_nonceB s z TestNonceManager.set_cache_noncec C sZ |j tdd� | jd�}| j� | _t| j|d�}|jd� |jdd� | j� dksVt �d S )N�NONCE_SPACE_RESERVATIONr i � i �0000000000002033)
�setattrr r* r r+ r �ensure_reservationr( r5 r )r � monkeypatch�
enc_cipher�managerr r r
� test_empty_cache_and_old_serverF s
z0TestNonceManager.test_empty_cache_and_old_serverc C sb |j tdd� | jd�}| j� | _d| j_t| j|d�}|jd� |jdd� | j � dks^t
�d S )Nr9 r i r: i r; ) r< r r* r r+ r r r= r( r5 r )r r> r? r@ r r r
�test_empty_cacheQ s
z!TestNonceManager.test_empty_cachec C sJ |j tdd� | jd�}| j� | _d | j_t| j|d�}|jd� |jdd� | j � dks^t
�| jjdksnt
�|jd� |j d� | j � dks�t
�| jjdks�t
�|jd� |j d� | j � dks�t
�| jjdks�t
�|jd� |j d� | j � d k� s�t
�| jjd
k�st
�|jd � |j d � | j � d k�s4t
�| jjd
k�sFt
�d S ) Nr9 r i r: r; i3 �
r# Z0000000000002063ic �@ Z00000000000020c3i� i i i i i i3 i i i3 iC i i i3 iC i� ) r< r r* r r+ r r r= r( r5 r r) )r r> r? r@ r r r
�test_empty_nonce] s2
z!TestNonceManager.test_empty_noncec C s| |j tdd� | jd�}| j� | _d| j_| jd� t| j|d�}|jd� |j dd� | j
� dksht �| jjdksxt �d S ) Nr9 r i �0000000000002000r: r; i3 i ) r< r r* r r+ r r8 r r= r( r5 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� | j
� dksht �| jjdksxt �d S ) Nr9 r i rF r: r; i3 i ) r< r r* r r+ r r8 r r= r( r5 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� | j � dks^t
�| jjdksnt
�d S ) Nr9 r i i r: r; i3 i ) r< r r* r r+ r r r= r( r5 r )r r> r? r@ r r r
� test_transaction_abort_no_cache� s
z0TestNonceManager.test_transaction_abort_no_cachec C sd |j tdd� | jd�}| j� | _| jd� t| j|d�}|jd� |jdd� | j � dks`t
�d S ) Nr9 r i rF i r: r; i ) r< r r* r r+ r8 r r= r( r5 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 � | j
� dksht �| jjdksxt �d S )
Nr9 r i i Z0000000000001000r: r; i3 i ) r< r r* r r+ r r8 r r= r( r5 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
� | j
� dksht �| jjdksxt �d| j_|jd � |j d� | j
� dks�t �| jjdks�t �|jd
� |j dd� | j
� d ks�t �| jjd ks�t �d S )Nr9 r i rF r: r; i3 i @ � � Z0000000000004035i5@ i i i i@ )
r< r r* r r+ r r8 r r= r( r5 r r) )r r> r? r@ r r r
�test_interleaved� s&
z!TestNonceManager.test_interleavedN)r r
r r r r* r, r5 r8 rA rB rE rG rH rI rJ rK rN r r r r
r s %
r )Zos.pathr0 r Zcryptor Z
crypto.noncesr Z
crypto.keyr Zhelpersr Zremoter r r r r r
�<module> s