XylotrechusZ
�
Ȓ�g� � �r � d dl mZ d dlmZ d dlZddlmZ ddlmZ dZ e� Z
d � Z d
� Z d � Z
d � Zd
� Zd� Zy)� )� annotations)�wrapsN� )�config� )�
_AsyncUtilTc �, � t j � y)z called when the test finishesN)� _async_util�close� � �K/opt/hc_python/lib64/python3.12/site-packages/sqlalchemy/testing/asyncio.py� _shutdownr # s � ����r
c �4 � t j | g|��i |��S �N)r
�run��fn�args�kwargss r �_run_coroutine_functionr ( s � � �?�?�2� /�� /�� /�/r
c �P � t s | |i |��S t j | g|��i |��S )a Run a function in an asyncio loop unconditionally.
This function is used for provisioning features like
testing a database connection for server info.
Note that for blocking IO database drivers, this means they block the
event loop.
)�ENABLE_ASYNCIOr
�run_in_greenletr s r �
_assume_asyncr , s2 � � ��4�"�6�"�"� � &� &�r� ;�D� ;�F� ;�;r
c � � t s | |i |��S t j rt j | g|��i |��S | |i |��S )a� Run a function in an asyncio loop if any current drivers might need it.
This function is used for provisioning features that take
place outside of a specific database driver being selected, so if the
current driver that happens to be used for the provisioning operation
is an async driver, it will run in asyncio and not fail.
Note that for blocking IO database drivers, this means they block the
event loop.
)r r � any_asyncr
r r s r �_maybe_async_provisioningr = sK � � ��4�"�6�"�"�
����*�*�2�?��?��?�?��4�"�6�"�"r
c � � t s | |i |��S t j j }|rt j
| g|��i |��S | |i |��S )z�Run a function in an asyncio loop if the current selected driver is
async.
This function is used for test setup/teardown and tests themselves
where the current DB driver is known.
)r r �_current�is_asyncr
r )r r r r! s r � _maybe_asyncr" R sT � � ��4�"�6�"�"����'�'�H���*�*�2�?��?��?�?��4�"�6�"�"r
c � � ��� t j � � r&t � ��fd��t � � ��� fd�� }|S t � � � fd�� }|S )z�Apply the _maybe_async function to an existing function and return
as a wrapped callable, supporting generator functions as well.
This is currently used for pytest fixtures that support generator use.
c �> �� t | � S # t $ r �cY S w xY wr )�next�
StopIteration)�gen�_stops �r � call_nextz'_maybe_async_wrapper.<locals>.call_nextq s& �� �
��C�y� �� �
� � �
�s �
� � c ? �J �K � �| i |��} t �|� }|�u ry |�� ��wr �r" )r r r' �valuer( r) r s ���r � wrap_fixturez*_maybe_async_wrapper.<locals>.wrap_fixturex s9 �� �� ��d�%�f�%�C��$�Y��4���E�>��� � �s � #c �" �� t �g| ��i |��S r r+ )r r r s �r r- z*_maybe_async_wrapper.<locals>.wrap_fixture� s �� � ��4�T�4�V�4� 4r
)�inspect�isgeneratorfunction�objectr )r r- r( r) s ` @@r �_maybe_async_wrapperr2 f s_ �� � �"�"�2�&���� �
�r�� �
� � ��
�r�� 5�
� 5� �r
)�
__future__r � functoolsr r/ � r �util.concurrencyr r r
r r r r r" r2 r r
r �<module>r7 sB ��* #� � � � )� ���l� ��
0�<�"#�*#�(!r