Your IP : 216.73.216.213
�
g fGG � @ sq d Z d d l m Z d d l Z d d l Z d d l Z d d l m Z m Z m Z d d l m
Z
m Z m Z d d l
Z
Gd d � d e j � Z Gd d � d e j � Z d
Z d Z e j d e j e � e j e � f � Z Gd
d � d � Z Gd d � d e e j � Z Gd d � d e e j � Z Gd d � d e j � Z d d � Z e d k rme � n d S)z Test cases for traceback module� )�StringION)� run_unittest�Error�captured_output)�TESTFN�unlink� cpython_onlyc @ 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 d � Z
d d � Z d d � Z d S) �SyntaxTracebackCasesc C sS y | � Wn5 | k
rB } z t j | | � SWYd d } ~ Xn
Xt d � � d S)Nz call did not raise exception)� traceback�format_exception_only�
ValueError)�self�func�exc�value� r �8/opt/alt/python34/lib64/python3.4/test/test_traceback.py�get_exception_format s
#z)SyntaxTracebackCases.get_exception_formatc C s t d d d � d S)Nzdef fact(x):
return x!
�?�exec)�compile)r
r r r �syntax_error_with_caret s z,SyntaxTracebackCases.syntax_error_with_caretc C s t d d d � d S)Nz1 +
r r )r )r
r r r �syntax_error_with_caret_2 s z.SyntaxTracebackCases.syntax_error_with_caret_2c C s t d d d � d S)Nz def spam():
print(1)
print(2)r r )r )r
r r r � syntax_error_bad_indentation s z1SyntaxTracebackCases.syntax_error_bad_indentationc C s t d d d � d S)Nu Python = "Ṕýţĥòñ" +r r )r )r
r r r �!syntax_error_with_caret_non_ascii" s z6SyntaxTracebackCases.syntax_error_with_caret_non_asciic C s t d d d � d S)Nz print(2)r r )r )r
r r r � syntax_error_bad_indentation2% s z2SyntaxTracebackCases.syntax_error_bad_indentation2c C sj | j | j t � } | j t | � d � | j | d j � d k � | j d | d � | j | d j d � | d j d � � | j | j t � } | j d | d � | j | d j
d � d � | j | d j d � | d j d � � | j | j t � } | j d | d � | j | d j
d � d � | j | d j d � | d j d � � d S) N� � z return x!�^� �!�
�+) r r � SyntaxError� assertEqual�len�
assertTrue�strip�assertIn�findr �countr )r
�errr r r �
test_caret( s * * z SyntaxTracebackCases.test_caretc C sU t d d � } t j t | � } | j t | � d � | j | d j � d � d S)N�error�x.py� �
bad syntax� r )r. r/ Nr0 )r# r
r r$ r% r' )r
r r+ r r r � test_nocaret< s z!SyntaxTracebackCases.test_nocaretc C s
| j | j t � } | j t | � d � | j | d j � d � | j d | d � | j | d j d � | d j d � � | j | j t � } | j t | � d � | j | d j � d � | j d | d � | j | d j d � | d j d � � d S)Nr r zprint(2)r r �)�p) r r �IndentationErrorr$ r% r' r( r) r )r
r+ r r r �test_bad_indentationB s * z)SyntaxTracebackCases.test_bad_indentationc C s5 t � } t j | j | � } | j | d g � d S)NzKeyboardInterrupt
)�KeyboardInterruptr
r � __class__r$ )r
�eZlstr r r �test_base_exceptionQ s z(SyntaxTracebackCases.test_base_exceptionc C s� Gd d � d t � } t j | | � � } | j t | � d � d | j } | j d
k ri | j } n d j | j | j g � } | j | d d | | f � d S) Nc @ s e Z d Z d d � Z d S)zESyntaxTracebackCases.test_format_exception_only_bad__str__.<locals>.Xc S s d d d S)Nr r r )r
r r r �__str__Y s zMSyntaxTracebackCases.test_format_exception_only_bad__str__.<locals>.X.__str__N)�__name__�
__module__� __qualname__r; r r r r �XX s r? r z<unprintable %s object>�__main__�builtins�.r z%s: %s
)z__main__zbuiltins)� Exceptionr
r r$ r% r< r= �join)r
r? r+ Z str_valueZstr_namer r r �%test_format_exception_only_bad__str__W s
z:SyntaxTracebackCases.test_format_exception_only_bad__str__c C s) t j d d � } | j | d g � d S)NzNone
)r
r r$ )r
r+ r r r �test_without_exceptiond s z+SyntaxTracebackCases.test_without_exceptionc sb d d l } d d l � d d l � d d l } � j � j d d g d � j d � j �} | j � \ } } t | d � j
� d � � � � � f d d � } | d d
d d � x� d D]� } | d k r� d
} n | d k r� d } n d } | d j | � | | d � | d j | � | | d � | d j | � | | d � q� W| d d d
d � d S)Nr z-cz&import sys; print(sys.stdout.encoding)�stdout�stderr�asciic s` z� t t d d | �} | j d j | | � � | j � � j � j t g d � j d � j �} | j � \ } } | j
� � j � } Wd t t � X| j
� d � } | j
� � } d j | � }
d j | � } � j d | | d
d j | d
| � � � j | d j |
� d
j | d |
� � � j | d | k d j | d | � � d S)N�w�encodingzq{0}if 1:
import traceback;
raise RuntimeError('{1}')
rG rH �backslashreplacezraise RuntimeError('{0}')zRuntimeError: {0}zline %sr z)Invalid line number: {0!r} instead of {1}r z.Invalid traceback line: {0!r} instead of {1!r}r1 z-Invalid error message: {0!r} instead of {1!r})�openr �write�format�close�Popen�
executable�PIPE�STDOUT� communicate�decode�
splitlinesr �encoder( r& �endswith) Z
firstlines�message�charset�lineno�output�processrG rH Zencoded_messageZ
message_asciiZerr_lineZerr_msg)�output_encodingr
�
subprocess�sysr r �do_testx s0
z7SyntaxTracebackCases.test_encoded_file.<locals>.do_test� Zfoor1 �
iso-8859-1�utf-8�GBKu 丂儀u hé hoz# coding: {0}
r z#!shebang
# coding: {0}
� z
# coding: {0}
z0
# coding: GBK
)zasciird re rf ) �tempfilera r` �osrQ rR rS rT rU �strrW rO ) r
rh ri r^ rG rH rb r[ �textr )r_ r
r` ra r �test_encoded_fileh s, 0 #
z&SyntaxTracebackCases.test_encoded_fileN)r< r= r> r r r r r r r, r2 r6 r: rE rF rl r r r r r
s
r c @ sm e Z d Z d d � Z e d d d � � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�TracebackFormatTestsc C s t d � � d S)NZblah)�KeyError)r
r r r �some_exception� s z#TracebackFormatTests.some_exceptionNc C s d d l m } y | j � Wnt k
r#t j � \ } } } | d k r_ | | j � n d d j t j | � � } t
� } | | | � | j � } t d � � } t j
| � Wd QXt
� }
t j
| d |
�t d � � } t j � Wd QXt j � } t
� }
t j d |
� Yn
Xt d � � | j | | � | j | j � |
j � � | j | j � |
j � � | j | |
j � � | j � } | j t | � d � | d } | d
d � \ } } | j | j d
� � | j | j d � � | j | j d � � d S)Nr )�traceback_printz#Traceback (most recent call last):
rc rH �filez&unable to create test traceback stringrg r � Tracebackz Filez raise���)� _testcapirp ro rn ra �exc_info�tb_nextrD r
� format_tbr �getvaluer �print_tb� print_exc�
format_excr r$ rW r% r& �
startswith)r
Z cleanup_funcrp �type_r �tbZ
traceback_fmtZfile_Z
python_fmtZtbstderrZtbfileZ excstderrZexcfmtZexcfileZtb_linesZbanner�locationZ source_liner r r �check_traceback_format� sB
z+TracebackFormatTests.check_traceback_formatc C s | j � d S)N)r� )r
r r r �test_traceback_format� s z*TracebackFormatTests.test_traceback_formatc C s d d � } | j | � d S)Nc S s | j j � d S)N)�tb_frame�clear)r~ r r r �
cleanup_tb� s zRTracebackFormatTests.test_traceback_format_with_cleared_frames.<locals>.cleanup_tb)r� )r
r� r r r �)test_traceback_format_with_cleared_frames� s z>TracebackFormatTests.test_traceback_format_with_cleared_framesc
C s� t d � � } t j t j d � � Wd QXt � } t j t j d � d | �| j | j � | j � � t j t j d � � } | j | j � d j | � � d S)NrH r rq rc )
r r
� print_stackra � _getframer r$ rx � format_stackrD )r
ZststderrZstfileZstfmtr r r �test_stack_format� s z&TracebackFormatTests.test_stack_formatc C s� d d � } t d � � } | � Wd QX| j j } | j | j � j � d d � d t | d f d d t | d f d
g � d S) Nc S s t j � d S)N)r
r� r r r r �prn� s z2TracebackFormatTests.test_print_stack.<locals>.prnrH r z) File "%s", line %d, in test_print_stackr1 z prn()z File "%s", line %d, in prnr z traceback.print_stack()���)r �__code__�co_firstlinenor$ rx rW �__file__)r
r� rH r\ r r r �test_print_stack� s
z%TracebackFormatTests.test_print_stackc C sa d d � } | � } | j j } | j | d d � d t | d f d t | d f g � d S)Nc S s
t j � S)N)r
r� r r r r �fmt s z3TracebackFormatTests.test_format_stack.<locals>.fmtr z> File "%s", line %d, in test_format_stack
result = fmt()
zA File "%s", line %d, in fmt
return traceback.format_stack()
r rs )r� r� r$ r� )r
r� �resultr\ r r r �test_format_stack s z&TracebackFormatTests.test_format_stack) r< r= r> ro r r� r� r� r� r� r� r r r r rm � s ,
rm zG
The above exception was the direct cause of the following exception:
zF
During handling of the above exception, another exception occurred:
z(%s|%s)c @ 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)�BaseExceptionReportingTestsc C sN t | t � r | Sy | � Wn) t k
rI } z | SWYd d } ~ Xn Xd S)N)�
isinstancerC )r
Zexception_or_callabler9 r r r �
get_exception s z)BaseExceptionReportingTests.get_exceptionc C s d d d S)Nr r r )r
r r r �zero_div% s z$BaseExceptionReportingTests.zero_divc C s_ | j � } | j | d j d � � | j d | d � | j | d j d � | d
� d S) Nr1 z Filez1/0 # In zero_divr r �ZeroDivisionError���rs ���r� )rW r& r| r( )r
�msg�linesr r r �check_zero_div( s z*BaseExceptionReportingTests.check_zero_divc C s� y d d Wn+ t k
r9 } z | } WYd d } ~ Xn X| j | � j � } | j t | � d � | j | d j d � � | j | d j d � � | j d | d � | j | d j d � � d S)
Nr r r rr z Filez 1/0 # Markerr r1 r� )r� �
get_reportrW r$ r% r& r| r( )r
�_r9 r� r r r � test_simple. s z'BaseExceptionReportingTests.test_simplec s� � f d d � � � f d d � } t j � j | � � } � j t | � d � � j | d t � � j | d � � j d | d � d S)
Nc sC y � j � Wn. t k
r> } z t | � WYd d } ~ Xn Xd S)N)r� r� rn )r9 )r
r r � inner_raise; s z;BaseExceptionReportingTests.test_cause.<locals>.inner_raisec s � � d S)Nr r )r� r r � outer_raise@ s z;BaseExceptionReportingTests.test_cause.<locals>.outer_raiser1 r r zinner_raise() # Markerr )�
boundaries�splitr� r$ r% �
cause_messager� r( )r
r� �blocksr )r� r
r �
test_cause: s z&BaseExceptionReportingTests.test_causec s� � f d d � � � f d d � } t j � j | � � } � j t | � d � � j | d t � � j | d � � j d | d � d S)
Nc s- y � j � Wn t k
r( t � Yn Xd S)N)r� r� rn r )r
r r r� I s
z=BaseExceptionReportingTests.test_context.<locals>.inner_raisec s � � d S)Nr r )r� r r r� N s z=BaseExceptionReportingTests.test_context.<locals>.outer_raiser1 r r zinner_raise() # Markerr )r� r� r� r$ r% �context_messager� r( )r
r� r� r )r� r
r � test_contextH s z(BaseExceptionReportingTests.test_contextc C s� y" y
t � Wn t d � Yn XWn+ t k
rO } z | } WYd d } ~ Xn X| j | � j � } | j t | � d � | j | d j d � � | j | d j d � � | j d | d � | j | d j d � � d S)
Nr r rr r z FilezZeroDivisionError from Noner r1 r� ) rC r� r� rW r$ r% r& r| r( )r
r� r9 r� r r r �test_context_suppressionV s
z4BaseExceptionReportingTests.test_context_suppressionc s� � f d d � � � f d d � } t j � j | � � } � j t | � d � � j | d t � � j | d � � j d | d � d S)
Nc sf y � j � Wn+ t k
r; } z | } WYd d } ~ Xn Xy t Wn t k
ra t | � Yn Xd S)N)r� r� Zxyzzy� NameErrorrn )Z_er9 )r
r r r� h s
zGBaseExceptionReportingTests.test_cause_and_context.<locals>.inner_raisec s � � d S)Nr r )r� r r r� q s zGBaseExceptionReportingTests.test_cause_and_context.<locals>.outer_raiser1 r r zinner_raise() # Markerr )r� r� r� r$ r% r� r� r( )r
r� r� r )r� r
r �test_cause_and_contexte s z2BaseExceptionReportingTests.test_cause_and_contextc s� � f d d � � � f d d � } t j � j | � � } � j t | � d � � j | d t � � j d | d � � j d | d � � j d
| d � � j | d � d S) Nc s~ yI y � j � Wn4 t k
rG } z | } t | � WYd d } ~ Xn XWn. t k
ry } z | | � WYd d } ~ Xn Xd S)N)r� r� rn )r9 �z)r
r r r� z s zEBaseExceptionReportingTests.test_cause_recursive.<locals>.inner_raisec s � � d S)Nr r )r� r r r� � s zEBaseExceptionReportingTests.test_cause_recursive.<locals>.outer_raiser1 r zraise KeyError from er z1/0zinner_raise() # Markerr ) r� r� r� r$ r% r� r( Z assertNotInr� )r
r� r� r )r� r
r �test_cause_recursivey s z0BaseExceptionReportingTests.test_cause_recursivec C sn d d � } | j | � j � } | j | d d � d d � } | j | � j � } | j | d d � d S) Nc S s t d d � � d S)Nrc r rg �hello)rc r rg r� )r# r r r r r9 � s zFBaseExceptionReportingTests.test_syntax_error_offset_at_eol.<locals>.er z ^c S s t d � d S)Nz x = 5 | 4 |)r r r r r r9 � s z ^rs rs )r� rW r$ )r
r9 r� r r r � test_syntax_error_offset_at_eol� s z;BaseExceptionReportingTests.test_syntax_error_offset_at_eolN)
r< r= r> r� r� r� r� r� r� r� r� r� r� r r r r r� s r� c @ s e Z d Z d d � Z d S)�PyExcReportingTestsc C s� | j | � } d j t j t | � | | j � � } t d � �# } t j t | � | | j � Wd QX| j | j � | � | S)Nrc rH )
r� rD r
�format_exception�type�
__traceback__r �print_exceptionr$ rx )r
r9 �sZsior r r r� � s !"z PyExcReportingTests.get_reportN)r< r= r> r� r r r r r� � s r� c @ s" e Z d Z e d d � � Z d S)�CExcReportingTestsc C sH d d l m } | j | � } t d � � } | | � Wd QX| j � S)Nr )�exception_printrH )rt r� r� r rx )r
r9 r� r� r r r r� � s
z CExcReportingTests.get_reportN)r< r= r> r r� r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�MiscTracebackCasesc s� � f d d � } � f d d � � d d � � y | � Wn t j � \ } } } Yn X| j j j j } | j t | j � d � t j | � | j t | j � d � d S) Nc s � � d S)Nr r )�middler r �outer� s z,MiscTracebackCases.test_clear.<locals>.outerc s � � d S)Nr r )�innerr r r� � s z-MiscTracebackCases.test_clear.<locals>.middlec S s d } d d d S)Nr r r )�ir r r r� � s z,MiscTracebackCases.test_clear.<locals>.innerr r ) ra ru rv r� r$ r% �f_localsr
� clear_frames)r
r� r} r r~ Z inner_framer )r� r� r �
test_clear� s
z MiscTracebackCases.test_clearc C se d d � } | � } | j j } | j | d d � t | d d d f t | d d d f g � d S)
Nc S s
t j � S)N)r
�
extract_stackr r r r �extract� s z6MiscTracebackCases.test_extract_stack.<locals>.extractr �test_extract_stackzresult = extract()r r� z return traceback.extract_stack()rs )r� r� r$ r� )r
r� r� r\ r r r r� � s z%MiscTracebackCases.test_extract_stackN)r< r= r> r� r� r r r r r� � s r� c C s t t � d S)N)r r< r r r r � test_main� s r� r@ ) �__doc__�ior ra Zunittest�reZ test.supportr r r r r r r
ZTestCaser rm r� r� r �escaper� r� r� r� r� r� r< r r r r �<module> s( �c%� (