← Back
Editing: 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,
Save File
Cancel