File: //home/arjun/projects/env/lib64/python3.10/site-packages/mock/__pycache__/backports.cpython-310.pyc
o
we�
� @ s� d dl Z e jdd� dk rAd dlZd dlZd dlmZ d dlmZmZm Z d dl
m Z dd� Z d d
� Z
d d � ZG d
d� de �ZdS d dlmZ d dl
mZ dS )� N� )� � )�
_is_coroutine)�ismethod�
isfunction� CO_COROUTINE)�TestCasec C s" t | tj�r| j} t | tj�s| S �N)�
isinstance� functools�partial�func)r � r �G/home/arjun/projects/env/lib/python3.10/site-packages/mock/backports.py�_unwrap_partial s �r c C s: t | �r | j} t | �st| �} t| �sdS t| jj|@ �S )z�Return true if ``f`` is a function (or a method or functools.partial
wrapper wrapping a function) whose code object has the given ``flag``
set in its flags.F)r �__func__r r �bool�__code__�co_flags)�f�flagr r r �_has_code_flag s �r c C s t | t�p t| dd�tu S )z}Return true if the object is a coroutine function.
Coroutine functions are defined with "async def" syntax.
r N)r r �getattrr )�objr r r �iscoroutinefunction s
�r c sD e Zd Zd
� fdd� Zdd� Zdd� Zdd � Zd� fd d � Z� ZS )�IsolatedAsyncioTestCase�runTestc s t � �|� d | _d | _d S r
)�super�__init__�_asyncioTestLoop�_asyncioCallsQueue)�self�
methodName�� __class__r r r ( s
z IsolatedAsyncioTestCase.__init__c � s@ �t �� | _}|�d � |�� I d H }|�� |d u s J �qr
)�asyncio�Queuer! �
set_result�get� task_done)r" �fut�queue�queryr r r �_asyncioLoopRunner- s �
�z*IsolatedAsyncioTestCase._asyncioLoopRunnerc C sX | j d u sJ �t�� }t�|� |�d� || _ |�� }|�| �|��| _|� |� d S )NT)
r r& �new_event_loop�set_event_loop� set_debug�
create_future� create_taskr. �_asyncioCallsTask�run_until_complete)r" �loopr+ r r r �_setupAsyncioLoop5 s
z)IsolatedAsyncioTestCase._setupAsyncioLoopc C sr | j d usJ �| j }d | _ | j�d � |�| j�� � z|�|�� � W t�d � |�� d S t�d � |�� w r
) r r! �
put_nowaitr5 �join�shutdown_asyncgensr& r0 �close)r" r6 r r r �_tearDownAsyncioLoop? s
�
z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s* | � � z t� �|�W | �� S | �� w r
)r7 r �runr< )r" �resultr$ r r r= M s zIsolatedAsyncioTestCase.run)r r
) �__name__�
__module__� __qualname__r r. r7 r< r= �
__classcell__r r r$ r r & s
r )r )r )�sys� version_infor& r �asyncio.coroutinesr �inspectr r r �unittestr r r r r r r r r �<module> s 1