U
|
Z ±dL> ã @ s< d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZm Z m
|
Z
|
d dlm Z d dl Z d dl Z ddlmZ ddlmZmZmZ d d
|
lmZ d d lmZ e jd k r²d d lmZ G dd deZ dd Z dd Z dd Z dd Z dd Z!dd 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,d0d1 Z-d2d3 Z.ej/j0e j1j2d4kd5d6d7d8 Z3d9d: Z4d;d< Z5d=d> Z6d?d@ Z7dRdBdCZ8dDdE Z9z d dl:Z:W n e;k
|
rÊ dAZ<Y nX dFZ<ej/j0e< dGd6Z=ee=dHdI Z>ee=dJdK Z?ee=dLdM Z@eej/j0edN A¡ dOd6dPdQ ZBdS )Sé N)ÚPath)Ú
|
extract_tbÚprint_exceptionÚformat_exception)Ú_cause_messageé )Úslowé )Ú
|
MultiErrorÚ concat_tbÚNonBaseMultiErroré )ÚTrioDeprecationWarning)Ú open_nursery)r é )ÚExceptionGroupc s( e Zd ZdZ fddZdd Z ZS )ÚNotHashableExceptionNc s t ¡ || _d S ©N)ÚsuperÚ__init__Úcode)Úselfr ©Ú __class__© úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_multierror.pyr s
|
z NotHashableException.__init__c C s t |tsdS | j|jkS )NF)Ú
|
isinstancer r )r Úotherr r r Ú__eq__" s
|
zNotHashableException.__eq__)Ú__name__Ú
|
__module__Ú __qualname__r r r Ú __classcell__r r r r r s r c à s t | d S r )r )r r r r Úraise_nothashable( s r# c C s
|
t d S r )Ú raiser1_2r r r r Úraiser1, s r% c C s
|
t d S r )Ú raiser1_3r r r r r$ 0 s r$ c C s t dd S )NZraiser1_string)Ú
|
ValueErrorr r r r r& 4 s r& c C s
|
t d S r )Ú raiser2_2r r r r Úraiser28 s r) c C s t dd S )NZraiser2_string)ÚKeyErrorr r r r r( < s r( c C s t d S r )Ú NameErrorr r r r Úraiser3@ s r, c
|
C s: z
|
| W n* t k
|
r4 } z | W Y ¢S d }~X Y nX d S r )Ú Exception)ÚraiserÚexcr r r Úget_excD s
|
r0 c C s
|
t | jS r )r0 Ú __traceback__)r. r r r Úget_tbK s r2 c C sÆ t t} t t}t| }t|}t| |}t||| ks>tt|| }t||| ks\tttd | |ksrttt| d |ksttd d d ksttt t|ks®ttt t|ksÂtd S r )r2 r% r) r r ÚAssertionError)Ztb1Ztb2Zentries1Zentries2Ztb12Ztb21r r r Útest_concat_tbO s
|
|
r4 c C s¨ t t} t t}t| g| ks"tt| |g}|j| |fks@tdt|ksPtdt|ks`tt t
|
¡ tt W 5 Q R X t t
|
¡ tt t g W 5 Q R X d S )Nr' )r0 r% r) r
|
r3 Ú
|
exceptionsÚstrÚreprÚpytestÚraisesÚ TypeErrorÚobjectr* r' )Úexc1Úexc2Úmr r r Útest_MultiErrori s r? c C s< t t f} t| }t|g}||ks*t|j| ks8td S r )r* r' r
|
r3 r5 )r5 ÚaÚbr r r Ú!test_MultiErrorOfSingleMultiErrory s
|
|
rB c à s t d} t d}t }| |ks"t| |ks.tt t¡> t 4 I d H }| td¡ | td¡ W 5 Q I d H R X W 5 Q R X d S )Né* i ) r r' r3 r8 r9 r
|
r Z
|
start_soonr# )r< r= Úexc3Znurseryr r r Útest_MultiErrorNotHashable s rE c C sN t tdt g} dd }t t¡ t || ¡}W 5 Q R X t|tsJtd S )NrC c S s t | trd S | S d S r )r r' ©r/ r r r Úhandle_ValueError s
|
z=test_MultiError_filter_NotHashable.<locals>.handle_ValueError) r
|
r r' r8 Úwarnsr Úfilterr r3 )ZexcsrG Z filtered_excsr r r Ú"test_MultiError_filter_NotHashable s
|
rJ c
|
C s` t t} t t}t t}zt| |gW n2 tk
|
rZ } zt||g W Y ¢S d }~X Y nX d S r )r0 r% r) r, r
|
Ú BaseException)r< r= rD Zm12r r r Ú make_tree s
|
rL c C s° | d ks|d kr | |ks t d S t| t|ks4t t| jt|jksLt t| j|j t| j|j t| tr¬t | j
|
t |j
|
kst t | j
|
|j
|
D ]\}}t|| qd S r ) r3 Útyper r1 Úassert_tree_eqÚ __cause__Ú __context__r r
|
Úlenr5 Úzip)Úm1Úm2Úe1Úe2r r r rN ´ s
|
rN c
|
C sz dd } t }t|| t t¡ t | |¡|ks8tW 5 Q R X t|t t }ztW n> tk
|
r t t¡ t | |¡|kstW 5 Q R X Y nX t|t dd }t t¡ t |t ¡}W 5 Q R X t |tsâtt
|
|j dksôtt |j d t s
|
tt |j d t s tt |j d jts8tt }t |j d j d tsZtt|jt|j d j t|j d j d j }dd }|| ||j d ||j d j d ||j d j |jd ksÜtt|j d jj}||ksütd
|
d }t }t t¡ t ||¡}W 5 Q R X ||j d ks@td d } t t¡ t | t ¡d ksltW 5 Q R X d S )Nc S s | S r r rF r r r Ú null_handlerà s z,test_MultiError_filter.<locals>.null_handlerc S s" t | trd S t | tr t S | S r ©r r' r* Ú RuntimeErrorrF r r r Ú simple_filter× s
|
|
|
z-test_MultiError_filter.<locals>.simple_filterr r r c S s t t| | | j d S r )r rM r1 rF r r r Úp÷ s z!test_MultiError_filter.<locals>.pc S s t | trd S | S r )r r+ rF r r r Úfilter_NameError s
|
z0test_MultiError_filter.<locals>.filter_NameErrorc S s d S r r rF r r r Ú
|
filter_all s z*test_MultiError_filter.<locals>.filter_all)rL rN r8 rH r r
|
rI r3 r' r rQ r5 rY r+ rP r* r r1 )
|
rW r> rZ Únew_mÚorigZorig_extractedr[ Z new_extractedr\ r] r r r Útest_MultiError_filter sZ
|
$ ÿþÿ r` c
|
C s^ dd } t t¡ t | ¡ W 5 Q R X W 5 Q R X t }t t¡:}t t¡$ t dd ¡
|
|W 5 Q R X W 5 Q R X W 5 Q R X |j|kst|j d k s t|j j
|
j j dks´t|j j d ksÄtd |_ t|t t t¡& t dd ¡ t W 5 Q R X W 5 Q R X dd }t t¡8}t t¡" t |¡ t W 5 Q R X W 5 Q R X W 5 Q R X |j}t|tsrtt|jd kstt|jd
|
tstt|jd ts²t|jr¾t|jd ksÎtt }t |_t t¡:}t t¡$ t d d ¡
|
|W 5 Q R X W 5 Q R X W 5 Q R X t|jjts:tt }t }||_t t¡:}t t¡$ t d d ¡
|
|W 5 Q R X W 5 Q R X W 5 Q R X |jj|ks¨t|jjr¶tdD ]}t }t }||_||_t }t t¡F}dd } t t¡( t | ¡ t||gW 5 Q R X W 5 Q R X W 5 Q R X |jj|ksDt|jj|ksºtqºd S )Nc S s d S r r ©Ú_r r r Únoop s z#test_MultiError_catch.<locals>.noopc S s | S r r rF r r r Ú<lambda># ó z'test_MultiError_catch.<locals>.<lambda>Útest_MultiError_catchc S s d S r r ra r r r rd / re c S s" t | trd S t | tr t S | S r rX rF r r r rZ 2 s
|
|
|
z,test_MultiError_catch.<locals>.simple_filterr r r c S s | S r r rF r r r rd L re c S s | S r r rF r r r rd T re )TFc S s t | trd S | S d S r )r rY rF r r r Úcatch_RuntimeErrora s
|
z1test_MultiError_catch.<locals>.catch_RuntimeError)r8 rH r r
|
ÚcatchrL r9 Úvaluer3 r1 Útb_frameÚf_codeÚco_nameÚtb_nextrN r rQ r5 rY r+ Ú__suppress_context__rP r' r* rO )
|
rc r> ÚexcinforZ r^ ÚvÚcontextZsuppress_contextZ
|
distractorrg r r r rf sj " $ " " *rf Úcpythonz$Only makes sense with refcounting GC)Úreasonc C s¦ t ¡ t ¡ } dd }dd }zjt t j¡ t t ¡8 t
|
t ¡" t |¡ | W 5 Q R X W 5 Q R X W 5 Q R X t ¡ t jrt W 5 t | ¡ t j ¡ X d S )Nc S s t ttttgd S r )r
|
r0 r% r) r r r r Ú
|
make_multiv s zFtest_MultiError_catch_doesnt_create_cyclic_garbage.<locals>.make_multic S s0 t | trt S t | tr t S ds,tdd S )NFz)only ValueError and KeyError should exist)r r' r- r* rY r3 rF r r r rZ z s
|
|
|
zItest_MultiError_catch_doesnt_create_cyclic_garbage.<locals>.simple_filter)ÚgcZcollectZ get_debugZ set_debugÚgarbageÚclearZ DEBUG_SAVEALLr8 r9 r
|
rH r rh r3 )Z old_flagsrt rZ r r r Ú2test_MultiError_catch_doesnt_create_cyclic_garbagen s $
|
rx c C sN d}t d | D ]8}t d| t |¡}| ||¡}|d k s@t| ¡ }qd S )Nr z looking for pattern matches...zchecking pattern:)ÚprintÚreÚcompileÚsearchr3 Úend)Z pattern_listÚstringÚoffsetÚpatternZreobjÚmatchr r r Úassert_match_in_seq s
|
|
r c C sD t ddgd t ddgd t t¡ t ddgd W 5 Q R X d S )Nr@ rA z xx a xx b xxz xx b xx a xx)r r8 r9 r3 r r r r Útest_assert_match_in_seq s r c C s$ t t t g} t| t ks tdS )zf
|
Test that MultiError() with at least one base exception will return a MultiError
|
object.
|
N)r
|
ÚZeroDivisionErrorÚKeyboardInterruptrM r3 rF r r r Útest_base_multierror s r c C s2 t t t g} t| tks tt| ts.tdS )zd
|
Test that MultiError() without base exceptions will return a NonBaseMultiError
|
object.
|
N)r
|
r r' rM r r3 r r rF r r r Útest_non_base_multierrorª s r Fc
|
C s dd l }t|jjj}ttjd | }ttj}td| d¡ d|kr\|d tj
|
¡}ng }| dt |¡ | dt |j¡ tj
|
|¡|d< td| d¡ |rÐ| ¡ dg}tjddd d
|
d d |¡ g}ntjdt |g}td | tj||tjtjd} td t| j d¡ | S )Nr Ztest_multierror_scriptszparent PYTHONPATH:Ú
|
PYTHONPATHzsubprocess PYTHONPATH:zexit()z-uz-mÚIPythonz--quickz!--TerminalIPythonApp.code_to_run=Ú
|
zrunning:)ÚenvÚstdoutÚstderrzprocess output:úutf-8)Útrior Ú__file__ÚparentÚdictÚosÚenvironry ÚgetÚsplitÚpathsepÚinsertr6 ÚjoinÚ read_textÚsysÚ
|
executableÚ
|
subprocessÚrunÚPIPEÚSTDOUTr Údecode)
|
ÚnameÚ use_ipythonr Z trio_pathZ script_pathr ÚppÚlinesÚcmdÚ completedr r r Ú
|
run_scriptµ s@
|
ù
|
|
ÿr¨ c C s& t ddddddddg| j d ¡ d S )
|
Nz in <module>r
|
ú --- 1 ---z
|
in exc1_fnr' ú --- 2 ---z
|
in exc2_fnr* r )r r r¡ ©r§ r r r Úcheck_simple_excepthookÝ s ø
|
|
õr¬ Tz need IPythonc C s t ddd} t| d S )Nzsimple_excepthook.pyT©r£ ©r¨ r¬ r« r r r Útest_ipython_exc_handler÷ s r¯ c C s t d} t| d S )Nz simple_excepthook_IPython.pyr® r« r r r Ú test_ipython_imported_but_unusedþ s r° c C sB t ddd} tdddddd g| j d
|
¡ d | j d
|
¡ks>td S ) Nzipython_custom_exc.pyTr ÚRuntimeWarningzIPython detectedzskip installing Trior
|
r' r* r úcustom sys.excepthook)r¨ r r r¡ r3 r« r r r Ú test_ipython_custom_exc_handler s ø
|
|
õr³ z4/usr/lib/python3/dist-packages/apport_python_hook.pyz)need Ubuntu with python3-apport installedc C s6 t d} | j d¡}d|ks ttddddg| d S )Nzapport_excepthook.pyr r² r© r* rª r' )r¨ r r¡ r3 r )r§ r r r r Ú.test_apport_excepthook_monkeypatch_interaction s
|
þr´ )F)Cru Úloggingr r Úpathlibr r8 Ú tracebackr r r r r rz Ztutilr Z _multierrorr
|
r r Ú r Z_corer Ú version_infoZexceptiongroupr r- r r# r% r$ r& r) r( r, r0 r2 r4 r? rB rE rJ rL rN r` rf ÚmarkZskipifÚimplementationr¢ rx r r r r r¨ r¬ r Ú ImportErrorZ have_ipythonZ need_ipythonr¯ r° r³ Úexistsr´ r r r r Ú<module> s
|
|
UW
|
ÿ
|
|
|
(
|
þ
|
