File: //opt/saltstack/salt/lib/python3.10/site-packages/cheroot/__pycache__/testing.cpython-310.pyc
o
;j� � @ s d Z ddlZddlZddlZddlZddlZddlmZm Z ddl
Z
ddl Z ddl
Z ddlmZ dZdZdZdZdZe jjeefdd�e jjeefe jjd �iZe d
d � �Ze
j d d
� �Z e
j dd� �Z e
j dd� �Z e
j dd� �Z G dd� d�Z!dd� Z"dd� Z#dd� Z$dS ) zAPytest fixtures and other helpers for doing testing by end-users.� N)�closing�contextmanager)�webtestz0.0.0.0z::� )� bind_addrZwsgi_app)r Zgatewayc c s� � t | �� }|�d�d }ttfD ]}z||f}| dd|i|��}W n ty, Y qw d|_tj|j d�}|�
� |j sGt �
d� |j r?z||fV W |�� |�� dS |�� |�� w )z/Set up and tear down a Cheroot server instance.r ���r )�targetg�������?N� )�config�copy�pop�ANY_INTERFACE_IPV6�ANY_INTERFACE_IPV4�OSErrorZshutdown_timeout� threading�ThreadZ
safe_start�start�ready�time�sleep�stop�join)Zserver_factoryZconfZ bind_port� interfaceZactual_bind_addrZ
httpserver�
server_threadr r �C/opt/saltstack/salt/lib/python3.10/site-packages/cheroot/testing.py�cheroot_server& s6 � �
� �
� �
r c c �B � t tjj��\} }| |fV W d � dS 1 sw Y dS )ztSet up and tear down a Cheroot WSGI server instance.
This emits a tuple of a thread and a server instance.
N)r �cheroot�wsgi�Server�r �srvr r r �thread_and_wsgi_serverH � � "�r" c c r )ztSet up and tear down a Cheroot HTTP server instance.
This emits a tuple of a thread and a server instance.
N)r r �server�
HTTPServerr r r r �thread_and_native_serverR r# r& c C � | \}}|S )z4Set up and tear down a Cheroot WSGI server instance.r )r" �_server_threadr! r r r � wsgi_server\ � r) c C r' )z4Set up and tear down a Cheroot HTTP server instance.r )r& r( r! r r r �
native_serverc r* r+ c @ s6 e Zd Zdd� Zdd� Z d dd �Zd
d � ZdS )
� _TestClientc C s* t |j�\| _| _| _|| _| �� | _d S )N)�_get_conn_datar �
_interface�_host�_port�server_instance�get_connection�_http_connection)�selfr$ r r r �__init__k s
�z_TestClient.__init__c C s6 | j � d| j� �}| jjd u rtjjntjj}||�S )N�:)r. r0 r1 Z ssl_adapter�http�client�HTTPConnection�HTTPSConnection)r4 �nameZconn_clsr r r r2 r s ��z_TestClient.get_connection�GETN�HTTP/1.1c C s" t j|||| j| j|p
| j|d�S )N)�method�headers�host�port� http_conn�protocol)r ZopenURLr/ r0 r3 )r4 �urir> r? rB rC r r r �request{ s �z_TestClient.requestc s � �fdd�}|S )Nc s � � � }�j| fd|i|��S )Nr> )�upperrE )rD �kwargsZ http_method�� attr_namer4 r r �_wrapper� s z)_TestClient.__getattr__.<locals>._wrapperr )r4 rI rJ r rH r � __getattr__� s z_TestClient.__getattr__)r< NNr= )�__name__�
__module__� __qualname__r5 r2 rE rK r r r r r, j s
� r, c
C s� z%t tjtjd���}|�| df� W d � W dS 1 s w Y W dS ty? } z|jtjkr4� W Y d }~dS d }~ww )N)�familyr TF)r �socket�AF_INET6�bindr �errnoZ
EADDRNOTAVAIL)r �sockZsock_errr r r �_probe_ipv6_sock� s ���
���rU c C sR t | t�r
| \}}n| d}}t�|�}d|v r$t|�s$d}d|v r$|}|||fS )Nr r6 z 127.0.0.1)�
isinstance�tupler r rU )r r@ rA r r r r r- � s
r- c C s t | �S )z5Create and return a test client for the given server.)r, )r$ r r r �get_server_client� s rX )%�__doc__rS � http.clientr7 rP r r �
contextlibr r ZpytestZcheroot.serverr Z cheroot.wsgiZ cheroot.testr ZEPHEMERAL_PORTZ NO_INTERFACEr r
ZSUCCESSFUL_SUBPROCESS_EXITr r r$ r% ZGatewayr
r Zfixturer" r& r) r+ r, rU r- rX r r r r �<module> sN ���
!
+