File: //usr/lib/python3/dist-packages/pbr/tests/__pycache__/base.cpython-38.pyc
U
jP�^�" � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
G dd� dej �Z G dd� dej
ej�ZG d d
� d
ej �Zd d � Zd
d� ZdS )z Common utilities used in testing� N)�content)�optionsc s( e Zd ZdZdd� Z� fdd�Z� ZS )�DiveDirzaDive into given directory and return back on cleanup.
:ivar path: The target directory.
c C s
|| _ d S �N)�path)�selfr � r �0/usr/lib/python3/dist-packages/pbr/tests/base.py�__init__= s zDiveDir.__init__c s0 t t| ��� | �tjt�� � t�| j� d S r )�superr �setUp�
addCleanup�os�chdir�getcwdr )r �� __class__r r r @ s z
DiveDir.setUp��__name__�
__module__� __qualname__�__doc__r
r �
__classcell__r r r r r 7 s r c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zg d d
fd d �Z� ZS )
� BaseTestCasec s
t t| ��� tj�dd�}z t|�}W n tk
rH td� d}Y nX |dkrf| � t
j |dd�� tj�d�t j
kr�| � t
�d��j}| � t
�d |�� tj�d
�t j
kr�| � t
�d ��j}| � t
�d |�� | � t
�d
��| _| � t
�� � | � t
�� � | � t
�� � | � t
�dd�� | � t
�� �j| _tj�| jd�| _t� tj�tj� t �d�| j� | � tj t�!� � t� | j� | � | j"� t#| dd��s| � t
�d�� tj�| jd�}t$|d��}|�%� }W 5 Q R X |�&dd�}t$|d��}|�'|� W 5 Q R X d S )NZOS_TEST_TIMEOUT� z=OS_TEST_TIMEOUT set to invalid value defaulting to no timeoutr T)ZgentleZOS_STDOUT_CAPTURE�stdoutz
sys.stdoutZOS_STDERR_CAPTURE�stderrz
sys.stderr�pbrZ PBR_VERSIONz0.0Z testpackageZ preversionedz setup.cfgZrtzversion = 0.1.dev� Zwt)(r r r r �environ�get�int�
ValueError�printZ
useFixture�fixturesZTimeoutr Z TRUE_VALUESZ StringStream�streamZ MonkeyPatchZ
FakeLoggerZ log_fixtureZ TempHomeDirZNestedTempfileZEnvironmentVariableZTempDirr Ztemp_dir�join� package_dir�shutilZcopytree�dirname�__file__r
r r �_discard_testpackage�getattr�open�read�replace�write)r Z test_timeoutr r Zsetup_cfg_pathZcfgr r r r r H sL
�� zBaseTestCase.setUpc C s. t tj�D ] }|dks |�d�r
tj|= q
d S )NZpbr_testpackagezpbr_testpackage.)�list�sys�modules�
startswith)r �kr r r r+ { s
�z!BaseTestCase._discard_testpackagec O s | j d|f|�S )Nr ��_run_cmd�r �args�kwargsr r r �run_pbr� s zBaseTestCase.run_pbrc O s | j tjd| f|�S )N)zsetup.py)r7 r2 �
executabler8 r r r � run_setup� s zBaseTestCase.run_setupTNc C s@ |p| j }t|gt|� |d�}|d r<|s<td|d ��|S )af Run a command in the root of the test working copy.
Runs a command, with the given argument list, in the root of the test
working copy--returns the stdout and stderr streams and the exit code
from the subprocess.
:param cwd: If falsy run within the test package dir, otherwise run
within the named path.
)�cwd� zCommand failed retcode=%s)r' r7 r1 � Exception)r �cmdr9 Z
allow_failr> �resultr r r r7 � s
zBaseTestCase._run_cmd) r r r r r+ r; r= r7 r r r r r r F s
3r c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�CapturedSubprocessa) Run a process and capture its output.
:attr stdout: The output (a string).
:attr stderr: The standard error (a string).
:attr returncode: The return code of the process.
Note that stdout and stderr are decoded from the bytestrings subprocess
returns using error=replace.
c sH t t| ��� || _|| _|| _tj| jd< tj| jd< tj| jd< dS )z�Create a CapturedSubprocess.
:param label: A label for the subprocess in the test log. E.g. 'foo'.
:param *args: The *args to pass to Popen.
:param **kwargs: The **kwargs to pass to Popen.
r �stdinr N)r rC r
�labelr9 r: �
subprocess�PIPE)r rE r9 r: r r r r
� s zCapturedSubprocess.__init__c s� t t| ��� tj| j| j�}|�� \}}|�dd�| _ |�dd�| _
| � | j d t
�| j �� | � | j d t
�| j
�� |j| _|jr�td| j| j|jf ��| �t| d� | �t| d� | �t| d� d S ) Nzutf-8r/ z-stdoutz-stderrz0Failed process args=%r, kwargs=%r, returncode=%s�out�err�
returncode)r rC r rF �Popenr9 r: � communicate�decoderH rI Z addDetailrE r Z text_contentrJ �AssertionErrorr
�delattr)r �procrH rI r r r r � s& ��zCapturedSubprocess.setUpr r r r r rC � s
rC c C sN t j| t jt jt j|d�}tdd� |�� D ��}|D ] }t|� q4||jf S )z�Run the command args in cwd.
:param args: The command to run e.g. ['git', 'status']
:param cwd: The directory to run the comamnd in.
:return: ((stdout, stderr), returncode)
)rD r r r> c s s | ]}|� d ��� V qdS )�latin1N)rM �strip)�.0�sr r r � <genexpr>� s z_run_cmd.<locals>.<genexpr>)rF rK rG �tuplerL r# rJ )r9 r> �pZstreamsZstream_contentr r r r7 � s �
r7 c C s@ t dddddgd � t dddddgd � t dddddgd � d S ) NZgitZconfigz--globalz
user.emailzexample@example.comz user.namezOpenStack Developerzuser.signingkeyr6 r r r r � _config_git� s � ���rX )r r r( rF r2 r$ Z
testresourcesZ testtoolsr r r ZFixturer ZTestCaseZResourcedTestCaser rC r7 rX r r r r �<module>( s T,