Your IP : 216.73.216.213
�
g f[, � @ s, d d l Z d d l Z d d l Z d d l Z d d l m 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 d d � Z e d k r(e � n d S)� N)�supportc @ 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 �=/opt/alt/python34/lib64/python3.4/test/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 s; | d k s$ | d k s$ | d k r7 | j | | � n 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 � } n y | j j | � } Wn1 t k
rg t | j � } | j j | � Yn X| j j | | t | � f � d S)zAdd an event to the log.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)z Remove calls to add_event().Nc s& g | ] } | d � k r | � q S)� r )�.0�item)�
disallowedr r �
<listcomp>6 s z*HookWatcher.get_events.<locals>.<listcomp>)r% r �__func__r r )r r )r) r �
get_events1 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 i e d
6e d 6e d 6e d 6e d 6e d 6Z d S)�ProfileSimulatorc C s# | | _ g | _ t j | � d S)N)�testcase�stackr r )r r. r r r r : 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_callC s zProfileSimulator.trace_callc C s! | j d | � | j j � d S)Nr )r r/ �pop)r r r r r � trace_returnG 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_exceptionK s z ProfileSimulator.trace_exceptionc C s d S)Nr )r r r r r �
trace_passO s zProfileSimulator.trace_passr r r Zc_callZc_returnZ c_exceptionN)
r r r r r r1 r3 r5 r6 r0 r r r r r- 9 s r- c @ s e Z d Z d d � Z d S)� TestCaseBasec C sQ t | | j � � } | | k rM | j d t j | � t j | � f � n d S)Nz'Expected events:
%s
Received events:
%s)�capture_events� new_watcherr4 �pprintZpformat)r �callableZexpectedr r r r � check_events] s zTestCaseBase.check_eventsN)r r r r< r r r r r7 \ s r7 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 r9 e 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 �fi s z*ProfileHookTestCase.test_simple.<locals>.fr r r )r% r< )r r? �f_identr r r � test_simpleh 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? q s z-ProfileHookTestCase.test_exception.<locals>.fr r r )r% r< )r r? r@ r r r �test_exceptionp 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? y s z4ProfileHookTestCase.test_caught_exception.<locals>.fr r r )r% r< )r r? r@ r r r �test_caught_exceptionx 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 rF 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_propogation.<locals>.fc s# z � | � Wd | j d � Xd S)Nzfalling through)r )r> )r? r r rF � s z9ProfileHookTestCase.test_exception_propogation.<locals>.gr r r& r zfalling through)r% r< )r rF r@ rH r )r? r �test_exception_propogation� s z.ProfileHookTestCase.test_exception_propogationc 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 rF � s z5ProfileHookTestCase.test_distant_exception.<locals>.gc s � � d S)Nr r )rF r r �h� s z5ProfileHookTestCase.test_distant_exception.<locals>.hc s � � d S)Nr r )rO r r �i� s z5ProfileHookTestCase.test_distant_exception.<locals>.ic s � � d S)Nr )r> )rP r r �j� s z5ProfileHookTestCase.test_distant_exception.<locals>.jr r r& rG � � r )r% r< )r rQ r@ rH �h_ident�i_ident�j_identr )r? rF rO rP 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)rP r r r r? � s z-ProfileHookTestCase.test_generator.<locals>.fc s x � � D] } q
Wd S)Nr )r> rP )r? r r rF � s z-ProfileHookTestCase.test_generator.<locals>.gr r r& r )r% r< )r rF r@ rH 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
Wt � d S)Nr& )rX �
StopIteration)rP r r r r? s z2ProfileHookTestCase.test_stop_iteration.<locals>.fc s x � � D] } q
Wd S)Nr )r> rP )r? r r rF s z2ProfileHookTestCase.test_stop_iteration.<locals>.gr r r& r )r% r< )r rF r@ rH r )r? r �test_stop_iteration s z'ProfileHookTestCase.test_stop_iterationN)r r r r9 rA rB rC rD rE rI rJ rK rL rN rW rY r[ r r r r r= d 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 r9 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 rA 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 rC . 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? 8 s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.fc s � � d S)Nr r )r? r r rF : s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.gc s � � d S)Nr r )rF r r rO < s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.hc s � � d S)Nr r )rO r r rP > s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.ic s � � d S)Nr )r> )rP r r rQ @ s z:ProfileSimulatorTestCase.test_distant_exception.<locals>.jr r r& rG rR rS r )r% r< )r rQ r@ rH rT rU rV r )r? rF rO rP r rW 7 s( z/ProfileSimulatorTestCase.test_distant_exceptionN)r r r r9 rA r] rC rW r r r r r\ s
r\ c C s4 t | d � r | j } n | j } | j | j f S)N�f_code)�hasattrr^ �__code__�co_firstlineno�co_name)Zfunction�coder r r r% T s r% c C s y | | � Wn Yn Xd S)Nr )r? r> r r r �protect\ s rd c C s� | d k r t � } n t j � } t j � z. t j | j � t | | � t j d � Wd | rr t j � n X| j � d d � S)Nr ���)
r �gc� isenabled�disabler r r rd �enabler, )r; r> Zold_gcr r r r8 c s
r8 c C s# d d l } | j t | � � d S)Nr )r: r8 )r; r: r r r � show_eventst s rj c C s t j t t t � d S)N)r Z run_unittestr r= r\ r r r r � test_mainy s rk �__main__)rf r: r ZunittestZtestr ZTestCaser r r- r7 r= r\ r% rd Z
protect_identr8 rj rk r r r r r �<module> s$ #�: