YfdZddlZddlmZddlmZddlmZGddZ dZ Gd d Z Gd d Z Gd dZ GddZdZeZdZdS)a I{Private} test utilities for use throughout Twisted's test suite. Unlike C{proto_helpers}, this is no exception to the don't-use-it-outside-Twisted-we-won't-maintain-compatibility rule! @note: Maintainers be aware: things in this module should be gradually promoted to more full-featured test helpers and exposed as public API as your maintenance time permits. In order to be public API though, they need their own test cases. N)BytesIO)minidom) FileWrapperc$eZdZdZdZdZdZdS)IOPumpzUtility to pump data between clients and servers for protocol testing. Perhaps this is a utility worthy of being in protocol.py? c>||_||_||_||_dS)N)clientserverclientIOserverIO)selfr r r r s H/var/www/html/env/lib/python3.11/site-packages/twisted/test/testutils.py__init__zIOPump.__init__s"      c\|r |dSdS)z,Pump until there is no more input or output.N)pump)r s rflushz IOPump.flush"s;iikk  iikk     rc&|jd|jd|j}|j}|jd|jd|j|j|D]}|j||D]}|j||s|rdSdS)zOMove data back and forth. Returns whether any data was moved. r)r seekr readtruncater dataReceivedr )r cDatasDatabytes rrz IOPump.pump's 1 1 ""$$ ""$$ 1 1         + +D K $ $T * * * * + +D K $ $T * * * *  E 11rN)__name__ __module__ __qualname____doc__rrrrrrrsK !!!  rrc:t}t}|t||t|t||||}|||S)z-Take two Protocol instances and connect them.)rmakeConnectionrrr)r r ciosiors rreturnConnectedr&>s~ ))C ))C +c**+++ +c**+++ &&#s + +DJJLLLJJLLL KrceZdZdZdZdS)XMLAssertionMixinz Test mixin defining a method for comparing serialized XML documents. Must be mixed in to a L{test case}. c|tj|tj|dS)z Verify that two strings represent the same XML document. @param first: An XML string. @type first: L{bytes} @param second: An XML string that should match C{first}. @type second: L{bytes} N) assertEqualdom parseStringtoxml)r firstseconds rassertXMLEqualz XMLAssertionMixin.assertXMLEqualSsW  OE " " ( ( * *COF,C,C,I,I,K,K     rN)rrrr r0r!rrr(r(Ls-      rr(c"eZdZdZdedefdZdS)_EqualzN A class the instances of which are equal to anything and everything. otherreturncdS)NTr!r r3s r__eq__z _Equal.__eq__gstrNrrrr objectboolr7r!rrr2r2bs@Ftrr2c"eZdZdZdedefdZdS) _NotEqualz> A class the instances of which are equal to nothing. r3r4cdS)NFr!r6s rr7z_NotEqual.__eq__psurNr8r!rrr<r<ks@Ftrr<ceZdZdZdZdS)ComparisonTestsMixina A mixin which defines a method for making assertions about the correctness of an implementation of C{==} and C{!=}. Use this to unit test objects which follow the common convention for C{==} and C{!=}: - The object compares equal to itself - The object cooperates with unrecognized types to allow them to implement the comparison - The object implements not-equal as the opposite of equal cZ|||k|||k|||k|||k|||k|||k||tk||tk||tk||tkdS)aU Assert that C{firstValueOne} is equal to C{secondValueOne} but not equal to C{valueOne} and that it defines equality cooperatively with other types it doesn't know about. @param firstValueOne: An object which is expected to compare as equal to C{secondValueOne} and not equal to C{valueTwo}. @param secondValueOne: A different object than C{firstValueOne} but which is expected to compare equal to that object. @param valueTwo: An object which is expected to compare as not equal to C{firstValueOne}. N) assertTrue assertFalser2r<)r firstValueOnesecondValueOnevalueTwos r"assertNormalEqualityImplementationz7ComparisonTestsMixin.assertNormalEqualityImplementations .  6777  7888 (2333 -7888 .8999  1222  1222 &((2333 )++5666  455555rN)rrrr rFr!rrr?r?ts-   6 6 6 6 6rr?cd}d} tjtj}|dd}n#t$rYnwxYw|r||S)z4Returns True if the system can bind an IPv6 address.NF)z::1rT)socketAF_INET6bindOSErrorclose)sockhas_ipv6s r _has_ipv6rOsw DH }V_-- *        Os5< A A c"tsd|_|S)Nz.Does not work on systems without IPv6 support.)HAS_IPV6skip)fs rskipWithoutIPv6rTs BA Hr)r rHiorxml.domrr+twisted.internet.protocolrrr&r(r2r<r?rOrQrTr!rrrXs^   """"""111111%%%%%%%%P           ,.6.6.6.6.6.6.6.6b" 9;;     r