File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/utils.cpython-310.pyc
o
�i�C � @ s& d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlZddlmZ ddlmZ ddlmZmZ zddlZW n eyg dZY nw ddlmZ ddlmZ ddlmZ dd lm Z dd
lm Z dd l m Z dd l m!Z! dd
l"m#Z# dd� Z$e$d�Z%e$d�Z&e$d�Z'e$d�Z(ddddddddd d d �
Z)d d � Z*d!d"�d#d$�Z+d%d&� Z,d'd(� Z-e!j.fd)d*�Z/d+d,� Z0G d-d.� d.e�Z1G d/d0� d0e�Z2G d1d2� d2�Z3G d3d4� d4e3e2�Z4d5d6�d7d8�Z5e6e d9��r=G d:d;� d;e
j7e�Z8G d<d=� d=e8e�Z9G d>d?� d?e9�Z:G d@dA� dAe3e:�Z;dBdC� Z<ej=dDdE� �Z>ej=d5d6�dFdG��Z?ej=dHdd5dI�dJdK��Z@dLdM� ZAG dNdO� dOejB�ZCG dPdQ� dQejD�ZEdRdS� ZFG dTdU� dUeG�ZHG dVdW� dW�ZIdXdY� ZJG dZd[� d[ejK�ZKej=d\d]� �ZLe jMe jNe jOfd^d_�ZPdS )`zUtilities shared by tests.� 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� | �� dS )z�Legacy API to run once through the event loop.
This is the recommended pattern for test code. It will poll the
selector once and run all callbacks scheduled in response to I/O
events.
N)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 ZdZd.� fdd� Zdd� Zdd� Z� fd d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Z� fd%d&�Zdd'�� fd(d)�
Zd*d+� Zd,d-� Z� ZS )/�TestLoopa� Loop for unittests.
It manages self time directly.
If something scheduled to be executed later then
on next loop iteration after all ready handlers done
generator passed to __init__ is calling.
Generator should be like this:
def gen():
...
when = yield ...
... = yield time_advance
Value returned by yield is absolute time of next scheduled handler.
Value passed to yield is time advance to move loop's time forward.
Nc 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 _ dS dS )zMove test time forward.Nr� )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 )zAdd a reader callback.)r� r� r� r r r �
add_reader� �
zTestLoop.add_readerc C � | � |� | �|�S )zRemove a reader callback.)r� r� r� r r r �
remove_reader� �
zTestLoop.remove_readerc G r� )zAdd a writer callback..)r� r� r� r r r �
add_writer� r� zTestLoop.add_writerc C r� )zRemove a writer callback.)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 �__doc__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 ZdZdd� ZdS )� MockPatternz�A regex based str with a fuzzy __eq__.
Use this helper with 'mock.assert_called_with', or anywhere
where a regex comparison between strings is needed.
For instance:
mock_call.assert_called_with(MockPattern('spam.*ham'))
c 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 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 � dV W t �| � dS t �| � w )zrContext manager to disable asyncio logger.
For example, it can be used to ignore warnings in debug mode.
r� N)r �level�setLevel�logging�CRITICAL)Z old_levelr r r �disable_logger1 s �r# c C s* t �tj�}| |_||_||_d|j_|S )z'Create a mock of a non-blocking socket.g )r Z MagicMock�socket�protor� �family�
gettimeoutr� )r% r� r& �sockr r r �mock_nonblocking_socket? s r) )Qr� r 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
�