Edit File: test_http_client.cpython-36.pyc
3 �k�`�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 �&/usr/lib/python3.6/test_http_client.py�_getTargetClass s z ListenerTests._getTargetClassc C s | j � � S )N)r )r r r r �_makeOne s zListenerTests._makeOnec C s | j � }| j|jd d �d � d S )N)r � assertEqual�status)r �instr r r � test_status s zListenerTests.test_statusc C sP | j � }z:tj}t� }|t_| j|jdd�d � | j|j� d� W d |t_X 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" | j � }| j|jd d d �d � d S )N)r r �response_header)r r r r r �test_response_header s z"ListenerTests.test_response_headerc C s | j � }| j|jd �d � d S )N)r r �done)r r r r r � test_done# s zListenerTests.test_donec C sH | j � }z2tj}t� }|t_|jdd� | j|j� d� W d |t_X 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 | j � }| j|jd �d � d S )N)r r �close)r r 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 kr| j � }i }| j� ||||d�S )N)�map)� _makeListenerr )r �listener�username�passwordZ socket_mapr r r r ; s zHTTPHandlerTests._makeOnec C s t � }|S )N)� DummyListener)r r( r r r r' F s z HTTPHandlerTests._makeListenerc C s" | j � }d|_| jt|jd� d S )N�abc)r r � assertRaises�AssertionError�get)r r r r r �test_get_url_not_NoneJ s z&HTTPHandlerTests.test_get_url_not_Nonec C s | j � }| jt|jdd� d S )Nznothttp://localhostz/abc)r r- �NotImplementedErrorr/ )r r r r r �test_get_bad_schemeO s z$HTTPHandlerTests.test_get_bad_schemec sn | j � }g �g � �fdd�|_� fdd�|_|jdd� | j|jd� | j�tjtjfg� | j� dg� d S ) Nc s � j | �S )N)�append)�arg)�socketsr r �<lambda>\ s z;HTTPHandlerTests.test_get_implied_port_80.<locals>.<lambda>c s � j | �S )N)r3 )�tup)�connectsr r r6 ] s zhttp://localhostz/abc/def�P � localhost)r: r9 ) r � create_socket�connectr/ r �port�socket�AF_INET� SOCK_STREAM)r r r )r8 r5 r �test_get_implied_port_80X s z)HTTPHandlerTests.test_get_implied_port_80c sn | j � }g �g � �fdd�|_� fdd�|_|jdd� | j|jd� | j�tjtjfg� | j� dg� d S ) Nc s � j | �S )N)r3 )r4 )r5 r r r6 g s z9HTTPHandlerTests.test_get_explicit_port.<locals>.<lambda>c s � j | �S )N)r3 )r7 )r8 r r r6 h s zhttp://localhost:8080z/abc/def� r: )r: rB ) r r; r< r/ r r= r> r? r@ )r r r )r8 r5 r �test_get_explicit_portc s z'HTTPHandlerTests.test_get_explicit_portc s` | j � }g �g � �fdd�|_� fdd�|_|jdd� | j�tjtjfg� | j� dg� d S )Nc s � j | �S )N)r3 )r4 )r5 r r r6 r s zGHTTPHandlerTests.test_get_explicit_unix_domain_socket.<locals>.<lambda>c s � j | �S )N)r3 )r7 )r8 r r r6 s s z unix:///a/b/cr% z/a/b/c)r r; r< r/ r r> ZAF_UNIXr@ )r r r )r8 r5 r �$test_get_explicit_unix_domain_socketn s z5HTTPHandlerTests.test_get_explicit_unix_domain_socketc sv | j � }g � � fdd�|_t� |_|j� | j|jjd � | j|jd� | j� dg� | j |jj� | j|j d� d S )Nc s � j d�S )NT)r3 r )�delsr r r6 { s z-HTTPHandlerTests.test_close.<locals>.<lambda>r TZCLOSED) r Z del_channel� DummySocketr> r r r( �closed� connected� assertTruer )r r r )rE r r x s zHTTPHandlerTests.test_closec sF ddl m} | j� }g � � fdd�|_|jdd� | j� d|g� d S )Nr )�CRLFc s � j | �S )N)r3 )�val)�pushesr r r6 � s z.HTTPHandlerTests.test_header.<locals>.<lambda>�namerK z name: val)r rJ r �push�headerr )r rJ r r )rL r � test_header� s z HTTPHandlerTests.test_headerc C s" | j � }d|_| j|j� d � d S )NT)r � error_handledr � handle_error)r r r r r �!test_handle_error_already_handled� s z2HTTPHandlerTests.test_handle_error_already_handledc sn | j � }g � � fdd�|_d|_| j|j� d � | j|jjd� | j|jjd� | j� dg� | j|j � d S )Nc s � j d�S )NT)r3 r )rG r r r6 � s z4HTTPHandlerTests.test_handle_error.<locals>.<lambda>�fooz"Cannot connect, error: None (None)T) r r r r rR r( � error_url� error_msgrI rQ )r r r )rG r �test_handle_error� s z"HTTPHandlerTests.test_handle_errorc sb | j � }g � � fdd�|_d|_d|_|j� | j|j� | j� ddddddd dd dddg � d S ) Nc s � j t| ��S )N)r3 r )rK )�pushedr r r6 � s zBHTTPHandlerTests.test_handle_connect_no_password.<locals>.<lambda>�/r: s GET / HTTP/1.1s s Host: localhosts Accept-Encoding: chunkeds Accept: */*s" User-agent: Supervisor HTTP Client)r rN �path�host�handle_connectrI rH r )r r r )rX r � test_handle_connect_no_password� s* z0HTTPHandlerTests.test_handle_connect_no_passwordc sr | j � }g � � fdd�|_d|_d|_d|_d|_|j� | j|j� | j � ddd dd dd dd dd dddg� d S )Nc s � j t| ��S )N)r3 r )rK )rX r r r6 � s zDHTTPHandlerTests.test_handle_connect_with_password.<locals>.<lambda>rY r: r* r) s GET / HTTP/1.1s s Host: localhosts Accept-Encoding: chunkeds Accept: */*s" User-agent: Supervisor HTTP Clients- Authorization: Basic dXNlcm5hbWU6cGFzc3dvcmQ=) r rN rZ r[ r* r) r\ rI rH r )r r r )rX r �!test_handle_connect_with_password� s2 z2HTTPHandlerTests.test_handle_connect_with_passwordc C s( | j � }|jd� | j|jjdg� d S )Nr )r r r r( �fed_data)r r r r r r � s zHTTPHandlerTests.test_feedc C sD | j � }|j|_d|_|jd� | j|jjdg� | j|jd� d S )Nr, rT �abcfoo� )r �body�part�buffer�collect_incoming_datar r( r_ )r r r r r �'test_collect_incoming_data_part_is_body� s z8HTTPHandlerTests.test_collect_incoming_data_part_is_bodyc C s@ | j � }d |_d|_|jd� | j|jjg � | j|jd� d S )Nr, rT r` )r rc rd re r r( r_ )r r r r r �+test_collect_incoming_data_part_is_not_body� s z<HTTPHandlerTests.test_collect_incoming_data_part_is_not_bodyc sH | j � }g � � fdd�|_d |_|j� | j� dg� | j|jd� d S )Nc s � j d�S )NT)r3 r )�partedr r r6 � s z8HTTPHandlerTests.test_found_terminator.<locals>.<lambda>Tra )r rc rd Zfound_terminatorr )r r r )rh r �test_found_terminator� s z&HTTPHandlerTests.test_found_terminatorc C s( | j � }d |_|j� | j|jd� d S )Nra )r rd �ignorer )r r r r r � test_ignore� s z HTTPHandlerTests.test_ignorec C s | j � }d|_| jt|j� d S )Ns NOTHTTP/1.0 200 OK)r rd r- � ValueError� status_line)r r r r r �$test_status_line_not_startswith_http� s z5HTTPHandlerTests.test_status_line_not_startswith_httpc C sT | j � }d|_|j� \}}}| j|d� | j|d� | j|d� | j|j|j� d S )Ns HTTP/1.0 200 OKs HTTP/1.0�� s OK)r rd rm r rc �headers)r r �versionr �reasonr r r �test_status_line_200� s z%HTTPHandlerTests.test_status_line_200c s� | j � }d|_g � � fdd�|_|j� \}}}| j|d� | j|d� | j|d� | j|j|j� | j|jjd� | j� dg� d S ) Ns HTTP/1.0 201 OKc s � j d�S )NT)r3 r )rG r r r6 s z;HTTPHandlerTests.test_status_line_not_200.<locals>.<lambda>s HTTP/1.0�� s OKz Cannot read, status code 201T) r rd r rm r rc rj r( rV )r r rq r rr r )rG r �test_status_line_not_200 s z)HTTPHandlerTests.test_status_line_not_200c sV | j � }d|_d|_d|_g � � fdd�|_|j� | j|j|j� | j� dg� d S )Nra s not chunked� c s � j | �S )N)r3 )�L)�termsr r r6 s zEHTTPHandlerTests.test_headers_empty_line_nonchunked.<locals>.<lambda>) r rd �encoding�length�set_terminatorrp r rc rb )r r r )rx r �"test_headers_empty_line_nonchunked s z3HTTPHandlerTests.test_headers_empty_line_nonchunkedc C s0 | j � }d|_d|_|j� | j|j|j� d S )Nra s chunked)r rd ry rp r rc � chunked_size)r r r r r � test_headers_empty_line_chunked s z0HTTPHandlerTests.test_headers_empty_line_chunkedc C s" | j � }d|_| j|j� d � d S )N� :)r rd r rp )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 | j � }d|_g � � fdd�|_|j� | j|jd� | j� dg� d S )Ns Transfer-Encoding: chunkedc s � j | |f�S )N)r3 )�n�v)� responsesr r r6 0 s zOHTTPHandlerTests.test_headers_nonempty_line_transfer_encoding.<locals>.<lambda>� chunked� transfer-encoding)r� r� )r rd r rp r ry )r r r )r� r �,test_headers_nonempty_line_transfer_encoding, s z=HTTPHandlerTests.test_headers_nonempty_line_transfer_encodingc sH | j � }d|_g � � fdd�|_|j� | j|jd� | j� dg� d S )Ns Content-Length: 3c s � j | |f�S )N)r3 )r� r� )r� r r r6 9 s zLHTTPHandlerTests.test_headers_nonempty_line_content_length.<locals>.<lambda>rv � content-length� 3)r� r� )r rd r rp r rz )r r r )r� r �)test_headers_nonempty_line_content_length5 s z:HTTPHandlerTests.test_headers_nonempty_line_content_lengthc s: | j � }d|_g � � fdd�|_|j� | j� dg� d S )Ns X-Test: abcc s � j | |f�S )N)r3 )r� r� )r� r r r6 B s zGHTTPHandlerTests.test_headers_nonempty_line_arbitrary.<locals>.<lambda>� x-test� abc)r� r� )r rd r rp r )r r r )r� r �$test_headers_nonempty_line_arbitrary> s z5HTTPHandlerTests.test_headers_nonempty_line_arbitraryc C s8 | j � }|jdd� | j|jjd� | j|jjd� d S )N� a� b)r r r r( �response_header_name�response_header_value)r r r r r r F s z%HTTPHandlerTests.test_response_headerc sB | j � }g � � fdd�|_|j� | j� dg� | j|jj� d S )Nc s � j d�S )NT)r3 r )rG r r r6 O s z,HTTPHandlerTests.test_body.<locals>.<lambda>T)r r rb r rI r( r )r r r )rG r � test_bodyL s zHTTPHandlerTests.test_bodyc C s" | j � }|j� | j|jj� d S )N)r r rI r( )r r r r r r T s zHTTPHandlerTests.test_donec C s6 | j � }d|_d|_| j|j� d � | j|jd� d S )Nra � )r rd rz r r} )r r r r r � test_chunked_size_empty_lineY s z-HTTPHandlerTests.test_chunked_size_empty_linec C sF | j � }d|_d|_| j|j� d � | j|jd� | j|j|j� d S )N� 0r� )r rd rz r r} rc �trailer)r r r r r �test_chunked_size_zero_size` s z,HTTPHandlerTests.test_chunked_size_zero_sizec sf | j � }d|_d|_g � � fdd�|_| j|j� d � | j|j|j� | j|jd� | j� dg� d S )Ns 10r� c s � j | �S )N)r3 )Zsz)rx r r r6 m s zAHTTPHandlerTests.test_chunked_size_nonzero_size.<locals>.<lambda>� � )r rd rz r{ r r} rc � chunked_body)r r r )rx r � test_chunked_size_nonzero_sizeh s z/HTTPHandlerTests.test_chunked_size_nonzero_sizec sv ddl m} | j� }d|_g �g � �fdd�|_� fdd�|_|j� | j�|g� | j� dg� | j|j|j � d S )Nr )rJ s bufferc s � j | �S )N)r3 )r� )rx r r r6 y s z4HTTPHandlerTests.test_chunked_body.<locals>.<lambda>c s � j | �S )N)r3 )r� )�linesr r r6 z s ) r rJ r rd r{ r r� r rc r} )r rJ r r )r� rx r �test_chunked_bodys s z"HTTPHandlerTests.test_chunked_bodyc C s" | j � }d|_| j|j� d � d S )Nra )r rd r r� )r r r r r �test_trailer_line_not_crlf� s z+HTTPHandlerTests.test_trailer_line_not_crlfc sn ddl m} | j� }||_g �g � �fdd�|_� fdd�|_| j|j� d � | j�dg� | j� dg� d S )Nr )rJ c s � j d�S )NT)r3 r )�donesr r r6 � s z9HTTPHandlerTests.test_trailer_line_crlf.<locals>.<lambda>c s � j d�S )NT)r3 r )�closesr r r6 � s T)r rJ r rd r r r r� )r rJ r r )r� r� r �test_trailer_line_crlf� s z'HTTPHandlerTests.test_trailer_line_crlf)Nr% N)(r r! r" r r r' r0 r2 rA rC rD r rP rS rW r] r^ r rf rg ri rk rn rs ru 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 )N)r_ )r r r r �__init__� s zDummyListener.__init__c C s || _ d S )N)rG )r r r r r r � s zDummyListener.closec C s || _ || _d S )N)rU rV )r r �msgr r r r � s zDummyListener.errorc C s | j j|� d S )N)r_ r3 )r r r r r r r � s zDummyListener.feedc C s || _ || _d S )N)Z status_urlZ status_int)r r �intr r r r � s zDummyListener.statusc C s || _ || _d S )N)r� r� )r r rM �valuer r r r � s z DummyListener.response_headerc C s d| _ d S )NT)r )r r r r r r � s zDummyListener.done) r r! r" rG rU rV r r� r r r r r r r r r r+ � s r+ c @ s e Zd ZdZdd� ZdS )rF Fc C s d| _ d S )NT)rG )r r r r r � s zDummySocket.closeN)r r! r" rG r r r r r rF � s rF ) r> r ZunittestZsupervisor.compatr r ZTestCaser r# �objectr+ rF r r r r �<module> s . ]
Back to File Manager