관리-도구
편집 파일: test_utils.cpython-36.opt-2.pyc
3 \�: � @ 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)�supportZwin32)� 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 �*/usr/lib64/python3.6/asyncio/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 r0 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 )r4 �format�argsr r r � log_messagel s z$SilentWSGIRequestHandler.log_messageN)�__name__� __module__� __qualname__r5 r8 r r r r r1 g s r1 c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�SilentWSGIServer� c s" t � j� \}}|j| j� ||fS )N)�super� get_request� settimeout�request_timeout)r4 �request� client_addr)� __class__r r r? t s z SilentWSGIServer.get_requestc C s d S )Nr )r4 rB �client_addressr r r � handle_errory s z SilentWSGIServer.handle_error)r9 r: r; rA r? rF � __classcell__r r )rD 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)r4 rB rE ZkeyfileZcertfile�contextZssockr r r �finish_request s z!SSLWSGIServerMixin.finish_requestN)r9 r: r; rM r r r r rH } s rH c @ s e Zd ZdS )� SSLWSGIServerN)r9 r: r; r r r r rN � s rN 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)rP rQ r )�environZstart_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>� s z"_run_test_server.<locals>.<lambda>)�target) r1 Zset_appZserver_address�address� threadingZThread�start�shutdownZ server_closer )rW rO � server_cls�server_ssl_clsrS Z server_classZ server_threadr )rT 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)r4 r r r rb � s zUnixHTTPServer.server_bindN)r9 r: r; rb 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^ rb Z setup_environ)r4 r r r rb � s zUnixWSGIServer.server_bindc s" t � j� \}}|j| j� |dfS )N� 127.0.0.1� )rd re )r> r? r@ rA )r4 rB rC )rD r r r? � s zUnixWSGIServer.get_request)r9 r: r; rA rb r? rG r r )rD r rc � s rc c @ s e Zd Zdd� ZdS )�SilentUnixWSGIServerc C s d S )Nr )r4 rB rE r r r rF � s z!SilentUnixWSGIServer.handle_errorN)r9 r: r; rF r r r r rf � s rf c @ s e Zd ZdS )�UnixSSLWSGIServerN)r9 r: r; r r r r rg � s rg c C s t j� �} | jS Q R X d S )N)�tempfileZNamedTemporaryFile�name)�filer r r �gen_unix_socket_path� s rk c c s< t � } z | V W d ytj| � W n tk r4 Y nX X d S )N)rk r �unlinkrK )r r r r �unix_socket_path� s rm c c s, t � � }t|| ttd�E d H W d Q R X d S )N)rW rO r[ r\ )rm r] rf rg )rO r r r r �run_test_unix_server� s rn z 127.0.0.1)�host�portrO c c s t | |f|ttd�E d H d S )N)rW rO r[ r\ )r] r< rN )ro rp rO r r r �run_test_server� s rq 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__)�baseZdctri 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)r4 r r r �__init__ s zTestSelector.__init__Nc C s t j|d||�}|| j|< |S )Nr )r Z SelectorKeyr} )r4 �fileobjr �data�keyr r r �register s zTestSelector.registerc C s | j j|�S )N)r} �pop)r4 r r r r � unregister s zTestSelector.unregisterc C s g S )Nr )r4 r- r r r �select s zTestSelector.selectc C s | j S )N)r} )r4 r r r �get_map s zTestSelector.get_map)N)r9 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)r4 r' )rD r r r~ ( s zTestLoop.__init__c C s | j S )N)r� )r4 r r r r+ ? s z TestLoop.timec C s |r| j |7 _ d S )N)r� )r4 �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)r4 )rD r r r% G s zTestLoop.closec G s t j||| �| j|< d S )N)r �Handler� )r4 �fd�callbackr7 r r r � _add_readerQ s zTestLoop._add_readerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)�remove_reader_countr� )r4 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)r4 r� r� r7 �handler r r � assert_reader\ s zTestLoop.assert_readerc C s || j krtd|� d���d S )Nzfd z is registered)r� r� )r4 r� r r r �assert_no_readerg s zTestLoop.assert_no_readerc G s t j||| �| j|< d S )N)r r� r� )r4 r� r� r7 r r r � _add_writerk s zTestLoop._add_writerc C s0 | j | d7 < || jkr(| j|= dS dS d S )Nr TF)�remove_writer_countr� )r4 r� r r r �_remove_writern s zTestLoop._remove_writerc G s | j | }d S )N)r� )r4 r� r� r7 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� RuntimeErrorr6 )r4 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� )r4 r� r� r7 r r r � add_reader� s zTestLoop.add_readerc C s | j |� | j|�S )N)r� r� )r4 r� r r r � remove_reader� s zTestLoop.remove_readerc G s | j |� | j||f|�� S )N)r� r� )r4 r� r� r7 r r r � add_writer� s zTestLoop.add_writerc C s | j |� | j|�S )N)r� r� )r4 r� r r r � remove_writer� s zTestLoop.remove_writerc C s t jt�| _t jt�| _d S )N)� collections� defaultdict�intr� r� )r4 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� )r4 �whenr� )rD r r r� � s zTestLoop._run_oncec s | j j|� t� j||f|�� S )N)r� �appendr> �call_at)r4 r� r� r7 )rD r r r� � s zTestLoop.call_atc C s d S )Nr )r4 Z event_listr r r �_process_events� s zTestLoop._process_eventsc C s d S )Nr )r4 r r r �_write_to_self� s zTestLoop._write_to_self)N)r9 r: r; r~ r+ r� r% r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rG r r )rD r r� s* r� c K s t jf ddgi| ��S )N�spec�__call__)r ZMock)�kwargsr r r rw � s rw c @ s e Zd Zdd� ZdS )� MockPatternc C s t tjt| �|tj��S )N)�bool�re�search�str�S)r4 �otherr r r �__eq__� s zMockPattern.__eq__N)r9 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_executorrZ r% )r& Zexecutorr r r � close_loop� s zTestCase.close_loopT)�cleanupc C s t jd � |r | j| j|� d S )N)r �set_event_loopZ addCleanupr� )r4 r& r� r r r r� � s zTestCase.set_event_loopNc C s t |�}| j|� |S )N)r� r� )r4 r' r& r r r � new_test_loop� s zTestCase.new_test_loopc C s | j t_ d S )N)�_get_running_loopr )r4 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 rU � s z TestCase.setUp.<locals>.<lambda>)r r� r Zthreading_setup�_thread_cleanup)r4 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)r4 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 )r4 r r r � __enter__� s z+TestCase.subTest.<locals>.EmptyCM.__enter__c W s d S )Nr )r4 �excr r r �__exit__� s z*TestCase.subTest.<locals>.EmptyCM.__exit__N)r9 r: r; r� r� r r r r �EmptyCM� s r� r )r4 r7 r� r� r r r �subTest� s zTestCase.subTest)N) r9 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�protorx �familyZ gettimeoutrs )r� rx r� Zsockr r r �mock_nonblocking_socket s r� c C s t jddd�S )Nz'asyncio.sslproto._is_sslproto_availableF)rs )r Zpatchr r r r �force_legacy_ssl_support s r� )r* )Mr� � contextlibr2 r� r r� r� r` r� rh rX r+ Zunittestr� r Z http.serverr Zwsgiref.simple_serverr r r � ImportErrorre r r r r r r Z coroutinesr �logr r r �platformZ windows_utilsr r rJ rI r" r) r. r0 r1 r< rH rN r] r ra r^ rc rf rg rk �contextmanagerrm rn rq r{ Z BaseSelectorr| Z BaseEventLoopr� rw r� r� r� r� r� Z IPPROTO_TCPZ SOCK_STREAMZAF_INETr� r� r r r r �<module> s� 4