o i9@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__rrhC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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 exceptionN)rfailr selfrrrremote_asynchronousException?z'SimpleRoot.remote_asynchronousExceptioncCtd)zC Fail synchronously with a non-pb.Error exception. zremote synchronous exceptionN)rr rrrremote_synchronousExceptionEz&SimpleRoot.remote_synchronousExceptioncCr)z@ Fail asynchronously with a pb.Error exception. zremote asynchronous errorN)rrrr rrrremote_asynchronousErrorKr#z#SimpleRoot.remote_asynchronousErrorcCr$)z? Fail synchronously with a pb.Error exception. zremote synchronous errorN)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!N)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?rrEunsafeTracebacksrZPBClientFactoryrIr ZconnectedServerZconnectedClientpumpr rr rsetUps     zPBConnTestCase.setUpN)rrrrJrLrrrrrCs 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 getRootObjectr5rKr[)r!r`rZr[r8rar)r\rZr[r`r!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. ZasynchronousExceptionTN)rdr r rrrtest_asynchronousExceptionz)PBFailureTests.test_asynchronousExceptioncCre)zw Like L{test_asynchronousException}, but for a method which raises an exception synchronously. synchronousExceptionTN)rdrr rrrtest_synchronousExceptionrgz(PBFailureTests.test_synchronousExceptioncCre)z Like L{test_asynchronousException}, but for a method which returns a Deferred failing with an L{pb.Error} subclass. ZasynchronousErrorFN)rdrr rrrtest_asynchronousErrorrgz%PBFailureTests.test_asynchronousErrorcCre)z} Like L{test_asynchronousError}, but for a method which synchronously raises a L{pb.Error} subclass. synchronousErrorFN)rdrr rrrtest_synchronousErrorrgz$PBFailureTests.test_synchronousErrorcCs||||Sr-)rT)r!resultexpectedResultrrr_successs zPBFailureTests._successcCs|j|j||fd|S)N)Z callbackArgs)Z addCallbacksro)r!Z remoteCallrnr\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)NrO)rTrUrS)rX)excr!rr gotFailuresz@PBFailureTests._testImpl..gotRootObj..gotFailure)rpr]ZaddBoth)objZfailureDeferredrrr\rqexpectedr`r!rr gotRootObjs  z,PBFailureTests._testImpl..gotRootObjNrb)r!r`rur\rqZ rootDeferredrvrrtr _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+rPrassertNotIsInstancerVstrassertIsInstancerWrr rr failureJelly z6PBFailureTests.test_jellyFailure..failureJellyrrzNrw)r!rrr rtest_jellyFailure z PBFailureTests.test_jellyFailurecrx)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}. cry)Nr{rr rrfailureDeferredJellyrzFPBFailureTests.test_deferredJellyFailure..failureDeferredJellyZ deferredJellyrNr)r!rrr rtest_deferredJellyFailurerz(PBFailureTests.test_deferredJellyFailurecrx)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.SynchronousErrorrTrVrr rrfailureUnjellyablezBPBFailureTests.test_unjellyableFailure..failureUnjellyablerkrNr)r!rrr rtest_unjellyableFailure z&PBFailureTests.test_unjellyableFailurecrx)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..failureUnknownZ unknownErrorrNr)r!rrr rtest_unknownFailurerz"PBFailureTests.test_unknownFailurecrx)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. cry)NrPrr|rVr}r~rWrr rrfailureSecurity,rz.failureSecuritysecurityrNr)r!rrr rtest_securityFailure%rz#PBFailureTests.test_securityFailurecrx)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. cry)Nrrr rrfailureDeferredSecurity;rzEPBFailureTests.test_deferredSecurity..failureDeferredSecurityZdeferredSecurityrNr)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)NrN)rPrZ NoSuchMethodrQrRrr rr failureNoSuchIs z>PBFailureTests.test_noSuchMethodFailure..failureNoSuchZnosuchrN)rwAttributeError)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)Nrh)r])ZrootObjrrr connectedZs z;PBFailureTests.test_copiedFailureLogging..connectedcs(t|t}t|ddS)N)rrXrSrrTrU)ZfailurerYr rr exception_s  z;PBFailureTests.test_copiedFailureLogging..exceptionN)rIrcr5r^rKr[)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. fooZinvokerc 3sNzdVWntjy}z |WYd}~dSd}~wwddS)NzRemoteError not raised)rZ RemoteErrorappendr)rqrr!rr generatorFuncqs zFPBFailureTests.test_throwExceptionIntoGenerator..generatorFuncNrOrascii)rrN)rCopyableFailurerrunjelly DummyInvokersend assertRaises StopIterationZthrowExceptionIntoGeneratorrTrUZ remoteTyperencodeargsZremoteTraceback)r!originalcopyrgenrqrrr test_throwExceptionIntoGeneratorgs z/PBFailureTests.test_throwExceptionIntoGeneratorr-)rrrr TestCaserTrQrdrfrirjrlrorprwrrrrrrrrrrrrrrMs&  rMc@seZdZejjZdZdS)PBFailureUnsafeTestsrON)rrrr rZ failIfEqualsrQrJrrrrrsrc@seZdZdZdZdS)rzZ A behaviorless object to be used as the invoker parameter to L{jelly.jelly}. N)rrrrZserializingPerspectiverrrrrsrc@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 rrZeroDivisionErrorZassertIscheckArithmeticErrorrrr)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!rZ copiedOnceZ derivativeZ 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 ExceptionrrrrZprintTracebackrformatrTgetvalue)r!rroutputrZexpectedOutputrrr test_printTracebackIncludesValues z5FailureJellyingTests.test_printTracebackIncludesValueN)rrrrrrrrrrrrs  rN)(riorZtwisted.internetrZtwisted.pythonrZtwisted.python.reflectrZtwisted.spreadrrrZtwisted.test.iosimr Z twisted.trialr rr rr*rrZ JellyableZ RemoteCopyrrZsetUnjellyableForClassZglobalSecurityZallowInstancesOfZRootrZPBServerFactoryr?rrCrMrrrrrrrs2         <j