← Back
Editing: async_case.cpython-312.pyc
� �֦ik � �J � d dl Z d dlZd dlZd dlZddlmZ dZ G d� de� Zy)� N� )�TestCaseTc � � � 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/lib64/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._callSetUpR 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._callTestMethodZ 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._callCleanupc 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._callAsyncf 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._callMaybeAsyncn 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._setupAsyncioRunnerx 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 � __unittestr r r r �<module>rd s' �� � � � � � � �E*�h� E*r
Save File
Cancel