File: //opt/python38/lib/python3.8/unittest/test/testmock/__pycache__/testasync.cpython-38.pyc
U
�i�_ב � @ sv d dl Z d dlZd dlZd dlZd dlmZmZmZmZm Z m
Z
m Z m Z m
Z
dd� ZG dd� d�ZG dd� d�Zd d
� Zd d � Zd
d� ZG dd� de�Ze� d�Ze� d�ZG dd� dej�ZG dd� dej�ZG dd� dej�Z G dd� dej�Z G dd � d ej�Z G d d � d ej�Z G d d � d ej �Z!G d!d"� d"ej�Z"G d#d$� d$ej�Z#G d%d&� d&ej�Z$G d'd(� d(ej�Z%dS ))� N) �ANY�call� AsyncMock�patch� MagicMock�Mock�create_autospec�sentinel� _CallListc C s t �d � d S �N)�asyncioZset_event_loop_policy� r
r
�?/opt/python38/lib/python3.8/unittest/test/testmock/testasync.py�tearDownModule
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
)�a�b�cr
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_async�r �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 �run�r+ �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� d�dd�t t� d�dd�� fdd���}t�|� � � �t�t�� d S )N� .async_func� �� return_value�.async_func_args� c � sh � � | jd� � � |jd� � �tt� � �tt� � � t� I d H d� � � tdddd�I 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� d�t� �� � �tt� W 5 Q R X � �t�t�� d S )Nr? )r r r r; r r( r2 r) r
r r
r r, � s z6AsyncPatchCMTest.test_async_def_cm.<locals>.test_async�r 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>.foo�r 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 )N�assert_awaited)r r( r2 r3 r r4 ZassertIn�dir�r 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_loop�Future�stop�closer r; )r Zloop�futrR 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 )NT�Zautospecc 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_method�r 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 �N��spec)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
dddd�g� |�� |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_args�await_args_list�assert_not_awaitedr r4 r( r) � iscoroutiner �assert_awaited_once�assert_awaited_once_with�assert_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� d�dd���} | dddd�}� �| jt� � �t�| �� � �t�|�� � �t �
|�� � � | j d� � � | j
g � � �| j� | �� |I d H W 5 Q R X � � | j d� � � | jtdddd�� � � | j
tdddd�g� | �� | 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 �subTest�r 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; � assertWarns�RuntimeWarningr( 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 T�Zspec_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_obj�cmr
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�}|d�I 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 |td�d�}| �t�� |d�I 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>.addition�err�Z 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�}|d�I 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�}|d�I 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 �item�er
r
r � test_add_side_effect_iterable� s
z,AsyncArguments.test_add_side_effect_iterablec � s` G dd� dt �}d|d�g}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>.inner��wraps)r r^ r rG rY r( �r r� rR r� r
r� r �test_wraps_coroutine� s
z#AsyncArguments.test_wraps_coroutinec � sN d�d� � �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 |�td�td�g� | �|jtd�td�g� | �|jtd�td�g� 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� d�ZG dd� d�ZG dd� d�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�AsyncContextManagerTestc @ s e Zd Zdd� Zdd� ZdS )z/AsyncContextManagerTest.WithAsyncContextManagerc � s
d| _ | S r� )Zentered�r �args�kwargsr
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 �d�4 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� �post�json)r �response�valr
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 sessionmock��nameZmagic_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� de�Zdd� Zdd� Zdd� Zd S )
�AsyncIteratorTestc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z#AsyncIteratorTest.WithAsyncIteratorc C s dddg| _ d S )NrT Z NormalFoo�baz)�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 t�d 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
)�.0�ir
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 )0�AsyncMockAssertc 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_called�assert_called_once�assert_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 d�g}| �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� td�td�g}| �| 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� tdddddd�g}t�� tjdddddd�g}| �|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|�� | j�d� W 5 Q R X t�| �� � d}| � t|�� | j�d� W 5 Q R X t�| �d�� | j�d� t�| �d�� | �t�� | j�d� 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�� | j�d� W 5 Q R X t�| �d�� | j�d� t�| �d�� | � t�� | j�d� 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�� | j�d� W 5 Q R X t�| �d�� | � t�� | j�d� W 5 Q R X t�| �d�� | j�d� t�| �d�� | j�d� 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 d�t d�g}| �t��}| j�|� W 5 Q R X | �t|jj�d� 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 d�t d�g}| �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�� | �td�t� d����}| j�
t � g� W 5 Q R X | � |j
j� | �td�t� d���� }| j�
t � t dd�g� 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)]z�Error 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% �format�re�escaper� 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