File: //opt/saltstack/salt/lib/python3.10/site-packages/cheroot/__pycache__/testing.cpython-310.pyc
o
�N�gL � @ s d Z ddlmZmZmZ eZddlmZ ddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlmZ ddlZddlmZ ddlZdZdZdZdZejjeefdd �ejj eefejj d
�iZ d d � Z e
j d
d� �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.� )�absolute_import�division�print_function)�closingN)� http_client)�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 sEt �
d� |j r=|V |�� dS )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)Zserver_factoryZconfZ bind_port� interfaceZactual_bind_addrZ
httpserverr r �C/opt/saltstack/salt/lib/python3.10/site-packages/cheroot/testing.py�cheroot_server$ s, � �
� �
� r c c � � t tjj�D ]} | V qdS )z4Set up and tear down a Cheroot WSGI server instance.N)r �cheroot�wsgi�Server�Zsrvr r r � wsgi_server@ � ��r! c c r )z4Set up and tear down a Cheroot HTTP server instance.N)r r �server�
HTTPServerr r r r �
native_serverG r" r% c @ s2 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__O s
�z_TestClient.__init__c C s2 dj | j| jd�}| jjd u rtjntj}||�S )Nz{interface}:{port})r �port)�formatr( r* r+ Z ssl_adapterr �HTTPConnection�HTTPSConnection)r. �nameZconn_clsr r r r, V s � ��z_TestClient.get_connection�GETN�HTTP/1.1c C s" t j|||| j| j|p
| j|d�S )N)�method�headers�hostr0 � http_conn�protocol)r ZopenURLr) r* r- )r. �urir7 r8 r: r; r r r �requestb s �z_TestClient.requestc s � �fdd�}|S )Nc s � � � }�j| fd|i|��S )Nr7 )�upperr= )r< �kwargsZ http_method�� attr_namer. r r �_wrappero s z)_TestClient.__getattr__.<locals>._wrapperr )r. rA rB r r@ r � __getattr__n s z_TestClient.__getattr__)r5 NNr6 )�__name__�
__module__� __qualname__r/ r, r= rC r r r r r& N s
� r& c
C s� z%t tjtjd���}|�| df� W d � W dS 1 s w Y W dS ttjfyB } z|jtjkr7� W Y d }~dS d }~ww )N)�familyr TF)r �socket�AF_INET6�bindr �error�errnoZ
EADDRNOTAVAIL)r �sockZsock_errr r r �_probe_ipv6_sockv s ���
���rN c C sR t | t�r
| \}}n| d}}t�|�}d|v r$t|�s$d}d|v r$|}|||fS )Nr �:z 127.0.0.1)�
isinstance�tupler r rN )r r9 r0 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 rR )'�__doc__Z
__future__r r r �typeZ
__metaclass__�
contextlibr rL rH r r ZpytestZ six.movesr Zcheroot.serverr Z cheroot.testr Z cheroot.wsgiZEPHEMERAL_PORTZ NO_INTERFACEr r r r r# r$ ZGatewayr r Zfixturer! r% r& rN r' rR r r r r �<module> sF ���
(