[XMc@sdZddlmZddlmZddlmZmZmZmZddlm Z m Z m Z ddlm Z ddl mZddlmZmZmZdd lmZmZd ejfd YZd ejfd YZdddYZdd dYZdefdYZdefdYZdefdYZdefdYZ defdYZ!defdYZ"dS(!sJ Assorted functionality which is commonly useful when writing unit tests. i(tStringIO(t implements(t ITransportt IConsumert IPushProducert IConnector(t IReactorTCPt IReactorSSLt IReactorUNIX(tIListeningPort(tbasic(tprotocolterrortaddress(t IPv4Addresst UNIXAddresstAccumulatingProtocolcBsKeZdZdZZdZdZdZdZ dZ dZ dZ RS(sg L{AccumulatingProtocol} is an L{IProtocol} implementation which collects the data delivered to it and can fire a Deferred when it is connected or disconnected. @ivar made: A flag indicating whether C{connectionMade} has been called. @ivar data: A string giving all the data passed to C{dataReceived}. @ivar closed: A flag indicated whether C{connectionLost} has been called. @ivar closedReason: The value of the I{reason} parameter passed to C{connectionLost}. @ivar closedDeferred: If set to a L{Deferred}, this will be fired when C{connectionLost} is called. itcCsVd|_|jdk rR|jjdk rR|jj}d|j_|j|ndS(Ni(tmadetfactorytNonetprotocolConnectionMadetcallback(tselftd((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytconnectionMade.s    cCs|j|7_dS(N(tdata(RR((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt dataReceived6scCsHd|_||_|jdk rD|jd}|_|jdndS(Ni(tclosedt closedReasontclosedDeferredRR(RtreasonR((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytconnectionLost9s   N( t__name__t __module__t__doc__RRRRRRRRRR (((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRs    tLineSendingProtocolcBs5eZeZedZdZdZdZRS(cCs ||_g|_||_dS(N(tlinestresponsetstart(RR%R'((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt__init__Ds  cCs#|jrt|j|jndS(N(R'tmaptsendLineR%(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRIs cCs<|js(t|j|jg|_n|jj|dS(N(R'R)R*R%R&tappend(Rtline((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt lineReceivedMs  cCs t|_dS(N(tTruetlostConn(RR((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR Ss( R!R"tFalseR/R.R(RR-R (((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR$As    tFakeDatagramTransportcBs&eZeZdZedZRS(cCs g|_dS(N(twritten(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(ZscCs|jj||fdS(N(R2R+(Rtpackettaddr((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytwrite]s(R!R"tobjecttnoAddrR(R5(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR1Ws  tStringTransportcBseZdZeeeeeZdZ dZ dZ dZ dZdddZdZdZdZdZdZdZd Zd Zd Zd Zd ZdZdZRS(s A transport implementation which buffers data in memory and keeps track of its other state without providing any behavior. L{StringTransport} has a number of attributes which are not part of any of the interfaces it claims to implement. These attributes are provided for testing purposes. Implementation code should not use any of these attributes; they are not provided by other transports. @ivar disconnecting: A C{bool} which is C{False} until L{loseConnection} is called, then C{True}. @ivar producer: If a producer is currently registered, C{producer} is a reference to it. Otherwise, C{None}. @ivar streaming: If a producer is currently registered, C{streaming} refers to the value of the second parameter passed to C{registerProducer}. @ivar hostAddr: C{None} or an object which will be returned as the host address of this transport. If C{None}, a nasty tuple will be returned instead. @ivar peerAddr: C{None} or an object which will be returned as the peer address of this transport. If C{None}, a nasty tuple will be returned instead. @ivar producerState: The state of this L{StringTransport} in its capacity as an L{IPushProducer}. One of C{'producing'}, C{'paused'}, or C{'stopped'}. @ivar io: A L{StringIO} which holds the data which has been written to this transport since the last call to L{clear}. Use L{value} instead of accessing this directly. t producingcCsG|j|dk r"||_n|dk r:||_nt|_dS(N(tclearRthostAddrtpeerAddrR.t connected(Rt hostAddresst peerAddress((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(s      cCst|_dS(s Discard all data written to this transport so far. This is not a transport method. It is intended for tests. Do not use it in implementation code. N(Rtio(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR:scCs |jjS(sJ Retrieve all data which has been buffered by this transport. This is not a transport method. It is intended for tests. Do not use it in implementation code. @return: A C{str} giving all data written to this transport since the last call to L{clear}. @rtype: C{str} (R@tgetvalue(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytvalues cCs2t|trtdn|jj|dS(NsData must not be unicode(t isinstancetunicodet TypeErrorR@R5(RR((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR5scCs|jjdj|dS(NR(R@R5tjoin(RR((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt writeSequencescCs t|_dS(s~ Close the connection. Does nothing besides toggle the C{disconnecting} instance variable to C{True}. N(R.t disconnecting(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytloseConnectionscCs)|jdkr"tjdddS|jS(NtTCPs 192.168.1.1i1(R<RR R(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytgetPeerscCs)|jdkr"tjdddS|jS(NRJs10.0.0.1i90(R;RR R(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytgetHostscCs4|jdk rtdn||_||_dS(NsCannot register two producers(tproducerRt RuntimeErrort streaming(RRMRO((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytregisterProducers cCs4|jdkrtdnd|_d|_dS(Ns5Cannot unregister a producer unless one is registered(RMRRNRO(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytunregisterProducers   cCs:|jrtdn|jdkr6tdndS(Ns,Cannot resume producing after loseConnectiontstoppeds Cannot resume a stopped producer(RHRNt producerState(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt _checkStates   cCs|jd|_dS(Ntpaused(RTRS(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytpauseProducings cCs d|_dS(NRR(RS(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt stopProducingscCs|jd|_dS(NR9(RTRS(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytresumeProducings N(R!R"R#RRRRR0RHRRMROR;R<RSR(R:RBR5RGRIRKRLRPRQRTRVRWRX(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR8as,"          t StringTransportWithDisconnectioncBseZdZRS(cCs2|jr.t|_|jjtjdndS(NsBye.(R=R0R R R tConnectionDone(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRIs  (R!R"RI(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRYstStringIOWithoutClosingcBseZdZdZRS(s* A StringIO that can't be closed. cCsdS(s Do nothing. N((R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytcloses(R!R"R#R\(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR[st _FakePortcBs<eZdZeedZdZdZdZRS(s A fake L{IListeningPort} to be used in tests. @ivar _hostAddress: The L{IAddress} this L{IListeningPort} is pretending to be listening on. cCs ||_dS(sz @param hostAddress: An L{IAddress} this L{IListeningPort} should pretend to be listening on. N(t _hostAddress(RR>((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(scCsdS(sQ Fake L{IListeningPort.startListening} that doesn't do anything. N((R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytstartListeningscCsdS(sP Fake L{IListeningPort.stopListening} that doesn't do anything. N((R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt stopListening!scCs|jS(sN Fake L{IListeningPort.getHost} that returns our L{IAddress}. (R^(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRL's( R!R"R#RR R(R_R`RL(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR] s     t_FakeConnectorcBsKeZdZeeeZdZdZdZ dZ dZ RS(s* A fake L{IConnector} that allows us to inspect if it has been told to stop connecting. @ivar stoppedConnecting: has this connector's L{FakeConnector.stopConnecting} method been invoked yet? @ivar _address: An L{IAddress} provider that represents our destination. cCs ||_dS(ss @param address: An L{IAddress} provider that represents this connector's destination. N(t_address(RR ((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(=scCs t|_dS(sv Implement L{IConnector.stopConnecting} and set L{FakeConnector.stoppedConnecting} to C{True} N(R.tstoppedConnecting(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytstopConnectingEscCsdS(s@ Implement L{IConnector.disconnect} as a no-op. N((R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt disconnectMscCsdS(s= Implement L{IConnector.connect} as a no-op. N((R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytconnectSscCs|jS(sp Implement L{IConnector.getDestination} to return the C{address} passed to C{__init__}. (Rb(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pytgetDestinationYs( R!R"R#RRR0RcR(RdReRfRg(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRa/s      t MemoryReactorcBseZdZeeeedZdddZdd dZ dddZ dd dZ dd d d Z dd d ZRS(s A fake reactor to be used in tests. This reactor doesn't actually do much that's useful yet. It accepts TCP connection setup attempts, but they will never succeed. @ivar tcpClients: a list that keeps track of connection attempts (ie, calls to C{connectTCP}). @type tcpClients: C{list} @ivar tcpServers: a list that keeps track of server listen attempts (ie, calls to C{listenTCP}). @type tcpServers: C{list} @ivar sslClients: a list that keeps track of connection attempts (ie, calls to C{connectSSL}). @type sslClients: C{list} @ivar sslServers: a list that keeps track of server listen attempts (ie, calls to C{listenSSL}). @type sslServers: C{list} @ivar unixClients: a list that keeps track of connection attempts (ie, calls to C{connectUNIX}). @type unixClients: C{list} @ivar unixServers: a list that keeps track of server listen attempts (ie, calls to C{listenUNIX}). @type unixServers: C{list} cCs:g|_g|_g|_g|_g|_g|_dS(s0 Initialize the tracking lists. N(t tcpClientst tcpServerst sslClientst sslServerst unixClientst unixServers(R((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(s      i2RcCs2|jj||||fttdd|S(si Fake L{reactor.listenTCP}, that logs the call and returns an L{IListeningPort}. RJs0.0.0.0(RjR+R]R(RtportRtbacklogt interface((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt listenTCPsicCs5|jj|||||fttd||S(sf Fake L{reactor.connectTCP}, that logs the call and returns an L{IConnector}. RJ(RiR+RaR(RthostRoRttimeoutt bindAddress((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt connectTCPscCs5|jj|||||fttdd|S(si Fake L{reactor.listenSSL}, that logs the call and returns an L{IListeningPort}. RJs0.0.0.0(RlR+R]R(RRoRtcontextFactoryRpRq((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt listenSSLs cCs8|jj||||||fttd||S(sf Fake L{reactor.connectSSL}, that logs the call and returns an L{IConnector}. RJ(RkR+RaR(RRsRoRRwRtRu((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt connectSSLs iicCs/|jj|||||ftt|S(sj Fake L{reactor.listenUNIX}, that logs the call and returns an L{IListeningPort}. (RnR+R]R(RR RRptmodetwantPID((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt listenUNIXscCs,|jj||||ftt|S(sg Fake L{reactor.connectUNIX}, that logs the call and returns an L{IConnector}. (RmR+RaR(RR RRttcheckPID((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyt connectUNIXsN(R!R"R#RRRRR(RrRRvRxRyR|R~(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRhbs      tRaisingMemoryReactorcBseZdZeeeed d dZdddZ dd dZ dddZ dd dZ dd d d Z dd d ZRS(s A fake reactor to be used in tests. It accepts TCP connection setup attempts, but they will fail. @ivar _listenException: An instance of an L{Exception} @ivar _connectException: An instance of an L{Exception} cCs||_||_dS(s @param listenException: An instance of an L{Exception} to raise when any C{listen} method is called. @param connectException: An instance of an L{Exception} to raise when any C{connect} method is called. N(t_listenExceptiont_connectException(RtlistenExceptiontconnectException((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR(s i2RcCs |jdS(sR Fake L{reactor.listenTCP}, that raises L{self._listenException}. N(R(RRoRRpRq((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRrsicCs |jdS(sT Fake L{reactor.connectTCP}, that raises L{self._connectException}. N(R(RRsRoRRtRu((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRvscCs |jdS(sR Fake L{reactor.listenSSL}, that raises L{self._listenException}. N(R(RRoRRwRpRq((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRxscCs |jdS(sT Fake L{reactor.connectSSL}, that raises L{self._connectException}. N(R(RRsRoRRwRtRu((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRysiicCs |jdS(sS Fake L{reactor.listenUNIX}, that raises L{self._listenException}. N(R(RR RRpRzR{((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR|scCs |jdS(sU Fake L{reactor.connectUNIX}, that raises L{self._connectException}. N(R(RR RRtR}((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyR~sN(R!R"R#RRRRRR(RrRvRxRyR|R~(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyRs N(((#R#Rtzope.interfaceRttwisted.internet.interfacesRRRRRRRR ttwisted.protocolsR ttwisted.internetR R R ttwisted.internet.addressRRtProtocolRt LineReceiverR$R1R8RYR[R6R]RaRhR(((s>/usr/lib/python2.7/dist-packages/twisted/test/proto_helpers.pyts$"*  %3h