Your IP : 216.73.216.85
3
� f � @ st d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z e j
d�Z G dd � d ej �Z
ed
krpej� dS ) zCTest InteractiveConsole and InteractiveInterpreter from code module� N)�dedent)� ExitStack)�mock)�support�codec @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TestInteractiveConsolec C s t j� | _| j� d S )N)r ZInteractiveConsole�console�mock_sys)�self� r �:/opt/alt/python36/lib64/python3.6/test/test_code_module.py�setUp s
z TestInteractiveConsole.setUpc C s� t � }| j|j� |jtjddd��| _|jtjd��| _|jtjd��| _tjdt j
t j
d�}|j|�| _ t
j t
j
kr�| j j
| j _ | j `| j `dS ) z.Mock system environment for InteractiveConsolez
code.inputT)Zcreatezcode.sys.stdoutzcode.sys.stderrzcode.sys)�wraps�specN)r Z
addCleanup�close�
enter_contextr Zpatch�infunc�stdout�stderrr �sys�sysmod�
excepthook�__excepthook__�ps1�ps2)r
�stackZprepatchr r r r s
z TestInteractiveConsole.mock_sysc C sL t d�| j_| jj� | j| jjd� d| j_| jj� | j| jjd� d S )N�Finishedz>>> z custom1> )�EOFErrorr � side_effectr �interact� assertEqualr r )r
r r r �test_ps1" s
z TestInteractiveConsole.test_ps1c C sL t d�| j_| jj� | j| jjd� d| j_| jj� | j| jjd� d S )Nr z... z custom2> ) r r r r r r r r r )r
r r r �test_ps2* s
z TestInteractiveConsole.test_ps2c C sR ddt d�g| j_| jj� x0t| jj�D ]}ddj|d �kr*P q*W t d��d S )Nz 'antioch'� r Zantioch� zno console stdout)
r r r r r �listr � method_calls�join�AssertionError)r
�callr r r �test_console_stderr2 s
z*TestInteractiveConsole.test_console_stderrc C sL dt d�g| j_| jj� x,| jjD ]}ddj|d �kr$P q$W td��d S )NZ undefinedr � NameErrorr# r$ z No syntax error from console) r r r r r r r& r' r( )r
r) r r r �test_syntax_error; s
z(TestInteractiveConsole.test_syntax_errorc C s: dt d�g| j_tj� }|| j_| jj� | j |j
� d S )Nzraise ValueError('')r ) r r r r ZMockr r r r Z
assertTrueZcalled)r
�hookr r r �test_sysexcepthookD s
z)TestInteractiveConsole.test_sysexcepthookc C s� t d�| j_| jjdd� | jt| jj�d� | jjd }| j|dd
i g� | jj � t d�| j_| jjdd� | jt| jj�d � d S ) Nr ZFoo)�banner� r �write�Foo
r# � )r2 )
r r r r r r �lenr r& �
reset_mock)r
Z banner_callr r r � test_bannerL s
z"TestInteractiveConsole.test_bannerc C s� t d�| j_| jjdd� | jt| jj�d� | jjd }d}| j|d|fi g� | jj � t d�| j_| jjddd� | jt| jj�d� | jj � d }t d�| j_| jjd|d� | jt| jj�d� | jjd }|d
}| j|d|fi g� d S ) Nr r# )r/ r3 r$ z"now exiting InteractiveConsole...
r1 )r/ Zexitmsgu bye! ζж�
)
r r r r r r r4 r r& r5 )r
Zerr_msg�expected�messager r r �
test_exit_msgZ s$
z$TestInteractiveConsole.test_exit_msgc C sJ dt d�g| j_| jj� djdd� | jjD ��}td�}| j ||� d S )Nz(raise ValueError('') from AttributeErrorr r# c s s | ]}d j |d �V qdS )r# r$ N)r' )�.0r) r r r � <genexpr>z s z7TestInteractiveConsole.test_cause_tb.<locals>.<genexpr>z�
AttributeError
The above exception was the direct cause of the following exception:
Traceback (most recent call last):
File "<console>", line 1, in <module>
ValueError
)
r r r r r r' r r& r �assertIn)r
�outputr8 r r r �
test_cause_tbv s
z$TestInteractiveConsole.test_cause_tbc C sJ dt d�g| j_| jj� djdd� | jjD ��}td�}| j ||� d S )Nztry: ham
except: eggs
r r# c s s | ]}d j |d �V qdS )r# r$ N)r' )r; r) r r r r< � s z9TestInteractiveConsole.test_context_tb.<locals>.<genexpr>ah
Traceback (most recent call last):
File "<console>", line 1, in <module>
NameError: name 'ham' is not defined
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "<console>", line 2, in <module>
NameError: name 'eggs' is not defined
)
r r r r r r' r r& r r= )r
r> r8 r r r �test_context_tb� s
z&TestInteractiveConsole.test_context_tbN)�__name__�
__module__� __qualname__r
r r! r" r* r, r. r6 r: r? r@ r r r r r s r �__main__)�__doc__r Zunittest�textwrapr �
contextlibr r Ztestr �
import_moduler ZTestCaser rA �mainr r r r �<module> s