o b@sdZddlZddlmZddlmZddlmZddlm Z edkr)ddl m Z ndd l m Z Gd d d Z Gd d d eZGddde jZdS)zK Tests for the internal implementation details of L{twisted.internet.udp}. N)udp)DatagramProtocol) platformType)unittestwin32)WSAEWOULDBLOCK) EWOULDBLOCKc@s(eZdZdZddZddZddZdS) 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. cCs||_d|_dSN)retvals connectedAddr)selfr rJ/usr/lib/python3/dist-packages/twisted/internet/test/test_udp_internals.py__init__ s zStringUDPSocket.__init__cCs ||_dSr )r )r addrrrrconnect$ zStringUDPSocket.connectcCs$|jd}t|tjr||dfS)zH Return (or raise) the next value from C{self.retvals}. rN)r pop isinstancesocketerror)r sizeretrrrrecvfrom's  zStringUDPSocket.recvfromN)__name__ __module__ __qualname____doc__rrrrrrrr s   r c@s eZdZdZddZddZdS) KeepReadsz% Accumulate reads in a list. cCs g|_dSr )reads)r rrrr6rzKeepReads.__init__cCs|j|dSr )r append)r datarrrrdatagramReceived9szKeepReads.datagramReceivedN)rrrrrr#rrrrr1s rc@s0eZdZdZddZddZddZdd Zd S) ErrorsTestsz/ Error handling tests for C{udp.Port}. cCstjd|tjjdt}td|}tddt ddt dg|_| | |j ddg| | |j gddS)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. iNresult123456)r%r&r') r_sockErrReadIgnorer! addCleanupremoverPortr rrdoRead assertEqualr r protocolportrrrtest_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)Nrrrrrresz5ErrorsTests.test_readImmediateError..Nab)r_sockErrReadRefuser!r)r*rconnectionRefusedr+r rrrr,r-r r.rrrtest_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. r2cs dS)NT)r!rrefusedrrr4s z>ErrorsTests.test_connectedReadImmediateError..Nr5r6z 127.0.0.1i'T)rr7r!r)r*rr8r+r rrrrr,r-r r.rr:r test_connectedReadImmediateErrorts   z,ErrorsTests.test_connectedReadImmediateErrorcCsJt}td|}tdtdg|_|tj|j||j dgdS)zG Socket reads with an unknown socket error are raised. Nsgoodi) rrr+r rr assertRaisesr,r-r r.rrrtest_readUnknownErrors  z!ErrorsTests.test_readUnknownErrorN)rrrrr1r9r<r>rrrrr$=s  r$)rrtwisted.internetrtwisted.internet.protocolrtwisted.python.runtimer twisted.trialrerrnorrr rSynchronousTestCaser$rrrrs