File: //usr/lib64/python3.9/site-packages/borg/testsuite/__pycache__/__init__.cpython-39.pyc
a
HZ�h�8 � @ sp d dl mZ d dlZd dlZd dlZz d dlZW n eyF dZY n0 d dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZddlmZ ddlmZ ddlmZ ddlmZmZmZ ddlmZ dd lmZmZm Z er�e ej d
�nd Z zd d l m!Z! W n dZ!Y n0 e ed
��p,e j�"d�Z#z>e �$� � Z%e�&e%j'ej(� W d � n1 �s`0 Y W n e)�y� d Z#Y n0 e�r�de*edg �v �r�d Z+nde
�,� v �r�dZ+ndZ+e j�"d��r�dZ+dd� Z-edd� �Z.ej/dd� �Z0ej/d d � �Z1ej/d d � �Z2ej/d d!� �Z3ej/d"d#� �Z4d$d%� Z5G d&d'� d'ej6�Z7G d(d)� d)�Z8G d*d+� d+�Z9G d,d-� d-�Z:dS ).� )�contextmanagerN� )�get_all)� get_flags)�umount)� EXIT_SUCCESS� EXIT_WARNING�
EXIT_ERROR)�platform)�llfuse� has_pyfuse3�
has_llfuse� st_mtime_nsF)�raisesZlchflags�linuxZ
HAVE_FUTIMENS�_have_functionsZ HAVE_UTIMES���i�����netbsd���c C s" t t| | ��}dt }||kS )zKcompare 2 timestamps (both in nanoseconds) whether they are (roughly) equal�
)�int�abs�st_mtime_ns_round)Zts_ns1Zts_ns2Zdiff_tsZdiff_max� r �=/usr/lib64/python3.9/site-packages/borg/testsuite/__init__.py�
same_ts_ns8 s
r c c s< t �� � } tj�| d�V W d � n1 s.0 Y d S )N�file)�tempfileZTemporaryDirectory�os�path�join)Ztempdirr r r �unopened_tempfile? s
r! c C s| t � �b} z>t�d| � tj| dd�rDt�| �dkrDW W d � dS W n tyX Y n0 W d � n1 sn0 Y dS )NZ somewhereF��follow_symlinksT)r! r �symlink�stat�readlink�OSError��filepathr r r �are_symlinks_supportedE s $r* c
C s� t td�sdS t� ��} t� ��}t| d��� zjt�| |� t�| �}t�|�}|j|j krfdkr�n n*|j|jkr�W W d � W d � dS W n t y� Y n0 W d � n1 s�0 Y W d � n1 s�0 Y dS )N�linkF�wr T)
�hasattrr r! �open�closer+ r% �st_nlink�st_inor' )Z file1pathZ file2pathZstat1Zstat2r r r �are_hardlinks_supportedQ s
(" Br2 c C s| t � �b} z t�| � W W d � dS ty6 Y n" tyF Y n tyV Y n0 W d � dS 1 sn0 Y d S )NTF)r! r �mkfifor' �NotImplementedError�AttributeErrorr( r r r �are_fifos_supportedd s
r6 c C s� t � ��} t� r t�d| � nt| d��� zHtj| ddd� tj| dd�}|jdkrp|j dkrpW W d � dS W n" t
y� Y n t y� Y n0 W d � dS 1 s�0 Y d S ) N� somethingr, )�� �� Fr" r8 r9 T) r! r* r r$ r. r/ �utimer% �st_atime�st_mtimer' r4 )r) � new_statsr r r �is_utime_fully_supporteds s r> c C s� t tjd�sdS t� ��} t� r,t�d| � nt| d��� ztd\}}}tj| ||fdd� tj| ||fdd� tj | dd�}|j
|kr�|j |kr�|j |kr�W W d � dS W n" t
y� Y n ty� Y n0 W d � dS 1 s�0 Y d S )N� st_birthtimeFr7 r, )i �k8i�Cm8i �n8r" T)r- r � stat_resultr! r* r$ r. r/ r: r% r? r< r; r' r4 )r) Z birthtime�mtimeZatimer= r r r � is_birthtime_fully_supported� s$
rB c sD d� t | t�r$� fdd�| �� D �S t | t�r@� fdd�| D �S d S )Ns security.selinuxc s i | ]\}}|� kr||�qS r r )�.0�k�v�Z SELINUX_KEYr r �
<dictcomp>� � z no_selinux.<locals>.<dictcomp>c s g | ]}|� kr|�qS r r )rC rD rF r r �
<listcomp>� rH z no_selinux.<locals>.<listcomp>)�
isinstance�dict�items�list)�xr rF r �
no_selinux� s
rO c @ s� e Zd ZdZejjZejjZ ejj
Z ejj Z
er:ee�ZnejjZedd� �Zdd� Zdd� Zdd d
�Zedd dd
�dd��Zdd�dd�Zedd� �Zd S )� BaseTestCasez
c c s> t j�|�rJ |� d���d V t j�|�s:J |� d���d S )Nz should not existz
should exist)r r �exists)�selfr r r r �assert_creates_file� s z BaseTestCase.assert_creates_filec K s" t �||�}| j|fi |�� d S �N)�filecmp�dircmp�_assert_dirs_equal_cmp)rR Zdir1Zdir2�kwargs�diffr r r �assert_dirs_equal� s z BaseTestCase.assert_dirs_equalc s. t � fdd�|D ��s*J d� � d|� ���d S )Nc 3 s | ]}t �� |�V qd S rT )�re�search)rC �line��expected_regexprr r � <genexpr>� rH z2BaseTestCase.assert_line_exists.<locals>.<genexpr>z
no match for z in )�any)rR �linesr_ r r^ r �assert_line_exists� s z BaseTestCase.assert_line_existsFc
sf | � |jg � | � |jg � | � |jg � | � |jg � |jD �]�}tj�|j |�}tj�|j
|�}tj |dd�� tj |dd��� j �j k}g d�} |r�tj�
|�s�| �d� |g� fdd�| D � }
|g�fdd�| D � } |
�dt� j�� | �dt�j�� |�s(|
�t|� �� | �t|��� t �� j��sLt �� j��sLd |
d < t ��j��spt ��j��spd | d < t� �r�|�r�|
�t� jd
�� | �t�jd
�� nV|�r�t�s�|
�t� jd �� | �t�jd �� n$|
�t� jt�� | �t�jt�� |�s0|
�tt |dd��� | �tt |dd��� | � |
| � q>|j � � D ]} | j | |||d � �qHd S )
NFr" )�st_uid�st_gid�st_rdevr0 c s g | ]}t � |��qS r ��getattr�rC �a)�s1r r rI � rH z7BaseTestCase._assert_dirs_equal_cmp.<locals>.<listcomp>c s g | ]}t � |��qS r rg ri )�s2r r rI � rH � � g e��Ar )� ignore_flags�
ignore_xattrs� ignore_ns) � assert_equal� left_only�
right_only�
diff_files� funny_files�commonr r r �left�rightr% �st_dev�isdir�append�insert�oct�st_moder �S_ISCHR�S_ISBLKr> r r �have_fuse_mtime_ns�roundr rO r �subdirs�valuesrW )
rR rY ro rp rq �filenameZpath1Zpath2Zfuse�attrsZd1Zd2Zsub_diffr )rk rl r rW � sN
z#BaseTestCase._assert_dirs_equal_cmpNT)�fork�os_forkc o s |d u rt �� }n
t�|� d||gt|� }|r�t�� dkr�t�� t�� dkr\t�d� z2| j|ddi|�� W t dt
j dd� t�d� q�t dt
j dd� t�d� 0 n.| j|d|i|�� |� dt
�tkr�d V d S | j|dd � d V t|� | j|dd � t�|� t�d
� d S ) N�mountr r� Fz<Fatal: borg mount did not daemonize properly. Force exiting.T)r �flushZ exit_code)�mountedg�������?)r Zmkdtempr �mkdirrM r� �setsid�_exit�cmd�print�sys�stderr�getr r �wait_for_mountstater �rmdir�time�sleep)rR �location�
mountpointr� r� �optionsrX �argsr r r �
fuse_mount� s:
� ��
zBaseTestCase.fuse_mount� )�timeoutc C sX |t � � 7 }|t � � kr8tj�|�|kr,dS t �d� q d�|rDdnd|�}t|��dS )z4Wait until a path meets specified mount point statusNg�������?zWaiting for {} of {}r� r )r� r r �ismountr� �format� TimeoutError)rR r� r� r� �messager r r r� , s z BaseTestCase.wait_for_mountstatec
c s� t j�d�r d| }d| }nPt j�d�r<d| }d| }n2t j�d�rZd| }d | }nd
t j }| �|� z:t�d | � t�|� d V W t�|� t�d
| � nt�|� t�d
| � 0 d S )a� Some paths need to be made read-only for testing
If the tests are executed inside a fakeroot environment, the
changes from chmod won't affect the real permissions of that
folder. This issue is circumvented by temporarily disabling
fakeroot with `LD_PRELOAD=`.
Using chmod to remove write permissions is not enough if the
tests are running with root privileges. Instead, the folder is
rendered immutable with chattr or chflags, respectively.
r zchattr +i "%s"zchattr -i "%s")�darwinZfreebsdr Zopenbsdzchflags uchg "%s"zchflags nouchg "%s"Zsunoszchmod S+vimmutable "%s"zchmod S-vimmutable "%s"z7Testing read-only repos is not supported on platform %sz LD_PRELOAD= chmod -R ugo-w "%s"Nz LD_PRELOAD= chmod -R ugo+w "%s")r� r
�
startswithZskipTestr �system)rR r Z
cmd_immutableZ cmd_mutabler� r r r � read_only6 s&
�
zBaseTestCase.read_only)FFF)N)�__name__�
__module__� __qualname__�__doc__�unittest�TestCaseZassertInZ assert_inZ assertNotInZ
assert_not_inZ assertEqualrr ZassertNotEqualZassert_not_equalr � staticmethodZ
assert_raisesZ assertRaisesr rS rZ rc rW r� r� r� r r r r rP � s$
09
rP c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� changedirc C s
|| _ d S rT )�dir)rR r� r r r �__init__Z s zchangedir.__init__c C s t �� | _t �| j� d S rT )r �getcwd�old�chdirr� )rR r r r � __enter__] s
zchangedir.__enter__c O s t �| j� d S rT )r r� r� )rR r� �kwr r r �__exit__a s zchangedir.__exit__N�r� r� r� r� r� r� r r r r r� Y s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�environment_variablec K s || _ i | _d S rT )r� �
old_values)rR r� r r r r� f s z environment_variable.__init__c C sL | j �� D ]<\}}tj�|�| j|< |d u r<tj�|d � q
|tj|< q
d S rT )r� rL r �environr� r� �pop)rR rD rE r r r r� j s
z environment_variable.__enter__c O s: | j �� D ]*\}}|d u r*tj�|d � q
|tj|< q
d S rT )r� rL r r� r� )rR r� r� rD rE r r r r� r s z environment_variable.__exit__Nr� r r r r r� e s r� c @ s" e Zd ZdZdd� Zddd�ZdS )�
FakeInputszASimulate multiple user inputs, can be used as input() replacementc C s
|| _ d S rT )�inputs)rR r� r r r r� | s zFakeInputs.__init__Nc C s@ |d urt |dd� z| j�d�W S ty: td �Y n0 d S )N� )�endr )r� r� r� �
IndexError�EOFError)rR �promptr r r �__call__ s zFakeInputs.__call__)N)r� r� r� r� r� r� r r r r r� z s r� );�
contextlibr rU � functoolsr �posix� ImportErrorr[ r% r� Z sysconfigr r� Zuuidr� Zxattrr r
r Zhelpersr r r r r� Z fuse_implr r r
r- ZEntryAttributesr� Zpytestr r� Z has_lchflagsZNamedTemporaryFiler Z set_flags�name� UF_NODUMPr' rh r Zget_config_varsr r! � lru_cacher* r2 r6 r> rB rO r� rP r� r� r� r r r r �<module> st
4
2