o bJ@sRdZddlmZddlmZddlmZddlmZm Z m Z Gdddej Z d S) z7 Test cases for L{twisted.protocols.haproxy.V1Parser}. )address)unittest) _v1parser)InvalidNetworkProtocolInvalidProxyHeaderMissingAddressDatac@steZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dddZ dS) V1ParserTestsz? Test L{twisted.protocols.haproxy.V1Parser} behaviour. returnNcC|ttjjddS)zT Test that an exception is raised when the PROXY header is missing. s NOTPROXY N) assertRaisesrrV1ParserparseselfrN/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_v1parser.pytest_missingPROXYHeaderValue z*V1ParserTests.test_missingPROXYHeaderValuecCr )zX Test that an exception is raised when the proto is not TCP or UNKNOWN. sPROXY WUTPROTO N)r rrr rrrrrtest_invalidNetworkProtocolrz)V1ParserTests.test_invalidNetworkProtocolcCr )zU Test that an exception is raised when the proto has no source data. s PROXY TCP4 Nr rrr rrrrrtest_missingSourceData'rz$V1ParserTests.test_missingSourceDatacCr )zU Test that an exception is raised when the proto has no destination. sPROXY TCP4 127.0.0.1 8080 8888Nrrrrrtest_missingDestData1rz"V1ParserTests.test_missingDestDatacCtjd}||jtjt|jtjsJt|jtjs J| |jj d| |jj d| |jj d| |jj ddS)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 rassertIsInstancesourcer IPv4Address isinstance destination assertEqualhostportrinforrrtest_fullParsingSuccess;z%V1ParserTests.test_fullParsingSuccesscCr)zK Test that parsing is successful for an IPv6 PROXY header. sPROXY TCP6 ::1 ::1 8080 8888z::1rrN) rr rrrr IPv6Addressr r!r"r#r$r%rrrtest_fullParsingSuccess_IPv6Jr(z*V1ParserTests.test_fullParsingSuccess_IPv6cCs(tjd}||j||jdS)zM Test that parsing is successful for a UNKNOWN PROXY header. s$PROXY UNKNOWN anything could go hereN)rr r assertIsNonerr!r%rrrtest_fullParsingSuccess_UNKNOWNYs  z-V1ParserTests.test_fullParsingSuccess_UNKNOWNcCst}|d\}}|||||d\}}|||||d\}}|||dus8J|dus>J||jtjt|jtjsOJt|j tjsXJ| |jj d| |jj d| |j j d| |j j ddS)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 assertFalserrrrr r!r"r#r$rparserr& remainingrrrtest_feedParsingcs$       zV1ParserTests.test_feedParsingcCs`t}|d\}}|||||d\}}|||||t|jddS)zN Test that parsing fails if no newline is found in 108 bytes. r-r.sd N)rr r/r0r rr1rrrtest_feedParsingTooLongzs    z%V1ParserTests.test_feedParsingTooLongcCs<t}|d\}}||||d||jdS)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"r0bufferr1rrrtest_feedParsingOverflows  z&V1ParserTests.test_feedParsingOverflow)r N)__name__ __module__ __qualname____doc__rrrrr'r*r,r4r5r8rrrrr s    r N) r<twisted.internetr twisted.trialrr _exceptionsrrrTestCaser rrrrs