D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
usr
/
lib
/
python3.9
/
site-packages
/
supervisor
/
tests
/
__pycache__
/
Filename :
test_http_client.cpython-39.pyc
back
Copy
a 0�g�5 � @ sx d dl Z d dlZd dlZd dlmZ d dlmZ G dd� dej�ZG dd� dej�ZG dd � d e �Z G d d � d e �Z dS ) � N)�as_bytes)�StringIOc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dS )� ListenerTestsc C s ddl m} |S )Nr )�Listener)�supervisor.http_clientr )�selfr � r �E/usr/lib/python3.9/site-packages/supervisor/tests/test_http_client.py�_getTargetClass s z ListenerTests._getTargetClassc C s | � � � S �N)r �r r r r �_makeOne s zListenerTests._makeOnec C s | � � }| �|�d d �d � d S r )r � assertEqual�status�r �instr r r � test_status s zListenerTests.test_statusc C sV | � � }z@tj}t� }|t_| �|�dd�d � | �|�� d� W |t_n|t_0 d S )N�url�errorz url error )r �sys�stderrr r r �getvalue)r r Z old_stderrr r r r � test_error s zListenerTests.test_errorc C s" | � � }| �|�d d d �d � d S r )r r �response_headerr r r r �test_response_header s z"ListenerTests.test_response_headerc C s | � � }| �|�d �d � d S r )r r �doner r r r � test_done# s zListenerTests.test_donec C sN | � � }z8tj}t� }|t_|�dd� | �|�� d� W |t_n|t_0 d S )Nr �data)r r �stdoutr �feedr r )r r Z old_stdoutr r r r � test_feed' s zListenerTests.test_feedc C s | � � }| �|�d �d � d S r )r r �closer r r r � test_close2 s zListenerTests.test_closeN) �__name__� __module__� __qualname__r r r r r r r r" r r r r r s r c @ s6 e Zd Zdd� ZdMdd�Zdd� Zd d � Zd d � Zd d� Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z d?d@� Z!dAdB� Z"dCdD� Z#dEdF� Z$dGdH� Z%dIdJ� Z&dKdL� Z'dS )N�HTTPHandlerTestsc C s ddl m} |S )Nr )� HTTPHandler)r r' )r r' r r r r 7 s z HTTPHandlerTests._getTargetClassN� c C s( |d u r| � � }i }| �� ||||d�S )N)�map)� _makeListenerr )r �listener�username�passwordZ socket_mapr r r r ; s �zHTTPHandlerTests._makeOnec C s t � }|S r )� DummyListener)r r+ r r r r* F s z HTTPHandlerTests._makeListenerc C s" | � � }d|_| �t|jd� d S )N�abc)r r � assertRaises�AssertionError�getr r r r �test_get_url_not_NoneJ s z&HTTPHandlerTests.test_get_url_not_Nonec C s | � � }| �t|jdd� d S )Nznothttp://localhostz/abc)r r0 �NotImplementedErrorr2 r r r r �test_get_bad_schemeO s �z$HTTPHandlerTests.test_get_bad_schemec sn | � � }g �g � �fdd�|_� fdd�|_|�dd� | �|jd� | ��tjtjfg� | �� dg� d S )Nc s � � | �S r ��append��arg��socketsr r �<lambda>\ � z;HTTPHandlerTests.test_get_implied_port_80.<locals>.<lambda>c s � � | �S r r6 ��tup��connectsr r r< ] r= zhttp://localhost�/abc/def�P )� localhostrC � r � create_socket�connectr2 r �port�socketZAF_INET� SOCK_STREAMr r �rA r; r �test_get_implied_port_80X s z)HTTPHandlerTests.test_get_implied_port_80c sn | � � }g �g � �fdd�|_� fdd�|_|�dd� | �|jd� | ��tjtjfg� | �� dg� d S )Nc s � � | �S r r6 r8 r: r r r< g r= z9HTTPHandlerTests.test_get_explicit_port.<locals>.<lambda>c s � � | �S r r6 r> r@ r r r< h r= zhttp://localhost:8080rB � )rD rM rE r r rK r �test_get_explicit_portc s z'HTTPHandlerTests.test_get_explicit_portc s` | � � }g �g � �fdd�|_� fdd�|_|�dd� | ��tjtjfg� | �� dg� d S )Nc s � � | �S r r6 r8 r: r r r< r r= zGHTTPHandlerTests.test_get_explicit_unix_domain_socket.<locals>.<lambda>c s � � | �S r r6 r> r@ r r r< s r= z unix:///a/b/cr( z/a/b/c)r rF rG r2 r rI ZAF_UNIXrJ r r rK r �$test_get_explicit_unix_domain_socketn s z5HTTPHandlerTests.test_get_explicit_unix_domain_socketc sv | � � }g � � fdd�|_t� |_|�� | �|jjd � | �|jd� | �� dg� | � |jj� | �|j d� d S )Nc s � � d�S �NTr6 r �Zdelsr r r< { r= z-HTTPHandlerTests.test_close.<locals>.<lambda>r TZCLOSED) r Z del_channel� DummySocketrI r! r r+ �closed� connected� assertTruer r r rQ r r" x s zHTTPHandlerTests.test_closec sF ddl m} | �� }g � � fdd�|_|�dd� | �� d|g� d S )Nr ��CRLFc s � � | �S r r6 ��val�Zpushesr r r< � r= z.HTTPHandlerTests.test_header.<locals>.<lambda>�namerY z name: val)r rW r �push�headerr �r rW r r rZ r � test_header� s z HTTPHandlerTests.test_headerc C s" | � � }d|_| �|�� d � d S rP )r � error_handledr � handle_errorr r r r �!test_handle_error_already_handled� s z2HTTPHandlerTests.test_handle_error_already_handledc sn | � � }g � � fdd�|_d|_| �|�� d � | �|jjd� | �|jjd� | �� dg� | �|j � d S )Nc s � � d�S rP r6 r �rS r r r< � r= z4HTTPHandlerTests.test_handle_error.<locals>.<lambda>�fooz"Cannot connect, error: None (None)T) r r! r r ra r+ � error_url� error_msgrU r` r r rc r �test_handle_error� s �z"HTTPHandlerTests.test_handle_errorc sN | � � }g � � fdd�|_d|_d|_|�� | �|j� | �� g d�� d S )Nc s � � t| ��S r �r7 r rX �Zpushedr r r< � r= zBHTTPHandlerTests.test_handle_connect_no_password.<locals>.<lambda>�/rD ) � GET / HTTP/1.1� � Host: localhostrl � Accept-Encoding: chunkedrl � Accept: */*rl �" User-agent: Supervisor HTTP Clientrl rl rl )r r\ �path�host�handle_connectrU rT r r r ri r � test_handle_connect_no_password� s �z0HTTPHandlerTests.test_handle_connect_no_passwordc sZ | � � }g � � fdd�|_d|_d|_d|_d|_|�� | �|j� | � � g d�� d S )Nc s � � t| ��S r rh rX ri r r r< � r= zDHTTPHandlerTests.test_handle_connect_with_password.<locals>.<lambda>rj rD r- r, )rk rl rm rl rn rl ro rl rp rl s- Authorization: Basic dXNlcm5hbWU6cGFzc3dvcmQ=rl rl rl ) r r\ rq rr r- r, rs rU rT r r r ri r �!test_handle_connect_with_password� s �z2HTTPHandlerTests.test_handle_connect_with_passwordc C s( | � � }|�d� | �|jjdg� d S )Nr )r r r r+ �fed_datar r r r r � s zHTTPHandlerTests.test_feedc C sD | � � }|j|_d|_|�d� | �|jjdg� | �|jd� d S )Nr/ rd �abcfoor= )r �body�part�buffer�collect_incoming_datar r+ rv r r r r �'test_collect_incoming_data_part_is_body� s z8HTTPHandlerTests.test_collect_incoming_data_part_is_bodyc C s@ | � � }d |_d|_|�d� | �|jjg � | �|jd� d S )Nr/ rd rw )r ry rz r{ r r+ rv r r r r �+test_collect_incoming_data_part_is_not_body� s z<HTTPHandlerTests.test_collect_incoming_data_part_is_not_bodyc sH | � � }g � � fdd�|_d |_|�� | �� dg� | �|jd� d S )Nc s � � d�S rP r6 r �Zpartedr r r< � r= z8HTTPHandlerTests.test_found_terminator.<locals>.<lambda>Tr= )r ry rz Zfound_terminatorr r r r~ r �test_found_terminator� s z&HTTPHandlerTests.test_found_terminatorc C s( | � � }d |_|�� | �|jd� d S �Nr= )r rz �ignorer r r r r � test_ignore� s z HTTPHandlerTests.test_ignorec C s | � � }d|_| �t|j� d S )Ns NOTHTTP/1.0 200 OK)r rz r0 � ValueError� status_liner r r r �$test_status_line_not_startswith_http� s z5HTTPHandlerTests.test_status_line_not_startswith_httpc C sT | � � }d|_|�� \}}}| �|d� | �|d� | �|d� | �|j|j� d S )Ns HTTP/1.0 200 OK� HTTP/1.0�� � OK)r rz r� r ry �headers�r r �versionr �reasonr r r �test_status_line_200� s z%HTTPHandlerTests.test_status_line_200c s� | � � }d|_g � � fdd�|_|�� \}}}| �|d� | �|d� | �|d� | �|j|j� | �|jjd� | �� dg� d S ) Ns HTTP/1.0 201 OKc s � � d�S rP r6 r rc r r r< r= z;HTTPHandlerTests.test_status_line_not_200.<locals>.<lambda>r� �� r� z Cannot read, status code 201T) r rz r! r� r ry r� r+ rf r� r rc r �test_status_line_not_200 s �z)HTTPHandlerTests.test_status_line_not_200c sV | � � }d|_d|_d|_g � � fdd�|_|�� | �|j|j� | �� dg� d S )Nr= s not chunked� c s � � | �S r r6 )�L��termsr r r< r= zEHTTPHandlerTests.test_headers_empty_line_nonchunked.<locals>.<lambda>) r rz �encoding�length�set_terminatorr� r ry rx r r r� r �"test_headers_empty_line_nonchunked s z3HTTPHandlerTests.test_headers_empty_line_nonchunkedc C s0 | � � }d|_d|_|�� | �|j|j� d S )Nr= � chunked)r rz r� r� r ry � chunked_sizer r r r � test_headers_empty_line_chunked s z0HTTPHandlerTests.test_headers_empty_line_chunkedc C s" | � � }d|_| �|�� d � d S )N� :)r rz r r� r r r r �+test_headers_nonempty_line_no_name_no_value' s z<HTTPHandlerTests.test_headers_nonempty_line_no_name_no_valuec sH | � � }d|_g � � fdd�|_|�� | �|jd� | �� dg� d S )Ns Transfer-Encoding: chunkedc s � � | |f�S r r6 ��n�v�Z responsesr r r< 0 r= zOHTTPHandlerTests.test_headers_nonempty_line_transfer_encoding.<locals>.<lambda>r� )s transfer-encodingr� )r rz r r� r r� r r r� r �,test_headers_nonempty_line_transfer_encoding, s z=HTTPHandlerTests.test_headers_nonempty_line_transfer_encodingc sH | � � }d|_g � � fdd�|_|�� | �|jd� | �� dg� d S )Ns Content-Length: 3c s � � | |f�S r r6 r� r� r r r< 9 r= zLHTTPHandlerTests.test_headers_nonempty_line_content_length.<locals>.<lambda>r� )s content-length� 3)r rz r r� r r� r r r� r �)test_headers_nonempty_line_content_length5 s z:HTTPHandlerTests.test_headers_nonempty_line_content_lengthc s: | � � }d|_g � � fdd�|_|�� | �� dg� d S )Ns X-Test: abcc s � � | |f�S r r6 r� r� r r r< B r= zGHTTPHandlerTests.test_headers_nonempty_line_arbitrary.<locals>.<lambda>)s x-tests abc)r rz r r� r r r r� r �$test_headers_nonempty_line_arbitrary> s z5HTTPHandlerTests.test_headers_nonempty_line_arbitraryc C s8 | � � }|�dd� | �|jjd� | �|jjd� d S )N� a� b)r r r r+ �response_header_name�response_header_valuer r r r r F s z%HTTPHandlerTests.test_response_headerc sB | � � }g � � fdd�|_|�� | �� dg� | �|jj� d S )Nc s � � d�S rP r6 r rc r r r< O r= z,HTTPHandlerTests.test_body.<locals>.<lambda>T)r r! rx r rU r+ r r r rc r � test_bodyL s zHTTPHandlerTests.test_bodyc C s" | � � }|�� | �|jj� d S r )r r rU r+ r r r r r T s zHTTPHandlerTests.test_donec C s6 | � � }d|_d|_| �|�� d � | �|jd� d S )Nr= � )r rz r� r r� r r r r � test_chunked_size_empty_lineY s z-HTTPHandlerTests.test_chunked_size_empty_linec C sF | � � }d|_d|_| �|�� d � | �|jd� | �|j|j� d S )N� 0r� )r rz r� r r� ry �trailerr r r r �test_chunked_size_zero_size` s z,HTTPHandlerTests.test_chunked_size_zero_sizec sf | � � }d|_d|_g � � fdd�|_| �|�� d � | �|j|j� | �|jd� | �� dg� d S )Ns 10r� c s � � | �S r r6 )Zszr� r r r< m r= zAHTTPHandlerTests.test_chunked_size_nonzero_size.<locals>.<lambda>� � )r rz r� r� r r� ry � chunked_bodyr r r� r � test_chunked_size_nonzero_sizeh s z/HTTPHandlerTests.test_chunked_size_nonzero_sizec sv ddl m} | �� }d|_g �g � �fdd�|_� fdd�|_|�� | ��|g� | �� dg� | �|j|j � d S )Nr rV s bufferc s � � | �S r r6 �r� r� r r r< y r= z4HTTPHandlerTests.test_chunked_body.<locals>.<lambda>c s � � | �S r r6 r� )�linesr r r< z r= ) r rW r rz r� r r� r ry r� r^ r )r� r� r �test_chunked_bodys s z"HTTPHandlerTests.test_chunked_bodyc C s" | � � }d|_| �|�� d � d S r� )r rz r r� r r r r �test_trailer_line_not_crlf� s z+HTTPHandlerTests.test_trailer_line_not_crlfc sn ddl m} | �� }||_g �g � �fdd�|_� fdd�|_| �|�� d � | ��dg� | �� dg� d S )Nr rV c s � � d�S rP r6 r )�donesr r r< � r= z9HTTPHandlerTests.test_trailer_line_crlf.<locals>.<lambda>c s � � d�S rP r6 r )�closesr r r< � r= T)r rW r rz r r! r r� r^ r )r� r� r �test_trailer_line_crlf� s z'HTTPHandlerTests.test_trailer_line_crlf)Nr( N)(r# r$ r% r r r* r3 r5 rL rN rO r" r_ rb rg rt ru r r| r} r r� r� r� r� r� r� r� r� r� r� r r� r r� r� r� r� r� r� r r r r r& 6 sJ r& c @ sT e Zd ZdZdZdZdZdd� Zdd� Zdd� Z d d � Z d d � Z d d� Z dd� ZdS )r. NFc C s g | _ d S r )rv r r r r �__init__� s zDummyListener.__init__c C s || _ d S r rc �r r r r r r! � s zDummyListener.closec C s || _ || _d S r )re rf )r r �msgr r r r � s zDummyListener.errorc C s | j �|� d S r )rv r7 )r r r r r r r � s zDummyListener.feedc C s || _ || _d S r )Z status_urlZ status_int)r r �intr r r r � s zDummyListener.statusc C s || _ || _d S r )r� r� )r r r[ �valuer r r r � s z DummyListener.response_headerc C s d| _ d S rP )r r� r r r r � s zDummyListener.done) r# r$ r% rS re rf r r� r! r r r r r r r r r. � s r. c @ s e Zd ZdZdd� ZdS )rR Fc C s d| _ d S rP rc r r r r r! � s zDummySocket.closeN)r# r$ r% rS r! r r r r rR � s rR ) rI r ZunittestZsupervisor.compatr r ZTestCaser r& �objectr. rR r r r r �<module> s . ]