tf9dZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZGd d eZGd d eZGd de jZGdde jZGdde je je jZGdde je jZe jeee jeee jeGdde jZGdde j Z!Gddej"Z#Gdde#Z$Gdde$Z%Gdd Z&Gd!d"ej"Z'd#S)$z! Tests for error handling in PB. )StringIO)defer)log)qual)flavorsjellypb)connectedServerAndClient)unittestceZdZdZdS)AsynchronousExceptionz Helper used to test remote methods which return Deferreds which fail with exceptions which are not L{pb.Error} subclasses. N__name__ __module__ __qualname____doc__a/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/spread/test/test_pbfailure.pyr r rr ceZdZdZdS)SynchronousExceptionzm Helper used to test remote methods which raise exceptions which are not L{pb.Error} subclasses. NrrrrrrrrrceZdZdZdS)AsynchronousErrorz Helper used to test remote methods which return Deferreds which fail with exceptions which are L{pb.Error} subclasses. Nrrrrrr"rrrceZdZdZdS)SynchronousErrorzi Helper used to test remote methods which raise exceptions which are L{pb.Error} subclasses. Nrrrrrr)rrrceZdZdS) JellyErrorNrrrrrrrr0DrrceZdZdS) SecurityErrorNrrrrr"r"4r rr"cTeZdZdZdZdZdZdZdZdZ dZ d Z d d Z d d Z d S) SimpleRootcDtjtdS)zD Fail asynchronously with a non-pb.Error exception. zremote asynchronous exception)rfailr selfs rremote_asynchronousExceptionz'SimpleRoot.remote_asynchronousException?sz/0OPPQQQrc td)zC Fail synchronously with a non-pb.Error exception. zremote synchronous exception)rr's rremote_synchronousExceptionz&SimpleRoot.remote_synchronousExceptionEs##ABBBrcDtjtdS)z@ Fail asynchronously with a pb.Error exception. zremote asynchronous error)rr&rr's rremote_asynchronousErrorz#SimpleRoot.remote_asynchronousErrorKsz+,GHHIIIrc td)z? Fail synchronously with a pb.Error exception. zremote synchronous error)rr's rremote_synchronousErrorz"SimpleRoot.remote_synchronousErrorQs9:::rcHGddtj}|d)z> Fail with error that is not known to client. ceZdZdS)4SimpleRoot.remote_unknownError..UnknownErrorNrrrr UnknownErrorr2\s Drr3zI'm not known to client!)r Error)r(r3s rremote_unknownErrorzSimpleRoot.remote_unknownErrorWsA      28   l5666rc.|dSN) raiseJellyr's r remote_jellyzSimpleRoot.remote_jellyas rc.|dSr7) raiseSecurityr's rremote_securityzSimpleRoot.remote_securityds rctj}||j|d|Sr7)rDeferred addCallbackr8callbackr(ds rremote_deferredJellyzSimpleRoot.remote_deferredJellygs: N   do&&& 4rctj}||j|d|Sr7)rr>r?r;r@rAs rremote_deferredSecurityz"SimpleRoot.remote_deferredSecurityms; N   d())) 4rNc td)NzI'm jellyable!)rr(resultss rr8zSimpleRoot.raiseJellyss)***rc td)Nz I'm secure!)r"rGs rr;zSimpleRoot.raiseSecurityvsM***rr7)rrrr)r+r-r/r5r9r<rCrEr8r;rrrr$r$>sRRR CCC JJJ ;;; 777  ++++++++++rr$ceZdZdZdZdZdS)SaveProtocolServerFactoryzd A L{pb.PBServerFactory} that saves the latest connected client in C{protocolInstance}. Nc||_dS)z3 Keep track of the given protocol. N)protocolInstance)r(protocols rclientConnectionMadez.SaveProtocolServerFactory.clientConnectionMades!)r)rrrrrMrOrrrrKrKzs5 )))))rrKceZdZdZdZdS)PBConnTestCaserctt_jj_t j_tfdfd\__ _ dS)Nc8jdSr7) serverFactory buildProtocolr'srz&PBConnTestCase.setUp..D&44T::rc8jdSr7) clientFactoryrUr'srrVz&PBConnTestCase.setUp..rWr) rKr$rTunsafeTracebacksr PBClientFactoryrYr connectedServerconnectedClientpumpr's`rsetUpzPBConnTestCase.setUpsw6z||DD.2.C+/11 % : : : : : : : :     IIIrN)rrrrZr_rrrrQrQs)      rrQceZdZejjZdZdZdZ dZ dZ dZ dZ dd Zd Zd Zd Zd ZdZdZdZdZdZdS)PBFailureTestscfdj}fd}||jdS)Nc||jdr8}t |d|j|j|jfS)NTraceback unavailable )trapcompare tracebackflushLoggedErrors assertEquallentypevalue)errerrs exceptionTypeflushr(s rebz)PBFailureTests._exceptionTest..ebsv HH] # # # LL(A B B B /--m<<  TA...Hci7 7rc\|}||Sr7) callRemote addErrback)rootrBrrmethods r gotRootObjectz4PBFailureTests._exceptionTest..gotRootObjects+''A LL   HrrY getRootObjectr?r^rq)r(rwrprqrBrxrrs```` @r_exceptionTestzPBFailureTests._exceptionTests 8 8 8 8 8 8 8   , , . .       m$$$ rc:|dtdS)z Test that a Deferred returned by a remote method which already has a Failure correctly has that error passed back to the calling side. asynchronousExceptionT)r{r r's rtest_asynchronousExceptionz)PBFailureTests.test_asynchronousExceptions ""#:.gotRootObj..gotFailures4$$S)?)?)D)D%E%EqIIIJr)rrtaddBoth)objfailureDeferredrrrrexpectedrwr(s r gotRootObjz,PBFailureTests._testImpl..gotRootObjsl"77v&&"O '' 333" "rNry)r(rwrrrr rootDeferredrs````` r _testImplzPBFailureTests._testImplsx )7799  # # # # # # # # #   ,,, rc<fd}dd|S)z Test that an exception which is a subclass of L{pb.Error} has more information passed across the network to the calling side. c|t|jt|j|jdS)N+rfrassertNotIsInstancerlstrassertIsInstancermr&r(s r failureJellyz6PBFailureTests.test_jellyFailure..failureJellysL IIj ! ! !  $ $TY 4 4 4  ! !$*di 8 8 82rrrr)r(rs` rtest_jellyFailurez PBFailureTests.test_jellyFailures4       ~~gr<888rc<fd}dd|S)z Test that a Deferred which fails with a L{pb.Error} is treated in the same way as a synchronously raised L{pb.Error}. c|t|jt|j|jdS)Nrrs rfailureDeferredJellyzFPBFailureTests.test_deferredJellyFailure..failureDeferredJellysL IIj ! ! !  $ $TY 4 4 4  ! !$*di 8 8 83r deferredJellyrr)r(rs` rtest_deferredJellyFailurez(PBFailureTests.test_deferredJellyFailures5       ~~os4HIIIrc<fd}dd|S)z A non-jellyable L{pb.Error} subclass raised by a remote method is turned into a Failure with a type set to the FQPN of the exception type. c>|jddS)Ns3twisted.spread.test.test_pbfailure.SynchronousErrorrjrlrs rfailureUnjellyablezBPBFailureTests.test_unjellyableFailure..failureUnjellyables*    Q   3rrrr)r(rs` rtest_unjellyableFailurez&PBFailureTests.test_unjellyableFailures6      ~~0#7IJJJrc<fd}dd|S)z Test that an exception which is a subclass of L{pb.Error} but not known on the client side has its type set properly. c>|jddS)Ns/twisted.spread.test.test_pbfailure.UnknownErrorrrs rfailureUnknownz:PBFailureTests.test_unknownFailure..failureUnknowns*    M   4r unknownErrorrr)r(rs` rtest_unknownFailurez"PBFailureTests.test_unknownFailures4       ~~ndNCCCrc<fd}dd|S)z Test that even if an exception is not explicitly jellyable (by being a L{pb.Jellyable} subclass), as long as it is an L{pb.Error} subclass it receives the same special treatment. c|t|jt|j|jdS)Nrfr"rrlrrrmrs rfailureSecurityz.failureSecurity,sL IIm $ $ $  $ $TY 4 4 4  ! !$*di 8 8 84rsecurityrr)r(rs` rtest_securityFailurez#PBFailureTests.test_securityFailure%s4      ~~j$@@@rc<fd}dd|S)z Test that a Deferred which fails with a L{pb.Error} which is not also a L{pb.Jellyable} is treated in the same way as a synchronously raised exception of the same type. c|t|jt|j|jdS)Nrrs rfailureDeferredSecurityzEPBFailureTests.test_deferredSecurity..failureDeferredSecurity;sL IIm $ $ $  $ $TY 4 4 4  ! !$*di 8 8 85rdeferredSecurityrr)r(rs` rtest_deferredSecurityz$PBFailureTests.test_deferredSecurity4s6      ~~0%9PQQQrcHfd}dd|tS)z Test that attempting to call a method which is not defined correctly results in an AttributeError on the calling side. c||tj|jddS)Nrd)rfr NoSuchMethodrgrhrs r failureNoSuchz>PBFailureTests.test_noSuchMethodFailure..failureNoSuchIs3 IIbo & & & LL)B C C C5rnosuchr)rAttributeError)r(rs` rtest_noSuchMethodFailurez'PBFailureTests.test_noSuchMethodFailureCs6       ~~h}nMMMrcj}d}||fd}||jdS)z Test that a copied failure received from a PB call can be logged locally. Note: this test needs some serious help: all it really tests is that log.err(copiedFailure) doesn't raise an exception. c,|dS)Nr)rt)rootObjs r connectedz;PBFailureTests.test_copiedFailureLogging..connectedZs%%&<== =rctj|t}t |ddS)N)rrnrirrjrk)failureror(s r exceptionz;PBFailureTests.test_copiedFailureLogging..exception_sI GG   ))*>??D   SYY * * * * *rN)rYrzr?rur^rq)r(rBrrs` rtest_copiedFailureLoggingz(PBFailureTests.test_copiedFailureLoggingPs   , , . . > > > i    + + + + + Y rctjtd}tjtj|t }gfd}|}|dt|j | tdd} |j ttd |jd |jd dS) z L{pb.CopiedFailure.throwExceptionIntoGenerator} will throw a L{RemoteError} into the given paused generator at the point where it last yielded. fooinvokerc3K dVddS#tj$r }|Yd}~dSd}~wwxYw)NzRemoteError not raised)r&r RemoteErrorappend)rrr(s r generatorFunczFPBFailureTests.test_throwExceptionIntoGenerator..generatorFuncqsy 4  233333> & & &  %%%%%%%%% &s AA  ANrerascii)rrd)r CopyableFailurerrunjelly DummyInvokersend assertRaises StopIterationthrowExceptionIntoGeneratorrjrk remoteTyperencodeargsremoteTraceback)r(originalcopyrgenrrs` @r test_throwExceptionIntoGeneratorz/PBFailureTests.test_throwExceptionIntoGeneratorgs% %nU&;&;<<}U[<>>JJJKK  4 4 4 4 4 4moo  -)I3OOO Y+++l n)=)=)D)DW)M)MNNN 8,,, ,.GHHHHHrr7)rrrr TestCaserjrgr{r~rrrrrrrrrrrrrrrrrrrarasF+G&YYYWWWRRRPPP0 9 9 9 J J J K K K D D D A A A R R R N N N.IIIIIrrac*eZdZejjZdZdS)PBFailureUnsafeTestsreN)rrrr r failIfEqualsrgrZrrrrrs!,GrrceZdZdZdZdS)rzZ A behaviorless object to be used as the invoker parameter to L{jelly.jelly}. N)rrrrserializingPerspectiverrrrrs" "rrc$eZdZdZdZdZdZdS)FailureJellyingTestsz: Tests for the interaction of jelly and failures. cFtjt}||tt||t t t jt j|t}||tt||t t dS)z An unjellied L{CopyableFailure} has a check method which behaves the same way as the original L{CopyableFailure}'s check method. rN r rZeroDivisionErrorassertIscheckArithmeticErrorrrr)r(rcopieds rtest_unjelliedFailureCheckz/FailureJellyingTests.test_unjelliedFailureChecks %&7&9&9:: hnn%6779JKKK hnn_55GGGu{8\^^LLLMM fll#4557HIII fll?33_EEEEErctjt}||tt||t t t jt j|t}tj|}t jt j|t}||tt||t t dS)aX The object which results from jellying a L{CopyableFailure}, unjellying the result, creating a new L{CopyableFailure} from the result of that, jellying it, and finally unjellying the result of that has a check method which behaves the same way as the original L{CopyableFailure}'s check method. rNr)r(r copiedOnce derivative copiedTwices rtest_twiceUnjelliedFailureCheckz4FailureJellyingTests.test_twiceUnjelliedFailureChecks%&7&9&9:: hnn%6779JKKK hnn_55GGG]5;x#P#P#PQQ ' 33 mEK LNN$S$S$STT  k''(9::&>??u{8\^^LLLMMf%%%OO MTT    ):):;;;;;rN)rrrrrrr rrrrrsQ F F FKKK"<<<<rs:""""""''''''----------777777"""""" I9rx     "BHbm        BHbm   *j111-777""=1119+9+9+9+9+9+9+9+x ) ) ) ) ) 2 ) ) )     X&   "fIfIfIfIfI^fIfIfIR> """"""""1<1<1<1<1<8,1<1<1<1<1