File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_runners.cpython-38.pyc
U
/D�g: � @ s` d dl Z d dlZd dlmZ ddlmZ G dd� de j�ZG dd� dej�Z G d d
� d
e �Z
dS ) � N)�mock� )�utilsc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestPolicyc C s || _ d | _d S �N)� loop_factory�loop)�selfr � r
�=/opt/cppython/lib/python3.8/test/test_asyncio/test_runners.py�__init__
s zTestPolicy.__init__c C s t �d S 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 k r|| _ 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)
�asyncioZ
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 �set_event_loop_policy�r Zpolicy�� __class__r
r r + s
zBaseTest.setUpc sH t �� }|jd k r0| �|j�� � | �|jj� t �d � t� �� d S r ) r Zget_event_loop_policyr �
assertTrueZ is_closedr r! r �tearDownr" r# r
r r& 1 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 �main> s z.RunTests.test_asyncio_run_return.<locals>.mainr) )� assertEqualr �run�r r, r
r
r �test_asyncio_run_return= s z RunTests.test_asyncio_run_returnc C s0 dd� }| � td�� t�|� � W 5 Q R X d S )Nc � s t �d�I d H td��d S )Nr �spam)r r+ �
ValueErrorr
r
r
r r, E s z.RunTests.test_asyncio_run_raises.<locals>.mainr1 )�assertRaisesRegexr2 r r. r/ r
r
r �test_asyncio_run_raisesD s z RunTests.test_asyncio_run_raisesc C sP ddd� hD ]>}| j |d��( | �td�� t�|� W 5 Q R X W 5 Q R X q d S )Nr c S s d S r r
r
r
r
r �<lambda>M � z5RunTests.test_asyncio_run_only_coro.<locals>.<lambda>)�objza coroutine was expected)ZsubTestr3 r2 r r. )r �or
r
r �test_asyncio_run_only_coroL s ��z#RunTests.test_asyncio_run_only_coroc sl � fdd�}t �|d�� t j|d�dd� t�ddd� ��& t �|d�� t j|d�dd� W 5 Q R X d S ) Nc � s t �� }� �|�� | � d S r )r r ZassertIsZ get_debug)Zexpectedr r r
r r, T s z-RunTests.test_asyncio_run_debug.<locals>.mainFT)�debugz!asyncio.coroutines._is_debug_modec S s dS r r
r
r
r
r r5 Z r6 z1RunTests.test_asyncio_run_debug.<locals>.<lambda>)r r. r Zpatchr/ r
r r �test_asyncio_run_debugS s z RunTests.test_asyncio_run_debugc s4 � fdd�� | � td�� t�� � � W 5 Q R X d S )Nc � s$ � � } zt�| � W 5 | � � X d S r )�closer r. )�coro�r, r
r r, _ 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_loop^ 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 �leftoverm s zARunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.leftoverc � s t �� � ��dS �N�{ )r � create_taskr
�rA �lo_taskr
r r, p s z=RunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.mainrC )r- r r. r% �doner/ r
rE r �&test_asyncio_run_cancels_hanging_tasksj 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 zt �d�I d H W n t jk
r2 dd Y nX d S )Nr@ r r )r r+ �CancelledErrorr
r
r
r rA | s zHRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.leftoverc � s t �� } � | _t ��� ��dS rB )r �get_running_loop�call_exception_handlerrD r �Zcall_exc_handler_mockrA rF r
r r, � s zDRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.mainrC zasyncio.run.*shutdown)�messageZtask� exception) r r r- r r. r% rG Zassert_called_with�
test_utilsZ MockPatternZMockInstanceOf�ZeroDivisionErrorr/ r
rL r �-test_asyncio_run_reports_hanging_tasks_errorsx s �z6RunTests.test_asyncio_run_reports_hanging_tasks_errorsc s� d �d �G dd� dt �� dd� ���fdd��� ��fdd�}| �� �� t�|� � W 5 Q R X | ���� � | ��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 rR c S s dV t �d�I d H q d S )Nr r* r
r
r
r �fidget� s zPRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.fidgetc � sB � � �z�2 z 3 d H W } q 6 W n t jk
r< dd Y nX d S )Nr r )r rI )Zthe_meaning_of_life)rS �spinnerr
r �spin� s zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.spinc � s& t �� } t�� | _t ��� ��� �d S r )r rJ r r rK rD r )rR �lazyboyrU r
r r, � s
zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.main)
� ExceptionZ assertRaisesr r. r% rG Z assertIsNone�ag_frameZ assertFalse�
ag_runningr/ r
)rR rS rV rU rT 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 r9 r; r? rH rQ rZ r
r
r
r r( ; s r( ) r Zunittestr � r rO ZAbstractEventLoopPolicyr ZTestCaser r( r
r
r
r �<module> s