Your IP : 216.73.216.85
���]�+ � @ s d d l Z d d l Z d d l Z d d l Z Gd d � d e j � Z Gd d � d � Z Gd d � d e � Z Gd d � d e j � Z Gd
d � d e � Z Gd d
� d
e � Z
d d � Z d d � Z e e � Z
d d d � Z d d � Z e d k r e j � d S)� Nc @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�TestGetProfilec C s t j d � d S)N)�sys�
setprofile)�self� r �8/opt/alt/python35/lib64/python3.5/test_sys_setprofile.py�setUp s zTestGetProfile.setUpc C s t j d � d S)N)r r )r r r r �tearDown s zTestGetProfile.tearDownc C s | j t j � � d S)N)Z assertIsNoner �
getprofile)r r r r �
test_empty s zTestGetProfile.test_emptyc C s3 d d � } t j | � | j t j � | � d S)Nc W s d S)Nr )�argsr r r �fn s z&TestGetProfile.test_setget.<locals>.fn)r r ZassertIsr
)r r
r r r � test_setget s
zTestGetProfile.test_setgetN)�__name__�
__module__� __qualname__r r r r r r r r r s r c @ sC e Z d Z d d � Z d d � Z d d d � Z d d � Z d S)
� HookWatcherc C s g | _ g | _ d S)N)�frames�events)r r r r �__init__ s zHookWatcher.__init__c C s8 | d k s$ | d k s$ | d k r4 | j | | � d S)N�call�return� exception)� add_event)r �frame�event�argr r r �callback s zHookWatcher.callbackNc
C s� | d k r t j d � } y | j j | � } Wn1 t k
rd t | j � } | j j | � Yn X| j j | | t | � f � d S)N� ) r � _getframer �index�
ValueError�len�appendr �ident)r r r Zframenor r r r # s
zHookWatcher.add_eventc sA t | j j � t t � g � d | _ � f d d � | j D� S)Nc s& g | ] } | d � k r | � q S)� r )�.0�item)�
disallowedr r �
<listcomp>5 s z*HookWatcher.get_events.<locals>.<listcomp>)r$ r �__func__r r )r r )r( r �
get_events0 s zHookWatcher.get_events)r r 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 d � Z d d � Z d d
� Z d d � Z d
e d e d e d e d e d e i Z d S)�ProfileSimulatorc C s# | | _ g | _ t j | � d S)N)�testcase�stackr r )r r- r r r r 9 s zProfileSimulator.__init__c C s | j | | | � d S)N)�dispatch)r r r r r r r r > s zProfileSimulator.callbackc C s$ | j d | � | j j | � d S)Nr )r r. r# )r r r r r �
trace_callB s zProfileSimulator.trace_callc C s! | j d | � | j j � d S)Nr )r r. �pop)r r r r r � trace_returnF s z ProfileSimulator.trace_returnc C s | j j d � d S)Nz2the profiler should never receive exception events)r- �fail)r r r r r �trace_exceptionJ s z ProfileSimulator.trace_exceptionc C s d S)Nr )r r r r r �
trace_passN s zProfileSimulator.trace_passr r r Zc_callZc_returnZ c_exceptionN)
r r r r r r0 r2 r4 r5 r/ r r r r r, 8 s r, c @ s e Z d Z d d � Z d S)� TestCaseBasec C sN t | | j � � } | | k rJ | j d t j | � t j | � f � d S)Nz'Expected events:
%s
Received events:
%s)�capture_events� new_watcherr3 �pprintZpformat)r �callableZexpectedr r r r � check_events\ s zTestCaseBase.check_eventsN)r r r r; r r r r r6 [ s r6 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 d � Z
d d � Z d d � Z d d � Z d S) �ProfileHookTestCasec C s t � S)N)r )r r r r r8 d s z ProfileHookTestCase.new_watcherc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s d S)Nr )�pr r r �fh s z*ProfileHookTestCase.test_simple.<locals>.fr r r )r$ r; )r r> �f_identr r r � test_simpleg s z ProfileHookTestCase.test_simplec C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s d d d S)Nr r r )r= r r r r> p s z-ProfileHookTestCase.test_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_exceptiono s z"ProfileHookTestCase.test_exceptionc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s y d d Wn Yn Xd S)Nr r r )r= r r r r> x s z4ProfileHookTestCase.test_caught_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_caught_exceptionw s z)ProfileHookTestCase.test_caught_exceptionc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s y d d Wn Yn Xd S)Nr r r )r= r r r r> � s z;ProfileHookTestCase.test_caught_nested_exception.<locals>.fr r r )r$ r; )r r> r? r r r � test_caught_nested_exception� s z0ProfileHookTestCase.test_caught_nested_exceptionc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s d d d S)Nr r r )r= r r r r> � s z4ProfileHookTestCase.test_nested_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_nested_exception� s z)ProfileHookTestCase.test_nested_exceptionc
s� d d � � � f d d � } t � � } t | � } | j | d d | f d d | f d d | f d d | f d d | f d d | f g � d S)
Nc S s d d d S)Nr r r )r= r r r r> � s z>ProfileHookTestCase.test_exception_in_except_clause.<locals>.fc s6 y � | � Wn! y � | � Wn Yn XYn Xd S)Nr )r= )r> r r �g� s z>ProfileHookTestCase.test_exception_in_except_clause.<locals>.gr r r% r � )r$ r; )r rE r? �g_identr )r> r � test_exception_in_except_clause� s z3ProfileHookTestCase.test_exception_in_except_clausec s� d d � � � f d d � } t � � } t | � } | j | d d | f d d | f d d | f d d | f d d | f g � d S)
Nc S s d d d S)Nr r r )r= r r r r> � s z9ProfileHookTestCase.test_exception_propagation.<locals>.fc s# z � | � Wd | j d � Xd S)Nzfalling through)r )r= )r> r r rE � s z9ProfileHookTestCase.test_exception_propagation.<locals>.gr r r% r zfalling through)r$ r; )r rE r? rG r )r> r �test_exception_propagation� s z.ProfileHookTestCase.test_exception_propagationc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s# y d d Wn d d Yn Xd S)Nr r r )r= r r r r> � s z/ProfileHookTestCase.test_raise_twice.<locals>.fr r r )r$ r; )r r> r? r r r �test_raise_twice� s z$ProfileHookTestCase.test_raise_twicec C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s y d d Wn � Yn Xd S)Nr r r )r= r r r r> � s z1ProfileHookTestCase.test_raise_reraise.<locals>.fr r r )r$ r; )r r> r? r r r �test_raise_reraise� s z&ProfileHookTestCase.test_raise_reraisec C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s
t � � d S)N)� Exception)r= r r r r> � s z)ProfileHookTestCase.test_raise.<locals>.fr r r )r$ r; )r r> r? r r r �
test_raise� s z ProfileHookTestCase.test_raisec s d d � � � f d d � � � f d d � � � f d d � � � f d d
� } t � � } t � � } t � � } t � � } t | � } | j | d d | f d
d | f d d | f d d | f d d | f d d | f d d | f d d | f d
d | f d d | f g
� d S)Nc S s d d d S)Nr r r r r r r r> � s z5ProfileHookTestCase.test_distant_exception.<locals>.fc s � � d S)Nr r )r> r r rE � s z5ProfileHookTestCase.test_distant_exception.<locals>.gc s � � d S)Nr r )rE r r �h� s z5ProfileHookTestCase.test_distant_exception.<locals>.hc s � � d S)Nr r )rN r r �i� s z5ProfileHookTestCase.test_distant_exception.<locals>.ic s � � d S)Nr )r= )rO r r �j� s z5ProfileHookTestCase.test_distant_exception.<locals>.jr r r% rF � � r )r$ r; )r rP r? rG �h_ident�i_ident�j_identr )r> rE rN rO r �test_distant_exception� s( z*ProfileHookTestCase.test_distant_exceptionc s� d d � � � f d d � } t � � } t | � } | j | d d | f d d | f d d | f d d | f d d | f d d | f d d | f d d | f g � d S) Nc s s x t d � D] } | Vq
Wd S)Nr% )�range)rO r r r r> � s z-ProfileHookTestCase.test_generator.<locals>.fc s x � � D] } q
Wd S)Nr )r= rO )r> r r rE � s z-ProfileHookTestCase.test_generator.<locals>.gr r r% r )r$ r; )r rE r? rG r )r> r �test_generator� s z"ProfileHookTestCase.test_generatorc s� d d � � � f d d � } t � � } t | � } | j | d d | f d d | f d d | f d d | f d d | f d d | f d d | f d d | f g � d S) Nc s s x t d � D] } | Vq
Wd S)Nr% )rW )rO r r r r> s z2ProfileHookTestCase.test_stop_iteration.<locals>.fc s x � � D] } q
Wd S)Nr )r= rO )r> r r rE s z2ProfileHookTestCase.test_stop_iteration.<locals>.gr r r% r )r$ r; )r rE r? rG r )r> r �test_stop_iteration s z'ProfileHookTestCase.test_stop_iterationN)r r r r8 r@ rA rB rC rD rH rI rJ rK rM rV rX rY r r r r r< c s r< c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �ProfileSimulatorTestCasec C s
t | � S)N)r, )r r r r r8 s z$ProfileSimulatorTestCase.new_watcherc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s d S)Nr )r= r r r r> s z/ProfileSimulatorTestCase.test_simple.<locals>.fr r r )r$ r; )r r> r? r r r r@ s z$ProfileSimulatorTestCase.test_simplec C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s d d d S)Nr r r )r= r r r r> % s z8ProfileSimulatorTestCase.test_basic_exception.<locals>.fr r r )r$ r; )r r> r? r r r �test_basic_exception$ s z-ProfileSimulatorTestCase.test_basic_exceptionc C sD d d � } t | � } | j | d d | f d d | f g � d S)Nc S s y d d Wn Yn Xd S)Nr r r )r= r r r r> - s z9ProfileSimulatorTestCase.test_caught_exception.<locals>.fr r r )r$ r; )r r> r? r r r rB , s z.ProfileSimulatorTestCase.test_caught_exceptionc s d d � � � f d d � � � f d d � � � f d d � � � f d d
� } t � � } t � � } t � � } t � � } t | � } | j | d d | f d
d | f d d | f d d | f d d | f d d | f d d | f d d | f d
d | f d d | f g
� d S)Nc S s d d d S)Nr r r r r r r r> 6 s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.fc s � � d S)Nr r )r> r r rE 8 s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.gc s � � d S)Nr r )rE r r rN : s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.hc s � � d S)Nr r )rN r r rO < s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.ic s � � d S)Nr )r= )rO r r rP > s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.jr r r% rF rQ rR r )r$ r; )r rP r? rG rS rT rU r )r> rE rN rO r rV 5 s( z/ProfileSimulatorTestCase.test_distant_exceptionN)r r r r8 r@ r[ rB rV r r r r rZ s
rZ c C s4 t | d � r | j } n | j } | j | j f S)N�f_code)�hasattrr\ �__code__�co_firstlineno�co_name)�function�coder r r r$ R s r$ c C s y | | � Wn Yn Xd S)Nr )r> r= r r r �protectZ s rc c C s� | d k r t � } t j � } t j � z. t j | j � t | | � t j d � Wd | rl t j � X| j � d d � S)Nr ���)
r �gc� isenabled�disabler r r rc �enabler+ )r: r= Zold_gcr r r r7 a s
r7 c C s# d d l } | j t | � � d S)Nr )r9 r7 )r: r9 r r r � show_eventsr s ri �__main__)re r9 r ZunittestZTestCaser r r, r6 r< rZ r$ rc Z
protect_identr7 ri r �mainr r r r �<module> s #�: