Your IP : 216.73.216.213
��Yf� � @ s� d Z d d l Z d d l Z d d l m Z d d l m Z d d l m Z d d l m Z e j
d � Z Gd d � d e j � Z
e d
k r� e j � d S) zCTest InteractiveConsole and InteractiveInterpreter from code module� N)�dedent)� ExitStack)�mock)�support�codec @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d S)�TestInteractiveConsolec C s t j � | _ | j � d S)N)r ZInteractiveConsole�console�mock_sys)�self� r �:/opt/alt/python35/lib64/python3.5/test/test_code_module.py�setUp s z TestInteractiveConsole.setUpc C s� t � } | j | j � | j t j d d d �� | _ | j t j d � � | _ | j t j d � � | _ t j d d t j
d t j
�} | j | � | _ t
j t
j
k r� | j j
| j _ d S)
z.Mock system environment for InteractiveConsolez
code.inputZcreateTzcode.sys.stdoutzcode.sys.stderrzcode.sys�wraps�specN)r Z
addCleanup�close�
enter_contextr Zpatch�infunc�stdout�stderrr �sys�sysmod�
excepthook�__excepthook__)r
�stackZprepatchr r r r s !z TestInteractiveConsole.mock_sysc C s9 t d � | j _ | j j � | j | j j d � d S)N�Finishedz>>> )�EOFErrorr � side_effectr �interact� assertEqualr Zps1)r
r r r �test_ps1 s
z TestInteractiveConsole.test_ps1c C s9 t d � | j _ | j j � | j | j j d � d S)Nr z... )r r r r r r r Zps2)r
r r r �test_ps2% s
z TestInteractiveConsole.test_ps2c C so d d t d � g | j _ | j j � x@ t | j j � D] } d d j | d � k r; Pq; Wt 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_stderr* s
z*TestInteractiveConsole.test_console_stderrc C sf d t d � g | j _ | j j � x: | j j D] } d d j | d � k r2 Pq2 Wt d � � 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_error3 s
z(TestInteractiveConsole.test_syntax_errorc C sQ d t d � g | j _ t j � } | | j _ | j j � | j | j
� d S)Nzraise ValueError('')r ) r r r r ZMockr r r r Z
assertTrueZcalled)r
�hookr r r �test_sysexcepthook< s
z)TestInteractiveConsole.test_sysexcepthookc C s� t d � | j _ | j j d d � | j t | j j � d � | j j d } | j | d d
i g � | j j � t d � | j _ | j j d d � | j t | j j � d � d S) Nr ZbannerZFoo� r �write�Foo
r! r" )r/ )
r r r r r r �lenr r$ Z
reset_mock)r
Z banner_callr r r � test_bannerD s
z"TestInteractiveConsole.test_bannerc C sg d t d � g | j _ | j j � d j d d � | j j D� � } t d � } | j | | � d S)Nz(raise ValueError('') from AttributeErrorr r! c s s" | ] } d j | d � Vq d S)r! r" N)r% )�.0r' r r r � <genexpr>V 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
�output�expectedr r r �
test_cause_tbR s
" z$TestInteractiveConsole.test_cause_tbc C sg d t d � g | j _ | j j � d j d d � | j j D� � } t d � } | j | | � d S)Nztry: ham
except: eggs
r r! c s s" | ] } d j | d � Vq d S)r! r" N)r% )r2 r' r r r r3 f 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 r4 )r
r5 r6 r r r �test_context_tbb s
"
z&TestInteractiveConsole.test_context_tbN)
�__name__�
__module__� __qualname__r
r r r r( r* r, r1 r7 r8 r r r r r s r �__main__)�__doc__r Zunittest�textwrapr �
contextlibr r Ztestr �
import_moduler ZTestCaser r9 �mainr r r r �<module> s i