+ jBRt^RIt^RIHt^RIHtHtHt^RIH t H t H t H t H t ^RIHtHt^RIHt^RIHt!RR ] P*4t!R R ] P.4t!R R ] P*4t!RR] P44t!RR]P84t!RR]P<4t!RR]P@4t!!RR]PD4t"!RR]P@4t#!RR]P@4t$!RR]P@4t%!RR] P*]PL4t'!R R!]P@4t(!R"R#]P@4t)!R$R%]4t*!R&R']PV4t,!R(R)]P@4t-R#)*z Test code for policies. N)StringIO) Interface implementedBy implementer)addressdeferprotocolreactortask)StringTransport StringTransportWithDisconnection)policies)unittestcDa]tRt^to^;ttRtRtRtRt Rt Rt Vt R#)SimpleProtocolcn\P!4Vn\P!4VnR#N)rDeferred dConnected dDisconnectedselfs&.IStubTransportrNNr0r1r2r3r4rNrrIStubTransportr rrc]tRt^tRtR#).StubTransportrNNrrNrr StubTransportrs rrN) rrrr ProtocolWrapperrProtocolrzmakeConnection assertTrue providedByrR)rrrprotowrappers& rtest_transportInterfaces%WrapperTests.test_transportInterfacess Y  ^ $   %  h../!!#**8+C+CD+I5Q}/ 11%//BCrc \4p\P!V4pVPRVP 44R#)zk L{WrappingFactory.logPrefix} is customized to mention both the original factory and the wrapping factory. zServer (WrappingFactory)N)rVr rz assertEqual logPrefix)rserverr~s& rtest_factoryLogPrefix"WrapperTests.test_factoryLogPrefixs5 **62 3W5F5F5HIrc !RR4pV!4p\P!V4pVPRVP44R#)z If the wrapped factory doesn't have a L{logPrefix} method, L{WrappingFactory.logPrefix} falls back to the factory class name. c]tRt^tRtR#)=WrapperTests.test_factoryLogPrefixFallback..NoFactoryrNNrrNrr NoFactoryrrrrzNoFactory (WrappingFactory)N)r rzrr)rrrr~s& rtest_factoryLogPrefixFallback*WrapperTests.test_factoryLogPrefixFallbacks?   **62 68I8I8KLrc \4p\P!V4pVP\P !RR^#44pVP RVP!44R#)zc L{ProtocolWrapper.logPrefix} is customized to mention both the original protocol and the wrapper. rxryzEchoProtocol (ProtocolWrapper)N)rVr rzr@rr{rr)rrr~rs& rtest_protocolLogPrefix#WrapperTests.test_protocolLogPrefixsU **62(()<).NoProtocolrNNrrNrr NoProtocolrrrrrxryzNoProtocol (ProtocolWrapper)N) rVrr rzr@rr{rr)rrrr~rs& rtest_protocolLogPrefixFallback+WrapperTests.test_protocolLogPrefixFallbacksd   $**62(()<)1WrapperTests.test_stopConsuming..s&--2ErTN)rrR stopConsumingr)rrrs& @rtest_stopConsumingWrapperTests.test_stopConsumingsD ""$*E' $(rc a.o!V3RlR4p\P!V!44p\4pVPV4VP SV.4R#)ze L{policies.WrappingFactory.startedConnecting} calls C{startedConnecting} on the underlying factory. c.<a]tRtRtoV3RltRtVtR#)4WrapperTests.test_startedConnecting..Factoryic*<SPV4R#rr)r connectorrs&&rstartedConnectingFWrapperTests.test_startedConnecting..Factory.startedConnectings i(rrNN)r0r1r2r3rr4r5r7rs@rFactoryrs ) )rrN)r rzrrr)rrrrrs& @rtest_startedConnecting#WrapperTests.test_startedConnecting sO  ) )**795H !!), )-rc a.o!V3RlR4p\P!V!44p\4p\4pVPW44VP SW43.4R#)zk L{policies.WrappingFactory.clientConnectionLost} calls C{clientConnectionLost} on the underlying factory. c.<a]tRtRtoV3RltRtVtR#)7WrapperTests.test_clientConnectionLost..Factoryi"c,<SPW34R#rrrrr'rs&&&rclientConnectionLostLWrapperTests.test_clientConnectionLost..Factory.clientConnectionLost# y12rrNN)r0r1r2r3rr4r5rs@rrr" 3 3rrN)r rzrrrrrrrr'rs& @rtest_clientConnectionLost&WrapperTests.test_clientConnectionLostsZ  3 3**795H $$Y7 9"5!67rc a.o!V3RlR4p\P!V!44p\4p\4pVPW44VP SW43.4R#)zo L{policies.WrappingFactory.clientConnectionFailed} calls C{clientConnectionFailed} on the underlying factory. c.<a]tRtRtoV3RltRtVtR#)9WrapperTests.test_clientConnectionFailed..Factoryi3c,<SPW34R#rrrs&&&rclientConnectionFailedPWrapperTests.test_clientConnectionFailed..Factory.clientConnectionFailed4rrrNN)r0r1r2r3rr4r5rs@rrr3rrrN)r rzrrrrs& @rtest_clientConnectionFailed(WrapperTests.test_clientConnectionFailed,sZ  3 3**795H &&y9 9"5!67rc `\P!\P!\44\P !44p\ 4pWnVPV4VPVP4VP4VPVP4R#)z L{policies.ProtocolWrapper.connectionLost} sets C{wrappedProtocol} to C{None} in order to break reference cycle between wrapper and wrapped protocols. N) r rrzrVrrr rassertIsNotNoner}loseConnectionrrs& rtest_breakReferenceCycle%WrapperTests.test_breakReferenceCycle=s **  $ $VX .0A0A0C 56 $y) W445  " '112rrNN)r0r1r2r3rWrrrrrrrrrrrrrrrrr4r5r6s@rrvrvrsf4D:J MQ O II 5 7 ). 8"8"33rrvc,a]tRtRtoRtRtRtVtR#)rziOcV#rrN)rrr<s&&&rrWrappingFactory.protocolPsrcz\PPV4VPP R4R#r)r rz startFactorydeferredr!rs&rrWrappingFactory.startFactorySs(  --d3 t$rrNN)r0r1r2r3rrr4r5r6s@rrzrzOs%%rrzc<a]tRtRtoRtRtRtRtRtRt Vt R#) ThrottlingTestsiXz* Tests for L{policies.ThrottlingFactory}. c aa a a a a aa\4pR\^44wo o o o \P!V^4o\ S4p\ P !4Vn\P!^VRR7oSP4Po V V V V 3RlpV V V VV3RlpV 3RlpV V 3RlpV V3RlpV V V3R lpVPPV4VPPV4VPPV4VPPV4VPPV4VPPV4VP#) z Full test using a custom server limiting number of connections. FIXME: https://twistedmatrix.com/trac/ticket/10012 This is a flaky test. c36"TFp\4xK R#5ir)r).0is& r -ThrottlingTests.test_limit..es=Hq.**Hsry) interfacec<\P!RS\S44SPP VV3Rl4SPP VV3Rl4SP #)rycF<\P!RS\S44#ryr connectTCPr9)rc2ns&rrAThrottlingTests.test_limit.._connect123..q',,[!\"=MNrcF<\P!RS\S44#rr)rc3r s&rrr tr r)r rr9r addCallbackr)resultsc1r rr s&r _connect123/ThrottlingTests.test_limit.._connect123nsT   {A|B/? @ MM % %N  MM % %N ## #rcB<SPSSS3Uu.uFqPNK up.RO4SPSSS3Uu.uFqPNK up.RO4SP\SPP 44^4V#uupiuupi)r)rrr)rr)rr r%len protocolskeys)rcrr rrtServers& r _check123-ThrottlingTests.test_limit.._check123xs   BB<@NACs BBcP<SPP4SP#r)rRrr)rrs&r_lose1*ThrottlingTests.test_limit.._lose1~s LL ' ' )## #rc^<\P!RS\S44SP#r)r rr9r)rc4r s&r _connect4-ThrottlingTests.test_limit.._connect4s$   {A|B/? @== rcx<SPSP^4SPSP^4V#)r)rr r%)rr!rs&r_check4+ThrottlingTests.test_limit.._check4s0   R\\1 -   R__a 0Nrc<SS3FpVPP4K \P!\P!SP 4SP SP .4#r)rRrr DeferredList maybeDeferred stopListeningr)rrr r!r<s& r_cleanup,ThrottlingTests.test_limit.._cleanups]V **,%%''8$$$$ r) rVranger r\rzrrrr listenTCPrportr)rr wrapTServerrrrr"r%r+rr rr!r r<rsf @@@@@@@r test_limitThrottlingTests.test_limit]s"=E!H=BB,,VQ7%g. $~~/    a D IIK   $ $    $  !   ((5((3((0((3((1((2###rc \4p\\P!4V4pVP \ P !RR^44p\4pW4nVP!V4VP!R.^,4VPVP4R4VPVP^4R#)zJ L{ThrottlingProtocol.writeSequence} is called on the underlying factory. rxrybytessbytesbytesbytesbytesN)rVrYr Clockr@rr{r rr writeSequencervaluewrittenThisSecond)rrrrrRs& rtest_writeSequence"ThrottlingTests.test_writeSequences+DJJL&A(()<)r?r@rG producingN)rVrYr r5r@rr{r rrr.rr7readThisSecondr]rB producerStateclearrCs& rtest_readLimitThrottlingTests.test_readLimits +DJJL&BO$$W%8%8 Q%OP - /  B -( -( %<= //4 d# //3 ))84 d# //3 ));7   -( -( %<= //4 d# //3 ))84 d# //3 ));7rrNN) r0r1r2r3rWr1r9rErMr4r5r6s@rrrXs%@$D 86<%8%8rrcBa]tRtRtoRtRtRtRtRtRt Rt Vt R #) TimeoutProtocolTestsiz( Tests for L{policies.TimeoutProtocol}. c \P!4p\P!4p\Vn\ WR4pVP \P!RRR44p\4pWEnVPV4WA3#)z Helper to set up an already connected protocol to be tested. @return: A new protocol with its attached clock. @rtype: Tuple of (L{policies.TimeoutProtocol}, L{task.Clock}) Nrxry90) r r5r ServerFactoryrror@rr{r r)rr]wrappedFactoryr~rrRs& rgetProtocolAndClock(TimeoutProtocolTests.getProtocolAndClockst !//1"0(E%%g&9&9%e&TU46 " Y'~rc VP4wrVP^4VPVP4VP VP P 4VP^4VP4VPVP4VP^4VP VP P 4R#)z" Will cancel the ongoing timeout. N) rU setTimeoutr timeoutCallrAr}r%rB cancelTimeoutrrsutr]s& rtest_cancelTimeout'TimeoutProtocolTests.test_cancelTimeouts--/  q S__- ,,99: a  #//*  a ,,99:rc VP4wrVPVP4VP4VP VP P 4R#)z, Does nothing if no timeout is already set. N)rUrrYrZrAr}r%r[s& rtest_cancelTimeoutNoTimeout0TimeoutProtocolTests.test_cancelTimeoutNoTimeout"sN--/  #//*  ,,99:rc VP4wrVPpVP^4VP^4VP VP 4VP 4R#)z0 Does nothing if no timeout is already reached. N)rUr}rXrBrr%rZ)rr\r] wrappedProtos& rtest_cancelTimeoutAlreadyCalled4TimeoutProtocolTests.test_cancelTimeoutAlreadyCalled.sW--/ **  q a  112 rc VP4wrVP^4VPP4VP 4VP VP P4R#)zU Does nothing if the timeout is cancelled from another part. Ex from another thread. N)rUrXrYcancelrZrAr}r%r[s& r"test_cancelTimeoutAlreadyCancelled7TimeoutProtocolTests.test_cancelTimeoutAlreadyCancelled<sX --/  q    ,,99:rrNN) r0r1r2r3rWrUr]r`rdrhr4r5r6s@rrPrPs(,;& ;  ; ;rrPc<a]tRtRtoRtRtRtRtRtRt Vt R#) TimeoutFactoryTestsiLz' Tests for L{policies.TimeoutFactory}. c \P!4Vn\P!4p\ Vn\ VPV^4VnVPP\P!RRR44Vn \4Vn VPVPnVPPVP4VPPVnR#)zY Create a testable, deterministic clock, and a set of server factory/protocol/transport. rxryrRN)r r5r]rrSrror~r@rr{rr rRrr}rc)rrTs& rsetUpTimeoutFactoryTests.setUpQs ZZ\ !//1"0-djj.!L \\//   {E :  :;"&** !!$..1 JJ66rc VPP.RO4VPVPP4VPPRR.4VP VPP4R#)z Make sure that when a TimeoutFactory accepts a connection, it will time out that connection if no data is read or written within the timeout period. 皙?N)rp??rsg?)r]pumprArcr%rrs&r test_timeout TimeoutFactoryTests.test_timeoutbs] 12 **778 c # ))667rc VPP.RO4VPVPP4VP P R4VPP.RO4VPVPP4VP PR.^,4VPP.RO4VPVPP4VPPRR.4VPVPP4R#)zz Make sure that writing data to a transport from a protocol constructed by a TimeoutFactory resets the timeout countdown. rpbytes bytes bytesr4g@N)rprrrsrprsrs) r]rtrArcr%rrSr6rrs&rtest_sendAvoidsTimeout*TimeoutFactoryTests.test_sendAvoidsTimeoutps ( **778 -. ( **778   (a0 ( **778 c # ))667rc VPP.RO4VPVPP4VP P R4VPP.RO4VPVPP4VPP.RO4VPVPP4R#)zB Make sure that receiving data also resets the timeout countdown. rxN)rprsrrry)r]rtrArcr%rr.rrs&rtest_receiveAvoidsTimeout-TimeoutFactoryTests.test_receiveAvoidsTimeouts ( **778  45 ( **778 ( ))667r)r]r~rrRrcN) r0r1r2r3rWrmrurzr}r4r5r6s@rrkrkLs#7" 88888rrkcTa]tRtRtoRt^tRtRtRtRt R Rlt R t R t R t VtR#) TimeoutTesteriz A testable protocol with timeout facility. @ivar timedOut: set to C{True} if a timeout has been detected. @type timedOut: C{bool} Fc WnR#)z6 Initialize the protocol with a C{task.Clock} object. Nrk)rr]s&&rrTimeoutTester.__init__s  rc <VPVP4R#)z# Upon connection, set the timeout. N)rXtimeOutrs&rr"TimeoutTester.connectionMades  %rc dVP4\PPW4R#)z Reset the timeout on data. N) resetTimeoutrrr.r,s&&rr.TimeoutTester.dataReceiveds$ &&t2rNc (VPR4R#)z4 On connection lost, cancel all timeout operations. N)rXr&s&&rr(TimeoutTester.connectionLosts rc RVnR#)zH Flags the timedOut variable to indicate the timeout of the connection. TN)timedOutrs&rtimeoutConnectionTimeoutTester.timeoutConnections  rc BVPP!W.VO5/VB#)z4 Override callLater to use the deterministic clock. rd)rtimeoutrhr_r`s&&&*,rreTimeoutTester.callLaters#zz##GCDCFCCr)r]rr)r0r1r2r3rWrrrr"r.r(rrer4r5r6s@rrrs;GH & 3  DDrrcTa]tRtRtoRtRtRtRtRtRt Rt R t R t R t VtR #) TimeoutMixinTestsiz% Tests for L{policies.TimeoutMixin}. c n\P!4Vn\VP4VnR#)zI Create a testable, deterministic clock and a C{TimeoutTester} instance. N)r r5r]rrrs&rrmTimeoutMixinTests.setUps!ZZ\ "4::. rc VPP^ 4VP\VPP 4^4R#)z Test that the callLater of the clock is used instead of L{reactor.callLater} N)rrXrrr]callsrs&rtest_overriddenCallLater*TimeoutMixinTests.test_overriddenCallLaters5 b! TZZ--.2rc TVPP\44VPP .RO4VP VPP 4VPP ^R.4VPVPP 4R#)zY Check that the protocol does timeout at the time specified by its C{timeOut} attribute. rsNrrrrsrs)rrr r]rtrArrrs&rruTimeoutMixinTests.test_timeoutsn !!/"34 *+ ,,- C!  ++,rc VPP\44VPP .RO4VP VPP 4VPPR4VPP .RO4VP VPP 4VPP ^R.4VPVPP 4R#)zF Check that receiving data is delaying the timeout of the connection. rss hello thereNr)rrsrsrr) rrr r]rtrArr.rrs&rtest_noTimeout TimeoutMixinTests.test_noTimeouts !!/"34 *+ ,,- / *+ ,,- C!  ++,rc RVPnVPP\44VPP ^4VP VPP^4VP P^R.4VPVPP4VP P^R.4VPVPP4R#)za Check that setting a new value for timeout cancel the previous value and install a new timeout. Ng?rq) rrrr rXrr]rtrArrrs&rtest_resetTimeout#TimeoutMixinTests.test_resetTimeouts "  !!/"34 a  ++Q/ C! ,,- C!  ++,rc r^VPnVPP\44VPP R4VP VPP4VP P.RO4VPVPP4R#)z? Setting the timeout to L{None} cancel any timeout operations. N)rrr) rrrr rXrr]rtrArrs&rr]$TimeoutMixinTests.test_cancelTimeoutsw  !!/"34 d# $**,,-  % ,,-rc ^VPnVPVPP^ 4^4VPVPPR4^ 4VP VPP^44VPVPP^4VPPR4R#)z= setTimeout should return the value of the previous timeout. N)rrrrXrrs&rtest_setTimeoutReturn'TimeoutMixinTests.test_setTimeoutReturns  ..r2A6 ..t4b9 $**//23 ++Q/ d#rc dVPP^4VPP4^,P 4VP VPP 4VPPR4VPVPP 4R#)z When the timeout was already cancelled from an external place, calling setTimeout with C{None} to explicitly cancel it will clean up the timeout without raising any exception. N)rrXr]getDelayedCallsrgrrrrs&r%test_setTimeoutCancleAlreadyCancelled7TimeoutMixinTests.test_setTimeoutCancleAlreadyCancelled,sx a  ""$Q'..0 TZZ//0 d# $**,,-r)r]rN)r0r1r2r3rWrmrrurrr]rrr4r5r6s@rrrs7/3 - -- . $..rrc0a]tRtRtoRtRtRtRtVtR#)!LimitTotalConnectionsFactoryTestsi=z/Tests for policies.LimitTotalConnectionsFactoryc\P!4p\PVnVP ^VP 4VP R4pVP ^VP 4VP R4pVP ^VP 4VPR4VP ^VP 4VPR4VP ^VP 4R#)rN)r LimitTotalConnectionsFactoryrrrconnectionCountr@r()rr~p1p2s& rtestConnectionCounting8LimitTotalConnectionsFactoryTests.testConnectionCounting@s779#,, G334 " "4 ( G334  " "4 ( G334 $ G334 $ G334rca\P!4o\PSn^SnSP R4pVP V4VP^SP4VPSP R44VP^SP4!V3RlR\P4pVSn RSn SP R4pVPR4VPSP4VP^SP4VPR4VP^SP4VPR4VP^SP4R#)rNc.<a]tRtRtoV3RltRtVtR#)RLimitTotalConnectionsFactoryTests.testConnectionLimiting..OverflowProtocoliec<RSnR#rE) overflowed)rr~s&rr"aLimitTotalConnectionsFactoryTests.testConnectionLimiting..OverflowProtocol.connectionMadefs %)"rrNN)r0r1r2r3r"r4r5)r7r~s@rOverflowProtocolres * *rrF)r rrrconnectionLimitr@rrrroverflowProtocolrrrr()rr<ropr~s& @rtestConnectionLimiting8LimitTotalConnectionsFactoryTests.testConnectionLimitingTsD779#,,"#  ! !$ ' Q G334 '//56 G334 *x00 *$4 "  " "4 ( $ **+ G334  G334 $ G334rrNN) r0r1r2r3rWrrr4r5r6s@rrr=s95($5$5rrc&a]tRtRtoRtRtVtR#)WriteSequenceEchoProtocoli{cVPR4R8wdVPPV.4R#\P W4R#)svector!N)findrRr6rCr.)rbytess&&rr.&WriteSequenceEchoProtocol.dataReceived|s5 ::j !R ' NN ( (% 1  % %d 2rrNN)r0r1r2r3r.r4r5r6s@rrr{s33rrc*a]tRtRtoRtRtRtVtR#)TestLoggingFactoryiNcbVPeQR4h\4VnVP#)Nzopen() called too many times)openFiler)rnames&&ropenTestLoggingFactory.opens+}}$D&DD$  }}r)r)r0r1r2r3rrr4r5r6s@rrrsHrrc6a]tRtRtoRtRtRtRtRtVt R#)LoggingFactoryTestsiz. Tests for L{policies.TrafficLoggingFactory}. c \4p\Vn\4p\ VR4pVP R 4pWBnVP V4VPP4pVPRV4VPVP44VPR4VPP4pVPRPR4V4VPRPR4V4VPVP4R4VP4VPR4VPP4pVPRPR.4V4VPVP4R4VP!4VPP4pVPRV4R #) zA Check the output produced by L{policies.TrafficLoggingFactory}. test*shere are some bytesz C 1: {!r}z S 1: {!r}s"prepare for vector! to the extremez SV 1: {!r}ConnectionDoneN1.2.3.4i.)rVrrr rr@rrgetvalueassertInrAr7r.formatrrLr)rrTtrr<vs& rtest_thingsGetLogged(LoggingFactoryTests.test_thingsGetLoggedsq "; , . ~v 6 OO- .   JJ   ! c1 # -. JJ   ! k(()?@!D k(()?@!D $:;   <= JJ   ! l))+P*QRTUV $IJ  JJ   ! &*rc \4p\VR4pVPVP^4VP R4VPVP^4RVnVP R4VPVP^4VP 4VPVP^4R#)z7 Test counter management with the resetCounter method. rNr)ri/)rVrr_counterr@r resetCounter)rrTrs& r test_counter LoggingFactoryTests.test_counters  ~v 6 Q' )* Q'  )* Q'  Q'rc aa .o.o VV 3RlpVP\RV4\P!4p\Vn\ P !VR4pVP\P!RRR44pVP\P!RRR44pR /3pR /3pVPWg.S4VPVPVP.S 4R#) z When the L{policies.TrafficLoggingFactory} builds a protocol, it automatically opens a unique log file for that protocol and attaches the logfile to the built protocol. c~<SPW34\4pV^,VnSPV4V#)z@ Mock for the open call to prevent actually opening a log file. )rrr)r_r`io open_calls open_rvaluess*, r mocked_openULoggingFactoryTests.test_loggingFactoryOpensLogfileAutomatically..mocked_opens:   tn -B1gBG    #IrrrrxryrRi:0N)ztest-1w)ztest-2r) patchbuiltinsrrSrr TrafficLoggingFactoryr@rr{rlogfile) rrrTr~ first_proto second_proto first_call second_callrrs & @@r,test_loggingFactoryOpensLogfileAutomatically@LoggingFactoryTests.test_loggingFactoryOpensLogfileAutomaticallys     8V[1!//1"000H++   {E : ,,   {E : &r* &+  *2J? +--|/C/CDlSrrNN) r0r1r2r3rWrrrr4r5r6s@rrrs! +D("$T$Trr).rWrrrzope.interfacerrrtwisted.internetrrrr r twisted.internet.testingr r twisted.protocolsr twisted.trialrrr ClientFactoryr9rCrSrVr\rYrqroTestCaservrzrrPrk TimeoutMixinrrrrrrrrNrrrsr @@DDV&"X&&(8)) #8$$ # X # #2 : :2*2X442*Z38$$Z3z%h..%Y8h''Y8xU;8,,U;pR8(++R8j.DH%%x'<'<.Dbh.))h.V;5(9(9;5|3 377\T(++\Tr