File: //usr/lib64/python3.6/site-packages/borg/testsuite/__pycache__/file_integrity.cpython-36.pyc
3
y��b� � @ sJ d dl Z ddlmZmZmZ G dd� d�ZG dd� d�ZG dd � d �ZdS )
� N� )�IntegrityCheckedFile� DetachedIntegrityCheckedFile�FileIntegrityErrorc @ s: e Zd Zdd� Zdd� Zdd� Zejjdd�d
d� �Z dS )�TestReadIntegrityFilec C s. |j d�}|jd� tjt|��d ks*t�d S )N�file�1234)�join�writer �read_integrity_file�str�AssertionError)�self�tmpdir�protected_file� r �&/usr/lib64/python3.6/file_integrity.py�test_no_integrity s
z'TestReadIntegrityFile.test_no_integrityc
C sL |j d�}|jd� |j d�jd� tjt�� tjt|�� W d Q R X d S )Nr r zfile.integrity� )r r
�pytest�raisesr r r r )r r r r r r �test_truncated_integrity
s
z.TestReadIntegrityFile.test_truncated_integrityc C s> |j d�}|jd� |j d�jd� tjt|��d ks:t�d S )Nr r zfile.integrityz4{"algorithm": "HMAC_SERIOUSHASH", "digests": "1234"})r r
r r r r
)r r r r r r �test_unknown_algorithm s
z,TestReadIntegrityFile.test_unknown_algorithm�json�4{"ALGORITHM": "HMAC_SERIOUSHASH", "digests": "1234"}�[]�1234.5�
"A string"� Invalid JSONc
C sL |j d�}|jd� |j d�j|� tjt�� tjt|�� W d Q R X d S )Nr r zfile.integrity)r r
r r r r r r )r r r r r r r �test_malformed s
z$TestReadIntegrityFile.test_malformedN)r r r r r )
�__name__�
__module__� __qualname__r r r r �mark� parametrizer r r r r r s r c @ sX e Zd Zejdd� �Zdd� Zdd� Zdd� Zej j
d d�d d
� �Z dd� Z dd� Z
dS )� TestDetachedIntegrityCheckedFilec C s4 t |jd��}t|dd��}|jd� W d Q R X |S )Nr T)r
s foo and bar)r r r r
)r r �path�fdr r r �integrity_protected_file* s z9TestDetachedIntegrityCheckedFile.integrity_protected_filec C sX |j d�jdd�st�|j d�jdd�s,t�t|dd��}|j� dksJt�W d Q R X d S )Nr T)r zfile.integrityF)r
s foo and bar)r Zcheckr
r �read)r r r( r' r r r � test_simple1 s z,TestDetachedIntegrityCheckedFile.test_simplec C sb t |d��}|jd� W d Q R X tjt��. t|dd��}|j� dksJt�W d Q R X W d Q R X d S )N�abs extra dataF)r
s foo and bar extra data)�openr
r r r r r) r
)r r( r' r r r �test_corrupted_file7 s
z4TestDetachedIntegrityCheckedFile.test_corrupted_filec C sl t |d��}|jd� W d Q R X tjt��8 t|dd�� }d}|jt|��|ksTt�W d Q R X W d Q R X d S )Nr+ s extra dataF)r
s foo and bar) r, r
r r r r r) �lenr
)r r( r' �datar r r � test_corrupted_file_partial_read> s zATestDetachedIntegrityCheckedFile.test_corrupted_file_partial_read�new_name�different_file� different_file.different_extc C st |j |�}|j d�j|� |j d�j|d � tjt��2 tt|�dd��}|j� dks\t�W d Q R X W d Q R X d S )Nr zfile.integrityz
.integrityF)r
s foo and bar) r �mover r r r r r) r
)r r r( r1 �new_pathr' r r r �test_renamed_fileF s
z2TestDetachedIntegrityCheckedFile.test_renamed_filec C sp |j d�}|jd�j|jd�� |jd�j|jd�� t|jd��}t|dd��}|j� dksbt�W d Q R X d S )NZanother_directoryr zfile.integrityF)r
s foo and bar)�mkdirr r4 r r r) r
)r r r( Znew_dirr5 r' r r r �test_moved_fileR s
z0TestDetachedIntegrityCheckedFile.test_moved_filec C s: |j d�j� t|dd��}|j� dks,t�W d Q R X d S )Nzfile.integrityF)r
s foo and bar)r �remover r) r
)r r r( r' r r r r Z s z2TestDetachedIntegrityCheckedFile.test_no_integrityN)r2 r3 )r r! r" r �fixturer( r* r- r0 r# r$ r6 r8 r r r r r r% ) s
r% c @ s@ e Zd Zejdd� �Zdd� Zdd� Zejj dd
�d
d � �Z
d S )�%TestDetachedIntegrityCheckedFilePartsc C sH t |jd��}t|dd��$}|jd� |jd� |jd� W d Q R X |S )Nr T)r
s foo and bar�fooparts other data)r r r r
� hash_part)r r r& r' r r r r( a s
z>TestDetachedIntegrityCheckedFileParts.integrity_protected_filec
C sP t |dd��:}d}|jt|��|ks(t�|jd� |j� dksBt�W d Q R X d S )NF)r
s foo and barr< s other data)r r) r. r
r= )r r( r' �data1r r r r* j s
z1TestDetachedIntegrityCheckedFileParts.test_simplec C sl t jt��X t|dd��@}d}|jt|��|ks4t�t jt�� |jd� W d Q R X W d Q R X W d Q R X d S )NF)r
s foo and barZbarpart)r r r r r) r. r
r= )r r( r' r> r r r �test_wrong_part_nameq s z:TestDetachedIntegrityCheckedFileParts.test_wrong_part_name� partial_readFTc C s� t |d��}|jd� W d Q R X tjt��t t|dd��\}d}y$|jt|��|ksVt�|j d� W n tk
r� ds~td��Y nX |s�|j� W d Q R X W d Q R X d S )Nr+ s% some extra stuff that does not belongF)r
s foo and barr< z9This part must not raise, since this part is still valid.)
r, r
r r r r r) r. r
r= )r r( r@ r' r> r r r �test_part_independence~ s z<TestDetachedIntegrityCheckedFileParts.test_part_independenceN)FT) r r! r" r r: r( r* r? r# r$ rA r r r r r; ` s
r; )r Zcrypto.file_integrityr r r r r% r; r r r r �<module> s "7