File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/utils.cpython-38.opt-2.pyc
U
/D�gQC � @ 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 �6/opt/cppython/lib/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�timeout�deadliner 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 )rE �format�argsr r r � log_message� s z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__rF rI r r r r rA � s rA c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�SilentWSGIServer� c s" t � �� \}}|�| j� ||fS r- ��super� get_request�
settimeout�request_timeout�rE �requestZ client_addr�� __class__r r rQ � s z SilentWSGIServer.get_requestc C s d S r- r �rE rU �client_addressr r r � handle_error� s z SilentWSGIServer.handle_error)rJ rK rL rS rQ rZ �
__classcell__r r rV r rM � s rM 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)rE rU rY �contextZssockr r r �finish_request� s z!SSLWSGIServerMixin.finish_requestN)rJ rK rL r_ r r r r r\ � s r\ c @ s e Zd ZdS )�
SSLWSGIServerN�rJ rK rL 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 )rg Zstart_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)
rA Zset_appZserver_address�address� threading�Thread�start�shutdownZ server_closer )rp rb �
server_cls�server_ssl_clsrk Z server_classZ
server_threadr )rl r2 r �_run_test_server� s
�
rw �AF_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_portrD r r r r~ � s zUnixHTTPServer.server_bindN)rJ rK rL r~ r r r r ry � s ry c s( e Zd ZdZdd� Z� fdd�Z� ZS )�UnixWSGIServerrN c C s t �| � | �� d S r- )ry r~ Z
setup_environrD r r r r~ � s
zUnixWSGIServer.server_bindc s" t � �� \}}|�| j� |dfS )N)rz � rO rT rV r r rQ � s zUnixWSGIServer.get_request)rJ rK rL rS r~ rQ r[ r r rV r r � s r c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S r- r rX r r r rZ � s z!SilentUnixWSGIServer.handle_errorN)rJ rK rL rZ r r r r r� � s r� c @ s e Zd ZdS )�UnixSSLWSGIServerNra r r r r r� � s r� c
C s& t �� �} | jW 5 Q R � S Q R X d S r- )�tempfile�NamedTemporaryFile�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)rp rb ru rv )r� rw r� r� )rb r r r r �run_test_unix_server s
�r� rz ��host�portrb c c s t | |f|ttd�E d H d S r� )rw rM 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__)�base�dctr� 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- ��keysrD r r r �__init__' s zTestSelector.__init__Nc C s t �|d||�}|| j|< |S )Nr )� selectors� SelectorKeyr� )rE �fileobjr ri �keyr r r �register* s
zTestSelector.registerc C s | j �|�S r- )r� �pop)rE r� r r r �
unregister/ s zTestSelector.unregisterc C s g S r- r )rE r; r r r �select2 s zTestSelector.selectc C s | j S r- r� rD r r r �get_map5 s zTestSelector.get_map)N)rJ rK rL 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��&� .>)rP r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr� � _selector�readers�writers�reset_counters�weakref�WeakValueDictionary� _transports)rE r3 rV r r r� L s
zTestLoop.__init__c C s | j S r- �r� rD r r r r7 c s z
TestLoop.timec C s |r| j |7 _ d S r- r� )rE �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)rP r0 r� r� �send�
StopIteration�AssertionErrorrD rV r r r0 k s
zTestLoop.closec G s t �||| d �| j|< d S r- )r �Handler� �rE �fd�callbackrH r r r � _add_readeru s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S �N� TF)�remove_reader_countr� �rE 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� � _callback�_args�rE r� r� rH �handler 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})
�
isinstancerc �fileno�AttributeError� TypeError�
ValueErrorrG r� �KeyError� RuntimeError)rE 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� defaultdictrc r� r� rD r r r r� � s zTestLoop.reset_countersc s6 t � �� | jD ]}| j�|�}| �|� qg | _d S r- )rP � _run_oncer� r� r� r� )rE �whenr� rV r r r� � s
zTestLoop._run_once)r^ c s& | j �|� t� j||f|�d|i�S )Nr^ )r� �appendrP �call_at)rE r� r� r^ rH rV r r r� � s zTestLoop.call_atc C s d S r- r )rE Z
event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S r- r rD r r r �_write_to_self� s zTestLoop._write_to_self)N)rJ rK rL r� r7 r� r0 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r[ r r rV 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�rE �otherr r r �__eq__� s zMockPattern.__eq__N)rJ rK rL r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MockInstanceOfc C s
|| _ d S r- )�_type)rE 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)rJ rK rL 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_executorrt r0 r Zmaybe_get_event_loop_policyZget_child_watcher�NotImplementedErrorr� �asyncioZThreadedChildWatcher�listZ_threads�valuesr )r2 �executorZpolicyZwatcher�threads�threadr r r �
close_loop� s zTestCase.close_loopT)�cleanupc C s t �d � |r | �| j|� d S r- )r �set_event_loopZ
addCleanupr� )rE r2 r� r r r r s
zTestCase.set_event_loopNc C s t |�}| �|� |S r- )r� r )rE r3 r2 r r r �
new_test_loop s
zTestCase.new_test_loopc C s | j t_ d S r- )�_get_running_loopr rD 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 rm " rn z TestCase.setUp.<locals>.<lambda>)r r r Zthreading_setup�_thread_cleanuprD 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 �
reap_childrenrD r r r �tearDown% s
zTestCase.tearDown)N)
rJ rK rL � 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 �level�setLevel�logging�CRITICAL)Z old_levelr r r �disable_logger3 s
r c C s* t �tj�}| |_||_||_d|j_|S )Ng )r Z MagicMock�socket�protor� �family�
gettimeoutr� )r r� r �sockr r r �mock_nonblocking_socketA s r )r6 )Mr� r� �
contextlibrB r
r r� r� r r| r r� rq r7 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 r= r@ rA rM r\ r` rw r r} ry r r� r� r� �contextmanagerr� r� r� r� � BaseSelectorr� Z
BaseEventLoopr� r� r� r� r� r� r� r � IPPROTO_TCP� SOCK_STREAM�AF_INETr r r r r �<module> s�
�
#
%
6
�