o ba@s<dZddlZddlmZddlmZmZmZmZm Z m Z ddl m Z m Z mZddlmZddlmZGdd d ZGd d d eejZGd d d eejZGdddeZGdddeZGdddejZeee d dGdddeZeee d dGdddeZeee d dGdddeZ dS)zG Tests for implementations of L{IReactorUDP} and L{IReactorMulticast}. N)skipIf)defererror interfacesprotocolreactorudp)Deferred gatherResults maybeDeferred)runtime)TestCasec@s0eZdZdZdZdZddZddZddZdS) MixinrNcCs g|_dSN)packetsselfr7/usr/lib/python3/dist-packages/twisted/test/test_udp.py__init__ zMixin.__init__cCs0d|_|jdur|jd}|_|ddSdSN)startedstartedDeferredcallbackrdrrr startProtocols  zMixin.startProtocolcCs d|_dSr)stoppedrrrr stopProtocol#rzMixin.stopProtocol) __name__ __module__ __qualname__rrrrrr rrrrrs rc@seZdZdZdZddZdS)ServerNrcCs:|j||f|jdur|jd}|_|ddSdSrrappendpacketReceivedr)rdataaddrrrrrdatagramReceived+s  zServer.datagramReceived)r!r"r#r'refusedr*rrrrr$'s r$c@s,eZdZdZdZddZddZddZdS) ClientNrcCs6|j||jdur|jd}|_|ddSdSrr%)rr(rrrrr*7s  zClient.datagramReceivedcCs,|jdur|jd}|_||||_dSr)rerrbackfailure)rr.rrrrconnectionFailed=s   zClient.connectionFailedcC2|jdur|jd}|_|tdd|_dSNyuprrr-rConnectionRefusedErrorr+rrrrconnectionRefusedC  zClient.connectionRefused)r!r"r#r'r+r*r/r5rrrrr,2s  r,c@eZdZddZdS) GoodClientcCr0r1r3rrrrr5Kr6zGoodClient.connectionRefusedN)r!r"r#r5rrrrr8J r8c@seZdZdZdS)BadClientErrorzf Raised by BadClient at the end of every datagramReceived call to try and screw stuff up. N)r!r"r#__doc__rrrrr:Rsr:c@s$eZdZdZdZddZddZdS) BadClientz A DatagramProtocol which always raises an exception from datagramReceived. Used to test error handling behavior in the reactor for that method. NcCs ||_dS)ze Set the Deferred which will be called back when datagramReceived is called. N)rrrrr setDeferredas zBadClient.setDeferredcCs*|jdur|jd}|_||td)NzApplication code is very buggy!)rrr:)rbytesr)rrrrr*hs  zBadClient.datagramReceived)r!r"r#r;rr=r*rrrrr<Ys  r<!This reactor does not support UDPc@sjeZdZddZddZddZddZd d Zee j d d kd ddZ ddZ ddZddZdS)UDPTestscs<t}t}|_tjd|ddfdd}||S)zu The C{type} of the host address of a listening L{DatagramProtocol}'s transport is C{"UDP"}. r 127.0.0.1 interfacecs}|jdS)NUDP)getHost assertEqualtype stopListening)ignoredr)prrr cbStartedzsz+UDPTests.test_oldAddress..cbStartedr$rr rr listenUDP addCallback)rserverrrLrrJrtest_oldAddressqs  zUDPTests.test_oldAddresscsRtt}_tjdddfdd}fdd}|||S)z The L{DatagramProtocol}'s C{startProtocol} and C{stopProtocol} methods are called when its transports starts and stops listening, respectively. rrArBcs$jdjdS)Nrr)rFrrrHrIport1rrPrrrLsz*UDPTests.test_startStop..cbStartedcsjddSr)rFrrR)rrPrr cbStoppedsz*UDPTests.test_startStop..cbStoppedrM)rrrLrUrrSrtest_startStops zUDPTests.test_startStopcsFtt}_tjddd}ddfdd}||S)zr Re-listening with the same L{DatagramProtocol} re-invokes the C{startProtocol} callback. rrArBcSs|SrrH)rIportrrrrLz'UDPTests.test_rebind..cbStartedcs*t}_tjddd}||SNrrArB)rr rrrNrO)rIrrKrLrPrrrUs z'UDPTests.test_rebind..cbStoppedrM)rrrKrUrr[r test_rebinds  zUDPTests.test_rebindcsXtt}_tjdddfdd}||fdd}|||S)z A L{CannotListenError} exception is raised when attempting to bind a second protocol instance to an already bound port rrArBcs>jt}jtjtjj |dddS)NrArB) rFrE transportr$ assertRaisesrCannotListenErrorrrNrX)rIserver2rXrrPrrrLs z*UDPTests.test_bindError..cbStartedcsSrrWrRrXrr cbFinishedrYz+UDPTests.test_bindError..cbFinishedrM)rrrLrcrrartest_bindErrors  zUDPTests.test_bindErrorcstt}_tjdddtt_fdd}||}fdd}||fdd }||fd d }|||S) z Datagrams can be sent with the transport's C{write} method and received via the C{datagramReceived} callback method. rrArBcstjddd_SrZ)rrNport2rR)client clientStartedrrrcbServerStartedsz2UDPTests.test_sendPackets..cbServerStartedcsjdjjj}j}t}_jd|j|jfddd|j|jffgfddtj d|gdd S) NrAhello)a)bNccs6rt}_|jjd|SdSNr)rr r'rOr]writepop)rInextClientWrite) cbClientSendrf clientWritesrPrrrqs  zHUDPTests.test_sendPackets..cbClientStarted..cbClientSendTfireOnOneErrback) r]connectrErXrr r'rnhost DeferredList)rIcAddrsAddr serverSend)rfrP)rqrrrcbClientStarteds   z2UDPTests.test_sendPackets..cbClientStartedcs`j}j}jd|j|jffg|j|jf}jd|fd|fd|fgdS)Nrirjrkrl)r]rErFrrvrX)rIrxry clientAddrrfrrPrrcbSendsFinisheds   z2UDPTests.test_sendPackets..cbSendsFinishedcs$tjtjtjjgddSNTrs)rrwr rHrerR)rTrrrrcs   z-UDPTests.test_sendPackets..cbFinished)r$rr rrrNr8rO)r serverStartedrhrr{r~rcr)rfrgrTrrPrtest_sendPacketss    zUDPTests.test_sendPacketsINFRASTRUCTUREAZUREPIPELINESz"Hangs on Pipelines due to firewallcstt}_tjdddtt}_tjdddtj||gdd}fdd}||fdd }|||S) z A L{ConnectionRefusedError} exception is raised when a connection attempt is actively refused by the other end. Note: This test assumes no one is listening on port 80 UDP. rrArBTrscs^t}_jddtdD]}jd|fjd|fdq|tj S)NrAP %d)rAr) rr rr]rurangern assertFailurerr4)rIr5ir}rrrLs  z2UDPTests.test_connectionRefused..cbStartedcs"tjtjtjgddSr)rrwr rHrR)rXrerrrc)s   z3UDPTests.test_connectionRefused..cbFinished) r8rr rrrNr$rwrO)rrgrrrLrcr)rfrXrerrPrtest_connectionRefuseds   zUDPTests.test_connectionRefusedcCsVt}tjd|dd}|tj|jjdd|jdd|t|jjdd| S)a  A call to the transport's connect method fails with an L{InvalidAddressError} when a non-IP address is passed as the host value. A call to a transport's connect method fails with a L{RuntimeError} when the transport is already connected. rrArB localhostr) r8rrNr^rInvalidAddressErrorr]ru RuntimeErrorrH)rrfrXrrrtest_badConnect5s zUDPTests.test_badConnectcstfdd}|ttjdddfdd}|tt dgfd d S) zr When datagramReceived raises an exception it is logged but the port is not disconnected. cs*t}t|ddt|fdS)z Flush the exceptions which the reactor should have logged and make sure they're actually there. z'Incorrectly found %d errors, expected 2N)flushLoggedErrorsr:rFlen)ignerrsrrr cbCompletedNs z8UDPTests.test_datagramReceivedError..cbCompletedrrArBcstjfddS)z Disconnect the port we started and pass on whatever was given to us in case it was a Failure. cSrrrresultrrbzHUDPTests.test_datagramReceivedError..cbCleanup..)rr rHaddBothrrbrr cbCleanup]sz6UDPTests.test_datagramReceivedError..cbCleanup<csszdWn tyYnwjddddf}t}|j |j j ffdd}fdd }| ||| jtd |jtd dS) a6 Send one packet to the listening BadClient. Set up a 0.1 second timeout to do re-transmits in case the packet is dropped. When two packets have been received by the BadClient, stop sending and let the finalDeferred's callbacks do some assertions. zNot enough packets receivedNzUDP Protocol lost its transportrrcs<|tdkrtdjddSdS)z A packet arrived. Cancel the timeout for it, record it, and maybe finish the test. rrN)cancelr&rr callLaterr)packet) finalDeferred makeAttemptsucceededAttempts timeoutCallrrcbPacketReceiveds    zRUDPTests.test_datagramReceivedError..makeAttempt..cbPacketReceivedcs dS)a The packet wasn't received quickly enough. Try sending another one. It doesn't matter if the packet for which this was the timeout eventually arrives: makeAttempt throws away the Deferred on which this function is the errback, so when datagramReceived callbacks, so it won't be on this Deferred, so it won't raise an AlreadyCalledError. Nr)err)rrrebPacketTimeouts zQUDPTests.test_datagramReceivedError..makeAttempt..ebPacketTimeoutg?z&Timed out in testDatagramReceivedError)fail Exceptionr-failIfIdenticalr]rorr r=rnrvrX addCallbacks addErrbackrrr TimeoutError)rpacketDeferredrr)r)attemptsrfrrrr)rrrss,     z8UDPTests.test_datagramReceivedError..makeAttempt) rr rOr<rrNrrElistr)rrrr)r)rrfrrrXrrrtest_datagramReceivedErrorGs     :z#UDPTests.test_datagramReceivedErrorcCsRGddd}tdt}||_|dd||jg}|t|ddS)ai C{''} is an alternative way to say C{'255.255.255.255'} ({socket.gethostbyname("")} returns C{'255.255.255.255'}), so because it becomes a valid IP address, no deprecation warning about passing hostnames to L{twisted.internet.udp.Port.write} needs to be emitted by C{write()} in this case. c@r7)z6UDPTests.test_NoWarningOnBroadcast..fakeSocketcSsdSrr)rfoobarrrrsendtosz=UDPTests.test_NoWarningOnBroadcast..fakeSocket.sendtoN)r!r"r#rrrrr fakeSocketr9rrstest)z iN) rPortr$socketrn flushWarningstest_NoWarningOnBroadcastrFr)rrrKwarningsrrrrs   z"UDPTests.test_NoWarningOnBroadcastN)r!r"r#rQrVr\rdrrosenvirongetrrrrrrrrr@osD ) ir@c@s8eZdZdZeedsdZddZddZ dd Z dS) ReactorShutdownInteractionTestsz!Test reactor shutdown interactionNr?cCs t|_tjd|jdd|_dS)zStart a UDP portrrArBN)r$rPrrNrXrrrrsetUpsz%ReactorShutdownInteractionTests.setUpcCs |jS)zStop the UDP port)rXrHrrrrtearDowns z(ReactorShutdownInteractionTests.tearDowncshtt}j_fdd}||fdd}|jjddjjjfS)z0Test reactor shutdown while in a recvfrom() loopcs jjtdjddSrm)rPr]connectionLostrrrrRfinishedrrrpktReces zQReactorShutdownInteractionTests.testShutdownFromDatagramReceived..pktRececs dSr)rrRrrr flushErrorss zUReactorShutdownInteractionTests.testShutdownFromDatagramReceived..flushErrorss@rA) rr rPr'rOr]rnrErX)rprrrrrr testShutdownFromDatagramReceiveds    z@ReactorShutdownInteractionTests.testShutdownFromDatagramReceived) r!r"r#r;r IReactorUDPrskiprrrrrrrrs  r'This reactor does not support multicastc@seZdZejddkrejrdZ e e dsdZ ddZ dd Zd d Zd d ZddZddZeejo>ej dddZddZeejdddZdS)MulticastTestsrrz Does not work on Azure PipelinesNrcCsNt|_t|_td|j|_td|j|_|jj d|jj j dS)NrrA) r$rPr,rfrlistenMulticastrTrer]rurErXrrrrr s zMulticastTests.setUpcCstt|jjt|jjgSr)r r rTrHrerrrrrs   zMulticastTests.tearDowncCsF|j|jfD]}||jd|jd||jdqdS)Nrr)rfrPrFr]getTTLsetTTL)rorrrtestTTLs  zMulticastTests.testTTLcsxjjdjjjjd}fdd}||fdd}||fdd}|||S) zv Test that after loopback mode has been set, multicast packets are delivered to their sender. r 225.0.0.250cs(t}j_jjddjf|S)Nrirr rPr'r]rnrXrIrr)rrrcbJoined*sz.MulticastTests.test_loopback..cbJoinedcsftjjdjjdjjdjjddjft }t d|j d|S)Nrrrir) rFrrPrr]setLoopbackModegetLoopbackModernrXr rrrrrrrcbPacket1sz.MulticastTests.test_loopback..cbPacketcstjjddSr)rFrrPrrRrrr cbNoPacket>sz0MulticastTests.test_loopback..cbNoPacket)rFrPr]rrE joinGrouprO)rjoinedrrrrrr test_loopback!s    zMulticastTests.test_loopbackcsjjjdjjdjjd}jjd}t||g}fdd}|||S)zK Test C{getOutgoingInterface} and C{setOutgoingInterface}. z0.0.0.0rAcs,jjdjjddS)NrA)rFrfr]getOutgoingInterfacerPrRrrr cbInterfacesPsz3MulticastTests.test_interface..cbInterfaces)rFrfr]rrPsetOutgoingInterfacer rO)rd1d2rrrrrtest_interfaceEs   zMulticastTests.test_interfacecsTjjd}fdd}||fdd}||fdd}|||S)zE Test that multicast a group can be joined and left. rcjjdSNr)rfr] leaveGrouprRrrr clientJoined]z3MulticastTests.test_joinLeave..clientJoinedcrr)rPr]rrRrrr clientLeftbrz1MulticastTests.test_joinLeave..clientLeftcrr)rPr]rrRrrr serverJoinedgrz3MulticastTests.test_joinLeave..serverJoined)rfr]rrO)rrrrrrrrtest_joinLeaveWs      zMulticastTests.test_joinLeavez2Windows' UDP multicast is not yet fully supported.cCs||jjdtjS)z Test that an attempt to join an address which is not a multicast address fails with L{error.MulticastJoinError}. rA)rrfr]rrMulticastJoinErrorrrrrtest_joinFailureos zMulticastTests.test_joinFailurecsvttdjjjjd}fdd}||fdd}||fdd}|||S) zn Test that a multicast group can be joined and messages sent to and received from it. rrcs&t}j_jddjf|S)N hello worldrrr)r)crrrrsz/MulticastTests.test_multicast..cbJoinedcsjjddddSNrr)rFrPrrRrrrrsz/MulticastTests.test_multicast..cbPacketcs tj}|fdd|S)Ncrrrr passthroughrrrrz@MulticastTests.test_multicast..cleanup..)r rHrOrr)rKrrcleanups z.MulticastTests.test_multicast..cleanup)r$rrrPr]rErrO)rrrrrr)r)rrKrrtest_multicast}s       zMulticastTests.test_multicastzjon non-linux platforms it appears multiple processes can listen, but not multiple sockets in same process?csttjdddjttjdddtjjg}fdd}| |fdd}| |fd d }| ||S) z Test that multiple sockets can listen on the same multicast port and that they both receive multicast messages directed to that address. rT)listenMultiplercs6t}_t}_jdft||gS)Nr)r r'r]rnr )rIrr) firstClientportno secondClienttheGrouprrrs   z5MulticastTests.test_multiListen..serverJoinedcs0jdddjddddSr)rFrrR)rrrrr gotPacketssz3MulticastTests.test_multiListen..gotPacketscs.ttjtjg}|fdd|S)NcrrrrrrrrrzBMulticastTests.test_multiListen..cleanup..)r r rHrOr) firstPort secondPortrrrsz0MulticastTests.test_multiListen..cleanup) r$rrrErXr rPr]rrOr)rrrrrr)rrrrrrrrtest_multiListens&      zMulticastTests.test_multiListen)r!r"r#rrrr platformisMacOSXrrIReactorMulticastrrrrrrrr isWindowsisVistarrrrrrrrs0 $   r)!r;runittestrtwisted.internetrrrrrrtwisted.internet.deferr r r twisted.pythonr twisted.trial.unittestr rDatagramProtocolr$ConnectedDatagramProtocolr,r8rr:r<rr@rrrrrrrs0     W6