U n5{i@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@steZdZdZeddddZddddZddd d Zddd d Zddd dZ ddddZ ddddZ dS)UnparseEndpointTestszg Tests to ensure that un-parsing an endpoint string round trips through escaping properly. N)inputreturncCs|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)selfrrN/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/protocols/haproxy/test/test_parser.pychecks zUnparseEndpointTests.checkrcCs|ddS)z% An individual word. wordNrrrrrtest_basicUnparse)sz&UnparseEndpointTests.test_basicUnparsecCs|ddS)z% Multiple arguments. zone:twoNrrrrrtest_multipleArguments/sz+UnparseEndpointTests.test_multipleArgumentscCs|ddS)z$ Keyword arguments. zaleph=one:bet=twoNrrrrr test_keywords5sz"UnparseEndpointTests.test_keywordscCs|ddS)z Escaped ":". zhello\:colon\:worldNrrrrrtest_colonInArgument;sz)UnparseEndpointTests.test_colonInArgumentcCs|ddS)z/ Escaped ":" in keyword value. zhello=\:Nrrrrrtest_colonInKeywordValueAsz-UnparseEndpointTests.test_colonInKeywordValuecCs|ddS)z. Escaped ":" in keyword name. z\:=helloNrrrrrtest_colonInKeywordNameGsz,UnparseEndpointTests.test_colonInKeywordName) __name__ __module__ __qualname____doc__strrrrrrr r!rrrrrs rc@sbeZdZdZeeeeeeee fe dddZ ddddZ ddd d Z ddd d ZdS) HAProxyServerParserTestsz@ Tests that the parser generates the correct endpoints. ) description expectedClassrcCsJt}t||}||tt|ts*t||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 ZassertIsInstancer isinstanceAssertionErrorZ_wrappedEndpointZassertIsZ_wrapperFactoryr)rr(r)ZreactorZendpointrrr onePrefixSs  z"HAProxyServerParserTests.onePrefixNrcCs|dtdS)zG Test if the parser generates a wrapped TCP4 endpoint. zhaproxy:tcp:8080N)r,rrrrr test_tcp4usz"HAProxyServerParserTests.test_tcp4cCs|dtdS)zG Test if the parser generates a wrapped TCP6 endpoint. zhaproxy:tcp6:8080N)r,rrrrr test_tcp6{sz"HAProxyServerParserTests.test_tcp6cCs|dtdS)zG Test if the parser generates a wrapped UNIX endpoint. z haproxy:unix:address=/tmp/socketN)r,rrrrr test_unixsz"HAProxyServerParserTests.test_unix)r"r#r$r%r&rrrrrrr,r-r.r/rrrrr'Ns "r'N)r%typingrrZtwisted.internet.endpointsrrrrrrr Ztwisted.internet.testingr Ztwisted.trial.unittestr ZTestCase_parserr _wrapperrrr'rrrrs     7