Your IP : 216.73.216.196
B
YI~d � @ sp d dl Z d dlZd dlZd dlmZ d dlZd dlmZ G dd� dej�ZG dd� dej�Z e
dkrle� � dS ) � N)�
script_helper)�mockc @ sP e Zd Zdd� Zdd� Zdd� Zdd� Ze�d �d
d � �Z e�d �d d
� �Z
dS )�TestScriptHelperc C s" t �dd�}| �d|d d� d S )Nz-czimport sys; sys.exit(0)r zreturn code was not 0)r �assert_python_ok� assertEqual)�self�t� r �</opt/alt/python37/lib64/python3.7/test/test_script_helper.py�test_assert_python_ok
s z&TestScriptHelper.test_assert_python_okc C s$ t �dd�\}}}| �d|d� d S )Nz-cz sys.exit(0)r zreturn code should not be 0)r �assert_python_failureZassertNotEqual)r Zrc�out�errr r r
�test_assert_python_failure s z+TestScriptHelper.test_assert_python_failurec C sL | � t��}t�dd� W d Q R X t|j�}| �d|� | jd|dd� d S )Nz-cz sys.exit(0)z
command line:zunexpected command line)�msg)� assertRaises�AssertionErrorr r �str� exception�assertIn)r �
error_context� error_msgr r r
� test_assert_python_ok_raises s
z-TestScriptHelper.test_assert_python_ok_raisesc C sL | � t��}t�dd� W d Q R X t|j�}| �d|� | jd|dd� d S )Nz-czimport sys; sys.exit(0)zProcess return code is 0
zunexpected command line.)r )r r r r r r r )r r r r r r
�!test_assert_python_failure_raises s
z2TestScriptHelper.test_assert_python_failure_raiseszsubprocess.Popenc C s� t jjtddd���}td�|_yt�ddd� W n4 tk
rd } z| �d|jd � W d d }~X Y nX | �d |j � | �d |j � |j
d d }| �t j |d � | �
d|� | �
d
|� | �d |� W d Q R X d S ) N� interpreter_requires_environmentF)� return_valuezbail out of unittestTz-c�Noner � z-Iz-E)r �patch�objectr � RuntimeError� side_effect�_assert_pythonr �args�
call_count� call_args�sys�
executabler � assertNotIn)r �
mock_popen�
mock_ire_funcr �
popen_commandr r r
�1test_assert_python_isolated_when_env_not_required& s
$ zBTestScriptHelper.test_assert_python_isolated_when_env_not_requiredc C s� t jjtddd��~}td�|_yt�ddd� W n4 tk
rd } z| �d|jd � W d d }~X Y nX |j d d }| �
d|� | �
d |� W d Q R X d S )
Nr T)r zbail out of unittestz-cr r z-Iz-E) r r r r r r! r" r r# r% r( )r r) r* r r+ r r r
�4test_assert_python_not_isolated_when_env_is_required8 s
$ zETestScriptHelper.test_assert_python_not_isolated_when_env_is_requiredN) �__name__�
__module__� __qualname__r r r r r r r, r- r r r r
r s r c @ sd e Zd Zdd� Zdd� Ze�d�dd� �Ze�d�dd � �Ze�d�d
d � �Z e�d�d d
� �Z
dS )�TestScriptHelperEnvironmentc C s | � ttd�� d tjd< d S )N�$__cached_interp_requires_environment)�
assertTrue�hasattrr �__dict__)r r r r
�setUpK s z!TestScriptHelperEnvironment.setUpc C s d t jd< d S )Nr2 )r r5 )r r r r
�tearDownQ s z$TestScriptHelperEnvironment.tearDownzsubprocess.check_callc C sd t j�tj��L tj�dd � t�dd�|_| � t
� � � | � t
� � � | � d|j
� W d Q R X d S )N�
PYTHONHOME� r )r r �dict�os�environ�pop�
subprocessZCalledProcessErrorr! r3 r r r r$ )r �mock_check_callr r r
�*test_interpreter_requires_environment_trueU s zFTestScriptHelperEnvironment.test_interpreter_requires_environment_truec C sP t j�tj��8 tj�dd � t�� | �t�� � | � d|j
� W d Q R X d S )Nr8 r ) r r r: r; r< r= r r � assertFalser r$ )r r? r r r
�+test_interpreter_requires_environment_false^ s
zGTestScriptHelperEnvironment.test_interpreter_requires_environment_falsec C s� t j�tj��r tj�dd � t�� | �t�� � | �t�� � | � d|j
� |j d d }| � t j
|d � | �d|� W d Q R X d S )Nr8 r r z-E)r r r: r; r< r= r r rA r r$ r% r& r' r )r r? Zcheck_call_commandr r r
�-test_interpreter_requires_environment_detailsg s zITestScriptHelperEnvironment.test_interpreter_requires_environment_detailsc C sR t j�tj��: dtjd< | �t�� � | �t�� � | �d|j � W d Q R X d S )NZ
MockedHomer8 r )
r r r: r; r< r3 r r r r$ )r r? r r r
�5test_interpreter_requires_environment_with_pythonhomes s
zQTestScriptHelperEnvironment.test_interpreter_requires_environment_with_pythonhomeN) r. r/ r0 r6 r7 r r r@ rB rC rD r r r r
r1 H s r1 �__main__) r> r& r; Z test.supportr Zunittestr ZTestCaser r1 r. �mainr r r r
�<module> s =4