U n5{i@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z e dkr^dd l m Zn dd l mZGd d d ZGd ddeZGddde jZdS)zK Tests for the internal implementation details of L{twisted.internet.udp}. ) annotationsN)udp)DatagramProtocol) platformType)unittestwin32)WSAEWOULDBLOCK) EWOULDBLOCKc@s@eZdZdZdddddZdddd d Zd d d ddZdS)StringUDPSocketa  A fake UDP socket object, which returns a fixed sequence of strings and/or socket errors. Useful for testing. @ivar retvals: A C{list} containing either strings or C{socket.error}s. @ivar connectedAddr: The address the socket is connected to. zlist[bytes | socket.error]None)retvalsreturncCs||_d|_dSN)r connectedAddr)selfr rL/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/internet/test/test_udp_internals.py__init__ szStringUDPSocket.__init__object)addrr cCs ||_dSr)r)rrrrrconnect$szStringUDPSocket.connectintztuple[bytes, None])sizer cCs$|jd}t|tjr||dfS)zH Return (or raise) the next value from C{self.retvals}. rN)r pop isinstancesocketerror)rrretrrrrecvfrom's  zStringUDPSocket.recvfromN)__name__ __module__ __qualname____doc__rrrrrrrr s r c@s0eZdZdZddddZddddd d Zd S) KeepReadsz% Accumulate reads in a list. r r cCs g|_dSr)reads)rrrrr6szKeepReads.__init__bytesr)datarr cCs|j|dSr)r%append)rr'rrrrdatagramReceived9szKeepReads.datagramReceivedN)rr r!r"rr)rrrrr#1sr#c@sHeZdZdZddddZddddZdddd Zddd d Zd S) ErrorsTestsz/ Error handling tests for C{udp.Port}. r r$cCstjd|tjjdt}td|}tddt ddt dg|_| | |j ddg| | |j dddgdS)z Socket reads with some good data followed by a socket error which can be ignored causes reading to stop, and no log messages to be logged. iNsresults123s456) rZ_sockErrReadIgnorer( addCleanupremover#Portr rrdoRead assertEqualr%rprotocolportrrrtest_socketReadNormalBs  z!ErrorsTests.test_socketReadNormalcCstjd|tjjdt}dd|_td|}tdt ddt t g|_ | | |jdg| | |jddgdS)z If the socket is unconnected, socket reads with an immediate connection refusal are ignored, and reading stops. The protocol's C{connectionRefused} method is not called. cSsddS)Nrrrrrrez5ErrorsTests.test_readImmediateError..Nab)r_sockErrReadRefuser(r+r,r#connectionRefusedr-r rrr r.r/r%r0rrrtest_readImmediateErrorXs   z#ErrorsTests.test_readImmediateErrorcstjd|tjjdt}gfdd|_td|}tdt ddt t g|_ | dd| ||jdg|d g| ||jddg|d gdS) z If the socket connected, socket reads with an immediate connection refusal are ignored, and reading stops. The protocol's C{connectionRefused} method is called. r4cs dS)NT)r(rZrefusedrrr6r7z>ErrorsTests.test_connectedReadImmediateError..Nr8r9z 127.0.0.1i'T)rr:r(r+r,r#r;r-r rrr rr.r/r%r0rr=r test_connectedReadImmediateErrorts    z,ErrorsTests.test_connectedReadImmediateErrorcCsJt}td|}tdtdg|_|tj|j||j dgdS)zG Socket reads with an unknown socket error are raised. Nsgoodi) r#rr-r rr assertRaisesr.r/r%r0rrrtest_readUnknownErrors  z!ErrorsTests.test_readUnknownErrorN)rr r!r"r3r<r>r@rrrrr*=s r*)r" __future__rrZtwisted.internetrZtwisted.internet.protocolrZtwisted.python.runtimerZ twisted.trialrerrnorr r r#ZSynchronousTestCaser*rrrrs