CasperSecurity
�
�ЦiY � �F � d dl Z d dlZd dlZd dlZddlmZ G d� de� Zy)� N� )�TestCasec � � � e Zd Zd� fd� Zd� Zd� Zd� Zd� Zd� Zd� Z d� Z
d � Z d
� Z d � Z
d � Zd
� Zd� fd� Z� fd�Zd� Z� xZS )�IsolatedAsyncioTestCasec �d �� t �| � |� d | _ t j � | _ y �N)�super�__init__�_asyncioRunner� contextvars� copy_context�_asyncioTestContext)�self�
methodName� __class__s ��*/usr/lib/python3.12/unittest/async_case.pyr
z IsolatedAsyncioTestCase.__init__# s) �� �
����$� "���#.�#;�#;�#=�� � c � � K � y �wr � �r s r �
asyncSetUpz"IsolatedAsyncioTestCase.asyncSetUp( � � �� � �� �c � � K � y �wr r r s r �
asyncTearDownz%IsolatedAsyncioTestCase.asyncTearDown+ r r c �0 � | j |g|��i |�� y r )�
addCleanup�r �func�args�kwargss r �addAsyncCleanupz'IsolatedAsyncioTestCase.addAsyncCleanup. s � � ����$� �� �1�&�1r c � � K � t |� } |j }|j } ||� � d{ ��� }| j ||ddd� |S # t $ r( t d|j
� d|j � d�� d�w xY w7 �O�w)z�Enters the supplied asynchronous context manager.
If successful, also adds its __aexit__ method as a cleanup
function and returns the result of the __aenter__ method.
�'�.zC' object does not support the asynchronous context manager protocolN)�type�
__aenter__� __aexit__�AttributeError� TypeError�
__module__� __qualname__r" )r �cm�cls�enter�exit�results r �enterAsyncContextz)IsolatedAsyncioTestCase.enterAsyncContext= s� � �� � �2�h�� '��N�N�E��=�=�D�
�R�y� �� � � �T�2�t�T�4�8��
��
� '��a���� /�q��1A�1A�0B� CT� U� �"&�
'� '�� !�s% � B�A � B�B�B�1A?�?Bc � � | j j � | j j | j � | j | j � y r )r �get_loopr �run�setUp�
_callAsyncr r s r �
_callSetUpz"IsolatedAsyncioTestCase._callSetUpQ s> � �
���$�$�&� � � �$�$�T�Z�Z�0� ������(r c �h � | j |� �!t j d|� d�t d�� y y )NzFIt is deprecated to return a value that is not None from a test case (�)� )�
stacklevel)�_callMaybeAsync�warnings�warn�DeprecationWarning)r �methods r �_callTestMethodz'IsolatedAsyncioTestCase._callTestMethodY s< � � � � �� '� 3� �M�M� (�(.�x�q�2�3E�RS�
U� 4r c � � | j | j � | j j | j � y r )r7 r r r5 �tearDownr s r �
_callTearDownz%IsolatedAsyncioTestCase._callTearDown^ s, � � ����*�*�+� � � �$�$�T�]�]�3r c �0 � | j |g|��i |�� y r )r= )r �functionr r! s r � _callCleanupz$IsolatedAsyncioTestCase._callCleanupb s � � �� � �X�7��7��7r c � � | j �J d� �t j |� s
J |�d�� �| j j ||i |��| j �� S )N�!asyncio runner is not initializedz is not an async function��context�r �inspect�iscoroutinefunctionr5 r r s r r7 z"IsolatedAsyncioTestCase._callAsynce so � ��"�"�.�S�0S�S�.��*�*�4�0�V�T�H�<U�2V�V�0��"�"�&�&� �$� !�&� !��,�,� '�
�
r c �� � | j �J d� �t j |� r-| j j ||i |��| j �� S | j j |g|��i |��S )NrJ rK rM r s r r= z'IsolatedAsyncioTestCase._callMaybeAsyncm s� � ��"�"�.�S�0S�S�.� � &� &�t� ,��&�&�*�*��d�%�f�%� �0�0� +� �
�
0�4�+�+�/�/��F�t�F�v�F� Fr c �d � | j �J d� �t j d�� }|| _ y )Nz%asyncio runner is already initializedT)�debug)r �asyncio�Runner�r �runners r �_setupAsyncioRunnerz+IsolatedAsyncioTestCase._setupAsyncioRunnerw s0 � ��"�"�*�S�,S�S�*�� � �d�+�� $��r c �< � | j }|j � y r )r �closerU s r �_tearDownAsyncioRunnerz.IsolatedAsyncioTestCase._tearDownAsyncioRunner| s � ��$�$��� � �r c � �� | j � t �| � |� | j � S # | j � w xY wr )rW r r5 rZ )r r1 r s �r r5 zIsolatedAsyncioTestCase.run� s9 �� � � � �"� *��7�;�v�&� � '� '� )��D� '� '� )�s �2 �Ac �b �� | j � t �| � � | j � y r )rW r rR rZ )r r s �r rR z IsolatedAsyncioTestCase.debug� s$ �� � � � �"�
��
�� �#�#�%r c �> � | j �| j � y y r )r rZ r s r �__del__z IsolatedAsyncioTestCase.__del__� s � � � � � *� � '� '� )� +r )�runTestr )�__name__r+ r, r
r r r" r2 r8 rB rE rH r7 r= rW rZ r5 rR r^ �
__classcell__)r s @r r r sU �� �4>�
�
�
2� �()�U�
4�8�
�G�%�
�*�&�
*r r )rS r rN r> �caser r r r r �<module>rc s" �� � � � � �E*�h� E*r