tfO@dZddlZddlmZddlmZddlmZmZddl m Z m Z ddl m Z mZddlmZmZmZmZdd lmZdd lmZmZdd lmZdd lmZdd lmZmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&GddeZ'GddZ(Gddee(e'Z)Gddee(e'Z*e+,e)-e+,e*-dS)zU Tests for implementations of L{IReactorUDP} and the UDP parts of L{IReactorSocket}. N) implementer) verifyObject)defererror) IPv4Address IPv6Address)Deferred maybeDeferred)IListeningPortILoggingContextIReactorSocket IReactorUDP)DatagramProtocol)LogObserverMixin findFreePort)ReactorBuilder)context) ILogContexterr) GoodClientServer)skipWithoutIPv6)SkipTestc$eZdZdZdZdZdZdS)DatagramTransportTestsMixinzP Mixin defining tests which apply to any port/datagram based transport. cp|}|}ttGddt}|}|||}d|jfz}||f|dddS)zu When a port starts, a message including a description of the associated protocol is logged. ceZdZdZdS)QDatagramTransportTestsMixin.test_startedListeningLogMessage..SomeProtocolcdS)NzCrazy Protocolselfs ]/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/internet/test/test_udp.py logPrefixz[DatagramTransportTestsMixin.test_startedListeningLogMessage..SomeProtocol.logPrefix1s''N)__name__ __module__ __qualname__r$r r%r# SomeProtocolr/s# ( ( ( ( (r%r)zCrazy Protocol starting on %drmessageN) observe buildReactorrr rgetListeningPortgetHostport assertEqual)r"loggedMessagesreactorr)protocolpexpectedMessages r#test_startedListeningLogMessagez;DatagramTransportTestsMixin.test_startedListeningLogMessage's ##%% _ % % ( ( ( ( (+ ( ( & % ( <>>  ! !'8 4 49QYY[[=My-IJJJJJr%c|||tdjd}fdfd}||||fdddS)z When a connection is lost a message is logged containing an address identifying the port and the fact that it was closed. z (UDP Port z Closed)c0dSNstopignoredr2s r# stopReactorzNDatagramTransportTestsMixin.test_connectionLostLogMessage..stopReactorDs LLNNNNNr%c`dd=tjdSr9)r stopListening addCallback)r1r4r>sr#doStopListeningzRDatagramTransportTestsMixin.test_connectionLostLogMessage..doStopListeningGs3qqq! !/ * * 6 6{ C C C C Cr%rr*N) r+r,r-rr.r/callWhenRunning runReactorr0)r"r5rBr1r4r2r>s @@@@r#test_connectionLostLogMessagez9DatagramTransportTestsMixin.test_connectionLostLogMessage:s ##%%  ! !'+;+=+= > >Aqyy{{'7AAA      D D D D D D D 000     /+^A->y-IJJJJJr%c`Gfddt}||}|||||j||j||jdS)z L{DatagramProtocol.stopProtocol} is called asynchronously (ie, not re-entrantly) when C{stopListening} is used to stop the datagram transport. c0eZdZdZdZdZdZdZfdZdS)VDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocolFcbd|_d|_|jd|_dS)NTF)startedinStartProtocol transportr@r!s r# startProtocolzdDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocol.startProtocol]s3# '+$,,...',$$$r%cVd|_|j|_dS)NT)stoppedrKstoppedInStartr;r"r2s r# stopProtocolzcDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocol.stopProtocolcs&# &*&:# r%N) r&r'r(rJrOrKrPrMrR)r2sr#DisconnectingProtocolrHWsQGG#O"N - - -        r%rSN) rr,r-rD assertTruerJrO assertFalserP)r"rSr3r2s @r#test_stopProtocolSchedulingz7DatagramTransportTestsMixin.test_stopProtocolSchedulingPs       $4   "##%%((** gx000     ())) ())) 011111r%N)r&r'r(__doc__r6rErVr r%r#rr"sQKKK&KKK,22222r%rceZdZdZdZdZedZdZdZ dZ dZ d Z ed Z ed Zd Zed ZedZdZdZdS)UDPPortTestsMixinzY Tests for L{IReactorUDP.listenUDP} and L{IReactorSocket.adoptDatagramPort}. c|}||t}|t t |dS)zY L{IReactorUDP.listenUDP} returns an object providing L{IListeningPort}. N)r,r-rrTrr r"r2r/s r#test_interfacez UDPPortTestsMixin.test_interfacexsS##%%$$W.>.@.@AA  ^T::;;;;;r%c ttj\}}|}||t ||}||td||dS)z L{IListeningPort.getHost} returns an L{IPv4Address} giving a dotted-quad of the IPv4 address the port is listening on as well as the port number. )type)r/ interfaceUDPN) rsocket SOCK_DGRAMr,r-rr0r.r)r"host portNumberr2r/s r# test_getHostzUDPPortTestsMixin.test_getHosts (V->???j##%%$$ %''jD%   UD*)M)MNNNNNr%c |}||td}|}||jd||tdS)zr L{IListeningPort.getHost} returns an L{IPv6Address} when listening on an IPv6 interface. ::1r_N)r,r-rr.r0rcassertIsInstancer)r"r2r/addrs r#test_getHostIPv6z"UDPPortTestsMixin.test_getHostIPv6sw ##%%$$W.>.@.@E$RR||~~ E*** dK00000r%c|}|tj|jt dddS)z An L{InvalidAddressError} is raised when trying to listen on an address that isn't a valid IPv4 or IPv6 address. rz example.comrhN)r, assertRaisesrInvalidAddressError listenUDPrrQs r#test_invalidInterfacez'UDPPortTestsMixin.test_invalidInterfacesW ##%%   %      #      r%cGddt}|d}|j}|}|}fd}|||t|fd|dd|j f dS) z Datagram transports implement L{ILoggingContext.logPrefix} to return a message reflecting the protocol they are running. c eZdZdZdZdZdS)IUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocolc:||_t|_dSr9)_prefixr system)r"prefixs r#__init__zRUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.__init__s% &jj r%c|jSr9)rur!s r#r$zSUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.logPrefixs |#r%c|jB|j}d|_|tjtddSdS)Nrv)rvcallbackrgetr)r"bytesrjrvs r#datagramReceivedzZUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.datagramReceivedsG;*![F"&DKOOGK $<$.gotSystems   5v > > > > >r%c,Sr9r:r<s r#z2UDPPortTestsMixin.test_logPrefix..sgllnnr%s some bytes 127.0.0.1N) rr,rvr-r.rA addErrbackrwriter/rD)r"rr3dr/addressrr2s` @r#test_logPrefixz UDPPortTestsMixin.test_logPrefixs H H H H H.> H H H##%%223EFF O$$Wh77,,.. ? ? ? ? ? i    S 4444555 =; "=>>>      r%cGddt}|}|j}|}|}dfd}|j||jt|jfd|d|j f dS)zH Write a sequence of L{bytes} to a L{DatagramProtocol}. ceZdZdZdZdS)DUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocolc,t|_dSr9)r rr!s r#rxzMUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocol.__init__s%ZZ r%c:|j|dSr9)rr{)r"datarjs r#r~zUUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocol.datagramReceiveds ##D)))))r%N)r&r'r(rxr~r r%r#SimpleDatagramProtocolrs2 ( ( ( * * * * *r%r)ssomesbytesstoswritecZd|dS)Nr%)r0join)r dataToWriter"s r#gotDataz5UDPPortTestsMixin.test_writeSequence..gotDatas+   SXXk22D 9 9 9 9 9r%c,Sr9r:r<s r#rz6UDPPortTestsMixin.test_writeSequence..s',,..r%rN) rr,rr-r.rArr writeSequencer/rD) r"rr3rr/rrrr2s ` @@r#test_writeSequencez$UDPPortTestsMixin.test_writeSequences  * * * * *%5 * * *##%%))++$$Wh77,,..:  : : : : : : '"""8888999 ;gl(CDDD      r%c|}||t}|t |jt |dS)zQ C{str()} on the listening port object includes the port number. N)r,r-rassertInstrr.r/r[s r#test_strzUDPPortTestsMixin.test_strsd##%%$$W.>.@.@AA c$,,..-..D :::::r%c|}||t}|t |jt|dS)zR C{repr()} on the listening port object includes the port number. N)r,r-rrreprr.r/rr[s r# test_reprzUDPPortTestsMixin.test_reprsd##%%$$W.>.@.@AA d4<<>>.//T;;;;;r%c | t tjx} _| dt tjx}_| dj} fd} fd}tj ||g}| || || t|  jd}||d|j|jffdS)zS Writing to an IPv6 UDP socket on the loopback interface succeeds. rgrhcjddjjft jx}_|S) Send a datagram from the client once it's started. @param ignored: a list of C{[None, None]}, which is ignored @returns: a deferred which fires when the server has received a datagram. spamrg)rLrr.r/rr packetReceivedr=serverReceivedclientservers r#cbClientStartedzDUDPPortTestsMixin.test_writeToIPv6Interface..cbClientStartedsP   " "7UF4D4L4L4N4N4S,T U U U5:^5E5E ENV2! !r%c0dSz Stop the reactor after a datagram is received. @param ignored: L{None}, which is ignored @returns: L{None} Nr:r<s r#cbServerReceivedzEUDPPortTestsMixin.test_writeToIPv6Interface..cbServerReceiveds LLNNNNNr%rrNr,rrr startedDeferredr-rrLr. gatherResultsrArrrDpacketsr0rcr/ r" serverStarted clientStartedcAddrrrrpacketrr2rs @@@r#test_writeToIPv6Interfacez+UDPPortTestsMixin.test_writeToIPv6Interfacesk ##%%161A1AA . gv???161A1AA . gv??? ((** " " " " " "        > ? ? o&&& &''' S    " 'EJ +C!DEEEEEr%c | t tjx} _| dt tjx}_| dj} fd} fd}tj ||g}| || || t|  jd}||d|j|jffdS)z An IPv6 address can be passed as the C{interface} argument to L{listenUDP}. The resulting Port accepts IPv6 datagrams. rgrhcjdjjjdt jx}_|S)rrgr)rLconnectr.r/rrr rrs r#rzMUDPPortTestsMixin.test_connectedWriteToIPv6Interface..cbClientStarted4sb   $ $UF,<,D,D,F,F,K L L L   " "7 + + +5:^5E5E ENV2! !r%c0dSrr:r<s r#rzNUDPPortTestsMixin.test_connectedWriteToIPv6Interface..cbServerReceivedBs LLNNNNNr%rrNrrs @@@r#"test_connectedWriteToIPv6Interfacez4UDPPortTestsMixin.test_connectedWriteToIPv6Interface$sk ##%%161A1AA . gv???161A1AA . gv??? ((** " " " " " "        > ? ? o&&& &''' S    " 'EJ +C!DEEEEEr%c|}||t}|tj|jdddS)zn Writing to a hostname instead of an IP address will raise an L{InvalidAddressError}. spam)example.invalidNr,r-rrmrrnrr[s r#/test_writingToHostnameRaisesInvalidAddressErrorzAUDPPortTestsMixin.test_writingToHostnameRaisesInvalidAddressErrorUsb ##%%$$W.>.@.@AA   %tz6;Q     r%c|}||td}|tj|jdddS)l Writing to an IPv6 address on an IPv4 socket will raise an L{InvalidAddressError}. rrhr)rgrNrr[s r#1test_writingToIPv6OnIPv4RaisesInvalidAddressErrorzCUDPPortTestsMixin.test_writingToIPv6OnIPv4RaisesInvalidAddressError`s[ ##%%$$W.>.@.@K$XX %3TZTTTTTr%c|}||td}|tj|jdddS)rrgrhr)rrNrr[s r#1test_writingToIPv4OnIPv6RaisesInvalidAddressErrorzCUDPPortTestsMixin.test_writingToIPv4OnIPv6RaisesInvalidAddressErrorjsg ##%%$$W.>.@.@E$RR   %tz6;K     r%c|}||t}|tj|jdddS)zq Connecting to a hostname instead of an IP address will raise an L{InvalidAddressError}. rrN)r,r-rrmrrnrr[s r#2test_connectingToHostnameRaisesInvalidAddressErrorzDUDPPortTestsMixin.test_connectingToHostnameRaisesInvalidAddressErrorvsX ##%%$$W.>.@.@AA %3T\CTVWXXXXXr%c|}||t}|d||dS)zk L{IListeningPort.setBroadcastAllowed} sets broadcast to be allowed on the socket. TN)r,r-rsetBroadcastAllowedrTgetBroadcastAllowedr[s r#test_allowBroadcastz%UDPPortTestsMixin.test_allowBroadcastsi ##%%$$W.>.@.@AA   &&& 002233333r%N)r&r'r(rWr\rerrkrprrrrrrrrrrrr r%r#rYrYrsM <<< O O O 1 1_ 1    "!"!"!H!!!8;;;<<<*F*F_*FX.F.F_.F`    UU_U   _  YYY44444r%rYc"eZdZdZefZ ddZdS)UDPServerTestsBuilderzM Run L{UDPPortTestsMixin} tests using newly created UDP sockets. r c4|||||S)aB Get a UDP port from a reactor. @param reactor: A reactor used to build the returned L{IListeningPort} provider. @type reactor: L{twisted.internet.interfaces.IReactorUDP} @see: L{twisted.internet.IReactorUDP.listenUDP} for other argument and return types. )r_ maxPacketSize)ro)r"r2r3r/r_rs r#r-z&UDPServerTestsBuilder.getListeningPorts*  (i}!   r%Nrrr)r&r'r(rWrrequiredInterfacesr-r r%r#rrsC &FJ      r%rc"eZdZdZefZ ddZdS)UDPFDServerTestsBuilderzC Run L{UDPPortTestsMixin} tests using adopted UDP sockets. rrrcjtj|rd|vr.tj}tj||dd}ntj}||f}tj|tj}|||d | | |j ||| | S#| | wxYwtd)a Get a UDP port from a reactor, wrapping an already-initialized file descriptor. @param reactor: A reactor used to build the returned L{IListeningPort} provider. @type reactor: L{twisted.internet.interfaces.IReactorSocket} @param port: A port number to which the adopted socket will be bound. @type port: C{int} @param interface: The local IPv4 or IPv6 address to which the adopted socket will be bound. defaults to '', ie all IPv4 addresses. @type interface: C{str} @see: L{twisted.internet.IReactorSocket.adoptDatagramPort} for other argument and return types. :rFz'Reactor does not provide IReactorSocket)r providedByraAF_INET6 getaddrinfoAF_INETrbbind setblockingadoptDatagramPortfilenofamilycloser) r"r2r3r/r_rdomainrportSocks r#r-z(UDPFDServerTestsBuilder.getListeningPorts!.  $W - - Fi ,Y==a@C$d+}VV->??H MM' " " "   ' ' ' !00OO%%x- !!!    !!!    DEE Es ".C99*D#Nr)r&r'r(rWr rr-r r%r#rrsJ)*FJ-F-F-F-F-F-Fr%r).rWrazope.interfacerzope.interface.verifyrtwisted.internetrrtwisted.internet.addressrrtwisted.internet.deferr r twisted.internet.interfacesr r r rtwisted.internet.protocolr&twisted.internet.test.connectionmixinsrr#twisted.internet.test.reactormixinsrtwisted.pythonrtwisted.python.logrrtwisted.test.test_udprrtwisted.test.testutilsrtwisted.trial.unittestrrrYrrglobalsupdatemakeTestCaseClassesr r%r#rs  &&&&&&......))))))))========:::::::: 766666QQQQQQQQ>>>>>>""""""////////44444444222222++++++M2M2M2M2M2"2M2M2M2`U4U4U4U4U4U4U4U4p     %'B   86F6F6F6F6F%'B6F6F6Fr &::<<=== (<<>>?????r%