ó [³XMc@s¸dZddlmZddlmZddlmZmZddlm Z ddl m Z ddl m Z mZddd „ƒYZd e e fd „ƒYZd e fd „ƒYZdS(s) Tests for L{twisted.internet.protocol}. iÿÿÿÿ(tFailure(tCancelledError(tProtocolt ClientCreator(tClock(tTestCase(t MemoryReactortStringTransporttMemoryConnectorcBseZeZd„ZRS(cCs t|_dS(N(tTruet _disconnected(tself((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt disconnects(t__name__t __module__tFalseR R (((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyRst"MemoryReactorWithConnectorsAndTimecBs2eZdZd„Zd„Zd„Zd„ZRS(ss An extension of L{MemoryReactor} which returns L{IConnector} providers from its C{connectTCP} method. cCs'tj|ƒtj|ƒg|_dS(N(Rt__init__Rt connectors(R ((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyRs  cOs0tj|||Žtƒ}|jj|ƒ|S(N(Rt connectTCPRRtappend(R tatkwt connector((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR$s cOs0tj|||Žtƒ}|jj|ƒ|S(N(Rt connectUNIXRRR(R RRR((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR+s cOs0tj|||Žtƒ}|jj|ƒ|S(N(Rt connectSSLRRR(R RRR((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR2s (R Rt__doc__RRRR(((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyRs    tClientCreatorTestscBsžeZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„ZRS(s? Tests for L{twisted.internet.protocol.ClientCreator}. cCs`dtfd„ƒY}tƒ}t||ƒ}|||ƒ}|jdƒ}|j||ƒdS(s Helper for implementing a test to verify that one of the I{connect} methods of L{ClientCreator} passes the right arguments to the right reactor method. @param check: A function which will be invoked with a reactor and a L{ClientCreator} instance and which should call one of the L{ClientCreator}'s I{connect} methods and assert that all of its arguments except for the factory are passed on as expected to the reactor. The factory should be returned. t SomeProtocolcBseZRS((R R(((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyRJsN(RRRt buildProtocoltNonetassertIsInstance(R tcheckRtreactortcctfactorytprotocol((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt_basicConnectTest>s  cs ‡fd†}ˆj|ƒdS(sä L{ClientCreator.connectTCP} calls C{reactor.connectTCP} with the host and port information passed to it, and with a factory which will construct the protocol passed to L{ClientCreator.__init__}. csx|jddddƒ|jjƒ\}}}}}ˆj|dƒˆj|dƒˆj|dƒˆj|dƒ|S(Ns example.comiÒiás1.2.3.4i”&(s1.2.3.4i”&(s1.2.3.4i”&(Rt tcpClientstpopt assertEquals(R!R"thosttportR#ttimeoutt bindAddress(R (sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR ZsN(R%(R R ((R sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_connectTCPTscs ‡fd†}ˆj|ƒdS(sÕ L{ClientCreator.connectUNIX} calls C{reactor.connectUNIX} with the filename passed to it, and with a factory which will construct the protocol passed to L{ClientCreator.__init__}. csb|jddtƒ|jjƒ\}}}}ˆj|dƒˆj|dƒˆj|tƒ|S(Ns/foo/bari{(RR t unixClientsR'R((R!R"taddressR#R+tcheckPID(R (sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR ks N(R%(R R ((R sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_connectUNIXescs ‡fd†}ˆj|ƒdS(sê L{ClientCreator.connectSSL} calls C{reactor.connectSSL} with the host, port, and context factory passed to it, and with a factory which will construct the protocol passed to L{ClientCreator.__init__}. c s—tƒ}|jdd|ddƒ|jjƒ\}}}}}}ˆj|dƒˆj|dƒˆj||ƒˆj|dƒˆj|dƒ|S(Ns example.comiÒiás4.3.2.1i.(s4.3.2.1i.(s4.3.2.1i.(tobjectRt sslClientsR'R(tassertIdentical( R!R"texpectedContextFactoryR)R*R#tcontextFactoryR+R,(R (sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR {s !N(R%(R R ((R sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_connectSSLus cCsmtƒ}t|tƒ}||ƒ}|jjƒ}|j|jƒ|jƒ|j|jƒ|j |t ƒS(s Helper for implementing a test to verify that cancellation of the L{Deferred} returned by one of L{ClientCreator}'s I{connect} methods is implemented to cancel the underlying connector. @param connect: A function which will be invoked with a L{ClientCreator} instance as an argument and which should call one its I{connect} methods and return the result. @return: A L{Deferred} which fires when the test is complete or fails if there is a problem. ( RRRRR't assertFalseR tcancelt assertTruet assertFailureR(R tconnectR!R"tdR((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt_cancelConnectTestˆs   cCsd„}|j|ƒS(s— The L{Deferred} returned by L{ClientCreator.connectTCP} can be cancelled to abort the connection attempt before it completes. cSs|jddƒS(Ns example.comiÒ(R(R"((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<¤s(R>(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectTCPŸs cCsd„}|j|ƒS(s— The L{Deferred} returned by L{ClientCreator.connectTCP} can be cancelled to abort the connection attempt before it completes. cSs |jdƒS(Ns/foo/bar(R(R"((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<®s(R>(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectUNIX©s cCsd„}|j|ƒS(s— The L{Deferred} returned by L{ClientCreator.connectTCP} can be cancelled to abort the connection attempt before it completes. cSs|jddtƒƒS(Ns example.comiÒ(RR2(R"((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<¸s(R>(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectSSL³s cCs’tƒ}t|tƒ}|||ƒ}|jjƒ}|jt|jƒƒdƒ|jƒ|j|jƒgƒ|j |j ƒ|j |t ƒS(sÍ Like L{_cancelConnectTest}, but for the case where the L{Deferred} is cancelled after the connection is set up but before it is fired with the resulting protocol instance. i( RRRRR'R(tlentgetDelayedCallsR9R:R R;R(R R<R!R"R=R((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt_cancelConnectTimeoutTest½s  cCsd„}|j|ƒS(s¡ L{ClientCreator.connectTCP} inserts a very short delayed call between the time the connection is established and the time the L{Deferred} returned from one of its connect methods actually fires. If the L{Deferred} is cancelled in this interval, the established connection is closed, the timeout is cancelled, and the L{Deferred} fails with L{CancelledError}. c SsY|jddƒ}|jjƒ\}}}}}|jdƒ}tƒ} |j| ƒ|S(Ns example.comiÒ(RR&R'RRRtmakeConnection( R!R"R=R)R*R#R+R,R$t transport((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<âs   (RD(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectTCPTimeoutÙs cCsd„}|j|ƒS(s¢ L{ClientCreator.connectUNIX} inserts a very short delayed call between the time the connection is established and the time the L{Deferred} returned from one of its connect methods actually fires. If the L{Deferred} is cancelled in this interval, the established connection is closed, the timeout is cancelled, and the L{Deferred} fails with L{CancelledError}. c SsS|jdƒ}|jjƒ\}}}}|jdƒ}tƒ}|j|ƒ|S(Ns/foo/bar(RR.R'RRRRE( R!R"R=R/R#R+R,R$RF((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<õs   (RD(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectUNIXTimeoutìs cCsd„}|j|ƒS(s¡ L{ClientCreator.connectSSL} inserts a very short delayed call between the time the connection is established and the time the L{Deferred} returned from one of its connect methods actually fires. If the L{Deferred} is cancelled in this interval, the established connection is closed, the timeout is cancelled, and the L{Deferred} fails with L{CancelledError}. c Ssb|jddtƒƒ}|jjƒ\}}}}}}|jdƒ} tƒ} | j| ƒ|S(Ns example.comiÒ(RR2R3R'RRRRE( R!R"R=R)R*R#R6R+t bindADdressR$RF((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<s !  (RD(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyttest_cancelConnectSSLTimeoutÿs cCs¤tƒ}t|tƒ}|||ƒ\}}|jjƒ}|j|ttdƒƒƒ|jt |j ƒƒdƒ|j ƒ|j|j ƒgƒ|j |t ƒS(sÌ Like L{_cancelConnectTest}, but for the case where the L{Deferred} is cancelled after the connection attempt has failed but before it is fired with the resulting failure. sSimulated failurei(RRRRR'tclientConnectionFailedRt ExceptionR(RBRCR9R;R(R R<R!R"R=R#R((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt_cancelConnectFailedTimeoutTests  cCsd„}|j|ƒS(sz Similar to L{test_cancelConnectTCPTimeout}, but for the case where the connection attempt fails. cSs:|jddƒ}|jjƒ\}}}}}||fS(Ns example.comiÒ(RR&R'(R!R"R=R)R*R#R+R,((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<0s(RM(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt"test_cancelConnectTCPFailedTimeout+s cCsd„}|j|ƒS(s{ Similar to L{test_cancelConnectUNIXTimeout}, but for the case where the connection attempt fails. cSs4|jdƒ}|jjƒ\}}}}||fS(Ns/foo/bar(RR.R'(R!R"R=R/R#R+R,((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<<s(RM(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt#test_cancelConnectUNIXFailedTimeout7s cCsd„}|j|ƒS(sz Similar to L{test_cancelConnectSSLTimeout}, but for the case where the connection attempt fails. c SsC|jddtƒƒ}|jjƒ\}}}}}}||fS(Ns example.comiÒ(RR2R3R'( R!R"R=R)R*R#R6R+RI((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR<Hs!(RM(R R<((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyt"test_cancelConnectSSLFailedTimeoutCs (R RRR%R-R1R7R>R?R@RARDRGRHRJRMRNRORP(((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyR:s"           N((Rttwisted.python.failureRttwisted.internet.deferRttwisted.internet.protocolRRttwisted.internet.taskRttwisted.trial.unittestRttwisted.test.proto_helpersRRRRR(((sG/usr/lib/python2.7/dist-packages/twisted/internet/test/test_protocol.pyts!