Your IP : 18.119.118.151
3
��2a�: �
@ 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 dl mZ d dlmZ d dlmZmZ y d dlZW n ek
r� dZY nX ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lm Z dd l m Z dd
l m Z d dl!m"Z" ej#dk�rDddl$m%Z% n d dlm%Z% dd� Z&e&d�Z'e&d�Z(dd� Z)dd� Z*dQdd�Z+d d � Z,G d d � d e�Z-G d d!� d!e�Z.G d"d#� d#�Z/G d$d%� d%e/e.�Z0d&d'�d(d)�Z1e2ed*��rVG d+d,� d,ej3e�Z4G d-d.� d.e4e�Z5G d/d0� d0e5�Z6G d1d2� d2e/e6�Z7d3d4� Z8ej9d5d6� �Z:ej9d&d'�d7d8��Z;ej9d9d d&d:�d;d<��Z<d=d>� Z=G d?d@� d@ej>�Z?G dAdB� dBej@�ZAdCdD� ZBG dEdF� dFeC�ZDdGdH� ZEG dIdJ� dJe jF�ZFej9dKdL� �ZGejHejIejJfdMdN�ZKdOdP� ZLdS )R� N)�mock)�
HTTPServer)�WSGIRequestHandler�
WSGIServer� )� base_events)�compat)�events)�futures)� selectors)�tasks)� coroutine)�logger)�support�win32)�
socketpairc C s` t td�r*tjjtj| �}tjj|�r*|S tjjtjjtj�d| �}tjj|�rT|S t | ��d S )N�
TEST_HOME_DIR�test)
�hasattrr �os�path�joinr �isfile�dirname�__file__�FileNotFoundError)�filename�fullname� r �//opt/alt/python36/lib64/python3.6/test_utils.py� data_file- s
r z ssl_cert.pemz ssl_key.pemc C s t d kr d S t jt j�S d S )N)�ssl�
SSLContextZPROTOCOL_SSLv23r r r r �dummy_ssl_context< s r# c
C s@ t dd� �}|� }| j|�}d|_z| j|� W d |j� X d S )Nc S s d S )Nr r r r r �onceD s zrun_briefly.<locals>.onceF)r
Z create_taskZ_log_destroy_pending�run_until_complete�close)�loopr$ �gen�tr r r � run_brieflyC s
r* � c C sT t j � | }xB|� sN|d k r8|t j � }|dkr8tj� �| jtjd| d�� qW d S )Nr g����MbP?)r' )�timer
� TimeoutErrorr% r Zsleep)r' Zpred�timeoutZdeadliner r r � run_untilR s r/ c C s | j | j� | j� d S )N)Z call_soon�stopZ run_forever)r' r r r �run_once\ s r1 c @ s e Zd Zdd� Zdd� ZdS )�SilentWSGIRequestHandlerc C s t j� S )N)�io�StringIO)�selfr r r �
get_stderri s z#SilentWSGIRequestHandler.get_stderrc G s d S )Nr )r5 �format�argsr r r � log_messagel s z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__r6 r9 r r r r r2 g s r2 c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�SilentWSGIServer� c s" t � j� \}}|j| j� ||fS )N)�super� get_request�
settimeout�request_timeout)r5 �request� client_addr)� __class__r r r@ t s z SilentWSGIServer.get_requestc C s d S )Nr )r5 rC �client_addressr r r � handle_errory s z SilentWSGIServer.handle_error)r: r; r< rB r@ rG �
__classcell__r r )rE r r= p s r= c @ s e Zd Zdd� ZdS )�SSLWSGIServerMixinc C s^ t }t}tj� }|j||� |j|dd�}y| j||| � |j� W n tk
rX Y nX d S )NT)Z server_side) �ONLYKEY�ONLYCERTr! r" Zload_cert_chainZ wrap_socketZRequestHandlerClassr& �OSError)r5 rC rF ZkeyfileZcertfile�contextZssockr r r �finish_request s z!SSLWSGIServerMixin.finish_requestN)r: r; r< rN r r r r rI } s rI c @ s e Zd ZdS )�
SSLWSGIServerN)r: r; r< r r r r rO � s rO F)�use_sslc
# sv dd� }|r|n|}|| t �� � j|� � j� _tj� fdd�d�}|j� z
� V W d � j� � j� |j � X d S )Nc S s d}dg}|||� dgS )Nz200 OK� Content-type�
text/plains Test message)rQ rR r )�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>� s z"_run_test_server.<locals>.<lambda>)�target)
r2 Zset_appZserver_address�address� threadingZThread�start�shutdownZ server_closer )rY rP �
server_cls�server_ssl_clsrU Z server_classZ
server_threadr )rV r �_run_test_server� s
r_ ZAF_UNIXc @ s e Zd Zdd� ZdS )�UnixHTTPServerc C s t jj| � d| _d| _d S )Nz 127.0.0.1�P )� socketserver�UnixStreamServer� server_bindZ server_nameZ server_port)r5 r r r rd � s zUnixHTTPServer.server_bindN)r: r; r< rd r r r r r` � s r` c s( e Zd ZdZdd� Z� fdd�Z� ZS )�UnixWSGIServerr> c C s t j| � | j� d S )N)r` rd Z
setup_environ)r5 r r r rd � s
zUnixWSGIServer.server_bindc s" t � j� \}}|j| j� |dfS )N� 127.0.0.1� )rf rg )r? r@ rA rB )r5 rC rD )rE r r r@ � s zUnixWSGIServer.get_request)r: r; r< rB rd r@ rH r r )rE r re � s re c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S )Nr )r5 rC rF r r r rG � s z!SilentUnixWSGIServer.handle_errorN)r: r; r< rG r r r r rh � s rh c @ s e Zd ZdS )�UnixSSLWSGIServerN)r: r; r< r r r r ri � s ri c C s t j� �} | jS Q R X d S )N)�tempfileZNamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s
rm c c s< t � } z
| V W d ytj| � W n tk
r4 Y nX X d S )N)rm r �unlinkrL )r r r r �unix_socket_path� s
ro c
c s, t � � }t|| ttd�E d H W d Q R X d S )N)rY rP r] r^ )ro r_ rh ri )rP r r r r �run_test_unix_server� s rp z 127.0.0.1)�host�portrP c c s t | |f|ttd�E d H d S )N)rY rP r] r^ )r_ r= rO )rq rr rP r r r �run_test_server� s
rs c C sP i }x4t | �D ](}|jd�r(|jd�r(qtd d�||< qW td| f| j |�� S )N�__)� return_valueZ TestProtocol)�dir�
startswith�endswith� MockCallback�type� __bases__)�baseZdctrk 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)r5 r r r �__init__ s zTestSelector.__init__Nc C s t j|d||�}|| j|< |S )Nr )r Z SelectorKeyr )r5 �fileobjr �data�keyr r r �register s
zTestSelector.registerc C s | j j|�S )N)r �pop)r5 r� r r r �
unregister s zTestSelector.unregisterc C s g S )Nr )r5 r. r r r �select s zTestSelector.selectc C s | j S )N)r )r5 r r r �get_map s zTestSelector.get_map)N)r: r; r< 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%�Z� fd&d'�Zd(d)� Zd*d+� Z� ZS )-�TestLoopNc sv t � j� |d kr"dd� }d| _nd| _|� | _t| j� d| _d| _g | _t� | _ i | _
i | _ | j � t
j� | _d S )Nc s s
d V d S )Nr r r r r r( , s z TestLoop.__init__.<locals>.genFTr g��&� .>)r? r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr~ Z _selector�readers�writers�reset_counters�weakref�WeakValueDictionary� _transports)r5 r( )rE r r r� ( s
zTestLoop.__init__c C s | j S )N)r� )r5 r r r r, ? s z
TestLoop.timec C s |r| j |7 _ d S )N)r� )r5 �advancer r r � advance_timeB s zTestLoop.advance_timec sB t � j� | jr>y| jjd� W n tk
r4 Y n
X td��d S )Nr z Time generator is not finished)r? r& r� r� �send�
StopIteration�AssertionError)r5 )rE r r r& G s
zTestLoop.closec G s t j||| �| j|< d S )N)r �Handler� )r5 �fd�callbackr8 r r r � _add_readerQ s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)�remove_reader_countr� )r5 r� r r r �_remove_readerT 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)r5 r� r� r8 �handler r r �
assert_reader\ s
zTestLoop.assert_readerc C s || j krtd|� d���d S )Nzfd z is registered)r� r� )r5 r� r r r �assert_no_readerg s
zTestLoop.assert_no_readerc G s t j||| �| j|< d S )N)r r� r� )r5 r� r� r8 r r r � _add_writerk s zTestLoop._add_writerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)�remove_writer_countr� )r5 r� r r r �_remove_writern s
zTestLoop._remove_writerc G s | j | }d S )N)r� )r5 r� r� r8 r� r r r �
assert_writerv s
zTestLoop.assert_writerc C s8 y| j | }W n tk
r" Y nX tdj||���d S )Nz.File descriptor {!r} is used by transport {!r})r� �KeyError� RuntimeErrorr7 )r5 r� Z transportr r r �_ensure_fd_no_transport~ s z TestLoop._ensure_fd_no_transportc G s | j |� | j||f|�� S )N)r� r� )r5 r� r� r8 r r r �
add_reader� s
zTestLoop.add_readerc C s | j |� | j|�S )N)r� r� )r5 r� r r r �
remove_reader� s
zTestLoop.remove_readerc G s | j |� | j||f|�� S )N)r� r� )r5 r� r� r8 r r r �
add_writer� s
zTestLoop.add_writerc C s | j |� | j|�S )N)r� r� )r5 r� r r r �
remove_writer� s
zTestLoop.remove_writerc C s t jt�| _t jt�| _d S )N)� collections� defaultdict�intr� r� )r5 r r r r� � s zTestLoop.reset_countersc s: t � j� x$| jD ]}| jj|�}| j|� qW g | _d S )N)r? � _run_oncer� r� r� r� )r5 �whenr� )rE r r r� � s
zTestLoop._run_oncec s | j j|� t� j||f|�� S )N)r� �appendr? �call_at)r5 r� r� r8 )rE r r r� � s zTestLoop.call_atc C s d S )Nr )r5 Z
event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S )Nr )r5 r r r �_write_to_self� s zTestLoop._write_to_self)N)r: r; r< r� r, r� r& r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rH r r )rE r r� s*
r� c K s t jf ddgi| ��S )N�spec�__call__)r ZMock)�kwargsr r r ry � s ry c @ s e Zd Zdd� ZdS )� MockPatternc C s t tjt| �|tj��S )N)�bool�re�search�str�S)r5 �otherr r r �__eq__� s zMockPattern.__eq__N)r: r; r< r� r r r r r� � s r� c C s$ t j| �}|d kr td| f ��|S )Nz unable to get the source of %r)r Z_get_function_source�
ValueError)�func�sourcer r r �get_function_source� s
r� c @ sV e Zd Zedd� �Zdd�dd�Zddd �Zd
d � Zd d
� Zdd� Z e
j sRdd� Z dS )�TestCasec C s& | j }|d k r|jdd� | j� d S )NT)�wait)Z_default_executorr\ r& )r' �executorr r r �
close_loop� s zTestCase.close_loopT)�cleanupc C s t jd � |r | j| j|� d S )N)r �set_event_loopZ
addCleanupr� )r5 r' r� r r r r� � s
zTestCase.set_event_loopNc C s t |�}| j|� |S )N)r� r� )r5 r( r' r r r �
new_test_loop� s
zTestCase.new_test_loopc C s | j t_ d S )N)�_get_running_loopr )r5 r r r �unpatch_get_running_loop� s z!TestCase.unpatch_get_running_loopc C s t j| _dd� t _tj� | _d S )Nc S s d S )Nr r r r r rW � s z TestCase.setUp.<locals>.<lambda>)r r� r Zthreading_setup�_thread_cleanup)r5 r r r �setUp� s
zTestCase.setUpc C sB | j � tjd � | jtj� d� | j� tj| j � tj
� d S )N)NNN) r� r r� Z assertEqual�sys�exc_infoZ
doCleanupsr Zthreading_cleanupr� Z
reap_children)r5 r r r �tearDown� s
zTestCase.tearDownc O s G dd� d�}|� S )Nc @ s e Zd Zdd� Zdd� ZdS )z!TestCase.subTest.<locals>.EmptyCMc S s d S )Nr )r5 r r r � __enter__� s z+TestCase.subTest.<locals>.EmptyCM.__enter__c W s d S )Nr )r5 �excr r r �__exit__� s z*TestCase.subTest.<locals>.EmptyCM.__exit__N)r: r; r< r� r� r r r r �EmptyCM� s r� r )r5 r8 r� r� r r r �subTest� s zTestCase.subTest)N)
r: r; r< � staticmethodr� r� r� r� r� r� r ZPY34r� r r r r r� � s
r� c c s2 t j} zt jtjd � d V W d t j| � X d S )Nr )r �levelZsetLevel�loggingZCRITICAL)Z old_levelr r r �disable_logger� s
r� c C s* t jtj�}| |_||_||_d|j_|S )Ng )r Z MagicMock�socket�protorz �familyZ
gettimeoutru )r� rz r� Zsockr r r �mock_nonblocking_socket s r� c C s t jddd�S )Nz'asyncio.sslproto._is_sslproto_availableF)ru )r Zpatchr r r r �force_legacy_ssl_support s r� )r+ )Mr� �
contextlibr3 r� r r� r� rb r� rj rZ r, Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r! � ImportErrorrg r r r r
r r Z
coroutinesr
�logr r r �platformZ
windows_utilsr r rK rJ r# r* r/ r1 r2 r= rI rO r_ r rc r` re rh ri rm �contextmanagerro rp rs r} Z BaseSelectorr~ Z
BaseEventLoopr� ry r� r� r� r� r� Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr� r� r r r r �<module> s�
4