Your IP : 216.73.216.85
3
��2a � @ sp d dl Z d dlZd dlZd dlmZ d dlZd dlmZ G dd� dej�ZG dd� dej�Z e
dkrlej � dS ) � N)�
script_helper)�mockc @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zejd �d
d � �Z ejd �d d
� �Z
dS )�TestScriptHelperc C s" t jdd�}| jd|d d� d S )Nz-czimport sys; sys.exit(0)r zreturn code was not 0)r �assert_python_ok� assertEqual)�self�t� r �7/opt/alt/python36/lib64/python3.6/test_script_helper.py�test_assert_python_ok
s z&TestScriptHelper.test_assert_python_okc C s$ t jdd�\}}}| jd|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 | j t��}tjdd� W d Q R X t|j�}| jd|� | 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 | j t��}tjdd� W d Q R X t|j�}| jd|� | 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�|_ytjddd� W n4 tk
rd } z| jd|jd � W Y d d }~X nX | jd |j � | jd |j � |j
d d }| jt j |d � | j
d|� | j
d
|� | jd |� 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�|_ytjddd� W n4 tk
rd } z| jd|jd � W Y d d }~X nX |j d d }| j
d|� | j
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� Zejd�dd� �Zejd�dd � �Zejd�d
d � �Z ejd�d d
� �Z
dS )�TestScriptHelperEnvironmentc C s | j 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 jjtj��L tjjdd � tjdd�|_| j t
j � � | j t
j � � | j 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 jjtj��8 tjjdd � tj� | jtj� � | j 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 jjtj��r tjjdd � tj� | jtj� � | jtj� � | j d|j
� |j d d }| j t j
|d � | jd|� 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 jjtj��: dtjd< | jtj� � | jtj� � | jd|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