File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_runners.cpython-310.pyc
o
�i� � @ s| d dl Z d dlZd dlmZ d dlmZ dd� ZG dd� de j�ZG dd � d ej �Z
G d
d � d e
�Z e d kr<e�
� dS dS )
� N)�mock)�utilsc C s t �d � d S �N)�asyncio�set_event_loop_policy� r r �;/usr/local/lib/python3.10/test/test_asyncio/test_runners.py�tearDownModule s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestPolicyc C s || _ d | _d S r )� loop_factory�loop)�selfr r r r �__init__ s
zTestPolicy.__init__c C s t �r )� RuntimeError�r
r r r �get_event_loop s zTestPolicy.get_event_loopc C s | � � S r )r r r r r �new_event_loop s zTestPolicy.new_event_loopc C s |d ur || _ d S d S r �r )r
r r r r �set_event_loop s
�zTestPolicy.set_event_loopN)�__name__�
__module__� __qualname__r r r r r r r r r
s
r
c s0 e Zd Zdd� Z� fdd�Z� fdd�Z� ZS )�BaseTestc sB t �� � t�� � _t�� � _d� jj_d� _� fdd�}|� _ � S )Nr Fc � s �d� _ d S �NT)�shutdown_ag_runr r r r �shutdown_asyncgens) s �
z-BaseTest.new_loop.<locals>.shutdown_asyncgens)
r Z
BaseEventLoopr �MockZ_process_events� _selector�selectZ return_valuer r )r
r r r r �new_loop" s
zBaseTest.new_loopc s" t � �� t| j�}t�|� d S r )�super�setUpr
r r r �r
Zpolicy�� __class__r r r! / s
zBaseTest.setUpc sH t �� }|jd ur| �|j�� � | �|jj� t �d � t� �� d S r ) r Zget_event_loop_policyr �
assertTrueZ is_closedr r r �tearDownr" r# r r r& 5 s
zBaseTest.tearDown)r r r r r! r& �
__classcell__r r r# r r s
r c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�RunTestsc C s dd� }| � t�|� �d� d S )Nc � s �t �d�I d H dS )Nr �* �r �sleepr r r r �mainB s �z.RunTests.test_asyncio_run_return.<locals>.mainr) )� assertEqualr �run�r
r, r r r �test_asyncio_run_returnA s z RunTests.test_asyncio_run_returnc C sF dd� }| � td�� t�|� � W d � d S 1 s w Y d S )Nc � s �t �d�I d H td��)Nr �spam)r r+ �
ValueErrorr r r r r, I s �z.RunTests.test_asyncio_run_raises.<locals>.mainr1 )�assertRaisesRegexr2 r r. r/ r r r �test_asyncio_run_raisesH s "�z RunTests.test_asyncio_run_raisesc
C sx ddd� hD ]3}| j |d��# | �td��
t�|� W d � n1 s%w Y W d � n1 s4w Y qd S )N� c S s d S r r r r r r �<lambda>Q � z5RunTests.test_asyncio_run_only_coro.<locals>.<lambda>)�objza coroutine was expected)ZsubTestr3 r2 r r. )r
�or r r �test_asyncio_run_only_coroP s �� �� ��z#RunTests.test_asyncio_run_only_coroc s� � fdd�}t �|d�� t j|d�dd� t�ddd� �� t �|d�� t j|d�dd� W d � d S 1 s:w Y d S ) Nc � s �t �� }� �|�� | � d S r )r r ZassertIsZ get_debug)Zexpectedr r r r r, X s �z-RunTests.test_asyncio_run_debug.<locals>.mainFT)�debugz!asyncio.coroutines._is_debug_modec S s dS r r r r r r r6 ^ r7 z1RunTests.test_asyncio_run_debug.<locals>.<lambda>)r r. r Zpatchr/ r r r �test_asyncio_run_debugW s "�z RunTests.test_asyncio_run_debugc sJ � fdd�� | � td�� t�� � � W d � d S 1 s w Y d S )Nc � s, �� � } z t �| � W | �� d S | �� w r )r r. �close)�coro�r, r r r, c s
� z9RunTests.test_asyncio_run_from_running_loop.<locals>.mainz cannot be called from a running)r3 r r r. r r r? r �"test_asyncio_run_from_running_loopb s �"�z+RunTests.test_asyncio_run_from_running_loopc s@ d �dd� � � �fdd�}| � t�|� �d� | ���� � d S )Nc � s �t �d�I d H d S )N皙�����?r* r r r r �leftoverq s �zARunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.leftoverc � s �t �� � ��dS �N�{ )r � create_taskr �rB �lo_taskr r r, t s � z=RunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.mainrD )r- r r. r% �doner/ r rF r �&test_asyncio_run_cancels_hanging_tasksn s
z/RunTests.test_asyncio_run_cancels_hanging_tasksc sh d �t �� � dd� �� ��fdd�}| �t�|� �d� | ���� � � �t� d��t�
t �d�� d S )Nc � s8 �z t �d�I d H W d S t jy dd Y d S w )NrA r5 r )r r+ �CancelledErrorr r r r rB � s ��zHRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.leftoverc � s �t �� } � | _t ��� ��dS rC )r �get_running_loop�call_exception_handlerrE r �Zcall_exc_handler_mockrB rG r r r, � s
� zDRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.mainrD zasyncio.run.*shutdown)�messageZtask� exception) r r r- r r. r% rH Zassert_called_with�
test_utilsZ MockPatternZMockInstanceOf�ZeroDivisionErrorr/ r rM r �-test_asyncio_run_reports_hanging_tasks_errors| s �z6RunTests.test_asyncio_run_reports_hanging_tasks_errorsc s� d �d �G dd� dt �� dd� ���fdd��� ��fdd�}| �� �� t�|� � W d � n1 s5w Y | ���� � | ��j� | ��j � d S ) Nc @ s e Zd ZdS )zSRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.FancyExitN)r r r r r r r � FancyExit� s rS c S s � dV t �d�I d H q)NTr5 r* r r r r �fidget� s
��zPRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.fidgetc � sB �� � �z
�2 z3 d H W } q6 W d S t jy dd Y d S w )Nr5 r )r rJ )Zthe_meaning_of_life)rT �spinnerr r �spin� s ���zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.spinc � s$ �t �� } t�� | _t ��� ��� �r )r rK r r rL rE r )rS �lazyboyrV r r r, � s
�
zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.main)
� ExceptionZ assertRaisesr r. r% rH Z assertIsNone�ag_frameZ assertFalse�
ag_runningr/ r )rS rT rW rV rU r �7test_asyncio_run_closes_gens_after_hanging_tasks_errors� s � z@RunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errorsN) r r r r0 r4 r: r< r@ rI rR r[ r r r r r( ? s r( �__main__)r Zunittestr Ztest.test_asyncior rP r ZAbstractEventLoopPolicyr
ZTestCaser r( r r, r r r r �<module> s ~ �