>Mc@sdZeZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZddlmZdd lmZmZmZmZmZdd lmZdd lmZmZmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,dej-dZ.defdYZ/dd-dYZ0de1fdYZ2dZ3dZ4de1fdYZ5de&fd YZ6d!efd"YZ7d#e&fd$YZ8d%e fd&YZ9d'e e,fd(YZ:d)efd*YZ;d+e fd,YZ<e=j>e9j?e=j>e:j?e=j>e<j?dS(.s. Tests for implementations of L{IReactorTCP}. iN(t implements(t verifyObject(tReactorBuilder(tDNSLookupError(tIResolverSimplet IConnectort IReactorFDSet(t IPv4Address(tDeferredt DeferredListtsucceedtfailt maybeDeferred(tTCP4ClientEndpoint(t ServerFactoryt ClientFactorytProtocol(tplatform(tFailure(tlog(tSkipTesttTestCase(t Connection(tClosingProtocol(tObjectModelIntegrationMixins 127.0.0.1cCsNtj}tj||}z!|j|df|jSWd|jXdS(s Ask the platform to allocate a free port on the specified interface, then release the socket and return the address which was allocated. @param interface: The local address to try to bind the port on. @type interface: C{str} @param type: The socket type which will use the resulting port. @return: A two-tuple of address and port, like that returned by L{socket.getsockname}. iN(tsockettAF_INETtbindt getsocknametclose(t interfacettypetfamilytprobe((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt findFreePort!s tStopcBs eZdZdZdZRS(sQ A client factory which stops a reactor when a connection attempt fails. cCs ||_dS(N(treactor(tselfR$((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt__init__;scCs|jjdS(N(R$tstop(R%t connectortreason((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytclientConnectionFailed?s(t__name__t __module__t__doc__R&R*(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR#7s t FakeResolvercBs*eZdZeedZdZRS(sR A resolver implementation based on a C{dict} mapping names to addresses. cCs ||_dS(N(tnames(R%R/((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR&JscCs>yt|j|SWn"tk r9ttd|SXdS(NsFakeResolver couldn't find (R R/tKeyErrorR R(R%tnamettimeout((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt getHostByNameNs (R+R,R-RRR&R3(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR.Ds  t_SimplePullProducercBs)eZdZdZdZdZRS(s A pull producer which writes one byte whenever it is resumed. For use by L{test_unregisterProducerAfterDisconnect}. cCs ||_dS(N(tconsumer(R%R5((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR&[scCsdS(N((R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt stopProducing_scCs!tjd|jjddS(NsProducer.resumeProducingtx(RtmsgR5twrite(R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytresumeProducingcs (R+R,R-R&R6R:(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR4Vs  cCsItj|r|jSd|jjkr2|jStd|fdS(sV Like L{IReactorFDSet.getWriters}, but with support for IOCP reactor as well. tIOCPsCannot find writers on %rN(Rt providedByt getWriterst __class__R+thandlest Exception(R$((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt _getWritersis  cCst}||_|S(s[ Helper function which provides the signature L{ServerFactory} should provide. (Rtprotocol(RBtfactory((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytserverFactoryForws  t FakeSocketcBs/eZdZdZdZdZdZRS(s A Fake Socket object icCs ||_dS(N(tdata(R%RF((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR&scCs ||_dS(N(tblocking(R%RG((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt setblockingscCs|jS(N(RF(R%tsize((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytrecvs(R+R,R-tfilenoR&RHRJ(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyREs   tTestFakeSocketcBs eZdZdZdZRS(sT Test that the FakeSocket can be used by the doRead method of L{Connection} cCs0td}|jd|j|jddS(NtsomeDatai(RERHt assertEqualsRG(R%tskt((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt test_blockings  cCs)td}|j|jdddS(NRMi (RERNRJ(R%RO((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt test_recvs (R+R,R-RPRQ(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRLs t FakeProtocolcBseZdZdZRS(sL An L{IProtocol} that returns a value from its dataReceived method. cCsdS(sq Return something other than C{None} to trigger a deprecation warning for that behavior. (((R%RF((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt dataReceiveds(R+R,R-RS(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRRstTCPConnectionTestscBseZdZdZRS(s@ Whitebox tests for L{twisted.internet.tcp.Connection}. cCstd}t}t||}|j|jtjg}|j|ddt|j|ddd|jt|ddS(s When an L{IProtocol} implementation that returns a value from its C{dataReceived} method, a deprecated warning is emitted. RMitcategorytmessagesReturning a value other than None from twisted.internet.test.test_tcp.FakeProtocol.dataReceived is deprecated since Twisted 11.0.0.iN( RERRRtdoReadt flushWarningsRSRNtDeprecationWarningtlen(R%RORBtconntwarnings((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_doReadWarningIsRaiseds    (R+R,R-R](((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRTstTCPClientTestsBuildercBsDeZdZdZdZdZdZdZdZRS(sG Builder defining tests relating to L{IReactorTCP.connectTCP}. cCs>|j}|jddt}|jtt|dS(sV L{IReactorTCP.connectTCP} returns an object providing L{IConnector}. s 127.0.0.1iN(t buildReactort connectTCPRt assertTrueRR(R%R$R(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_interfaces cCsEt\}}|j}|j||t||j|dS(sl The reactor can be stopped by a client factory's C{clientConnectionFailed} method. N(R"R_R`R#t runReactor(R%thosttportR$((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt'test_clientConnectionFailedStopsReactors cs#t\}}|jjdttd|}|j}id d6d d6dtffdY}t}||_j d|jj |dd |fj t id d6|j |jdtd d ||jdtd d |j d S( s A client's transport's C{getHost} and C{getPeer} return L{IPv4Address} instances which give the dotted-quad string form of the local and remote endpoints of the connection respectively. iRRdtpeert CheckAddresscseZfdZRS(cs.|jd<|jd<jdS(NRdRg(tgetHosttgetPeerR'(R%t transport(t addressesR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytmakeConnections(R+R,Rm((RlR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRhst localhostt bindAddresss 127.0.0.1tTCPN(R"R_t listenTCPRDRRitNoneR#RBR`RetinstallResolverR.Rct assertEqualR(R%RdRetservert serverAddressRht clientFactory((R$RlsB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_addressess(      cs|jjdtt}gdtffdY}t}||_jd|jj|j |j dS(s This test checks that we correctly get notifications event for a client. This ought to prevent a regression under Windows using the GTK2 reactor. See #3925. itCheckConnectioncseZfdZRS(csj|jdS(N(tappendR'(R%(t connectedR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytconnectionMades (R+R,R|((R{R$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRyss 127.0.0.1N( R_RqRDRR#RBR`RiRetrunRa(R%RuRyRw((R$R{sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_connectEvents    cs|jjdtt}tjtjjfdgdt ffdY}t }||_ j d|j j||j|jdddS(sY If a producer is unregistered from a L{ITCPTransport} provider after the transport has been disconnected (by the peer) and after L{ITCPTransport.loseConnection} has been called, the transport is not re-added to the reactor as a writer as would be necessary if the transport were still connected. ics jS(N(R'(tign(R$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt,stClientProtocolcs5eZdZdZfdZdZRS(s Protocol to connect, register a producer, try to lose the connection, wait for the server to disconnect from us, and then unregister the producer. cSs:tjd|jjt|jt|jjdS(NsClientProtocol.connectionMade(RR8RktregisterProducerR4tFalsetloseConnection(R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR|6s  csDtjd|jj|jtkjddS(NsClientProtocol.connectionLost(RR8t unregisterRzRkRAtcallbackRr(R%R)(R$tfinishedtwriting(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytconnectionLost<s  cSstjd|jjdS(NsClientProtocol unregister(RR8RktunregisterProducer(R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRBs (R+R,R-R|RR((R$RR(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR0s s 127.0.0.1s/Transport was writing after unregisterProducer.N(R_RqRDRRt addErrbackRterrt addCallbackRRRBR`RiReRct assertFalse(R%ReRRw((R$RRsB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt&test_unregisterProducerAfterDisconnects  "   cs|jddg}jj}||krKtjrKtdndd dYjdtt}t j t j j fddtffd Y}t}||_jd |jj||jd S( s If L{ITCPTransport.loseConnection} is called while a producer is registered with the transport, the connection is closed after the producer is unregistered. t Glib2Reactort Gtk2Reactors=A pygobject/pygtk bug disables this functionality on Windows.tProducercBseZdZRS(cSstjddS(NsProducer.resumeProducing(RR8(R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR:as(R+R,R:(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR`sics jS(N(R'(R(R$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRhsRcsAeZdZfdZfdZfdZRS(s Protocol to connect, register a producer, try to lose the connection, unregister the producer, and wait for the connection to actually be lost. csPtjd|jjt|jjjdjd|jdS(NsClientProtocol.connectionMadei(RR8RkRRRt callLaterR(R%(RR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR|ps  cs@tjd|jjjdjttddS(NsClientProtocol unregisterg?sConnection was not lost(RR8RkRRterrbackRR@(R%(RR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRzs    cstjdjddS(NsClientProtocol.connectionLost(RR8RRr(R%R)(R(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs (R+R,R-R|RR((R$RR(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRjs  s 127.0.0.1N((R_R>R+Rt isWindowsRRqRDRRRRRRRRBR`RiReRc(R%tskippedReactorstreactorClassNameReRRw((R$RRsB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_disconnectWhileProducingNs      "  ( R+R,R-RbRfRxR~RR(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR^s %  /tTCPPortTestsBuildercBs2eZdZdZdZdZdZRS(s, Tests for L{IReactorRCP.listenTCP} cCs|jdtS(s/ Get a TCP port from a reactor i(RqR(R%R$((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytgetListeningPortscCsd|jjfS(sI Get the expected connection lost message for a TCP port s(TCP Port %s Closed)(RiRe(R%Re((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytgetExpectedConnectionLostLogMsgscsgfd|j|j|j}tjfdfd}j|j|j|dS(s When a connection is lost, an informative message should be logged (see L{getExpectedConnectionLostLogMsg}): an address identifying the port and the fact that it was closed. csjtj|dS(N(RzRttextFromEventDict(t eventDict(tloggedMessages(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytlogConnectionLostMsgscstjjdS(N(RtremoveObserverR'(tignored(RR$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyt stopReactors cs'tjtjjdS(N(Rt addObserverR t stopListeningR((tpRR(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytdoStopListenings N(R_RRRRtcallWhenRunningR}tassertIn(R%texpectedMessageR((RRR$RRsB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_connectionLostLogMsgs    cCs,|j}|j|}|j|dS(sz The L{IListeningPort} object is an instance of a class with no classic classes in its hierarchy. N(R_RtassertFullyNewStyle(R%R$Re((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_allNewStyles (R+R,R-RRRR(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs    tStopStartReadingProtocolcBs)eZdZdZdZdZRS(sD Protocol that pauses and resumes the transport a few times cCsd|_|jddS(Nti(RFtpauseResumeProducing(R%((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR|s cCsi|jj|jj|rC|jjjd|j|dn"|jjjd|jjj|dS(s? Toggle transport read state, then count down. iiN( RktpauseProducingR:RCR$RRtreadyR(R%tcounter((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs  cCsWtjdt||j|7_t|jdkrS|jjj|jndS(Nsgot dataiii@(RR8RZRFRCR'R(R%RF((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRSs(R+R,R-R|RRS(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs  tTCPConnectionTestsBuildercBseZdZdZRS(sP Builder defining tests relating to L{twisted.internet.tcp.Connection}. c s#tj_ddg}jj}||kr[tjr[tdnt_ t _ t _ j d}|jj}fd}fdtd|}t}t|_ t|j|j gj||}j|S(st This test verifies transport socket read state after multiple pause/resumeProducing calls. RRs.This test is broken on gtk/glib under Windows.icsj|dj|d|dd|ddf}|djjdddd jj||jfdS( sL Send several IOCPReactor's buffers' worth of data. iiR7iitycs jS(N(R'(R(R$(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyR si i (RaRkR9R'R(tprotosRe(R$R%tsftcleanup(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytproceeds  cs[j|dddd dtt|djjt|djjt|jgS( s Make sure IOCPReactor didn't start several WSARecv operations that clobbered each other's results. R7iiRsdid not get the right dataiii i (RNR R RkRR(RFRRe(R%(sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs s 127.0.0.1(RR_R$R>R+RRRRRBRRR'RqRiReR RRR tconnectRRc( R%RRRReRtcctcftd((R$R%RRsB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyttest_stopStartReadings(           * (R+R,R-R(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pyRs((@R-Rt __metaclass__Rtzope.interfaceRtzope.interface.verifyRt#twisted.internet.test.reactormixinsRttwisted.internet.errorRttwisted.internet.interfacesRRRttwisted.internet.addressRttwisted.internet.deferRR R R R ttwisted.internet.endpointsR ttwisted.internet.protocolRRRttwisted.python.runtimeRttwisted.python.failureRttwisted.pythonRttwisted.trial.unittestRRttwisted.internet.tcpRttwisted.test.test_tcpRttwisted.internet.test.test_coreRt SOCK_STREAMR"R#R.tobjectR4RARDRERLRRRTR^RRRtglobalstupdatetmakeTestCaseClasses(((sB/usr/lib/python2.7/dist-packages/twisted/internet/test/test_tcp.pytsF (    <7