U n5{i¡ã@sRdZddlmZddlmZddlmZddlmZm Z m Z Gdd„dej ƒZ d S) z7 Test cases for L{twisted.protocols.haproxy.V1Parser}. é)Úaddress)Úunittesté)Ú _v1parser)ÚInvalidNetworkProtocolÚInvalidProxyHeaderÚMissingAddressDatac@sœeZdZdZddœdd„Zddœdd„Zddœdd „Zddœd d „Zddœd d „Zddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z ddœdd„Z dS)Ú V1ParserTestsz? Test L{twisted.protocols.haproxy.V1Parser} behaviour. N)ÚreturncCs| ttjjd¡dS)zT Test that an exception is raised when the PROXY header is missing. s NOTPROXY N)Ú assertRaisesrrÚV1ParserÚparse©Úself©rúP/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/protocols/haproxy/test/test_v1parser.pyÚtest_missingPROXYHeaderValues ýz*V1ParserTests.test_missingPROXYHeaderValuecCs| ttjjd¡dS)zX Test that an exception is raised when the proto is not TCP or UNKNOWN. sPROXY WUTPROTO N)r rrr r rrrrÚtest_invalidNetworkProtocols ýz)V1ParserTests.test_invalidNetworkProtocolcCs| ttjjd¡dS)zU Test that an exception is raised when the proto has no source data. s PROXY TCP4 N©r rrr r rrrrÚtest_missingSourceData's ýz$V1ParserTests.test_missingSourceDatacCs| ttjjd¡dS)zU Test that an exception is raised when the proto has no destination. sPROXY TCP4 127.0.0.1 8080 8888NrrrrrÚtest_missingDestData1s ýz"V1ParserTests.test_missingDestDatacCs„tj d¡}| |jtj¡t|jtjƒs.t‚t|j tjƒs@t‚|  |jj d¡|  |jj d¡|  |j j d¡|  |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) rr r ÚassertIsInstanceÚsourcerÚ IPv4AddressÚ isinstanceÚAssertionErrorÚ destinationÚ assertEqualÚhostÚport©rÚinforrrÚtest_fullParsingSuccess;sÿz%V1ParserTests.test_fullParsingSuccesscCs„tj d¡}| |jtj¡t|jtjƒs.t‚t|j tjƒs@t‚|  |jj d¡|  |jj d¡|  |j j d¡|  |j j d¡dS)zK Test that parsing is successful for an IPv6 PROXY header. sPROXY TCP6 ::1 ::1 8080 8888z::1rrN) rr r rrrÚ IPv6Addressrrrr r!r"r#rrrÚtest_fullParsingSuccess_IPv6Jsÿz*V1ParserTests.test_fullParsingSuccess_IPv6cCs(tj d¡}| |j¡| |j¡dS)zM Test that parsing is successful for a UNKNOWN PROXY header. s$PROXY UNKNOWN anything could go hereN)rr r Z assertIsNonerrr#rrrÚtest_fullParsingSuccess_UNKNOWNYs ÿ z-V1ParserTests.test_fullParsingSuccess_UNKNOWNcCsôt ¡}| d¡\}}| |¡| |¡| d¡\}}| |¡| |¡| d¡\}}| |¡|dk spt‚|dk s|t‚| |jtj¡t |jtjƒsžt‚t |j tjƒs°t‚|  |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 Nrrr)rr ÚfeedÚ assertFalserrrrrrrr r!r"©rÚparserr$Ú remainingrrrÚtest_feedParsingcs$       zV1ParserTests.test_feedParsingcCs`t ¡}| d¡\}}| |¡| |¡| d¡\}}| |¡| |¡| t|jd¡dS)zN Test that parsing fails if no newline is found in 108 bytes. r)r*sd N)rr r+r,r rr-rrrÚtest_feedParsingTooLongzs    ýz%V1ParserTests.test_feedParsingTooLongcCs<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 / sHTTP/1.1 GET / N)rr r+Ú assertTruer r,Úbufferr-rrrÚtest_feedParsingOverflow‹sÿ  z&V1ParserTests.test_feedParsingOverflow)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr%r'r(r0r1r4rrrrr s     r N) r8Ztwisted.internetrZ twisted.trialrÚrÚ _exceptionsrrrZTestCaser rrrrÚs