File: //lib/python3/dist-packages/twisted/protocols/haproxy/test/__pycache__/test_v1parser.cpython-38.pyc
U
W[V � @ sR d Z ddlmZ ddlmZ ddlmZmZmZ ddl m
Z
G dd� dej �Z d S )
z7
Test cases for L{twisted.protocols.haproxy.V1Parser}.
� )�unittest)�address� )�InvalidProxyHeader�InvalidNetworkProtocol�MissingAddressData)� _v1parserc @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�
V1ParserTestsz?
Test L{twisted.protocols.haproxy.V1Parser} behaviour.
c C s | � ttjjd� dS )zT
Test that an exception is raised when the PROXY header is missing.
s NOTPROXY N)� assertRaisesr r �V1Parser�parse��self� r �N/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_v1parser.py� test_missingPROXYHeaderValue s
�z*V1ParserTests.test_missingPROXYHeaderValuec C s | � ttjjd� dS )zX
Test that an exception is raised when the proto is not TCP or UNKNOWN.
s PROXY WUTPROTO N)r
r r r r r
r r r �test_invalidNetworkProtocol! s
�z)V1ParserTests.test_invalidNetworkProtocolc C s | � ttjjd� dS )zU
Test that an exception is raised when the proto has no source data.
s PROXY TCP4 N�r
r r r r r
r r r �test_missingSourceData, s
�z$V1ParserTests.test_missingSourceDatac C s | � ttjjd� dS )zU
Test that an exception is raised when the proto has no destination.
s PROXY TCP4 127.0.0.1 8080 8888Nr r
r r r �test_missingDestData7 s
�z"V1ParserTests.test_missingDestDatac C s` t j�d�}| �|jtj� | �|jjd� | �|jj d� | �|j
jd� | �|j
j d� dS )zE
Test that parsing is successful for a PROXY header.
s( PROXY TCP4 127.0.0.1 127.0.0.1 8080 8888� 127.0.0.1� �" N) r r r �assertIsInstance�sourcer � IPv4Address� assertEqual�host�port� destination�r �infor r r �test_fullParsingSuccessB s �z%V1ParserTests.test_fullParsingSuccessc C s` t j�d�}| �|jtj� | �|jjd� | �|jj d� | �|j
jd� | �|j
j d� dS )zK
Test that parsing is successful for an IPv6 PROXY header.
s PROXY TCP6 ::1 ::1 8080 8888s ::1r r N) r r r r r r Z IPv6Addressr r r r r r r r � test_fullParsingSuccess_IPv6P s �z*V1ParserTests.test_fullParsingSuccess_IPv6c C s( t j�d�}| �|j� | �|j� dS )zM
Test that parsing is successful for a UNKNOWN PROXY header.
s$ PROXY UNKNOWN anything could go hereN)r r r Z assertIsNoner r r r r r � test_fullParsingSuccess_UNKNOWN^ s
� z-V1ParserTests.test_fullParsingSuccess_UNKNOWNc C s� t �� }|�d�\}}| �|� | �|� |�d�\}}| �|� | �|� |�d�\}}| �|� | �|jtj� | �|jj d� | �|jj
d� | �|j j d� | �|j j
d� dS )zE
Test that parsing happens when fed a complete line.
� PROXY TCP4 127.0.0.1 127.0.0.1 � 8080 8888s
r r r N) r r �feed� assertFalser r r r r r r r �r �parserr! Z remainingr r r �test_feedParsingi s
z V1ParserTests.test_feedParsingc C s` t �� }|�d�\}}| �|� | �|� |�d�\}}| �|� | �|� | �t|jd� dS )zN
Test that parsing fails if no newline is found in 108 bytes.
r% r& sd N)r r r' r( r
r r) r r r �test_feedParsingTooLong} s
�z%V1ParserTests.test_feedParsingTooLongc C s< t �� }|�d�\}}| �|� | �|d� | �|j� dS )zH
Test that parsing leaves overflow bytes in the buffer.
s: PROXY TCP4 127.0.0.1 127.0.0.1 8080 8888
HTTP/1.1 GET /
s HTTP/1.1 GET /
N)r r r' Z
assertTruer r( �bufferr) r r r �test_feedParsingOverflow� s �
z&V1ParserTests.test_feedParsingOverflowN)�__name__�
__module__� __qualname__�__doc__r r r r r" r# r$ r+ r, r. r r r r r s r N)
r2 Z
twisted.trialr Ztwisted.internetr � _exceptionsr r r � r ZTestCaser r r r r �<module> s