Your IP : 216.73.216.213
U
i�fQC � @ 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 ek
r� dZY nX 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*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��rhG d9d:� d:e j3e�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� dZejG�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 �</opt/alt/python38/lib64/python3.8/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 kr d S t �t j�S d S �N)r" r# Z PROTOCOL_TLSr r r r �dummy_ssl_contextY s r. c C s< dd� }|� }| � |�}d|_z| �|� W 5 |�� X d S )Nc � s d S r- r r r r r �oncea s zrun_briefly.<locals>.onceF)Z create_taskZ_log_destroy_pending�close�run_until_complete)�loopr/ �gen�tr r r � run_briefly` s
r5 � c C sL t �� | }|� sH|d k r6|t �� }|dkr6t�� �| �t�d�� q d S )Nr g����MbP?)�time� monotonicr � TimeoutErrorr1 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 )rD �format�argsr r r � log_message� s z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__rE rH r r r r r@ � s r@ c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�SilentWSGIServer� c s" t � �� \}}|�| j� ||fS r- ��super� get_requestZ
settimeout�request_timeout�rD �requestZ client_addr�� __class__r r rP � s z SilentWSGIServer.get_requestc C s d S r- r �rD rS �client_addressr r r � handle_error� s z SilentWSGIServer.handle_error)rI rJ rK rQ rP rX �
__classcell__r r rT r rL � s rL c @ s e Zd Zdd� ZdS )�SSLWSGIServerMixinc C sV t �� }|�tt� |j|dd�}z| �||| � |�� W n tk
rP Y nX d S )NT)Z server_side) r" r# r$ r% r& Z wrap_socketZRequestHandlerClassr0 �OSError)rD rS rW �contextZssockr r r �finish_request� s z!SSLWSGIServerMixin.finish_requestN)rI rJ rK r] r r r r rZ � s rZ c @ s e Zd ZdS )�
SSLWSGIServerN�rI rJ rK r r r r r^ � s r^ F)�use_sslc # s� dd� ��fdd�}|r |n|}|| t �� � �|� � j� _tj� fdd�d�}|�� z
� V W 5 � �� � �� |� � X 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 )re 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 )rm r` �
server_cls�server_ssl_clsrh Z server_classZ
server_threadr )ri r2 r �_run_test_server� s
�
rs 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_portrC r r r ry � s zUnixHTTPServer.server_bindN)rI rJ rK ry r r r r rt � s rt c s( e Zd ZdZdd� Z� fdd�Z� ZS )�UnixWSGIServerrM c C s t �| � | �� d S r- )rt ry Z
setup_environrC r r r ry � s
zUnixWSGIServer.server_bindc s" t � �� \}}|�| j� |dfS )N)ru � rN rR rT r r rP � s zUnixWSGIServer.get_request)rI rJ rK rQ ry rP rY r r rT r rz � s rz c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S r- r rV r r r rX � s z!SilentUnixWSGIServer.handle_errorN)rI rJ rK rX r r r r r| � s r| c @ s e Zd ZdS )�UnixSSLWSGIServerNr_ r r r r r} � s r} c
C s& t �� �} | jW 5 Q R � S Q R X d S r- )�tempfileZNamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s
r� c c s< t � } z
| V W 5 zt�| � W n tk
r4 Y nX X d S r- )r� r �unlinkr[ )r r r r �unix_socket_path s
r� c c s, t � � }t|| ttd�E d H W 5 Q R X d S �N)rm r` rq rr )r� rs r| r} )r` r r r r �run_test_unix_server s
�r� ru �ZhostZportr` c c s t | |f|ttd�E d H d S r� )rs rL 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- ��keysrC r r r �__init__' s zTestSelector.__init__Nc C s t �|d||�}|| j|< |S )Nr )� selectorsZ SelectorKeyr� )rD �fileobjr rg �keyr r r �register* s
zTestSelector.registerc C s | j �|�S r- )r� �pop)rD r� r r r �
unregister/ s zTestSelector.unregisterc C s g S r- r )rD r; r r r �select2 s zTestSelector.selectc C s | j S r- r� rC r r r �get_map5 s zTestSelector.get_map)N)rI rJ rK 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 r- r r r r r r3 P s z TestLoop.__init__.<locals>.genFTr g��&� .>)rO r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr� Z _selector�readers�writers�reset_counters�weakrefZWeakValueDictionary� _transports)rD r3 rT r r r� L s
zTestLoop.__init__c C s | j S r- �r� rC r r r r7 c s z
TestLoop.timec C s |r| j |7 _ d S r- r� )rD �advancer r r � advance_timef s zTestLoop.advance_timec sB t � �� | jr>z| j�d� W n tk
r4 Y n
X td��d S )Nr z Time generator is not finished)rO r0 r� r� �send�
StopIteration�AssertionErrorrC rT r r r0 k s
zTestLoop.closec G s t �||| d �| j|< d S r- )r �Handler� �rD �fd�callbackrG r r r � _add_readeru s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S �N� TF)�remove_reader_countr� �rD r� r r r �_remove_readerx s
zTestLoop._remove_readerc G sh || j krtd|� 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�rD r� r� rG Zhandler r r �
assert_reader� s
�
�zTestLoop.assert_readerc C s || j krtd|� 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 < || jkr(| 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 s~ t |t�sFzt|�� �}W n* tttfk
rD td�|��d �Y nX z| j| }W n tk
rh Y nX t d�||���d S )NzInvalid file object: {!r}z.File descriptor {!r} is used by transport {!r})
�
isinstancera �fileno�AttributeError� TypeError�
ValueErrorrF r� �KeyError� RuntimeError)rD r� Z transportr r r �_ensure_fd_no_transport� s&
�� ��z TestLoop._ensure_fd_no_transportc G s | � |� | j||f|�� 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||f|�� 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� defaultdictra r� r� rC r r r r� � s zTestLoop.reset_countersc s6 t � �� | jD ]}| j�|�}| �|� qg | _d S r- )rO � _run_oncer� r� r� r� )rD �whenr� rT r r r� � s
zTestLoop._run_once)r\ c s& | j �|� t� j||f|�d|i�S )Nr\ )r� �appendrO �call_at)rD r� r� r\ rG rT r r r� � s zTestLoop.call_atc C s d S r- r )rD Z
event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S r- r rC r r r �_write_to_self� s zTestLoop._write_to_self)N)rI rJ rK r� r7 r� r0 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rY r r rT 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�rD �otherr r r �__eq__� s zMockPattern.__eq__N)rI rJ rK r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MockInstanceOfc C s
|| _ d S r- )�_type)rD 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)rI rJ rK 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� | �� t�� }|d k r�z |�� }W n tk
rR Y n.X t|tj �r�t
|j � � �}|D ] }|�
� qrd S )NT)�wait)Z_default_executorrp r0 r Zmaybe_get_event_loop_policyZget_child_watcher�NotImplementedErrorr� �asyncioZThreadedChildWatcher�listZ_threads�valuesr )r2 ZexecutorZpolicyZwatcherZthreads�threadr r r �
close_loop� s zTestCase.close_loopT)�cleanupc C s t �d � |r | �| j|� d S r- )r �set_event_loopZ
addCleanupr� )rD r2 r� r r r r� s
zTestCase.set_event_loopNc C s t |�}| �|� |S r- )r� r� )rD r3 r2 r r r �
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S r- )�_get_running_loopr rC 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 rj " rk z TestCase.setUp.<locals>.<lambda>)r r� r Zthreading_setup�_thread_cleanuprC 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_childrenrC r r r �tearDown% s
zTestCase.tearDown)N)
rI rJ rK � staticmethodr� r� r� r� r� r� r r r r r� � s
r� c c s2 t j} zt �tjd � d V W 5 t �| � X d S )Nr� )r �levelZsetLevel�loggingZCRITICAL)Z old_levelr r r �disable_logger3 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_socketA s r� )r6 )Mr� r� �
contextlibrA r� r r� r� r� rw r� r~ rn r7 Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r" � ImportErrorr r r r r
Z asyncio.logr Ztestr r r% r& ZSIGNED_CERTFILEZ
SIGNING_CAZPEERCERTr* r, r. r5 r< r? r@ rL rZ r^ rs r rx rt rz 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�
�
#
%
6
�