VhO@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+jYe)j[e+jYe*j[y)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"eZdZdZdZdZdZy)DatagramTransportTestsMixinzP Mixin defining tests which apply to any port/datagram based transport. c&|j}|j}ttGddt}|}|j ||}d|j jfz}|j|f|ddy)zu When a port starts, a message including a description of the associated protocol is logged. ceZdZdZy)QDatagramTransportTestsMixin.test_startedListeningLogMessage..SomeProtocolcy)NzCrazy Protocolselfs N/home/dcms/DCMS/lib/python3.12/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 49QYY[=M=My-IJr%cJ|j|j|jtdj j d}fdfd}j ||j|j|fddy)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)c&jyNstopignoredr2s r# stopReactorzNDatagramTransportTestsMixin.test_connectionLostLogMessage..stopReactorDs LLNr%cXdd=tjjyr9)r stopListening addCallback)r1r4r>sr#doStopListeningzRDatagramTransportTestsMixin.test_connectionLostLogMessage..doStopListeningGs"q! !// * 6 6{ Cr%rr*N) r+r,r-rr.r/callWhenRunning runReactorr0)r"r5rBr1r4r2r>s @@@@r#test_connectionLostLogMessagez9DatagramTransportTestsMixin.test_connectionLostLogMessage:s ##%  ! !'+;+= >&qyy{'7'7&8A  D 0   /+^A->y-IJr%c>Gfddt}|j|}|j||j|j |j |j |j |j|jy)z L{DatagramProtocol.stopProtocol} is called asynchronously (ie, not re-entrantly) when C{stopListening} is used to stop the datagram transport. c.eZdZdZdZdZdZdZfdZy)VDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocolFcbd|_d|_|jjd|_y)NTF)startedinStartProtocol transportr@r!s r# startProtocolzdDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocol.startProtocol]s)# '+$,,.',$r%cVd|_|j|_jy)NT)stoppedrKstoppedInStartr;r"r2s r# stopProtocolzcDatagramTransportTestsMixin.test_stopProtocolScheduling..DisconnectingProtocol.stopProtocolcs!# &*&:&:# r%N) r&r'r(rJrOrKrPrMrR)r2sr#DisconnectingProtocolrHWs!GG#O"N -  r%rSN) rr,r-rD assertTruerJrO assertFalserP)r"rSr3r2s @r#test_stopProtocolSchedulingz7DatagramTransportTestsMixin.test_stopProtocolSchedulingPs| $4 "##%(* gx0   (() (() 001r%N)r&r'r(__doc__r6rErVr r%r#rr"sK&K,2r%rceZdZdZdZdZedZdZdZ dZ dZ d Z ed Z ed Zd Zed ZedZdZdZy)UDPPortTestsMixinzY Tests for L{IReactorUDP.listenUDP} and L{IReactorSocket.adoptDatagramPort}. c|j}|j|t}|jt t |y)zY L{IReactorUDP.listenUDP} returns an object providing L{IListeningPort}. N)r,r-rrTrr r"r2r/s r#test_interfacez UDPPortTestsMixin.test_interfacexs;##%$$W.>.@A  ^T:;r%cttj\}}|j}|j |t ||}|j |jtd||y)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_getHostsj (V->->?j##%$$ %'jD%  UD*)MNr%c|j}|j|td}|j}|j |j d|j |ty)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_getHostIPv6s\ ##%$$W.>.@E$R||~ E* dK0r%c|j}|jtj|jt ddy)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_invalidInterfacesC ##%   % %      #  r%cGddt}j|d}|j}j|}|j }fd}|j ||j t|j fd|jdd|jfjy) z Datagram transports implement L{ILoggingContext.logPrefix} to return a message reflecting the protocol they are running. ceZdZdZdZdZy)IUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocolc0||_t|_yr9)_prefixr system)r"prefixs r#__init__zRUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.__init__s% &j r%c|jSr9)rur!s r#r$zSUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.logPrefixs ||#r%c|j?|j}d|_|jtjtdyy)Nrv)rvcallbackrgetr)r"bytesrjrvs r#datagramReceivedzZUDPPortTestsMixin.test_logPrefix..CustomLogPrefixDatagramProtocol.datagramReceiveds;;;*![[F"&DKOOGKK $.gotSystems   5v >r%c$jSr9r:r<s r#z2UDPPortTestsMixin.test_logPrefix..s gllnr%s some bytes 127.0.0.1N) rr,rvr-r.rA addErrbackrwriter/rD)r"rr3dr/addressrr2s` @r#test_logPrefixz UDPPortTestsMixin.test_logPrefixs H.> H##%23EF OO$$Wh7,,. ? i  S 45 =; "=>  r%cGddt}j|}|j}j|}|j }dfd}|j ||j t|j fd|jd|jfjy)zH Write a sequence of L{bytes} to a L{DatagramProtocol}. ceZdZdZdZy)DUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocolc"t|_yr9)r rr!s r#rxzMUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocol.__init__s %Z r%c:|jj|yr9)rr{)r"datarjs r#r~zUUDPPortTestsMixin.test_writeSequence..SimpleDatagramProtocol.datagramReceiveds ##D)r%N)r&r'r(rxr~r r%r#SimpleDatagramProtocolrs  ( *r%r)ssomesbytesstoswritecHjdj|y)Nr%)r0join)r dataToWriter"s r#gotDataz5UDPPortTestsMixin.test_writeSequence..gotDatas   SXXk2D 9r%c$jSr9r: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 *##%)+$$Wh7,,.:  : '"89 ;gll(CD  r%c|j}|j|t}|jt |j j t |y)zQ C{str()} on the listening port object includes the port number. N)r,r-rassertInstrr.r/r[s r#test_strzUDPPortTestsMixin.test_strsK##%$$W.>.@A c$,,.--.D :r%c|j}|j|t}|jt |j j t|y)zR C{repr()} on the listening port object includes the port number. N)r,r-rrreprr.r/rr[s r# test_reprzUDPPortTestsMixin.test_reprsK##%$$W.>.@A d4<<>../T;r%c |j t tjx} _|j dt tjx}_|j djj} fd} fd}tj||g}|j||j||jt|j  jd}|j|d|j |j"ffy)zS Writing to an IPv6 UDP socket on the loopback interface succeeds. rgrhcjjddjjjft j x}_|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..cbClientStartedsO    " "7UF4D4D4L4L4N4S4S,T U5:^^5E ENV2! !r%c&jyz 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 LLNr%rrNr,rrr startedDeferredr-rrLr. gatherResultsrArrrDpacketsr0rcr/ r" serverStarted clientStartedcAddrrrrpacketrr2rs @@@r#test_writeToIPv6Interfacez+UDPPortTestsMixin.test_writeToIPv6Interfaces  ##%161AA . gv?161AA . gv?  ((* "     > ? o& &' S  " 'EJJ +C!DEr%c |j t tjx} _|j dt tjx}_|j djj} fd} fd}tj||g}|j||j||jt|j  jd}|j|d|j |j"ffy)z An IPv6 address can be passed as the C{interface} argument to L{listenUDP}. The resulting Port accepts IPv6 datagrams. rgrhcjjdjjjjj dt j x}_|S)rrgr)rLconnectr.r/rrr rrs r#rzMUDPPortTestsMixin.test_connectedWriteToIPv6Interface..cbClientStarted4s^    $ $UF,<,<,D,D,F,K,K L    " "7 +5:^^5E ENV2! !r%c&jyrr:r<s r#rzNUDPPortTestsMixin.test_connectedWriteToIPv6Interface..cbServerReceivedBs LLNr%rrNrrs @@@r#"test_connectedWriteToIPv6Interfacez4UDPPortTestsMixin.test_connectedWriteToIPv6Interface$s  ##%161AA . gv?161AA . gv?  ((* "     > ? o& &' S  " 'EJJ +C!DEr%c|j}|j|t}|jtj |j ddy)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_writingToHostnameRaisesInvalidAddressErrorUsJ ##%$$W.>.@A   % %tzz6;Q r%c|j}|j|td}|jtj |j ddy)l Writing to an IPv6 address on an IPv4 socket will raise an L{InvalidAddressError}. rrhr)rgrNrr[s r#1test_writingToIPv6OnIPv4RaisesInvalidAddressErrorzCUDPPortTestsMixin.test_writingToIPv6OnIPv4RaisesInvalidAddressError`sK ##%$$W.>.@K$X %33TZZTr%c|j}|j|td}|jtj |j ddy)rrgrhr)rrNrr[s r#1test_writingToIPv4OnIPv6RaisesInvalidAddressErrorzCUDPPortTestsMixin.test_writingToIPv4OnIPv6RaisesInvalidAddressErrorjsO ##%$$W.>.@E$R   % %tzz6;K r%c|j}|j|t}|jtj |j ddy)zq Connecting to a hostname instead of an IP address will raise an L{InvalidAddressError}. rrN)r,r-rrmrrnrr[s r#2test_connectingToHostnameRaisesInvalidAddressErrorzDUDPPortTestsMixin.test_connectingToHostnameRaisesInvalidAddressErrorvsH ##%$$W.>.@A %33T\\CTVWXr%c|j}|j|t}|jd|j |j y)zk L{IListeningPort.setBroadcastAllowed} sets broadcast to be allowed on the socket. TN)r,r-rsetBroadcastAllowedrTgetBroadcastAllowedr[s r#test_allowBroadcastz%UDPPortTestsMixin.test_allowBroadcastsK ##%$$W.>.@A   & 0023r%N)r&r'r(rWr\rerrkrprrrrrrrrrrrr r%r#rYrYrs < O 1 1  "!H!8;<*F*FX.F.F`  UU    Y4r%rYc eZdZdZefZ ddZy)UDPServerTestsBuilderzM Run L{UDPPortTestsMixin} tests using newly created UDP sockets. c,|j||||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%Nri )r&r'r(rWrrequiredInterfacesr-r r%r#rrs &FJ r%rc eZdZdZefZ ddZy)UDPFDServerTestsBuilderzC Run L{UDPPortTestsMixin} tests using adopted UDP sockets. c<tj|rd|vr-tj}tj||dd}ntj }||f}tj|tj }|j||jd |j|j|j|||j|jStd#|j|jwxYw)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 -i ,,Y=a@C$d+}}VV->->?H MM' "   ' !00OO%x- ! DE E ! s "+C99"DNr)r&r'r(rWr rr-r r%r#rrs)*FJ-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 &.)=: 7Q>"/42+M2"2M2`U4U4p %'B 86F%'B6Fr &::<= (<<>?r%