Your IP : 216.73.216.213
a
XC?h D � @ 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 dlZd dlmZ d dlmZ d dlmZmZ z d dlZW n ey� dZY n0 d dl mZ d dl mZ d dl mZ d dl mZ d d l m 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*e j+fd'd(�Z,d)d*� Z-G d+d,� d,e�Z.G d-d.� d.e�Z/G d/d0� d0�Z0G d1d2� d2e0e/�Z1d3d4�d5d6�Z2e3ed7��rjG d8d9� d9e j4e�Z5G d:d;� d;e5e�Z6G d<d=� d=e6�Z7G d>d?� d?e0e7�Z8d@dA� Z9ej:dBdC� �Z;ej:d3d4�dDdE��Z<ej:dFd d3dG�dHdI��Z=dJdK� Z>G dLdM� dMej?�Z@G dNdO� dOejA�ZBdPdQ� ZCG dRdS� dSeD�ZEG dTdU� dU�ZFdVdW� ZGG dXdY� dYejH�ZHej:dZd[� �ZIejJejKejLfd\d]�ZMdS )^� 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 �</opt/alt/python39/lib64/python3.9/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)�)Z countryNameZXY))�organizationNamez Python Software Foundation CA))�
commonNamez
our-ca-serverzOct 28 14:23:16 2037 GMTzAug 29 14:23:16 2018 GMTZCB2D80995A69525C)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_sslcontextI s
r* T)�disable_verifyc C s" t �t j�}d|_| r t j|_|S r! )r" r# ZPROTOCOL_TLS_CLIENTr' r( r) )r+ Zclient_contextr r r �simple_client_sslcontextQ s
r, c C s t d u r d S t �t j�S d S �N)r" r# Z PROTOCOL_TLSr r r r �dummy_ssl_contextY s r. c C sD dd� }|� }| � |�}d|_z| �|� W |�� n
|�� 0 d S )Nc � s d S r- r r r r r �oncea s zrun_briefly.<locals>.onceF)Z create_taskZ_log_destroy_pending�run_until_complete�close)�loopr/ �gen�tr r r � run_briefly` s
r5 c C sL t �� | }|� sH|d ur6|t �� }|dkr6t�� �| �t�d�� q d S )Nr g����MbP?)�time� monotonicr � TimeoutErrorr0 r
�sleep)r2 Zpred�timeoutZdeadliner r r � run_untiln s r; c C s | � | j� | �� d S r- )Z call_soon�stopZ run_forever�r2 r r r �run_oncex s r> c @ s e Zd Zdd� Zdd� ZdS )�SilentWSGIRequestHandlerc C s t �� S r- )�io�StringIO��selfr r r �
get_stderr� s z#SilentWSGIRequestHandler.get_stderrc G s d S r- r )rC �format�argsr r r � log_message� s z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__rD rG r r r r r? � s r? c s* e Zd ZejZ� fdd�Zdd� Z� ZS )�SilentWSGIServerc s" t � �� \}}|�| j� ||fS r- ��super� get_requestZ
settimeout�request_timeout�rC �requestZ client_addr�� __class__r r rN � s z SilentWSGIServer.get_requestc C s d S r- r �rC rQ �client_addressr r r � handle_error� s z SilentWSGIServer.handle_error) rH rI rJ r �LOOPBACK_TIMEOUTrO rN rV �
__classcell__r r rR r rK � s rK c @ s e Zd Zdd� ZdS )�SSLWSGIServerMixinc C sT t �� }|�tt� |j|dd�}z| �||| � |�� W n tyN Y n0 d S )NT)Z server_side) r" r# r$ r% r& Z wrap_socketZRequestHandlerClassr1 �OSError)rC rQ rU �contextZssockr r r �finish_request� s z!SSLWSGIServerMixin.finish_requestN)rH rI rJ r\ r r r r rY � s rY c @ s e Zd ZdS )�
SSLWSGIServerN�rH rI rJ r r r r r] � s r] F)�use_sslc # s� dd� ��fdd�}|r |n|}|| t �� � �|� � j� _tj� fdd�d�}|�� z"� V W � �� � �� |� � n� �� � �� |� � 0 d S )Nc s s< t | d �}|r8| d �t|d��}|V |t|�8 }q d S )NZCONTENT_LENGTHz
wsgi.inputi )�int�read�min�len)�environ�size�datar r r r2 � s
z _run_test_server.<locals>.loopc s2 d}dg}|||� | d dkr(� | �S dgS d S )Nz200 OK)z Content-typez
text/plainZ PATH_INFOz/loops Test messager )rd Zstart_responseZstatusZheadersr= 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 )rl r_ �
server_cls�server_ssl_clsrg Z server_classZ
server_threadr )rh r2 r �_run_test_server� s$
�
�rr ZAF_UNIXc @ s e Zd Zdd� ZdS )�UnixHTTPServerc C s t j�| � d| _d| _d S )N� 127.0.0.1�P )� socketserver�UnixStreamServer� server_bindZ server_nameZ server_portrB r r r rx � s zUnixHTTPServer.server_bindN)rH rI rJ rx r r r r rs � s rs c s* e Zd ZejZdd� Z� fdd�Z� ZS )�UnixWSGIServerc C s t �| � | �� d S r- )rs rx Z
setup_environrB r r r rx � s
zUnixWSGIServer.server_bindc s" t � �� \}}|�| j� |dfS )N)rt � rL rP rR r r rN � s zUnixWSGIServer.get_request) rH rI rJ r rW rO rx rN rX r r rR r ry � s ry c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S r- r rT r r r rV � s z!SilentUnixWSGIServer.handle_errorN)rH rI rJ rV r r r r r{ � s r{ c @ s e Zd ZdS )�UnixSSLWSGIServerNr^ r r r r r| � s r| c C s2 t �� �} | jW d � S 1 s$0 Y d S r- )�tempfileZNamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s
r� c c s\ t � } z,| V W zt�| � W qX ty0 Y qX0 n$zt�| � W n tyT Y n0 0 d S r- )r� r �unlinkrZ )r r r r �unix_socket_path s � r� c c s@ t � �&}t|| ttd�E d H W d � n1 s20 Y d S �N)rl r_ rp rq )r� rr r{ r| )r_ r r r r �run_test_unix_server s
�r� rt �ZhostZportr_ c c s t | |f|ttd�E d H d S r� )rr rK r] r� r r r �run_test_server s
�r� c C sL i }t | �D ](}|�d�r&|�d�r&q td d�||< q td| f| j |�� S )N�__)� return_valueZ TestProtocol)�dir�
startswith�endswith� MockCallback�type� __bases__)�baseZdctr~ 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 r- ��keysrB r r r �__init__' s zTestSelector.__init__Nc C s t �|d||�}|| j|< |S )Nr )� selectorsZ SelectorKeyr� )rC �fileobjr rf �keyr r r �register* s
zTestSelector.registerc C s | j �|�S r- )r� �pop)rC r� r r r �
unregister/ s zTestSelector.unregisterc C s g S r- r )rC r: r r r �select2 s zTestSelector.selectc C s | j S r- r� rB r r r �get_map5 s zTestSelector.get_map)N)rH rI rJ 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 u r"dd� }d| _nd| _|� | _t| j� d| _d| _g | _t� | _ i | _
i | _ | � � t
�� | _d S )Nc s s
d V d S r- r r r r r r3 P s z TestLoop.__init__.<locals>.genFTr g��&� .>)rM r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr� Z _selector�readers�writers�reset_counters�weakrefZWeakValueDictionary� _transports)rC r3 rR r r r� L s
zTestLoop.__init__c C s | j S r- �r� rB r r r r6 c s z
TestLoop.timec C s |r| j |7 _ d S r- r� )rC �advancer r r � advance_timef s zTestLoop.advance_timec s@ t � �� | jr<z| j�d� W n ty2 Y n
0 td��d S )Nr z Time generator is not finished)rM r1 r� r� �send�
StopIteration�AssertionErrorrB rR r r r1 k s
zTestLoop.closec G s t �||| d �| j|< d S r- )r �Handler� �rC �fd�callbackrF r r r � _add_readeru s zTestLoop._add_readerc C s0 | j | d7 < || jv r(| j|= dS dS d S �N� TF)�remove_reader_countr� �rC r� r r r �_remove_readerx s
zTestLoop._remove_readerc G sh || j vrtd|� d���| j | }|j|krDtd|j� d|� ���|j|krdtd|j� d|� ���d S )N�fd z is not registeredzunexpected callback: z != zunexpected callback args: )r� r� Z _callbackZ_args�rC r� r� rF Zhandler r r �
assert_reader� s
�
�zTestLoop.assert_readerc C s || j v rtd|� d���d S )Nr� z is registered)r� r� r� r r r �assert_no_reader� s
zTestLoop.assert_no_readerc G s t �||| d �| j|< d S r- )r r� r� r� r r r � _add_writer� s zTestLoop._add_writerc C s0 | j | d7 < || jv r(| j|= dS dS d S r� )�remove_writer_countr� r� r r r �_remove_writer� s
zTestLoop._remove_writerc G s | j | }d S r- )r� r� r r r �
assert_writer� s
zTestLoop.assert_writerc
C sz t |t�sDzt|�� �}W n( tttfyB td�|��d �Y n0 z| j| }W n tyd Y n0 t d�||���d S )NzInvalid file object: {!r}z.File descriptor {!r} is used by transport {!r})
�
isinstancer` �fileno�AttributeError� TypeError�
ValueErrorrE r� �KeyError� RuntimeError)rC r� Z transportr r r �_ensure_fd_no_transport� s$
�� ��z TestLoop._ensure_fd_no_transportc G s | � |� | j||g|�R � S r- )r� r� r� r r r �
add_reader� s
zTestLoop.add_readerc C s | � |� | �|�S r- )r� r� r� r r r �
remove_reader� s
zTestLoop.remove_readerc G s | � |� | j||g|�R � S r- )r� r� r� r r r �
add_writer� s
zTestLoop.add_writerc C s | � |� | �|�S r- )r� r� r� r r r �
remove_writer� s
zTestLoop.remove_writerc C s t �t�| _t �t�| _d S r- )� collections� defaultdictr` r� r� rB r r r r� � s zTestLoop.reset_countersc s6 t � �� | jD ]}| j�|�}| �|� qg | _d S r- )rM � _run_oncer� r� r� r� )rC �whenr� rR r r r� � s
zTestLoop._run_once)r[ c s( | j �|� t� j||g|�R d|i�S )Nr[ )r� �appendrM �call_at)rC r� r� r[ rF rR r r r� � s zTestLoop.call_atc C s d S r- r )rC Z
event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S r- r rB r r r �_write_to_self� s zTestLoop._write_to_self)N)rH rI rJ r� r6 r� r1 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rX r r rR r r� 9 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 r- )�bool�re�search�str�S�rC �otherr r r �__eq__� s zMockPattern.__eq__N)rH rI rJ r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MockInstanceOfc C s
|| _ d S r- )�_type)rC r� r r r r� � s zMockInstanceOf.__init__c C s t || j�S r- )r� r� r� r r r r� � s zMockInstanceOf.__eq__N)rH rI rJ r� r� r r r r r� � s r� c C s$ t �| �}|d u r 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 ur0| �� s"| �| �� � n| j jdd� | �� t�� }|d ur�z |�� }W n t yf Y n.0 t
|t j �r�t
|j�� �}|D ] }|�� q�d S )NT)�wait)Z_default_executorZ is_closedr0 Zshutdown_default_executorro r1 r Zmaybe_get_event_loop_policyZget_child_watcher�NotImplementedErrorr� �asyncioZThreadedChildWatcher�listZ_threads�valuesr )r2 ZpolicyZwatcherZthreads�threadr r r �
close_loop� s
zTestCase.close_loopT)�cleanupc C s t �d � |r | �| j|� d S r- )r �set_event_loopZ
addCleanupr� )rC r2 r� r r r r� s
zTestCase.set_event_loopNc C s t |�}| �|� |S r- )r� r� )rC r3 r2 r r r �
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S r- )�_get_running_loopr rB 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 r- r r r r r ri $ rj z TestCase.setUp.<locals>.<lambda>)r r� r Zthreading_setup�_thread_cleanuprB 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_childrenrB r r r �tearDown' s
zTestCase.tearDown)N)
rH rI rJ � staticmethodr� r� r� r� r� r� r r r r r� � s
r� c c s< t j} z$t �tjd � d V W t �| � n t �| � 0 d S )Nr� )r �levelZsetLevel�loggingZCRITICAL)Z old_levelr r r �disable_logger5 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_socketC s r� )Nr� r� �
contextlibr@ r� r r� r� r� rv r� r} rm r6 Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r" � ImportErrorr r r r r
Z asyncio.logr �testr r r% r& ZSIGNED_CERTFILEZ
SIGNING_CAZPEERCERTr* r, r. r5 Z
SHORT_TIMEOUTr; r> r? rK rY r] rr r rw rs ry r{ r| r� �contextmanagerr� r� r� r� Z BaseSelectorr� Z
BaseEventLoopr� r� r� r� r� r� r� r� Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr� r r r r �<module> s�
�
#
%
8
�