U
|
¸ý°d ( ã @ s
|
d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m Z d dl m
|
Z
|
dd l m Z zDd d
|
l mZ d d l mZ d d l mZ ed ddZedddZdZW n ek
|
rÈ d Z ZZY nX eedk dG dd de Zeedk dG dd de ZdS )é )Úprint_functionN)Úpartial)Ú
|
skipUnless)ÚskipIf)Úgreenlet)Ú
|
getcurrenté )ÚTestCase)ÚContext)Ú
|
ContextVar)Ú copy_contextÚid)ÚdefaultÚvarzContextVar not supportedc @ sl e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd ZdS )ÚContextVarsTestsc O s t j||S ©N)r Úrun)ÚselfÚargsÚkwargs© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\greenlet/tests/test_contextvars.pyÚ _new_ctx_run" s z ContextVarsTests._new_ctx_runc C sb t }|d kr | | ¡ ¡ n| | ¡ |¡ | |¡ tdD ] }|| ¡ d7 < | q>d S )Né r )ÚID_VARÚ assertIsNoneÚgetÚ assertEqualÚsetÚrange)r Ú greenlet_idÚcallbackÚcountsÚexpectZctx_varÚ_r r r Ú
|
_increment% s
|
zContextVarsTests._incrementc sV t d¡ t j tdd tdD fddtddD }|D ](}dkr`t |_qJd krJt j|_qJtd
|
D ]*}t ¡ d7 < |D ] }| ¡ qq|d krÈt |D ] }| ¡ q¸n|D ] }| ¡ qÌ|D ]4}
|
|j ¡ d kr |j¡ qÞ |j¡ qÞd kr8 ddddd d ¡ n t ¡ td
|
g¡ d S )Nr c s s | ]}|d fV qdS )r Nr ©Ú.0Úir r r Ú <genexpr>4 s z1ContextVarsTests._test_context.<locals>.<genexpr>é c
|
sX g | ]P}t td kr$tt jjnj| dkr>|d ndkrJdnddqS )r Úsharer )r r r N)r r! r" r# )r r r r r% r& ©r! r" Ú propagate_byr r r Ú
|
<listcomp>6 s óýþöz2ContextVarsTests._test_context.<locals>.<listcomp>r r r+ r r é )r r r é é )r r r ÚswitchÚdictr r Ú
|
gr_contextr ÚreversedÚ
|
assertTrueZdeadr ZassertIsNotNoner Úvalues)r r- ZletsZletr( r r, r Ú _test_context0 s8
|
ò
|
|
|
|
z ContextVarsTests._test_contextc C s | | jd¡ d S )Nr ©r r8 ©r r r r Ú&test_context_propagated_by_context_runl s z7ContextVarsTests.test_context_propagated_by_context_runc C s | | jd¡ d S )Nr r9 r: r r r Ú,test_context_propagated_by_setting_attributeo s z=ContextVarsTests.test_context_propagated_by_setting_attributec C s | | jd ¡ d S r r9 r: r r r Útest_context_not_propagatedr s z,ContextVarsTests.test_context_not_propagatedc C s | | jd¡ d S )Nr+ r9 r: r r r Útest_context_sharedu s z$ContextVarsTests.test_context_sharedc C s@ t t j}t t j}| t j¡ | t j¡ | ¡ d S r )r r r r2 r ©r Úlet1Zlet2r r r Útest_break_ctxvarsx s
|
z#ContextVarsTests.test_break_ctxvarsc C sL t t j}t t j}t |_t |_| ¡ | ¡ | ¡ | ¡ d S r )r r r2 r r4 r? r r r Ú9test_not_broken_if_using_attribute_instead_of_context_run s zJContextVarsTests.test_not_broken_if_using_attribute_instead_of_context_runc s t d ¡ fdd}t| td¡
|
`W 5 Q R X j¡ ¡ \}} | j¡ |t d¡ |t d¡ |
|
t j ¡d¡ | _ ¡ j|¡ d _ ¡ j|¡ d _ ¡ j¡ d t ¡ t |¡d¡ t |¡d¡ d S )Nc sX t ¡ ¡ j¡ t d¡ jt¡ t ¡ d¡ jt d¡ j} d _ t ¡ ¡ j¡ t d¡ jt¡ t ¡ d¡ jt d¡ j}t j
|
| |f¡ t ¡ d¡ | _ t ¡ d¡ t j
|
¡ t ¡ ¡ j¡ | _ t ¡ d¡ t j
|
¡ t ¡ ¡ j¡ d S )Nr r ) r r r r4 r ZassertIsInstancer
|
r r Úparentr2 )Ú old_contextÚ new_context©Úgrr r r Útarget s6
|
|
zFContextVarsTests.test_context_assignment_while_running.<locals>.targetz can't delete context attributer r )r r r ÚassertRaisesRegexÚAttributeErrorr4 r r2 ÚassertIsr r r ÚgcZcollectÚsysÚ getrefcount)r rH rD rE r rF r Ú%test_context_assignment_while_running s0
|
) z6ContextVarsTests.test_context_assignment_while_runningc sX dd l }t d ¡ t | ¡ | ¡ | ¡ | ¡ g fdd fdd}|j|dd}| ¡ d¡ d }| t d ¡ t
|
|d
|
W 5 Q R X | t d ¡ d |_ W 5 Q R X ¡ d¡ | |j ¡ | |j t d ¡ d |_ ¡ | d¡ | |d g¡ | |j ¡ |_ | |j ¡ d d
= d }d }d S ) Nr c sD t d¡ ¡ d¡ t d¡ t j ¡ t ¡ ¡ d S )Nr é
|
r )ÚVAR_VARr Úwaitr rC r2 Úappendr r )ÚholderÚ
|
is_runningÚshould_suspendr r Úgreenlet_in_thread_fnß s
|
|
|
zXContextVarsTests.test_context_assignment_different_thread.<locals>.greenlet_in_thread_fnc sF t } | _ | ¡ | ¡ ¡ d¡ | ¡ ~ t d S )NrP )r r4 rS r2 r rR )rG )ÚctxÚ did_suspendrW rT Ú should_exitr r Ú thread_fnç s
|
|
zLContextVarsTests.test_context_assignment_different_thread.<locals>.thread_fnT)rH ÚdaemonrP zrunning in a differentr4 r )Ú threadingrQ r r
|
ÚEventÚThreadÚstartrR rI Ú
|
ValueErrorÚgetattrr4 rK r Újoinr )r r] r[ ÚthreadrG r )rX rY rW rT rU rZ rV r Ú(test_context_assignment_different_threadÔ s@
|
|
|
|
|
z9ContextVarsTests.test_context_assignment_different_threadc C s( t }| td¡ | |_W 5 Q R X d S )Nz6greenlet context must be a contextvars.Context or None)r rI Ú TypeErrorr4 )r Úgr r r Ú"test_context_assignment_wrong_type s
|
ÿz3ContextVarsTests.test_context_assignment_wrong_typeN)Ú__name__Ú
|
__module__Ú __qualname__r r% r8 r; r< r= r> rA rB rO re rh r r r r r s <
|
|
HBr zContextVar supportedc @ s e Zd Zdd ZdS )ÚNoContextVarsTestsc C s¢ t t j}| t|d¡ | t¡ t|d W 5 Q R X | t¡ d |_W 5 Q R X | ¡ | t¡ t|d W 5 Q R X | t¡ d |_W 5 Q R X ~d S )Nr4 ) r r r2 Z assertFalseÚhasattrÚ assertRaisesrJ rb r4 )r r@ r r r Útest_contextvars_errors s z*NoContextVarsTests.test_contextvars_errorsN)ri rj rk ro r r r r rl s rl )Ú
|
__future__r rL rM Ú functoolsr Zunittestr r r r Ú r Z contextvarsr
|
r r r rQ Ú ImportErrorr rl r r r r Ú<module> s, }
|
