Current File : //usr/local/lib/python3.8/unittest/test/testmock/__pycache__/testasync.cpython-38.pyc
U
qaב @ sv d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z m
Z
dd ZG dd dZG dd dZd d
Zd d Zd
d ZG dd deZe dZe dZG dd dejZG dd dejZG dd dejZ G dd dejZ G dd d ejZ G d d d ejZ G d d d ej Z!G d!d" d"ejZ"G d#d$ d$ejZ#G d%d& d&ejZ$G d'd( d(ejZ%dS )) N) ANYcall AsyncMockpatch MagicMockMockcreate_autospecsentinel _CallListc C s t d d S N)asyncioZset_event_loop_policy r
r
</usr/local/lib/python3.8/unittest/test/testmock/testasync.pytearDownModule
s r c @ s< e Zd Zdd Zdd Zdd Zedd Zed d
Z d S )
AsyncClassc C s d S r r
selfr
r
r __init__ s zAsyncClass.__init__c s d S r r
r r
r
r async_method s zAsyncClass.async_methodc C s d S r r
r r
r
r
normal_method s zAsyncClass.normal_methodc s d S r r
)clsr
r
r async_class_method s z AsyncClass.async_class_methodc s d S r r
r
r
r
r async_static_method s z AsyncClass.async_static_methodN)
__name__
__module__ __qualname__r r r classmethodr staticmethodr r
r
r
r r s
r c @ s e Zd Zdd ZdS )AwaitableClassc c s
d V d S r r
r r
r
r __await__ s zAwaitableClass.__await__N)r r r r r
r
r
r r s r c s d S r r
r
r
r
r
async_func# s r c s d S r r
)abcr
r
r async_func_args& s r$ c C s d S r r
r
r
r
r normal_func) s r% c @ s e Zd Zdd ZdS ) NormalClassc C s d S r r
r r
r
r r! - s z
NormalClass.aN)r r r r! r
r
r
r r& , s r& z .AsyncClassz .NormalClassc @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )AsyncPatchDecoratorTestc s" t td fdd}| d S )Nr c s t| d S r )
assertTruer iscoroutinefunction mock_methodr r
r
test_async7 s zLAsyncPatchDecoratorTest.test_is_coroutine_function_patch.<locals>.test_asyncr objectr r r, r
r r test_is_coroutine_function_patch6 s
z8AsyncPatchDecoratorTest.test_is_coroutine_function_patchc sB t td fdd}t t d fdd}| | d S )Nr c s$ | } t| t| d S r r( inspect isawaitabler runr+ mr r
r r, = s z?AsyncPatchDecoratorTest.test_is_async_patch.<locals>.test_asyncz
.async_methodc s$ | } t| t| d S r r1 r5 r r
r test_no_parent_attributeC s zMAsyncPatchDecoratorTest.test_is_async_patch.<locals>.test_no_parent_attribute)r r. r async_foo_name)r r, r7 r
r r test_is_async_patch< s
z+AsyncPatchDecoratorTest.test_is_async_patchc s" t td fdd}| d S )Nr c s | t d S r assertIsInstancer r* r r
r r, M s zCAsyncPatchDecoratorTest.test_is_AsyncMock_patch.<locals>.test_asyncr- r/ r
r r test_is_AsyncMock_patchL s
z/AsyncPatchDecoratorTest.test_is_AsyncMock_patchc s" t td fdd}| d S )Nr c s | t d S r r: r* r r
r r, T s zPAsyncPatchDecoratorTest.test_is_AsyncMock_patch_staticmethod.<locals>.test_asyncr- r/ r
r r $test_is_AsyncMock_patch_staticmethodS s
z<AsyncPatchDecoratorTest.test_is_AsyncMock_patch_staticmethodc s" t td fdd}| d S )Nr c s | t d S r r: r* r r
r r, [ s zOAsyncPatchDecoratorTest.test_is_AsyncMock_patch_classmethod.<locals>.test_asyncr- r/ r
r r #test_is_AsyncMock_patch_classmethodZ s
z;AsyncPatchDecoratorTest.test_is_AsyncMock_patch_classmethodc sP t t dddt t ddd fdd}t| tt d S )N .async_func return_value.async_func_args c sh | jd |jd tt tt t I d H d tddddI d H d d S )Nr$ r r@ rD r# ) assertEqualZ
_mock_namer; r r r$ )Zfunc_args_mockZ func_mockr r
r r, b s z@AsyncPatchDecoratorTest.test_async_def_patch.<locals>.test_async)r r r r4 r( r2 r) r r/ r
r r test_async_def_patcha s
z,AsyncPatchDecoratorTest.test_async_def_patchN) r r r r0 r9 r< r= r> rH r
r
r
r r' 5 s r' c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
AsyncPatchCMTestc s fdd}| d S )Nc s, t td} t| W 5 Q R X d S Nr )r r. r r( r r) r* r r
r r, t s z>AsyncPatchCMTest.test_is_async_function_cm.<locals>.test_asyncr
r/ r
r r test_is_async_function_cms s z*AsyncPatchCMTest.test_is_async_function_cmc s fdd}| d S )Nc s< t td&} | } t| t| W 5 Q R X d S rJ )r r. r r( r2 r3 r r4 r5 r r
r r, { s z5AsyncPatchCMTest.test_is_async_cm.<locals>.test_asyncr
r/ r
r r test_is_async_cmz s z!AsyncPatchCMTest.test_is_async_cmc s fdd}| d S )Nc s( t td} | t W 5 Q R X d S rJ )r r. r r; r r* r r
r r, s z9AsyncPatchCMTest.test_is_AsyncMock_cm.<locals>.test_asyncr
r/ r
r r test_is_AsyncMock_cm s z%AsyncPatchCMTest.test_is_AsyncMock_cmc s fdd}t | d S )Nc s> t t dt tt W 5 Q R X tt d S )Nr? )r r r r; r r( r2 r) r
r r
r r, s z6AsyncPatchCMTest.test_async_def_cm.<locals>.test_asyncr r4 r/ r
r r test_async_def_cm s z"AsyncPatchCMTest.test_async_def_cmN)r r r rK rL rM rO r
r
r
r rI r s rI c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S )
AsyncMockTestc C s t }| t| d S r )r r( r r) r mockr
r
r test_iscoroutinefunction_default s z.AsyncMockTest.test_iscoroutinefunction_defaultc C s4 dd }t |}| t| | t| d S )Nc s d S r r
r
r
r
r foo z<AsyncMockTest.test_iscoroutinefunction_function.<locals>.foor r( r r) r2 r rT rR r
r
r !test_iscoroutinefunction_function s z/AsyncMockTest.test_iscoroutinefunction_functionc C s: t }| }| t| t| | dt| d S )Nassert_awaited)r r( r2 r3 r r4 ZassertIndirr rR r6 r
r
r test_isawaitable s
z AsyncMockTest.test_isawaitablec C s4 dd }t |}| t| | t| d S )Nc S s d S r r
r
r
r
r rT rU zCAsyncMockTest.test_iscoroutinefunction_normal_function.<locals>.foorV rW r
r
r (test_iscoroutinefunction_normal_function s z6AsyncMockTest.test_iscoroutinefunction_normal_functionc C sD t }t | t }| | t|}| |t j d S r )r Znew_event_loopZset_event_loopFuturestopcloser r; )r ZloopfutrR r
r
r test_future_isfuture s
z"AsyncMockTest.test_future_isfutureN)r r r rS rX r\ r] rb r
r
r
r rP s
rP c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) AsyncAutospecTestc s@ t tdd fdd}t tdd fdd}| | d S )NTZautospecc s | jt | t d S r r; r r r r* r r
r r, s z=AsyncAutospecTest.test_is_AsyncMock_patch.<locals>.test_asyncc s | jt d S r )r; r r r* r r
r test_normal_method s zEAsyncAutospecTest.test_is_AsyncMock_patch.<locals>.test_normal_methodr r8 )r r, rf r
r r r< s
z)AsyncAutospecTest.test_is_AsyncMock_patchc C s& | t ttdd W 5 Q R X d S )NT)instance) assertRaises RuntimeErrorr r r r
r
r test_create_autospec_instance s z/AsyncAutospecTest.test_create_autospec_instancec C s t t d}| t |t d S Nspec)r r r; r )r Zawaitable_mockr
r
r $test_create_autospec_awaitable_class s z6AsyncAutospecTest.test_create_autospec_awaitable_classc s t t}|dddd fdd}| |jd | |j | |jg | t | |
t | |
t | |jd | |jt
dddd | |jt
ddddg | |jdddd |jdddd | d S )Nr@ rD rE rF c s I d H d S r r
r
awaitabler
r main s z4AsyncAutospecTest.test_create_autospec.<locals>.mainr )r r$ rG await_count assertIsNone
await_argsawait_args_listassert_not_awaitedr r4 r( r) iscoroutiner assert_awaited_onceassert_awaited_once_withassert_awaited_withrY )r rn rr r
rp r test_create_autospec s" z&AsyncAutospecTest.test_create_autospecc s fdd}t | d S )Nc sH t t ddd} | dddd} | jt t| t| t
| | j d | j
g | j | |I d H W 5 Q R X | j d | jtdddd | j
tddddg | | jdddd | jdddd | | | j d | j | j
g d S ) NrC Trd r@ rD rE rF r )r r r; rR r r( r r) rx r2 r3 rG rs rv rt ru rw r ry rz r{ rY
reset_mock)r+ rq r r
r r, s, z>AsyncAutospecTest.test_patch_with_autospec.<locals>.test_asyncrN r/ r
r r test_patch_with_autospec s z*AsyncAutospecTest.test_patch_with_autospecN)r r r r< rk ro r| r~ r
r
r
r rc s
rc c @ sd e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dS )
AsyncSpecTestc
sB fdd}t tfD ](} d| || W 5 Q R X qd S )Nc s( | t } |jt |jt d S r )r r; r r r r ) mock_typerR r r
r
inner_test s zCAsyncSpecTest.test_spec_normal_methods_on_class.<locals>.inner_testztest method types with r r subTestr r r r
r r !test_spec_normal_methods_on_class s z/AsyncSpecTest.test_spec_normal_methods_on_classc C s( t t}| |jt | |jt d S r )r r r; r r r rQ r
r
r +test_spec_normal_methods_on_class_with_mock s z9AsyncSpecTest.test_spec_normal_methods_on_class_with_mockc
sD fdd}t ttfD ](} d| || W 5 Q R X qd S )Nc sX | t d} || t t| W 5 Q R X | td} || d S rl r r; assertWarnsRuntimeWarningr( r2 r3 r% r
async_mockZ sync_mockr r
r r s
z8AsyncSpecTest.test_spec_mock_type_kw.<locals>.inner_testztest spec kwarg with r r r r r r
r r test_spec_mock_type_kw s
z$AsyncSpecTest.test_spec_mock_type_kwc
sD fdd}t ttfD ](} d| || W 5 Q R X qd S )Nc sT | t } || t t| W 5 Q R X | t} || d S r r r r r
r r % s z@AsyncSpecTest.test_spec_mock_type_positional.<locals>.inner_testztest spec positional with r r r
r r test_spec_mock_type_positional$ s
z,AsyncSpecTest.test_spec_mock_type_positionalc C s: t td}| |t | }| t| t| d S rl r r% r; r( r2 r3 r r4 r[ r
r
r test_spec_as_normal_kw_AsyncMock3 s
z.AsyncSpecTest.test_spec_as_normal_kw_AsyncMockc C s8 t t}| |t | }| t| t| d S r r r[ r
r
r (test_spec_as_normal_positional_AsyncMock: s
z6AsyncSpecTest.test_spec_as_normal_positional_AsyncMockc s& t jtddd fdd}| d S )Nr Trm c s | t d S r r: r* r r
r r, B s z6AsyncSpecTest.test_spec_async_mock.<locals>.test_asyncr- r/ r
r r test_spec_async_mockA s z"AsyncSpecTest.test_spec_async_mockc s" t tdd fdd}| d S )NTrm c s | t | jt d S r )r; r r r r* r r
r r, I s zIAsyncSpecTest.test_spec_parent_not_async_attribute_is.<locals>.test_asyncrg r/ r
r r 'test_spec_parent_not_async_attribute_isH s
z5AsyncSpecTest.test_spec_parent_not_async_attribute_isc s( t jtdtjd fdd}| d S )Nr rm c s0 | t t| t| d S r )r; r assertFalser2 rx r3 r* r r
r test_async_attributeQ s zFAsyncSpecTest.test_target_async_spec_not.<locals>.test_async_attribute)r r. r r& r! )r r r
r r test_target_async_spec_notP s z(AsyncSpecTest.test_target_async_spec_notc s& t jtdtd fdd}| d S )Nr! rm c s | t d S r r: )Zmock_async_funcr r
r test_attribute_not_async_spec_isZ s zUAsyncSpecTest.test_target_not_async_spec_is.<locals>.test_attribute_not_async_spec_is)r r. r& r )r r r
r r test_target_not_async_spec_isY s z+AsyncSpecTest.test_target_not_async_spec_isc s" t ttd fdd}| d S )Nrm c s | jt | t d S r re )ZMockNormalClassr r
r test_async_attributes_coroutines` s zRAsyncSpecTest.test_spec_async_attributes.<locals>.test_async_attributes_coroutines)r normal_foo_namer )r r r
r r test_spec_async_attributes_ s
z(AsyncSpecTest.test_spec_async_attributesN)r r r r r r r r r r r r r r r
r
r
r r s
r c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
AsyncSpecSetTestc s t jtddd fdd}d S )Nr TZspec_setc s | t d S r r: )r r r
r r, j s z<AsyncSpecSetTest.test_is_AsyncMock_patch.<locals>.test_asyncr- r/ r
r r r< i s z(AsyncSpecSetTest.test_is_AsyncMock_patchc C s, t tjd}| t| | |t d S Nr )r r r r( r r) r; rQ r
r
r test_is_async_AsyncMockn s z(AsyncSpecSetTest.test_is_async_AsyncMockc C sZ t td}| t|j | t|j | |jt | |jt | |t d S r )
r r r( r r) r r r r; r rQ r
r
r test_is_child_AsyncMocks s
z(AsyncSpecSetTest.test_is_child_AsyncMockc C s> t }t dd d|_t|d}| |t W 5 Q R X d S )Nc S s | S r r
xr
r
r <lambda>} rU z=AsyncSpecSetTest.test_magicmock_lambda_spec.<locals>.<lambda>rm mock_func)r r r r. r; )r Zmock_objcmr
r
r test_magicmock_lambda_spec{ s z+AsyncSpecSetTest.test_magicmock_lambda_specN)r r r r< r r r r
r
r
r r h s r c @ st e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd ZdS ) AsyncArgumentsc s2 dd }t |dd}|dI d H }| |d d S )Nc s |d S Nr@ r
)r varr
r
r addition s z6AsyncArguments.test_add_return_value.<locals>.addition
rA r rG )r r rR outputr
r
r test_add_return_value s z$AsyncArguments.test_add_return_valuec s@ dd }t |tdd}| t |dI d H W 5 Q R X d S )Nc s | d S r r
r r
r
r r s z?AsyncArguments.test_add_side_effect_exception.<locals>.additionerrZ side_effectr )r Exceptionri )r r rR r
r
r test_add_side_effect_exception s z-AsyncArguments.test_add_side_effect_exceptionc s0 dd }t |d}|dI d H }| |d d S )Nc s | d S r r
r r
r
r r s z?AsyncArguments.test_add_side_effect_coroutine.<locals>.additionr r r r r rR resultr
r
r test_add_side_effect_coroutine s
z-AsyncArguments.test_add_side_effect_coroutinec s0 dd }t |d}|dI d H }| |d d S )Nc S s | d S r r
r r
r
r r s zEAsyncArguments.test_add_side_effect_normal_function.<locals>.additionr r r r r r
r
r $test_add_side_effect_normal_function s
z3AsyncArguments.test_add_side_effect_normal_functionc sX dddg}t |d}|D ]}| | I d H | q| t}| I d H W 5 Q R X d S )Nr@ rD rE r )r rG ri StopAsyncIteration)r valsrR itemer
r
r test_add_side_effect_iterable s
z,AsyncArguments.test_add_side_effect_iterablec s` G dd dt }d|dg}t|d}| | I d H d | |}| I d H W 5 Q R X d S )Nc @ s e Zd ZdS )zOAsyncArguments.test_add_side_effect_exception_iterable.<locals>.SampleExceptionN)r r r r
r
r
r SampleException s r r@ rT r )r r rG ri )r r r rR r r
r
r 'test_add_side_effect_exception_iterable s
z6AsyncArguments.test_add_side_effect_exception_iterablec s0 t dd}t |d}| I d H }| || d S )Nr rA )r assertIs)r valuerR r r
r
r test_return_value_AsyncMock s
z*AsyncArguments.test_return_value_AsyncMockc s: t }|d t|d}| I d H }| |t j d S )NrA )r r^
set_resultr r; r ra rR r r
r
r test_return_value_awaitable s
z*AsyncArguments.test_return_value_awaitablec s^ t }|d t|gd}| I d H }| |t j | t | I d H W 5 Q R X d S )Nr )r r^ r r r; ri r r r
r
r !test_side_effect_awaitable_values s
z0AsyncArguments.test_side_effect_awaitable_valuesc s0 t dd}t |d}| I d H }| |d d S )Nr rA r r )r ZeffectrR r r
r
r test_side_effect_is_AsyncMock s
z,AsyncArguments.test_side_effect_is_AsyncMockc sR t d fdd}t|d}| I d H }| | | | d S )NFc s d S NTr
r
Zranr r
r inner s z2AsyncArguments.test_wraps_coroutine.<locals>.innerwraps)r r^ r rG rY r( r r rR r r
r r test_wraps_coroutine s
z#AsyncArguments.test_wraps_coroutinec sN dd fdd}t |d}| I d H }| | | | d S )Nr@ Fc s d S r r
r
r r
r r s z8AsyncArguments.test_wraps_normal_function.<locals>.innerr )r rG rY r( r r
r r test_wraps_normal_function s
z)AsyncArguments.test_wraps_normal_functionc sx t }|d}|d}|I d H |I d H |tdtdg | |jtdtdg | |jtdtdg d S )NrD r@ )r assert_has_awaitsr rG rv call_args_list)r r Zmock2Zmock1r
r
r test_await_args_list_order s
z)AsyncArguments.test_await_args_list_orderN)r r r r r r r r r r r r r r r r r
r
r
r r s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) AsyncMagicMethodsc C sB t }| |jt | |jt | |jt | |jt d S r )r r;
__aenter__r __aexit__ __anext__ __aiter__r Zm_mockr
r
r +test_async_magic_methods_return_async_mocks s
z=AsyncMagicMethods.test_async_magic_methods_return_async_mocksc C sB t }| |jt | |jt | |jt | |jt d S r )r r; __enter__r __exit____next____len__r a_mockr
r
r *test_sync_magic_methods_return_magic_mocks s
z<AsyncMagicMethods.test_sync_magic_methods_return_magic_mocksc C s: t }| t|d | t|d | t|d d S )Nr r r )r r( hasattrr r
r
r &test_magicmock_has_async_magic_methods s z8AsyncMagicMethods.test_magicmock_has_async_magic_methodsc C sJ t }| t|d | t|d | t|d | t|d d S )Nr r r r )r r( r r r
r
r %test_asyncmock_has_sync_magic_methods s
z7AsyncMagicMethods.test_asyncmock_has_sync_magic_methodsc C sJ t }| |jt | |jt | t|j | t|j d S r )r r; r r r r( r r) r r
r
r &test_magic_methods_are_async_functions s
z8AsyncMagicMethods.test_magic_methods_are_async_functionsN)r r r r r r r r r
r
r
r r s
r c @ s^ e Zd ZG dd dZG dd dZG dd dZdd Zd d
Zd d Zd
d Z dd Z
dS )AsyncContextManagerTestc @ s e Zd Zdd Zdd ZdS )z/AsyncContextManagerTest.WithAsyncContextManagerc s
d| _ | S r )Zenteredr argskwargsr
r
r r & s z:AsyncContextManagerTest.WithAsyncContextManager.__aenter__c s
d| _ d S r )Zexitedr r
r
r r * s z9AsyncContextManagerTest.WithAsyncContextManager.__aexit__N)r r r r r r
r
r
r WithAsyncContextManager% s r c @ s e Zd Zdd Zdd ZdS )z.AsyncContextManagerTest.WithSyncContextManagerc O s | S r r
r r
r
r r . s z8AsyncContextManagerTest.WithSyncContextManager.__enter__c O s d S r r
r r
r
r r 1 s z7AsyncContextManagerTest.WithSyncContextManager.__exit__N)r r r r r r
r
r
r WithSyncContextManager- s r c @ s e Zd Zdd Zdd ZdS )z&AsyncContextManagerTest.ProductionCodec C s
d | _ d S r )sessionr r
r
r r 6 s z/AsyncContextManagerTest.ProductionCode.__init__c sJ | j d4 I d H &}| I d H }|W 5 Q I d H R S Q I d H R X d S )Nzhttps://python.org)r postjson)r responsevalr
r
r rr 9 s z+AsyncContextManagerTest.ProductionCode.mainN)r r r r rr r
r
r
r ProductionCode4 s r c
sB fdd}t tfD ](} d| || W 5 Q R X qd S )Nc sl }tdd|_| dd}tdd}tddid|_||j_||jj_t |
} |ddi d S )NZ sessionmocknameZmagic_cmr r { rA ) r r r r r r rB r r r4 rr rG )r Zpcr r r r r
r r ? s
zKAsyncContextManagerTest.test_set_return_value_of_aenter.<locals>.inner_testz%test set return value of aenter with r r r
r r test_set_return_value_of_aenter> s z7AsyncContextManagerTest.test_set_return_value_of_aenterc
sB fdd}t tfD ](} d| || W 5 Q R X qd S )Nc sx d }| | fdd}t| } jj jj j j | d S )NFc
s& 4 I d H
} d W 5 Q I d H R X | S r r
r calledZcm_mockr
r use_context_managerT s zqAsyncContextManagerTest.test_mock_supports_async_context_manager.<locals>.inner_test.<locals>.use_context_manager) r r r4 r( r r r rY Z assertIsNot)r r r Z cm_resultr r r r O s
zTAsyncContextManagerTest.test_mock_supports_async_context_manager.<locals>.inner_testz!test context manager magics with r r r
r r (test_mock_supports_async_context_managerN s z@AsyncContextManagerTest.test_mock_supports_async_context_managerc sB | }t| t }| j_ fdd}| t| | d S )Nc s4 4 I d H } | W 5 Q I d H R S Q I d H R X d S r r
r
mock_instancer
r r n s z^AsyncContextManagerTest.test_mock_customize_async_context_manager.<locals>.use_context_manager)r r r. r rB r r r4 )r rh Zexpected_resultr r
r r )test_mock_customize_async_context_managerg s zAAsyncContextManagerTest.test_mock_customize_async_context_managerc sl d d fdd}fdd}| }t||_|_fdd}t| | | d S )NFc s d d S r r
r ) enter_calledr
r enter_coroutinex s ziAsyncContextManagerTest.test_mock_customize_async_context_manager_with_coroutine.<locals>.enter_coroutinec s d d S r r
r ) exit_calledr
r exit_coroutine| s zhAsyncContextManagerTest.test_mock_customize_async_context_manager_with_coroutine.<locals>.exit_coroutinec
s" 4 I d H W 5 Q I d H R X d S r r
r
r r
r r s zmAsyncContextManagerTest.test_mock_customize_async_context_manager_with_coroutine.<locals>.use_context_manager)r r r r r r4 r( )r r r rh r r
)r r r r 8test_mock_customize_async_context_manager_with_coroutinet s
zPAsyncContextManagerTest.test_mock_customize_async_context_manager_with_coroutinec C s@ dd }| }t|}| t t|| W 5 Q R X d S )Nc
s( | 4 I d H t W 5 Q I d H R X d S r ) TypeError)Zcontext_managerr
r
r raise_in s zYAsyncContextManagerTest.test_context_manager_raise_exception_by_default.<locals>.raise_in)r r ri r r r4 )r r rh r r
r
r /test_context_manager_raise_exception_by_default s
zGAsyncContextManagerTest.test_context_manager_raise_exception_by_defaultN) r r r r r r r r r r r r
r
r
r r $ s
r c @ s4 e Zd ZG dd deZdd Zdd Zdd Zd S )
AsyncIteratorTestc @ s$ e Zd Zdd Zdd Zdd ZdS )z#AsyncIteratorTest.WithAsyncIteratorc C s dddg| _ d S )NrT Z NormalFoobaz)itemsr r
r
r r s z,AsyncIteratorTest.WithAsyncIterator.__init__c C s | S r r
r r
r
r r s z-AsyncIteratorTest.WithAsyncIterator.__aiter__c s* z | j W S tk
r Y nX td S r )r pop
IndexErrorr r r
r
r r s
z-AsyncIteratorTest.WithAsyncIterator.__anext__N)r r r r r r r
r
r
r WithAsyncIterator s r c sF t dd dddg j_ fdd}t| }| |dddg d S )NZtesterr r@ rD rE c s dd 2 I d H S )Nc s g | z3 d H W }|q6 S r r
).0ir
r
r
<listcomp> s zOAsyncIteratorTest.test_aiter_set_return_value.<locals>.main.<locals>.<listcomp>r
r
Z mock_iterr
r rr s z;AsyncIteratorTest.test_aiter_set_return_value.<locals>.main)r r rB r r4 rG )r rr r r
r r test_aiter_set_return_value s
z-AsyncIteratorTest.test_aiter_set_return_valuec
sB fdd}t tfD ](} d| || W 5 Q R X qd S )Nc s\ }| |} t|j t|j t|j t|j d S r )r r r r) r r( r )r rh r r r
r r s zIAsyncIteratorTest.test_mock_aiter_and_anext_asyncmock.<locals>.inner_testz%test aiter and anext corourtine with r r r
r r #test_mock_aiter_and_anext_asyncmock s z5AsyncIteratorTest.test_mock_aiter_and_anext_asyncmockc
s dd dddg fdd} fdd } fd
d }t tfD ]p}d | || W 5 Q R X d
| || W 5 Q R X d| || W 5 Q R X qHd S )Nc s& g }| 2 z3 d H W }| | q6 |S r )append)iteratorZ accumulatorr r
r
r iterate s z6AsyncIteratorTest.test_mock_async_for.<locals>.iterateZFOOZBARZBAZc s& | }t |g d S r )r rG r r4 r r )r r r
r test_default s z;AsyncIteratorTest.test_mock_async_for.<locals>.test_defaultc s6 | } d d |j_t| d S r )r r rB rG r r4 r Zexpectedr r r
r test_set_return_value s zDAsyncIteratorTest.test_mock_async_for.<locals>.test_set_return_valuec s: | }t d d |j_t| d S r )r iterr rB rG r r4 r r r
r test_set_return_value_iter s zIAsyncIteratorTest.test_mock_async_for.<locals>.test_set_return_value_iterzdefault value with zset return_value with z set return_value iterator with r )r r r r r r
r r test_mock_async_for s
z%AsyncIteratorTest.test_mock_async_forN)r r r r. r r r r r
r
r
r r s r c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/S )0AsyncMockAssertc C s t | _d S r )r rR r r
r
r setUp s zAsyncMockAssert.setUpc s | j ||I d H d S r )rR r r
r
r _runnable_test s z AsyncMockAssert._runnable_testc s
|I d H S r r
)r coroutiner
r
r _await_coroutine s z AsyncMockAssert._await_coroutinec C s t t}| t | W 5 Q R X | t|j |j |j |j
| t |
W 5 Q R X | t |j
W 5 Q R X d S r )r r r r r r( r r)
assert_calledassert_called_onceassert_called_once_withri AssertionErrorrY rQ r
r
r "test_assert_called_but_not_awaited s
z2AsyncMockAssert.test_assert_called_but_not_awaitedc C s t t}| }|j |j |j | t |j W 5 Q R X t
| | |j |j |j |j
d S r )r r r r" r# r$ ri r% rY r r4 r! ry rz )r rR Zmock_coroutiner
r
r test_assert_called_then_awaited s
z/AsyncMockAssert.test_assert_called_then_awaitedc C sf | t | j W 5 Q R X | t | j W 5 Q R X t| | j | j d S r )
ri r% rR rY r" r r4 r r# ry r r
r
r +test_assert_called_and_awaited_at_same_time
s
z;AsyncMockAssert.test_assert_called_and_awaited_at_same_timec C s t t}| }| t | W 5 Q R X | t |j W 5 Q R X |j t
| | |j |j d S r )
r r r r r ri r% rY r" r r4 r! ry r rR r r
r
r )test_assert_called_twice_and_awaited_once s
z9AsyncMockAssert.test_assert_called_twice_and_awaited_oncec C s^ t t}| }|j t| | | t t| | W 5 Q R X |j d S r )
r r r r# r r4 r! ri rj rY r) r
r
r )test_assert_called_once_and_awaited_twice& s
z9AsyncMockAssert.test_assert_called_once_and_awaited_twicec C s | t | j W 5 Q R X | t | j W 5 Q R X | t t| | j W 5 Q R X | t | j W 5 Q R X | t | j W 5 Q R X d S r ) ri r% rR rY r" r r r4 r! r r
r
r "test_assert_awaited_but_not_called0 s z2AsyncMockAssert.test_assert_awaited_but_not_calledc C s\ t dg}| t | d W 5 Q R X | j| | t | j| W 5 Q R X d S NrT )r r r rR Zassert_has_callsri r% r )r Zkallsr
r
r test_assert_has_calls_not_awaits> s
z0AsyncMockAssert.test_assert_has_calls_not_awaitsc C s | t | W 5 Q R X ddi fg}| | jj| | t | d | d W 5 Q R X t tdtdg}| | jj| d S )N r
rT r )r r rR rG
mock_callsr )r kalls_empty
mock_kallsr
r
r 0test_assert_has_mock_calls_on_async_mock_no_specG s
z@AsyncMockAssert.test_assert_has_mock_calls_on_async_mock_no_specc C s t t}| t | W 5 Q R X ddi fg}| |jj| | |jt g | t |jdddddd W 5 Q R X t tddddddg}t tjddddddg}| |jj| | |j| d S ) Nr/ r
r@ rD rE r )r! r" )r r r r r rG r0 r )r Z a_class_mockr1 Z method_kallsr2 r
r
r 2test_assert_has_mock_calls_on_async_mock_with_specU s zBAsyncMockAssert.test_assert_has_mock_calls_on_async_mock_with_specc C s | t* | jjdd d | jjjdtjd W 5 Q R X | | jjdddd ifdd d
tjifgd | | jjjdd d
tjifgd d S ) NrE )fishr )cake something)rE r6 zsomething_else.something)r r7 z#method calls not recorded correctly) r r rR r8 Zsomething_elser ZCakerG method_callsr r
r
r test_async_method_calls_recordedf s
z0AsyncMockAssert.test_async_method_calls_recordedc s fdd}| j t& dd j dd W 5 Q R X j | j tt} t | |jddd W 5 Q R X | || d S )Nc s@ d}|D ]2}t | |} |t |t |g qd S )N)r r9 r0 )getattrr; r
listrG )rR namesr attrr r
r assert_attrsv s
z:AsyncMockAssert.test_async_arg_lists.<locals>.assert_attrsr@ rD rE )r! )rR r r r} r r r )r r? r r
r r test_async_arg_listsu s
z$AsyncMockAssert.test_async_arg_listsc C s< | t | j W 5 Q R X t| | j d S r )ri r% rR rY r r4 r r r
r
r test_assert_awaited s z#AsyncMockAssert.test_assert_awaitedc C sj | t | j W 5 Q R X t| | j t| | t | j W 5 Q R X d S r )ri r% rR ry r r4 r r r
r
r test_assert_awaited_once s
z(AsyncMockAssert.test_assert_awaited_oncec C s d}| t| | jd W 5 Q R X t| d}| t| | jd W 5 Q R X t| d | jd t| d | t | jd W 5 Q R X d S )Nz Not awaitedrT zexpected await not found
SomethingElse)assertRaisesRegexr% rR r{ r r4 r ri )r msgr
r
r test_assert_awaited_with s z(AsyncMockAssert.test_assert_awaited_withc C st | t | jd W 5 Q R X t| d | jd t| d | t | jd W 5 Q R X d S r- )ri r% rR rz r r4 r r r
r
r test_assert_awaited_once_with s z-AsyncMockAssert.test_assert_awaited_once_withc C s | t | jd W 5 Q R X t| d | t | jd W 5 Q R X t| d | jd t| d | jd d S )NrT r rC )ri r% rR Zassert_any_awaitr r4 r r r
r
r test_assert_any_wait s z$AsyncMockAssert.test_assert_any_waitc C s t dt dg}| t}| j| W 5 Q R X | t|jjd t
| d | t | j| W 5 Q R X t
| d | t | j| W 5 Q R X t
| d | j| t
| d | j| d S )NrT r r@ rC ) r ri r% rR r rG len exceptionr r r4 r )r callsr r
r
r test_assert_has_awaits_no_order s z/AsyncMockAssert.test_assert_has_awaits_no_orderc C s t dt dg}| t | jj|dd W 5 Q R X t| d | t | jj|dd W 5 Q R X t| d | t | jj|dd W 5 Q R X t| d | jj|dd t| d | jj|dd d S )NrT r T)Z any_orderZbamfZqux)r ri r% rR r r r4 r )r rK r
r
r test_assert_has_awaits_ordered s z.AsyncMockAssert.test_assert_has_awaits_orderedc C s< | j t| | t | j W 5 Q R X d S r )rR rw r r4 r ri r% r r
r
r test_assert_not_awaited s
z'AsyncMockAssert.test_assert_not_awaitedc C s d dd}t |d| _t| d | tdt d}| j
t g W 5 Q R X | |j
j | tdt d }| j
t t ddg W 5 Q R X | |j
jt d S )
Nc s d S r r
r r
r
r f rU zIAsyncMockAssert.test_assert_has_awaits_not_matching_spec_error.<locals>.frm r@ z^{}$z6Awaits not found.
Expected: [call()]
Actual: [call(1)]zError processing expected awaits.
Errors: [None, TypeError('too many positional arguments')]
Expected: [call(), call(1, 2)]
Actual: [call(1)]rD )N)r rR r r4 r rD r% formatreescaper r rt rJ __cause__r; r )r rO r r
r
r .test_assert_has_awaits_not_matching_spec_error s.
"z>AsyncMockAssert.test_assert_has_awaits_not_matching_spec_errorN)r r r r r r! r& r' r( r* r+ r, r. r3 r5 r: r@ rA rB rF rG rH rL rM rN rT r
r
r
r r s.
r )&r r2 rQ ZunittestZ
unittest.mockr r r r r r r r r
r r r r r$ r% r. r& r r8 r ZTestCaser' rI rP rc r r ZIsolatedAsyncioTestCaser r r r r r
r
r
r <module> s0 ,
=!"Pc{&uL
Mini Shell