o >h9@srdZddlmZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZGd d d eZGd d d eZGd dde jZGddde jZGddde je je jZGddde je jZe eee eee jeGddde jZGddde j Z!Gdddej"Z#Gddde#Z$Gddde$Z%Gdd d Z&Gd!d"d"ej"Z'd#S)$z! Tests for error handling in PB. )StringIO)defer)log)qual)flavorsjellypb)connectedServerAndClient)unittestc@eZdZdZdS)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__rr/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/spread/test/test_pbfailure.pyr r c@r )SynchronousExceptionzm Helper used to test remote methods which raise exceptions which are not L{pb.Error} subclasses. Nr rrrrrrrc@r )AsynchronousErrorz Helper used to test remote methods which return Deferreds which fail with exceptions which are L{pb.Error} subclasses. Nr rrrrr"rrc@r )SynchronousErrorzi Helper used to test remote methods which raise exceptions which are L{pb.Error} subclasses. Nr rrrrr)rrc@ eZdZdS) JellyErrorNrrrrrrrr0rc@r) SecurityErrorNrrrrrr4rrc@sheZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dddZ dddZ dS) SimpleRootcCttdS)zD Fail asynchronously with a non-pb.Error exception. zremote asynchronous exception)rfailr selfrrrremote_asynchronousException?z'SimpleRoot.remote_asynchronousExceptioncCtd)zC Fail synchronously with a non-pb.Error exception. zremote synchronous exception)rr rrrremote_synchronousExceptionEz&SimpleRoot.remote_synchronousExceptioncCr)z@ Fail asynchronously with a pb.Error exception. zremote asynchronous error)rrrr rrrremote_asynchronousErrorKr#z#SimpleRoot.remote_asynchronousErrorcCr$)z? Fail synchronously with a pb.Error exception. zremote synchronous error)rr rrrremote_synchronousErrorQr&z"SimpleRoot.remote_synchronousErrorcCsGdddtj}|d)z> Fail with error that is not known to client. c@r)z4SimpleRoot.remote_unknownError..UnknownErrorNrrrrr UnknownError\rr)zI'm not known to client!)rError)r!r)rrrremote_unknownErrorWszSimpleRoot.remote_unknownErrorcC |dSN) raiseJellyr rrr remote_jellya zSimpleRoot.remote_jellycCr,r-) raiseSecurityr rrrremote_securitydr0zSimpleRoot.remote_securitycC"t}||j|d|Sr-)rDeferred addCallbackr.callbackr!drrrremote_deferredJellyg  zSimpleRoot.remote_deferredJellycCr3r-)rr4r5r1r6r7rrrremote_deferredSecuritymr:z"SimpleRoot.remote_deferredSecurityNcCr$)NzI'm jellyable!)rr!resultsrrrr.szSimpleRoot.raiseJellycCr$)Nz I'm secure!)rr<rrrr1vr>zSimpleRoot.raiseSecurityr-)rrrr"r%r'r(r+r/r2r9r;r.r1rrrrr>s  rc@seZdZdZdZddZdS)SaveProtocolServerFactoryzd A L{pb.PBServerFactory} that saves the latest connected client in C{protocolInstance}. NcCs ||_dS)z3 Keep track of the given protocol. N)protocolInstance)r!protocolrrrclientConnectionMades z.SaveProtocolServerFactory.clientConnectionMade)rrrrr@rBrrrrr?zs r?c@seZdZdZddZdS)PBConnTestCasercsJtt_jj_t_tfddfdd\__ _ dS)Nc jdSr-) serverFactory buildProtocolrr rr z&PBConnTestCase.setUp..crDr-) clientFactoryrFrr rrrGrH) r?rrEunsafeTracebacksrPBClientFactoryrIr connectedServerconnectedClientpumpr rr rsetUps     zPBConnTestCase.setUpN)rrrrJrOrrrrrCs rCc@seZdZejjZddZddZddZ ddZ d d Z d d Z d dZ d$ddZddZddZddZddZddZddZddZd d!Zd"d#ZdS)%PBFailureTestscs@fddj}fdd}||jdS)NcsF||jdr}t|d|j|j|jfS)NTraceback unavailable )trapcompare tracebackflushLoggedErrors assertEquallentypevalue)errerrs) exceptionTypeflushr!rrebs  z)PBFailureTests._exceptionTest..ebcs|}||Sr-) callRemote addErrback)rootr8)r_methodrr gotRootObjects  z4PBFailureTests._exceptionTest..gotRootObjectrI getRootObjectr5rNr^)r!rcr]r^r8rdr)r_r]r^rcr!r_exceptionTests   zPBFailureTests._exceptionTestcC|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)rgr r rrrtest_asynchronousExceptionz)PBFailureTests.test_asynchronousExceptioncCrh)zw Like L{test_asynchronousException}, but for a method which raises an exception synchronously. synchronousExceptionT)rgrr rrrtest_synchronousExceptionrkz(PBFailureTests.test_synchronousExceptioncCrh)z Like L{test_asynchronousException}, but for a method which returns a Deferred failing with an L{pb.Error} subclass. asynchronousErrorF)rgrr rrrtest_asynchronousErrorrkz%PBFailureTests.test_asynchronousErrorcCrh)z} Like L{test_asynchronousError}, but for a method which synchronously raises a L{pb.Error} subclass. synchronousErrorF)rgrr rrrtest_synchronousErrorrkz$PBFailureTests.test_synchronousErrorcCs||||Sr-)rW)r!resultexpectedResultrrr_successs zPBFailureTests._successcCs|j|j||fd|S)N) callbackArgs) addCallbacksrt)r! remoteCallrsr_rrr_addFailingCallbackssz#PBFailureTests._addFailingCallbacksNcs6j}fdd}||jdS)z Call the given remote method and attach the given errback to the resulting Deferred. If C{exc} is not None, also assert that one exception of that type was logged. cs8|}durfdd}|||S)Ncstd|S)NrR)rWrXrV)r[)excr!rr gotFailuresz@PBFailureTests._testImpl..gotRootObj..gotFailure)rxr`addBoth)objfailureDeferredrzr_ryexpectedrcr!rr gotRootObjs  z,PBFailureTests._testImpl..gotRootObjNre)r!rcrr_ry rootDeferredrrr~r _testImpls  zPBFailureTests._testImplcfdd}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+rSrassertNotIsInstancerYstrassertIsInstancerZrr rr failureJelly z6PBFailureTests.test_jellyFailure..failureJellyrrr)r!rrr rtest_jellyFailure z PBFailureTests.test_jellyFailurecr)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}. cr)Nrrr rrfailureDeferredJellyrzFPBFailureTests.test_deferredJellyFailure..failureDeferredJelly deferredJellyrr)r!rrr rtest_deferredJellyFailurerz(PBFailureTests.test_deferredJellyFailurecr)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.SynchronousErrorrWrYrr rrfailureUnjellyablezBPBFailureTests.test_unjellyableFailure..failureUnjellyablerprr)r!rrr rtest_unjellyableFailure z&PBFailureTests.test_unjellyableFailurecr)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. cr)Ns/twisted.spread.test.test_pbfailure.UnknownErrorrrr rrfailureUnknownrz:PBFailureTests.test_unknownFailure..failureUnknown unknownErrorrr)r!rrr rtest_unknownFailurerz"PBFailureTests.test_unknownFailurecr)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. cr)NrSrrrYrrrZrr rrfailureSecurity,rz.failureSecuritysecurityrr)r!rrr rtest_securityFailure%rz#PBFailureTests.test_securityFailurecr)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. cr)Nrrr rrfailureDeferredSecurity;rzEPBFailureTests.test_deferredSecurity..failureDeferredSecuritydeferredSecurityrr)r!rrr rtest_deferredSecurity4rz$PBFailureTests.test_deferredSecuritycsfdd}dd|tS)z Test that attempting to call a method which is not defined correctly results in an AttributeError on the calling side. cs|tj|jddS)NrQ)rSr NoSuchMethodrTrUrr rr failureNoSuchIs z>PBFailureTests.test_noSuchMethodFailure..failureNoSuchnosuchr)rAttributeError)r!rrr rtest_noSuchMethodFailureCs z'PBFailureTests.test_noSuchMethodFailurecs@j}dd}||fdd}||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. cSs |dS)Nrl)r`)rootObjrrr connectedZs z;PBFailureTests.test_copiedFailureLogging..connectedcs(t|t}t|ddS)N)rr[rVrrWrX)failurer\r rr exception_s  z;PBFailureTests.test_copiedFailureLogging..exceptionN)rIrfr5rarNr^)r!r8rrrr rtest_copiedFailureLoggingPs    z(PBFailureTests.test_copiedFailureLoggingcsttd}ttj|td}gfdd}|}|dt|j | t dd} |j t td |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. fooinvokerc 3sNzdVWntjy}z |WYd}~dSd}~wwddS)NzRemoteError not raised)r RemoteErrorappendr)ryrr!rr generatorFuncqs zFPBFailureTests.test_throwExceptionIntoGenerator..generatorFuncNrRrascii)rrQ)rCopyableFailurerrunjelly DummyInvokersend assertRaises StopIterationthrowExceptionIntoGeneratorrWrX remoteTyperencodeargsremoteTraceback)r!originalcopyrgenryrrr test_throwExceptionIntoGeneratorgs z/PBFailureTests.test_throwExceptionIntoGeneratorr-)rrrr TestCaserWrTrgrjrmrorqrtrxrrrrrrrrrrrrrrrPs&  rPc@seZdZejjZdZdS)PBFailureUnsafeTestsrRN)rrrr r failIfEqualsrTrJrrrrrsrc@seZdZdZdZdS)rzZ A behaviorless object to be used as the invoker parameter to L{jelly.jelly}. N)rrrrserializingPerspectiverrrrrsrc@s(eZdZdZddZddZddZdS) FailureJellyingTestsz: Tests for the interaction of jelly and failures. cCsntt}||tt||ttttj|td}||tt||ttdS)z An unjellied L{CopyableFailure} has a check method which behaves the same way as the original L{CopyableFailure}'s check method. rN rrZeroDivisionErrorassertIscheckArithmeticErrorrrr)r!rcopiedrrrtest_unjelliedFailureChecks z/FailureJellyingTests.test_unjelliedFailureCheckcCstt}||tt||ttttj|td}t|}ttj|td}||tt||ttdS)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 copiedTwicerrrtest_twiceUnjelliedFailureChecks  z4FailureJellyingTests.test_twiceUnjelliedFailureCheckcCsZttd}ttj|td}t}||tt}d |}| || dS)a When L{CopiedFailure.printTraceback} is used to print a copied failure which was unjellied from a L{CopyableFailure} with C{unsafeTracebacks} set to C{False}, the string representation of the exception value is included in the output. z some reasonrz.Traceback from remote host -- {}: some reason N) rr ExceptionrrrrprintTracebackrformatrWgetvalue)r!rroutputrexpectedOutputrrr test_printTracebackIncludesValues z5FailureJellyingTests.test_printTracebackIncludesValueN)rrrrrrrrrrrrs  rN)(riortwisted.internetrtwisted.pythonrtwisted.python.reflectrtwisted.spreadrrrtwisted.test.iosimr twisted.trialr rr rr*rr Jellyable RemoteCopyrrsetUnjellyableForClassglobalSecurityallowInstancesOfRootrPBServerFactoryr?rrCrPrrrrrrrs2         <j