Your IP : 216.73.216.213
U
i�f � @ st d Z ddlZddlZddlZddlmZ ddlZddlmZ G dd� dej�Z G dd� dej�Z
e d krpe� � dS )
zEUnittests for test.support.script_helper. Who tests the test helper?� 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 )N�-c�import sys; sys.exit(0)r zreturn code was not 0)r �assert_python_ok� assertEqual)�self�t� r �</opt/alt/python38/lib64/python3.8/test/test_script_helper.py�test_assert_python_ok
s z&TestScriptHelper.test_assert_python_okc C s$ t �dd�\}}}| �d|d� d S )Nr � 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 5 Q R X t|j�}| �d|� | jd|dd� d S )Nr r z
command line:zunexpected command line��msg)� assertRaises�AssertionErrorr r �str� exception�assertIn�r Z
error_contextZ error_msgr r r � test_assert_python_ok_raises s
z-TestScriptHelper.test_assert_python_ok_raisesc C sL | � t��}t�dd� W 5 Q R X t|j�}| �d|� | jd|dd� d S )Nr r zProcess return code is 0
zunexpected command line.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�|_zt�ddd� W n4 tk
rd } z| �d|jd � W 5 d }~X Y nX | �d |j � | �d |j � |j
d d }| �t j |d � | �
d|� | �
d
|� | �d |� W 5 Q R X d S ) N� interpreter_requires_environmentF�Z return_value�bail out of unittestTr �Noner � �-I�-E)r �patch�objectr � RuntimeError� side_effect�_assert_pythonr �args�
call_count� call_args�sys�
executabler � assertNotIn�r Z
mock_popenZ
mock_ire_funcr Z
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�|_zt�ddd� W n4 tk
rd } z| �d|jd � W 5 d}~X Y nX |j d d }| �
d |� | �
d
|� W 5 Q R X dS ) z>Ensure that -I is not passed when the environment is required.r Tr r r r r Nr" 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$ r0 r1 r r r r r s
r c @ sh e Zd ZdZdd� Zdd� Ze�d�dd� �Ze�d�d d
� �Z e�d�d d � �Z
e�d�d
d� �Z dS )�TestScriptHelperEnvironmentz5Code coverage for interpreter_requires_environment().c C s | � ttd�� d tjd< d S �NZ$__cached_interp_requires_environment)�
assertTrue�hasattrr �__dict__�r r r r �setUpK s �z!TestScriptHelperEnvironment.setUpc C s d t jd< d S r6 )r r9 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 5 Q R X d S )N�
PYTHONHOME� r! )r r$ �dict�os�environ�pop�
subprocessZCalledProcessErrorr' r7 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 5 Q R X d S )Nr= r! ) r r$ r? r@ rA rB r r � assertFalser r* rD 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 5 Q R X d S )Nr= r! r r# )r r$ r? r@ rA rB r r rG r r* r+ r, r- r )r rE 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 5 Q R X d S )NZ
MockedHomer= r )
r r$ r? r@ rA r7 r r r r* rD r r r �5test_interpreter_requires_environment_with_pythonhomes s
zQTestScriptHelperEnvironment.test_interpreter_requires_environment_with_pythonhomeN) r2 r3 r4 �__doc__r; r< r r$ rF rH rI rJ r r r r r5 H s
r5 �__main__)
rK rC r, r@ Z test.supportr Zunittestr ZTestCaser r5 r2 �mainr r r r �<module> s =4