U  W[-ć@s¤dZddlmZmZddlZddlmZddlmZddl m Z ddl m Z e dkrbdd l mZn dd l mZGd d „d eƒZGd d„deƒZGdd„dejƒZdS)zK Tests for the internal implementation details of L{twisted.internet.udp}. é)ŚdivisionŚabsolute_importN)Śunittest)ŚDatagramProtocol)Śudp)Ś platformTypeZwin32)ŚWSAEWOULDBLOCK)Ś EWOULDBLOCKc@s(eZdZdZdd„Zdd„Zdd„ZdS) Ś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|_dS©N)ŚretvalsŚ connectedAddr)Śselfr ©rśJ/usr/lib/python3/dist-packages/twisted/internet/test/test_udp_internals.pyŚ__init__"szStringUDPSocket.__init__cCs ||_dSr )r )rŚaddrrrrŚconnect'szStringUDPSocket.connectcCs$|j d”}t|tjƒr|‚|dfS)zH Return (or raise) the next value from C{self.retvals}. rN)r ŚpopŚ isinstanceŚsocketŚerror)rŚsizeZretrrrŚrecvfrom+s  zStringUDPSocket.recvfromN)Ś__name__Ś __module__Ś __qualname__Ś__doc__rrrrrrrr s r c@s eZdZdZdd„Zdd„ZdS)Ś KeepReadsz% Accumulate reads in a list. cCs g|_dSr )Śreads)rrrrr;szKeepReads.__init__cCs|j |”dSr )rŚappend)rŚdatarrrrŚdatagramReceived?szKeepReads.datagramReceivedN)rrrrrr"rrrrr6src@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) Ś ErrorsTestsz/ Error handling tests for C{udp.Port}. cCsˆtj d”| tjjd”tƒ}t d|”}tddt  d”dt  d”gƒ|_|  ”|  |j ddg”|  ”|  |j dddg”dS)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. iØä’’Nsresults123s456) rZ_sockErrReadIgnorer Ś addCleanupŚremoverŚPortr rrŚdoReadŚ assertEqualr©rZprotocolZportrrrŚtest_socketReadNormalIs  ’’z!ErrorsTests.test_socketReadNormalcCsŒtj d”| tjjd”tƒ}dd„|_t d|”}tdt   d”dt   t ”gƒ|_ |  ”|  |jdg”|  ”|  |jddg”dS)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)NérrrrrrŚmóz5ErrorsTests.test_readImmediateError..Nóaób)rŚ_sockErrReadRefuser r$r%rŚconnectionRefusedr&r rrr r'r(rr)rrrŚtest_readImmediateError`s   ’z#ErrorsTests.test_readImmediateErrorcs¼tj d”| tjjd”tƒ}g‰‡fdd„|_t d|”}tdt   d”dt   t ”gƒ|_ |  dd”|  ”| |jdg”| ˆd g”|  ”| |jddg”| ˆd g”dS) 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. r+cs ˆ d”S)NT)r r©Zrefusedrrr-‰r.z>ErrorsTests.test_connectedReadImmediateError..Nr/r0z 127.0.0.1i'T)rr1r r$r%rr2r&r rrr rr'r(rr)rr4rŚ test_connectedReadImmediateError|s   ’ z,ErrorsTests.test_connectedReadImmediateErrorcCsJtƒ}t d|”}tdt d”gƒ|_| tj|j”| |j dg”dS)zG Socket reads with an unknown socket error are raised. NsgoodiĒś’’) rrr&r rrZ assertRaisesr'r(rr)rrrŚtest_readUnknownError›s  z!ErrorsTests.test_readUnknownErrorN)rrrrr*r3r5r6rrrrr#Ds r#)rZ __future__rrrZ twisted.trialrZtwisted.internet.protocolrZtwisted.internetrZtwisted.python.runtimerŚerrnorr Śobjectr rZSynchronousTestCaser#rrrrŚs