File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/utils.cpython-310.opt-2.pyc
o
�i�C � @ 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 zd dlZW n eyf dZY nw 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 l!m"Z" d
d� Z#e#d�Z$e#d�Z%e#d�Z&e#d�Z'dddddddddd d �
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�Z0G d.d/� d/e�Z1G d0d1� d1�Z2G d2d3� d3e2e1�Z3d4d5�d6d7�Z4e5ed8��r<G d9d:� d:e j6e�Z7G d;d<� d<e7e�Z8G d=d>� d>e8�Z9G d?d@� d@e2e9�Z:dAdB� Z;ej<dCdD� �Z=ej<d4d5�dEdF��Z>ej<dGd d4dH�dIdJ��Z?dKdL� Z@G dMdN� dNejA�ZBG dOdP� dPejC�ZDdQdR� ZEG dSdT� dTeF�ZGG dUdV� dV�ZHdWdX� ZIG dYdZ� dZejJ�ZJej<d[d\� �ZKejLejMejNfd]d^�ZOdS )_� N)�mock)�
HTTPServer)�WSGIRequestHandler�
WSGIServer)� base_events)�events)�format_helpers)�futures)�tasks)�logger)�support)�threading_helperc C sZ t td�rtj�tj| �}tj�|�r|S tj�tj�t�d| �}tj�|�r)|S t | ��)N�
TEST_HOME_DIRz..)
�hasattrr �os�path�joinr �isfile�dirname�__file__�FileNotFoundError)�filename�fullname� r �4/usr/local/lib/python3.10/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 serialNumber�subjectZsubjectAltName�versionc C s* t �t j�} | �tt� d| _t j| _| S �NF) �ssl�
SSLContext�PROTOCOL_TLS_SERVER�load_cert_chain�ONLYCERT�ONLYKEY�check_hostname� CERT_NONE� verify_mode)Zserver_contextr r r �simple_server_sslcontextJ s
r- T��disable_verifyc C s" t �t j�}d|_| rt j|_|S r# )r$ r% ZPROTOCOL_TLS_CLIENTr* r+ r, )r/ Zclient_contextr r r �simple_client_sslcontextR s
r0 c C s t d u rd S tdd�S )NTr. )r$ r0 r r r r �dummy_ssl_contextZ s
r1 c C sB dd� }|� }| � |�}d|_z | �|� W |�� d S |�� w )Nc � s �d S �Nr r r r r �onceb s �zrun_briefly.<locals>.onceF)Z create_taskZ_log_destroy_pending�run_until_complete�close)�loopr3 �gen�tr r r � run_brieflya s
r9 c C sT t �� | }|� s(|d ur|t �� }|dkrt�� �| �t�d�� |� r d S d S )Nr g����MbP?)�time� monotonicr � TimeoutErrorr4 r
�sleep)r6 Zpred�timeout�deadliner r r � run_untilo s �r@ c C s | � | j� | �� d S r2 )Z call_soon�stopZ run_forever�r6 r r r �run_oncey s rC c @ � e Zd Zdd� Zdd� ZdS )�SilentWSGIRequestHandlerc C s t �� S r2 )�io�StringIO��selfr r r �
get_stderr� s z#SilentWSGIRequestHandler.get_stderrc G � d S r2 r )rI �format�argsr r r � log_message� � z$SilentWSGIRequestHandler.log_messageN)�__name__�
__module__� __qualname__rJ rN r r r r rE � s rE c s* e Zd ZejZ� fdd�Zdd� Z� ZS )�SilentWSGIServerc s" t � �� \}}|�| j� ||fS r2 ��super� get_request�
settimeout�request_timeout�rI �requestZ client_addr�� __class__r r rV � s z SilentWSGIServer.get_requestc C rK r2 r �rI rZ �client_addressr r r � handle_error� rO z SilentWSGIServer.handle_error) rP rQ rR r �LOOPBACK_TIMEOUTrX rV r_ �
__classcell__r r r[ r rS � s rS c @ � e Zd Zdd� ZdS )�SSLWSGIServerMixinc C sX t �t j�}|�tt� |j|dd�}z| �||| � |�� W d S t y+ Y d S w )NT)Z server_side)
r$ r% r& r' r( r) Z wrap_socketZRequestHandlerClassr5 �OSError)rI rZ r^ �contextZssockr r r �finish_request� s �z!SSLWSGIServerMixin.finish_requestN)rP rQ rR rf r r r r rc � � rc c @ � e Zd ZdS )�
SSLWSGIServerN�rP rQ rR r r r r ri � � ri F)�use_sslc # s� � dd� ��fdd�}|r|n|}|| t �� � �|� � j� _tj� fdd�d�}|�� z� V W � �� � �� |� � d S � �� � �� |� � w )Nc s sD � t | d �}|r | d �t|d��}|V |t|�8 }|s d S d S )NZCONTENT_LENGTHz
wsgi.inputi )�int�read�min�len)�environ�size�datar r r r6 � s � �z _run_test_server.<locals>.loopc s. d}dg}|||� | d dkr� | �S dgS )Nz200 OK)z Content-typez
text/plainZ PATH_INFOz/loops Test messager )rq Zstart_response�statusZheadersrB 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)
rE Zset_appZserver_address�address� threading�Thread�start�shutdownZ server_closer )ry rl �
server_cls�server_ssl_clsru Z server_classZ
server_threadr )rv r6 r �_run_test_server� s& �
� �
r� �AF_UNIXc @ rb )�UnixHTTPServerc C s t j�| � d| _d| _d S )N� 127.0.0.1�P )� socketserver�UnixStreamServer� server_bindZ server_nameZ server_portrH r r r r� � s
zUnixHTTPServer.server_bindN)rP rQ rR r� r r r r r� � rg r� c s* e Zd ZejZdd� Z� fdd�Z� ZS )�UnixWSGIServerc C s t �| � | �� d S r2 )r� r� Z
setup_environrH r r r r� � s
zUnixWSGIServer.server_bindc s" t � �� \}}|�| j� |dfS )N)r� � rT rY r[ r r rV � s zUnixWSGIServer.get_request) rP rQ rR r r` rX r� rV ra r r r[ r r� � s r� c @ rb )�SilentUnixWSGIServerc C rK r2 r r] r r r r_ � rO z!SilentUnixWSGIServer.handle_errorN)rP rQ rR r_ r r r r r� � rg r� c @ rh )�UnixSSLWSGIServerNrj r r r r r� � rk r� c C s2 t �� � } | jW d � S 1 sw Y d S r2 )�tempfile�NamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s
$�r� c c sZ � t � } z| V W zt�| � W d S ty Y d S w zt�| � W w ty, Y w w r2 )r� r �unlinkrd )r r r r �unix_socket_path s � �� �r� c c sD � t � �}t|| ttd�E d H W d � d S 1 sw Y d S �N)ry rl r~ r )r� r� r� r� )rl r r r r �run_test_unix_server
s ��"�r� r� ��host�portrl c c s � t | |f|ttd�E d H d S r� )r� rS ri r� r r r �run_test_server s
�
�r� c C sL i }t | �D ]}|�d�r|�d�rqtd d�||< qtd| 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 r2 ��keysrH r r r �__init__( �
zTestSelector.__init__Nc C s t �|d||�}|| j|< |S )Nr )� selectors� SelectorKeyr� )rI �fileobjr rs �keyr r r �register+ s
zTestSelector.registerc C s | j �|�S r2 )r� �pop)rI r� r r r �
unregister0 � zTestSelector.unregisterc C s g S r2 r )rI r> r r r �select3 rO zTestSelector.selectc C � | j S r2 r� rH r r r �get_map6 � zTestSelector.get_mapr2 )rP rQ rR r� r� r� r� r� r r r r r� & s
r� c s� e Zd Z d-� 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 rdd� }d| _nd| _|� | _t| j� d| _d| _g | _t� | _ i | _
i | _ | � � t
�� | _d S )Nc s s � d V d S r2 r r r r r r7 Q s �
z TestLoop.__init__.<locals>.genFTr g��&� .>)rU r� �_check_on_close�_gen�next�_timeZ_clock_resolution�_timersr� � _selector�readers�writers�reset_counters�weakref�WeakValueDictionary� _transports)rI r7 r[ r r r� M s
zTestLoop.__init__c C r� r2 �r� rH r r r r: d r� z
TestLoop.timec C s |r | j |7 _ d S d S r2 r� )rI �advancer r r � advance_timeg s �zTestLoop.advance_timec s@ t � �� | jr z | j�d� W td�� ty Y d S w d S )Nr z Time generator is not finished)rU r5 r� r� �send�
StopIteration�AssertionErrorrH r[ r r r5 l s
���zTestLoop.closec G � t �||| d �| j|< d S r2 )r �Handler� �rI �fd�callbackrM r r r � _add_readerv � zTestLoop._add_readerc C �, | j | d7 < || jv r| j|= dS dS �N� TF)�remove_reader_countr� �rI r� r r r �_remove_readery �
zTestLoop._remove_readerc G sh || j vr
td|� d���| j | }|j|kr"td|j� d|� ���|j|kr2td|j� d|� ���d S )N�fd � is not registeredzunexpected callback: � != zunexpected callback args: )r� r� � _callback�_args�rI r� r� rM �handler r r �
assert_reader� s
�
��zTestLoop.assert_readerc C s || j v r
td|� d���d S )Nr� z is registered)r� r� r� r r r �assert_no_reader� s
�zTestLoop.assert_no_readerc G r� r2 )r r� r� r� r r r � _add_writer� r� zTestLoop._add_writerc C r� r� )�remove_writer_countr� r� r r r �_remove_writer� r� zTestLoop._remove_writerc G sd || j vr
td|� d���| j | }|j|kr!t|j�d|����|j|kr0t|j�d|����d S )Nr� r� r� )r� r� r� r� r� r r r �
assert_writer� s
�zTestLoop.assert_writerc
C st t |t�s zt|�� �}W n tttfy td�|��d �w z| j| }W n
ty1 Y d S w t d�||���)NzInvalid file object: {!r}z.File descriptor {!r} is used by transport {!r})
�
isinstancerm �fileno�AttributeError� TypeError�
ValueErrorrL r� �KeyError� RuntimeError)rI r� Z transportr r r �_ensure_fd_no_transport� s(
��� ���z TestLoop._ensure_fd_no_transportc G � | � |� | j||g|�R � S r2 )r� r� r� r r r �
add_reader� �
zTestLoop.add_readerc C � | � |� | �|�S r2 )r� r� r� r r r �
remove_reader� �
zTestLoop.remove_readerc G r� r2 )r� r� r� r r r �
add_writer� r� zTestLoop.add_writerc C r� r2 )r� r� r� r r r �
remove_writer� r� zTestLoop.remove_writerc C s t �t�| _t �t�| _d S r2 )� collections� defaultdictrm r� r� rH r r r r� � s zTestLoop.reset_countersc s6 t � �� | jD ]
}| j�|�}| �|� qg | _d S r2 )rU � _run_oncer� r� r� r� )rI �whenr� r[ r r r� � s
zTestLoop._run_once)re c s( | j �|� t� j||g|�R d|i�S )Nre )r� �appendrU �call_at)rI r� r� re rM r[ r r r� � s zTestLoop.call_atc C rK r2 r )rI Z
event_listr r r �_process_events� rO zTestLoop._process_eventsc C rK r2 r rH r r r �_write_to_self� rO zTestLoop._write_to_selfr2 )rP rQ rR r� r: r� r5 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� ra r r r[ r r� : s.
r� c K s t jdddgi| ��S )N�spec�__call__r )r ZMock)�kwargsr r r r� � s r� c @ s e Zd Z dd� ZdS )� MockPatternc C s t t�t| �|tj��S r2 )�bool�re�search�str�S�rI �otherr r r �__eq__� s zMockPattern.__eq__N)rP rQ rR r r r r r r� � s r� c @ rD )�MockInstanceOfc C s
|| _ d S r2 )�_type)rI r� r r r r� � r� zMockInstanceOf.__init__c C s t || j�S r2 )r� r r r r r r � r� zMockInstanceOf.__eq__N)rP rQ rR r� r r r r r r � s r c C s$ t �| �}|d u rtd| 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 @ s@ e Zd Zedd� �Zdd�dd�Zddd �Zd
d � Zd d
� ZdS )�TestCasec C s� | j d ur| �� s| �| �� � n| j jdd� | �� t�� }|d urKz|�� }W n
t y4 Y d S w t
|t j �rMt
|j�� �}|D ]
}|�� qDd S d S d S )NT)�wait)Z_default_executorZ is_closedr4 Zshutdown_default_executorr} r5 r Zmaybe_get_event_loop_policyZget_child_watcher�NotImplementedErrorr� �asyncioZThreadedChildWatcher�listZ_threads�valuesr )r6 ZpolicyZwatcher�threads�threadr r r �
close_loop s&
�
�zTestCase.close_loopT)�cleanupc C s4 |d u rt d��t�d � |r| �| j|� d S d S )Nz loop is None)r� r �set_event_loopZ
addCleanupr )rI r6 r r r r r s
�zTestCase.set_event_loopNc C s t |�}| �|� |S r2 )r� r )rI r7 r6 r r r �
new_test_loop s
zTestCase.new_test_loopc C s t �� | _d S r2 )r
Zthreading_setup�_thread_cleanuprH r r r �setUp" s zTestCase.setUpc C s: t �d � | �t�� d� | �� tj| j� t �
� d S )N)NNN) r r Z assertEqual�sys�exc_infoZ
doCleanupsr
Zthreading_cleanupr r �
reap_childrenrH r r r �tearDown% s
zTestCase.tearDownr2 ) rP rQ rR � staticmethodr r r r r r r r r r � s
r c c s> � t j} zt �tjd � d V W t �| � d S t �| � w )Nr� )r �level�setLevel�logging�CRITICAL)Z old_levelr r r �disable_logger1 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_socket? s r( )Pr r� �
contextlibrF r r r� r� r# r� r r� rz r: Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r$ � ImportErrorr r r r r
Z asyncio.logr �testr Z test.supportr
r r( r) ZSIGNED_CERTFILEZ
SIGNING_CAZPEERCERTr- r0 r1 r9 Z
SHORT_TIMEOUTr@ rC rE rS rc ri r� r r� r� 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� � �
#
&
2
�