File: //lib/python3/dist-packages/twisted/protocols/haproxy/test/__pycache__/test_wrapper.cpython-310.pyc
o
�b�6 � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZm Z ddl
m Z ddl mZ dd l
mZ G d
d � d e �ZG d d
� d
ej�ZG dd� dej�ZdS )z>
Test cases for L{twisted.protocols.haproxy.HAProxyProtocol}.
� )�Optional)�mock)�address)�Factory�Protocol)� StringTransportWithDisconnection)�unittest� )�HAProxyWrappingFactoryc @ s, e Zd ZdZd dd�Zdeddfdd�ZdS )
�StaticProtocolz6
Protocol stand-in that maintains test state.
�returnNc C s d | _ d | _d| _d| _d S )N� F)�source� destination�data� disconnected)�self� r �M/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_wrapper.py�__init__ s
zStaticProtocol.__init__r c C s4 | j sJ �| j �� | _| j �� | _| j|7 _d S )N)� transport�getPeerr �getHostr r )r r r r r � dataReceived s
zStaticProtocol.dataReceived�r N)�__name__�
__module__� __qualname__�__doc__r �bytesr r r r r r s
r c @ s` e Zd ZdZddd�Zddd�Zddd �Zdd
d �Zdd d
�Zddd�Z ddd�Z
ddd�Z dS )� HAProxyWrappingFactoryV1Testsz]
Test L{twisted.protocols.haproxy.HAProxyWrappingFactory} with v1 PROXY
headers.
r Nc 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_invalidPartialHeaderDisconnects7 s �
zBHAProxyWrappingFactoryV1Tests.test_invalidPartialHeaderDisconnectsc C sl t t�t��}|�t�ddd��}ttj j
tj j d�}|� |� | �
|�� tj j
� | �
|�� tj j � dS )z�
Before any data is received the HAProxy protocol will return the same peer
and host as the IP connection.
r" � 127.0.0.1r$ )� hostAddress� peerAddressN)r
r r& r r' r r( r r �sentinel� host_address� peer_addressr* � assertEqualr r r- r r r � test_preDataReceived_getPeerHostF s ��
z>HAProxyWrappingFactoryV1Tests.test_preDataReceived_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 IPv4 headers result in the correct host and peer data.
r" r2 r$ s& PROXY TCP4 1.1.1.1 2.2.2.2 8080 8888
z1.1.1.1z2.2.2.2�" N)r
r r& r r' r r( r r* r r8 r �host�port�wrappedProtocolr r r- r r r �(test_validIPv4HeaderResolves_getPeerHostW �6 �
� � � �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
z::2r; N)r
r r& r r' r � IPv6Addressr r* r r8 r r< r= r> r r r- r r r �(test_validIPv6HeaderResolves_getPeerHostw r@ 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" rB r$ s, PROXY TCP6 ::1 ::2 8080 8888
HTTP/1.1 / GET� HTTP/1.1 / GETN�
r
r r& r r' r rC r r* r r8 r> 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" rB r$ � PROXY TCP6 ::1 ::2 � 8080 8888
HTTP/1.1 / GETrF NrG 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" rB r$ rJ rK s
s HTTP/1.1 / GET
NrG r- r r r �,test_overflowBytesSentToWrappedProtocolAfter� s �
zJHAProxyWrappingFactoryV1Tests.test_overflowBytesSentToWrappedProtocolAfterr ) r r r r r0 r1 r9 r? rD rH rL rM r r r r r # s
r c @ sd e Zd ZdZdZdZdZde e Zddd �Zdd
d �Z dd d
�Z
ddd�Z ddd�Z ddd�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 �r Nc C s` t t�t��}|�t�ddd��}t� }||_|� |� |�
d| j dd� � | � |j
� dS )r! r" rB r$ � � N)r
r r& r r' r rC r r) r* r �
IPV4HEADERr+ r, r- r r r r0 � 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" r2 r$ r; N)r
r r& r r' r r( r r* r rQ r8 r r< r= r> r r r- r r r r? �6 �
� � � �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 )rA r" rB r$ z0:0:0:0:0:0:0:1r; N)r
r r& r r' r r( r r* r �
IPV6HEADERr8 r r< r= r> r r r- r r r rD ( rR 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 � UNIXAddressr r* r �
UNIXHEADERr8 r �namer> r r r- r r r �(test_validUNIXHeaderResolves_getPeerHostH s" �
� �zFHAProxyWrappingFactoryV2Tests.test_validUNIXHeaderResolves_getPeerHostc C sV t t�t��}|�t�ddd��}t� }|�|� |� | j
d � | � |j j
d� dS )rE r" rB r$ rF N�r
r r& r r' r rC r r* r rS r8 r> r r- r r r rH ^ s �
zEHAProxyWrappingFactoryV2Tests.test_overflowBytesSentToWrappedProtocolc C sr t t�t��}|�t�ddd��}t� }|�|� |� | j
dd� � |� | j
dd� d � | � |j j
d� dS )rI r" rB r$ N� rF rX r- r r r rL k s �
zKHAProxyWrappingFactoryV2Tests.test_overflowBytesSentToWrappedProtocolChunksr )r r r r rQ rS �
_SOCK_PATHrU r0 r? rD rW rH rL r r r r rN � s( ��� � ��
rN N)r �typingr r r �twisted.internetr �twisted.internet.protocolr r �twisted.test.proto_helpersr �
twisted.trial�_wrapperr
r �TestCaser rN r r r r �<module> s