File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/utils.cpython-37.opt-2.pyc
B
��g@ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dl
mZ d dlmZ d dlmZmZ y d dlZW n ek
r� dZY nX d dlmZ d dlmZ d dlmZ d dlmZ d d lm Z d d
l m Z d d l m Z d d
� Z!e!d�Z"e!d�Z#e!d�Z$e!d�Z%ddddddddddd �
Z&d d � Z'd d �d!d"�Z(d#d$� Z)d%d&� Z*d_d(d)�Z+d*d+� Z,G d,d-� d-e�Z-G d.d/� d/e�Z.G d0d1� d1�Z/G d2d3� d3e/e.�Z0d4d5�d6d7�Z1e2ed8��r`G d9d:� d:ej3e�Z4G d;d<� d<e4e�Z5G d=d>� d>e5�Z6G d?d@� d@e/e6�Z7dAdB� Z8ej9dCdD� �Z:ej9d4d5�dEdF��Z;ej9dGd d4dH�dIdJ��Z<dKdL� Z=G dMdN� dNej>�Z?G dOdP� dPej@�ZAdQdR� ZBG dSdT� dTeC�ZDG dUdV� dV�ZEdWdX� ZFG dYdZ� dZe
jG�ZGej9d[d\� �ZHejIejJejKfd]d^�ZLdS )`� N)�mock)�
HTTPServer)�WSGIRequestHandler�
WSGIServer)� base_events)�events)�format_helpers)�futures)�tasks)�logger)�supportc C s^ t td�r*tj�tj| �}tj�|�r*|S tj�tj�t�d| �}tj�|�rR|S t | ��d S )N�
TEST_HOME_DIRz..)
�hasattrr �os�path�joinr
�isfile�dirname�__file__�FileNotFoundError)�filename�fullname� r �3/usr/local/lib/python3.7/test/test_asyncio/utils.py� data_file& s
r z ssl_cert.pemz ssl_key.pemz keycert3.pemz pycacert.pem)z)http://testca.pythontest.net/testca/ocsp/)z0http://testca.pythontest.net/testca/pycacert.cer)z2http://testca.pythontest.net/testca/revocation.crl)))� countryName�XY))�organizationNamez Python Software Foundation CA))�
commonNamez
our-ca-serverzJul 7 14:23:16 2028 GMTzAug 29 14:23:16 2018 GMTZCB2D80995A69525C)))r r ))Z localityNamezCastle Anthrax))r zPython Software Foundation))r � localhost))ZDNSr � )
ZOCSPZ caIssuersZcrlDistributionPointsZissuerZnotAfterZ notBeforeZ serialNumberZsubjectZsubjectAltName�versionc C s* t �t j�} | �tt� d| _t j| _| S )NF) �ssl�
SSLContextZPROTOCOL_TLS_SERVER�load_cert_chain�ONLYCERT�ONLYKEY�check_hostname� CERT_NONE� verify_mode)Zserver_contextr r r �simple_server_sslcontextH s
r* T)�disable_verifyc C s" t �t j�}d|_| r t j|_|S )NF)r" r# ZPROTOCOL_TLS_CLIENTr' r( r) )r+ Zclient_contextr r r �simple_client_sslcontextP s
r, c C s t d kr d S t �t j�S d S )N)r" r# Z PROTOCOL_TLSr r r r �dummy_ssl_contextX s r- c C s< dd� }|� }| � |�}d|_z| �|� W d |�� X d S )Nc � s d S )Nr r r r r �once` s zrun_briefly.<locals>.onceF)Z create_taskZ_log_destroy_pending�run_until_complete�close)�loopr. �gen�tr r r � run_briefly_ s
r4 � c C sT t �� | }xB|� sN|d k r8|t �� }|dkr8t�� �| �tjd| d�� qW d S )Nr g����MbP?)r1 )�timeZ monotonicr � TimeoutErrorr/ r
Zsleep)r1 Zpred�timeoutZdeadliner r r � run_untilm s r9 c C s | � | j� | �� d S )N)Z call_soon�stopZ run_forever)r1 r r r �run_oncew s r; c @ s e Zd Zdd� Zdd� ZdS )�SilentWSGIRequestHandlerc C s t �� S )N)�io�StringIO)�selfr r r �
get_stderr� s z#SilentWSGIRequestHandler.get_stderrc G s d S )Nr )r? �format�argsr r r � log_message� s z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__r@ rC r r r r r< � s r< c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�SilentWSGIServer� c s" t � �� \}}|�| j� ||fS )N)�super� get_request�
settimeout�request_timeout)r? �request� client_addr)� __class__r r rJ � s z SilentWSGIServer.get_requestc C s d S )Nr )r? rM �client_addressr r r � handle_error� s z SilentWSGIServer.handle_error)rD rE rF rL rJ rQ �
__classcell__r r )rO r rG � s rG c @ s e Zd Zdd� ZdS )�SSLWSGIServerMixinc C sV t �� }|�tt� |j|dd�}y| �||| � |�� W n tk
rP Y nX d S )NT)Z server_side) r" r# r$ r% r& Z wrap_socketZRequestHandlerClassr0 �OSError)r? rM rP �contextZssockr r r �finish_request� s z!SSLWSGIServerMixin.finish_requestN)rD rE rF rV r r r r rS � s rS c @ s e Zd ZdS )�
SSLWSGIServerN)rD rE rF r r r r rW � s rW F)�use_sslc # sv dd� }|r|n|}|| t �� � �|� � j� _tj� fdd�d�}|�� z
� V W d � �� � �� |� � X d S )Nc S s d}dg}|||� dgS )Nz200 OK)z Content-typez
text/plains Test messager )�environZstart_response�statusZheadersr r r �app� s
z _run_test_server.<locals>.appc s � j dd�S )Ng�������?)Z
poll_interval)Z
serve_foreverr )�httpdr r �<lambda>� � z"_run_test_server.<locals>.<lambda>)�target)
r< Zset_appZserver_address�address� threadingZThread�start�shutdownZ server_closer )r` rX �
server_cls�server_ssl_clsr[ Z server_classZ
server_threadr )r\ r �_run_test_server� s
rf ZAF_UNIXc @ s e Zd Zdd� ZdS )�UnixHTTPServerc C s t j�| � d| _d| _d S )Nz 127.0.0.1�P )� socketserver�UnixStreamServer� server_bindZ server_nameZ server_port)r? r r r rk � s zUnixHTTPServer.server_bindN)rD rE rF rk r r r r rg � s rg c s( e Zd ZdZdd� Z� fdd�Z� ZS )�UnixWSGIServerrH c C s t �| � | �� d S )N)rg rk Z
setup_environ)r? r r r rk � s
zUnixWSGIServer.server_bindc s" t � �� \}}|�| j� |dfS )N)z 127.0.0.1� )rI rJ rK rL )r? rM rN )rO r r rJ � s zUnixWSGIServer.get_request)rD rE rF rL rk rJ rR r r )rO r rl � s rl c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S )Nr )r? rM rP r r r rQ � s z!SilentUnixWSGIServer.handle_errorN)rD rE rF rQ r r r r rn � s rn c @ s e Zd ZdS )�UnixSSLWSGIServerN)rD rE rF r r r r ro � s ro c C s t �� �} | jS Q R X d S )N)�tempfileZNamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s
rs c c s< t � } z
| V W d yt�| � W n tk
r4 Y nX X d S )N)rs r �unlinkrT )r r r r �unix_socket_path� s
ru c c s, t � � }t|| ttd�E d H W d Q R X d S )N)r` rX rd re )ru rf rn ro )rX r r r r �run_test_unix_server s rv z 127.0.0.1)�host�portrX c c s t | |f|ttd�E d H d S )N)r` rX rd re )rf rG rW )rw rx rX r r r �run_test_server s
ry c C sP i }x4t | �D ](}|�d�r(|�d�r(qtd d�||< qW td| f| j |�� S )N�__)� return_valueZ TestProtocol)�dir�
startswith�endswith� MockCallback�type� __bases__)�baseZdctrq r r r �make_test_protocol s r� c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � ZdS )
� TestSelectorc C s
i | _ d S )N)�keys)r? r r r �__init__ s zTestSelector.__init__Nc C s t �|d||�}|| j|< |S )Nr )� selectorsZ SelectorKeyr� )r? �fileobjr �data�keyr r r �register s
zTestSelector.registerc C s | j �|�S )N)r� �pop)r? r� r r r �
unregister$ s zTestSelector.unregisterc C s g S )Nr )r? r8 r r r �select' s zTestSelector.selectc C s | j S )N)r� )r? r r r �get_map* s zTestSelector.get_map)N)rD rE rF r� r� r� r� r� r r r r r� s
r� c s� e Zd Zd-� fdd� Zdd� Zdd� Z� fdd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Z� fd$d%�Zdd&�� fd'd(�
Zd)d*� Zd+d,� Z� ZS ).�TestLoopNc sv t � �� |d kr"dd� }d| _nd| _|� | _t| j� d| _d| _g | _t� | _ i | _
i | _ | � � t
�� | _d S )Nc s s
d V d S )Nr r r r r r2 E s z TestLoop.__init__.<locals>.genFTr g��&� .>)rI r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr� Z _selector�readers�writers�reset_counters�weakrefZWeakValueDictionary� _transports)r? r2 )rO r r r� A s
zTestLoop.__init__c C s | j S )N)r� )r? r r r r6 X s z
TestLoop.timec C s |r| j |7 _ d S )N)r� )r? �advancer r r � advance_time[ s zTestLoop.advance_timec sB t � �� | jr>y| j�d� W n tk
r4 Y n
X td��d S )Nr z Time generator is not finished)rI r0 r� r� �send�
StopIteration�AssertionError)r? )rO r r r0 ` s
zTestLoop.closec G s t �||| d �| j|< d S )N)r �Handler� )r? �fd�callbackrB r r r � _add_readerj s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S )N� TF)�remove_reader_countr� )r? r� r r r �_remove_readerm s
zTestLoop._remove_readerc G sh || j krtd|� d���| j | }|j|krDtd|j� d|� ���|j|krdtd|j� d|� ���d S )Nzfd z is not registeredzunexpected callback: z != zunexpected callback args: )r� r� Z _callbackZ_args)r? r� r� rB �handler r r �
assert_readeru s
zTestLoop.assert_readerc C s || j krtd|� d���d S )Nzfd z is registered)r� r� )r? r� r r r �assert_no_reader� s
zTestLoop.assert_no_readerc G s t �||| d �| j|< d S )N)r r� r� )r? r� r� rB r r r � _add_writer� s zTestLoop._add_writerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr� TF)�remove_writer_countr� )r? r� r r r �_remove_writer� s
zTestLoop._remove_writerc G s | j | }d S )N)r� )r? r� r� rB r� r r r �
assert_writer� s
zTestLoop.assert_writerc
C s~ t |t�sFyt|�� �}W n* tttfk
rD td�|��d �Y nX y| j| }W n tk
rh Y nX t d�||���d S )NzInvalid file object: {!r}z.File descriptor {!r} is used by transport {!r})
�
isinstance�int�fileno�AttributeError� TypeError�
ValueErrorrA r� �KeyError� RuntimeError)r? r� Z transportr r r �_ensure_fd_no_transport� s
z TestLoop._ensure_fd_no_transportc G s | � |� | j||f|�� S )N)r� r� )r? r� r� rB r r r �
add_reader� s
zTestLoop.add_readerc C s | � |� | �|�S )N)r� r� )r? r� r r r �
remove_reader� s
zTestLoop.remove_readerc G s | � |� | j||f|�� S )N)r� r� )r? r� r� rB r r r �
add_writer� s
zTestLoop.add_writerc C s | � |� | �|�S )N)r� r� )r? r� r r r �
remove_writer� s
zTestLoop.remove_writerc C s t �t�| _t �t�| _d S )N)� collections� defaultdictr� r� r� )r? r r r r� � s zTestLoop.reset_countersc s: t � �� x$| jD ]}| j�|�}| �|� qW g | _d S )N)rI � _run_oncer� r� r� r� )r? �whenr� )rO r r r� � s
zTestLoop._run_once)rU c s& | j �|� t� j||f|�d|i�S )NrU )r� �appendrI �call_at)r? r� r� rU rB )rO r r r� � s zTestLoop.call_atc C s d S )Nr )r? Z
event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S )Nr )r? r r r �_write_to_self� s zTestLoop._write_to_self)N)rD rE rF r� r6 r� r0 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rR r r )rO r r� . s*
r� c K s t jf ddgi| ��S )N�spec�__call__)r ZMock)�kwargsr r r r � s r c @ s e Zd Zdd� ZdS )� MockPatternc C s t t�t| �|tj��S )N)�bool�re�search�str�S)r? �otherr r r �__eq__� s zMockPattern.__eq__N)rD rE rF r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MockInstanceOfc C s
|| _ d S )N)�_type)r? r� r r r r� � s zMockInstanceOf.__init__c C s t || j�S )N)r� r� )r? r� r r r r� � s zMockInstanceOf.__eq__N)rD rE rF r� r� r r r r r� � s r� c C s$ t �| �}|d kr td| f ��|S )Nz unable to get the source of %r)r Z_get_function_sourcer� )�func�sourcer r r �get_function_source� s
r� c @ sH e Zd Zedd� �Zdd�dd�Zddd �Zd
d � Zd d
� Zdd� Z dS )�TestCasec C s& | j }|d k r|jdd� | �� d S )NT)�wait)Z_default_executorrc r0 )r1 �executorr r r �
close_loop� s zTestCase.close_loopT)�cleanupc C s t �d � |r | �| j|� d S )N)r �set_event_loopZ
addCleanupr� )r? r1 r� r r r r� � s
zTestCase.set_event_loopNc C s t |�}| �|� |S )N)r� r� )r? r2 r1 r r r �
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S )N)�_get_running_loopr )r? r r r �unpatch_get_running_loop s z!TestCase.unpatch_get_running_loopc C s t j| _dd� t _t�� | _d S )Nc S s d S )Nr r r r r r] r^ z TestCase.setUp.<locals>.<lambda>)r r� r Zthreading_setup�_thread_cleanup)r? r r r �setUp s
zTestCase.setUpc C sB | � � t�d � | �t�� d� | �� tj| j � t�
� d S )N)NNN) r� r r� Z assertEqual�sys�exc_infoZ
doCleanupsr Zthreading_cleanupr� Z
reap_children)r? r r r �tearDown s
zTestCase.tearDown)N)
rD rE rF � staticmethodr� r� r� r� r� r� r r r r r� � s
r� c c s2 t j} zt �tjd � d V W d t �| � X d S )Nr� )r �levelZsetLevel�loggingZCRITICAL)Z old_levelr r r �disable_logger s
r� c C s* t �tj�}| |_||_||_d|j_|S )Ng )r Z MagicMock�socket�protor� �familyZ
gettimeoutr{ )r� r� r� Zsockr r r �mock_nonblocking_socket* s r� )r5 )Mr� �
contextlibr= r� r r� r� r� ri r� rp ra r6 Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r" � ImportErrorZasyncior r r r r
Z asyncio.logr Ztestr r r% r& ZSIGNED_CERTFILEZ
SIGNING_CAZPEERCERTr* r, r- r4 r9 r; r< rG rS rW rf r rj rg rl rn ro rs �contextmanagerru rv ry r� Z BaseSelectorr� Z
BaseEventLoopr� r r� r� r� r� r� r� Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr� r r r r �<module> s�
%
*