File: //lib/python3/dist-packages/twisted/protocols/haproxy/test/__pycache__/test_wrapper.cpython-38.pyc
U
W[�2 � @ s| d Z ddlmZ ddlmZ ddlmZmZ ddlm Z ddl
m Z G dd � d e�Z G d
d � d ej
�ZG d d
� d
ej
�ZdS )z>
Test cases for L{twisted.protocols.haproxy.HAProxyProtocol}.
� )�unittest)�address)�Protocol�Factory)� StringTransportWithDisconnection� )�HAProxyWrappingFactoryc @ s e Zd ZdZdd� Zdd� ZdS )�StaticProtocolz6
Protocol stand-in that maintains test state.
c C s d | _ d | _d| _d| _d S )N� F)�source� destination�dataZ disconnected)�self� r �M/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_wrapper.py�__init__ s zStaticProtocol.__init__c C s* | j �� | _| j �� | _| j|7 _d S )N)� transport�getPeerr �getHostr r
)r r
r r r � dataReceived s zStaticProtocol.dataReceivedN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� HAProxyWrappingFactoryV1Testsz]
Test L{twisted.protocols.haproxy.HAProxyWrappingFactory} with v1 PROXY
headers.
c C sR t t�t��}|�t�ddd��}t� }||_|� |� |�
d� | � |j � dS )�J
Test if invalid headers result in connectionLost events.
�TCP� 127.1.1.1� s NOTPROXY anything can go here
N�
r r � forProtocolr �
buildProtocolr � IPv4Addressr �protocol�makeConnectionr � assertFalse� connected�r �factory�protor r r r � test_invalidHeaderDisconnects* s �
z;HAProxyWrappingFactoryV1Tests.test_invalidHeaderDisconnectsc C s\ t t�t��}|�t�ddd��}t� }||_|� |� |�
d� |�
d� | � |j � dS )r r r r s PROXY TCP4 1.1.1.1
s 2.2.2.2 8080
Nr r' r r r �$test_invalidPartialHeaderDisconnects9 s �
zBHAProxyWrappingFactoryV1Tests.test_invalidPartialHeaderDisconnectsc C s� t t�t��}|�t�ddd��}t� }|�|� |� d� | �
|� � j d� | �
|� � j
d� | �
|jj� � j d� | �
|jj� � j
d� | �
|�� j d� | �
|�� j
d� | �
|jj�� j d� | �
|jj�� j
d� dS ) �P
Test if IPv4 headers result in the correct host and peer data.
r � 127.0.0.1r s& PROXY TCP4 1.1.1.1 2.2.2.2 8080 8888
s 1.1.1.1s 2.2.2.2�" N)r r r r r! r r" r r$ r � assertEqualr �host�port�wrappedProtocolr r r' r r r �(test_validIPv4HeaderResolves_getPeerHostI s6 �
� � � �zFHAProxyWrappingFactoryV1Tests.test_validIPv4HeaderResolves_getPeerHostc C s� t t�t��}|�t�ddd��}t� }|�|� |� d� | �
|� � j d� | �
|� � j
d� | �
|jj� � j d� | �
|jj� � j
d� | �
|�� j d� | �
|�� j
d� | �
|jj�� j d� | �
|jj�� j
d� dS )�P
Test if IPv6 headers result in the correct host and peer data.
r � ::1r s PROXY TCP6 ::1 ::2 8080 8888
s ::2r. N)r r r r r! r � IPv6Addressr r$ r r/ r r0 r1 r2 r r r' r r r �(test_validIPv6HeaderResolves_getPeerHostj s6 �
� � � �zFHAProxyWrappingFactoryV1Tests.test_validIPv6HeaderResolves_getPeerHostc C sP t t�t��}|�t�ddd��}t� }|�|� |� d� | �
|j j d� dS )�N
Test if non-header bytes are passed to the wrapped protocol.
r r5 r s, PROXY TCP6 ::1 ::2 8080 8888
HTTP/1.1 / GET� HTTP/1.1 / GETN�
r r r r r! r r6 r r$ r r/ r2 r
r' r r r �'test_overflowBytesSentToWrappedProtocol� s �
zEHAProxyWrappingFactoryV1Tests.test_overflowBytesSentToWrappedProtocolc C sZ t t�t��}|�t�ddd��}t� }|�|� |� d� |� d� | �
|j j d� dS )�K
Test if header streaming passes extra data appropriately.
r r5 r � PROXY TCP6 ::1 ::2 � 8080 8888
HTTP/1.1 / GETr9 Nr: r' r r r �-test_overflowBytesSentToWrappedProtocolChunks� s �
zKHAProxyWrappingFactoryV1Tests.test_overflowBytesSentToWrappedProtocolChunksc C sd t t�t��}|�t�ddd��}t� }|�|� |� d� |� d� |� d� | �
|j j d� dS ) zT
Test if wrapper writes all data to wrapped protocol after parsing.
r r5 r r= r> s
s HTTP/1.1 / GET
Nr: r' r r r �,test_overflowBytesSentToWrappedProtocolAfter� s �
zJHAProxyWrappingFactoryV1Tests.test_overflowBytesSentToWrappedProtocolAfterN) r r r r r* r+ r3 r7 r; r? r@ r r r r r $ s !!r c @ sX e Zd ZdZdZdZdZde e Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )� HAProxyWrappingFactoryV2Testsz]
Test L{twisted.protocols.haproxy.HAProxyWrappingFactory} with v2 PROXY
headers.
s
QUIT
! �"�s4
QUIT
!! $ �"�sl /home/tests/mysockets/sock s
QUIT
!1 �c C s` t t�t��}|�t�ddd��}t� }||_|� |� |�
d| j dd� � | � |j
� dS )r r r5 r � � N)r r r r r! r r6 r r# r$ r �
IPV4HEADERr% r& r' r r r r* � s �
z;HAProxyWrappingFactoryV2Tests.test_invalidHeaderDisconnectsc C s� t t�t��}|�t�ddd��}t� }|�|� |� | j
� | � |� � j
d� | � |� � jd� | � |jj� � j
d� | � |jj� � jd� | � |�� j
d� | � |�� jd� | � |jj�� j
d� | � |jj�� jd� dS )r, r r- r r. N)r r r r r! r r" r r$ r rD r/ r r0 r1 r2 r r r' r r r r3 � s6 �
� � � �zFHAProxyWrappingFactoryV2Tests.test_validIPv4HeaderResolves_getPeerHostc C s� t t�t��}|�t�ddd��}t� }|�|� |� | j
� | � |� � j
d� | � |� � jd� | � |jj� � j
d� | � |jj� � jd� | � |�� j
d� | � |�� jd� | � |jj�� j
d� | � |jj�� jd� dS )r4 r r5 r s 0:0:0:0:0:0:0:1r. N)r r r r r! r r" r r$ r �
IPV6HEADERr/ r r0 r1 r2 r r r' r r r r7 s6 �
� � � �zFHAProxyWrappingFactoryV2Tests.test_validIPv6HeaderResolves_getPeerHostc C s� t t�t��}|�t�d��}t� }|�|� |� | j
� | � |� � j
d� | � |jj� � j
d� | � |�� j
d� | � |jj�� j
d� dS )zP
Test if UNIX headers result in the correct host and peer data.
s /home/test/sockets/server.socks /home/tests/mysockets/sockN)r r r r r! r Z UNIXAddressr r$ r �
UNIXHEADERr/ r �namer2 r r r' r r r �(test_validUNIXHeaderResolves_getPeerHost> s" �
� �zFHAProxyWrappingFactoryV2Tests.test_validUNIXHeaderResolves_getPeerHostc C sV t t�t��}|�t�ddd��}t� }|�|� |� | j
d � | � |j j
d� dS )r8 r r5 r r9 N�r r r r r! r r6 r r$ r rE r/ r2 r
r' r r r r; U s �
zEHAProxyWrappingFactoryV2Tests.test_overflowBytesSentToWrappedProtocolc C sr t t�t��}|�t�ddd��}t� }|�|� |� | j
dd� � |� | j
dd� d � | � |j j
d� dS )r< r r5 r N� r9 rI r' r r r r? c s �
zKHAProxyWrappingFactoryV2Tests.test_overflowBytesSentToWrappedProtocolChunksN)r r r r rD rE Z
_SOCK_PATHrF r* r3 r7 rH r; r? r r r r rA � s$ ����
� !!rA N)r Z
twisted.trialr Ztwisted.internetr Ztwisted.internet.protocolr r Ztwisted.test.proto_helpersr Z_wrapperr r ZTestCaser rA r r r r �<module> s