o b@sdZddlmZmZddlmZmZmZmZ m Z m Z ddl m Z ddlmZddlmZddlmZGd d d eZGd d d eZd S)z1 Tests for L{twisted.protocols.haproxy._parser}. )TypeUnion)TCP4ServerEndpointTCP6ServerEndpointUNIXServerEndpoint_parse_WrapperServerEndpointserverFromString) MemoryReactor)SynchronousTestCase)unparseEndpoint)HAProxyWrappingFactoryc@s^eZdZdZdeddfddZdddZdd d Zdd d Zdd dZ dddZ dddZ dS)UnparseEndpointTestszg Tests to ensure that un-parsing an endpoint string round trips through escaping properly. inputreturnNcCs|tt||dS)aA Check that the input unparses into the output, raising an assertion error if it doesn't. @param input: an input in endpoint-string-description format. (To ensure determinism, keyword arguments should be in alphabetical order.) @type input: native L{str} N) assertEqualr parseEndpoint)selfrrL/usr/lib/python3/dist-packages/twisted/protocols/haproxy/test/test_parser.pychecks zUnparseEndpointTests.checkcC|ddS)z% An individual word. wordNrrrrrtest_basicUnparse)z&UnparseEndpointTests.test_basicUnparsecCr)z% Multiple arguments. zone:twoNrrrrrtest_multipleArguments/rz+UnparseEndpointTests.test_multipleArgumentscCr)z$ Keyword arguments. zaleph=one:bet=twoNrrrrr test_keywords5rz"UnparseEndpointTests.test_keywordscCr)z Escaped ":". zhello\:colon\:worldNrrrrrtest_colonInArgument;rz)UnparseEndpointTests.test_colonInArgumentcCr)z/ Escaped ":" in keyword value. zhello=\:Nrrrrrtest_colonInKeywordValueArz-UnparseEndpointTests.test_colonInKeywordValuecCr)z. Escaped ":" in keyword name. z\:=helloNrrrrrtest_colonInKeywordNameGrz,UnparseEndpointTests.test_colonInKeywordNamerN) __name__ __module__ __qualname____doc__strrrrrr r!r"rrrrrs    rc@sZeZdZdZdedeeeeeee fde fddZ ddd Z dd d Z dd d ZdS)HAProxyServerParserTestsz@ Tests that the parser generates the correct endpoints. description expectedClassrcCsJt}t||}||tt|tsJ||j|||jt|S)a% Test the C{haproxy} enpdoint prefix against one sub-endpoint type. @param description: A string endpoint description beginning with C{haproxy}. @type description: native L{str} @param expectedClass: the expected sub-endpoint class given the description. @type expectedClass: L{type} @return: the parsed endpoint @rtype: L{IStreamServerEndpoint} @raise twisted.trial.unittest.Failtest: if the parsed endpoint doesn't match expectations. ) r r assertIsInstancer isinstance_wrappedEndpointassertIs_wrapperFactoryr)rr*r+reactorendpointrrr onePrefixSs  z"HAProxyServerParserTests.onePrefixNcC|dtdS)zG Test if the parser generates a wrapped TCP4 endpoint. zhaproxy:tcp:8080N)r3rrrrr test_tcp4uz"HAProxyServerParserTests.test_tcp4cCr4)zG Test if the parser generates a wrapped TCP6 endpoint. zhaproxy:tcp6:8080N)r3rrrrr test_tcp6{r6z"HAProxyServerParserTests.test_tcp6cCr4)zG Test if the parser generates a wrapped UNIX endpoint. z haproxy:unix:address=/tmp/socketN)r3rrrrr test_unixr6z"HAProxyServerParserTests.test_unixr#)r$r%r&r'r(rrrrrrr3r5r7r8rrrrr)Ns" " r)N)r'typingrrtwisted.internet.endpointsrrrrrrr twisted.test.proto_helpersr twisted.trial.unittestr TestCase_parserr _wrapperrrr)rrrrs     7