File: //usr/local/lib/python3.10/test/__pycache__/test_code_module.cpython-310.opt-1.pyc
o
�i# � @ sx d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlm Z e �
d�Z G dd � d ej �Z
ed
kr:e�� dS dS ) zCTest InteractiveConsole and InteractiveInterpreter from code module� N)�dedent)� ExitStack)�mock)�
import_helper�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 �� | _| �� d S )N)r ZInteractiveConsole�console�mock_sys��self� r �2/usr/local/lib/python3.10/test/test_code_module.py�setUp s
z TestInteractiveConsole.setUpc C s� t � }| �|j� |�tjddd��| _|�t�d��| _|�t�d��| _tjdt j
t j
d�}|�|�| _ t
j t
j
u rB| 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_| j�� | �| jjd� d| j_| j�� | �| jjd� d S )N�Finishedz>>> z custom1> )�EOFErrorr � side_effectr �interact� assertEqualr r r
r r r
�test_ps1# �
z TestInteractiveConsole.test_ps1c C sL t d�| j_| j�� | �| jjd� d| j_| j�� | �| jjd� d S )Nr z... z custom2> ) r r r r r r! r r r r
r r r
�test_ps2+ r# z TestInteractiveConsole.test_ps2c C sN ddt d�g| j_| j�� t| jj�D ]}dd�|d �v r" d S qt d��)Nz 'antioch'� r Zantioch� zno console stdout)
r r r r r �listr � method_calls�join�AssertionError�r �callr r r
�test_console_stderr3 s
�z*TestInteractiveConsole.test_console_stderrc C sH dt d�g| j_| j�� | jjD ]}dd�|d �v r d S qtd��)NZ undefinedr � NameErrorr% r& z No syntax error from console) 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_t�� }|| j_| j�� | � |j
� d S )Nzraise ValueError('')r ) r r r r ZMockr r r r Z
assertTrueZcalled)r �hookr r r
�test_sysexcepthookE s �
z)TestInteractiveConsole.test_sysexcepthookc C s� t d�| j_| jjdd� | �t| jj�d� | jjd }| �|ddi g� | j� � t d�| j_| jjdd� | �t| jj�d � d S )
Nr ZFoo��banner� r �write)zFoo
r% � �
r r r r r r! �lenr r( Z
reset_mock)r Z banner_callr r r
� test_bannerM s
z"TestInteractiveConsole.test_bannerc C s� t d�| j_| jjdd� | �t| jj�d� | jjd }d}| �|d|fi g� | j� � t d�| j_| jjddd� | �t| jj�d� | j� � d }t d�| j_| jjd|d� | �t| jj�d� | jjd }|d
}| �|d|fi g� d S ) Nr r% r2 r6 r& z"now exiting InteractiveConsole...
r5 )r3 Zexitmsgu bye! ζж�
r7 )r �err_msg�expected�messager r r
�
test_exit_msg[ s&
� z$TestInteractiveConsole.test_exit_msgc C �J dt d�g| j_| j�� d�dd� | jjD ��}td�}| � ||� d S )Nz(raise ValueError('') from AttributeErrorr r% c s � � | ]
}d � |d �V qdS �r% r& N�r) ��.0r, r r r
� <genexpr>{ � � 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 ZassertIn�r �outputr< r r r
�
test_cause_tbw s �
z$TestInteractiveConsole.test_cause_tbc C r? )Nztry: ham
except: eggs
r r% c s r@ rA rB rC r r r
rE � rF 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
rG rH r r r
�test_context_tb� s �
z&TestInteractiveConsole.test_context_tbN)�__name__�
__module__� __qualname__r r r" r$ r- r/ r1 r9 r> rJ rK r r r r
r
s r �__main__)�__doc__r Zunittest�textwrapr �
contextlibr r Z test.supportr �
import_moduler ZTestCaser rL �mainr r r r
�<module> s
�