File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/test_runners.cpython-37.opt-1.pyc
B
��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 �:/usr/local/lib/python3.7/test/test_asyncio/test_runners.py�__init__
s zTestPolicy.__init__c C s t �d S )N)� RuntimeError)r r r r
�get_event_loop s zTestPolicy.get_event_loopc C s | � � S )N)r )r r r r
�new_event_loop s zTestPolicy.new_event_loopc C s |d k r|| _ d S )N)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_eventsZ _selectorZselectZ return_valuer r )r r r )r r
�new_loop s
zBaseTest.new_loopc s" t � �� t| j�}t�|� d S )N)�super�setUpr r r �set_event_loop_policy)r �policy)� __class__r r
r + s
zBaseTest.setUpc sH t �� }|jd k r0| �|j�� � | �|jj� t �d � t� �� d S )N) r Zget_event_loop_policyr �
assertTrueZ is_closedr r r �tearDown)r r )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 d 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>.mainr( )�assertRaisesRegexr) r r& )r r$ r r r
�test_asyncio_run_raisesD s z RunTests.test_asyncio_run_raisesc C sT xNddd� hD ]>}| j |d��( | �td�� t�|� W d Q R X W d Q R X qW d S )Nr c S s d S )Nr r r r r
�<lambda>M � z5RunTests.test_asyncio_run_only_coro.<locals>.<lambda>)�objza coroutine was expected)ZsubTestr* r) r r& )r �or r r
�test_asyncio_run_only_coroL s
z#RunTests.test_asyncio_run_only_coroc s0 � fdd�}t �|d�� t j|d�dd� d S )Nc � s t �� }� �|�� | � d S )N)r r
ZassertIsZ get_debug)Zexpectedr )r r r
r$ T s z-RunTests.test_asyncio_run_debug.<locals>.mainFT)�debug)r r& )r r$ r )r r
�test_asyncio_run_debugS s z RunTests.test_asyncio_run_debugc s4 � fdd�� | � td�� t�� � � W d Q R X d S )Nc � s$ � � } zt �| � W d | �� X d S )N)r r& �close)�coro)r$ r r
r$ \ s z9RunTests.test_asyncio_run_from_running_loop.<locals>.mainz cannot be called from a running)r* 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 )Ng�������?)r r# r r r r
�leftoverj s zARunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.leftoverc � s t �� � ��dS )N�{ )r � create_taskr )r6 �lo_taskr r
r$ m s z=RunTests.test_asyncio_run_cancels_hanging_tasks.<locals>.mainr7 )r% r r& r �done)r r$ r )r6 r9 r
�&test_asyncio_run_cancels_hanging_tasksg 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 yt �d�I d H W n t jk
r2 dd Y nX d S )Ng�������?r r )r r# �CancelledErrorr r r r
r6 y s zHRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.leftoverc � s t �� } � | _t ��� ��dS )Nr7 )r �get_running_loop�call_exception_handlerr8 )r )�call_exc_handler_mockr6 r9 r r
r$ s zDRunTests.test_asyncio_run_reports_hanging_tasks_errors.<locals>.mainr7 zasyncio.run.*shutdown)�messageZtaskZ exception) r r r% r r& r r: Zassert_called_with�
test_utilsZ MockPatternZMockInstanceOf�ZeroDivisionError)r r$ r )r? r6 r9 r
�-test_asyncio_run_reports_hanging_tasks_errorsu s z6RunTests.test_asyncio_run_reports_hanging_tasks_errorsc s� d �d �G dd� dt �� dd� ���fdd��� ��fdd�}| �� �� t�|� � W d 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 rD c S s xdV t �d�I d H qW d S )Nr )r r# r r r r
�fidget� s zPRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.fidgetc � sZ � � �y0x*�2 y 3 d H } W q t k
s(X q Y W W n tjk
rT dd Y nX d S )Nr r )�StopAsyncIterationr r< )Zthe_meaning_of_life)rE �spinnerr r
�spin� s zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.spinc � s& t �� } t�� | _t ��� ��� �d S )N)r r= r r r> r8 )r )rD �lazyboyrH r r
r$ � s
zNRunTests.test_asyncio_run_closes_gens_after_hanging_tasks_errors.<locals>.main)
� ExceptionZ assertRaisesr r& r r: Z assertIsNone�ag_frameZ assertFalse�
ag_running)r r$ r )rD rE rI rH rG 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 r' r+ r0 r2 r5 r; rC rM r r r r
r! ; s r! ) r Zunittestr � r rA ZAbstractEventLoopPolicyr ZTestCaser r! r r r r
�<module> s