U 2ܟj9@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@seZdZdZdS)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__rrF/tmp/pip-unpacked-wheel-4b529toj/twisted/spread/test/test_pbfailure.pyr sr c@seZdZdZdS)SynchronousExceptionzm Helper used to test remote methods which raise exceptions which are not L{pb.Error} subclasses. Nr rrrrrsrc@seZdZdZdS)AsynchronousErrorz Helper used to test remote methods which return Deferreds which fail with exceptions which are L{pb.Error} subclasses. Nr rrrrr"src@seZdZdZdS)SynchronousErrorzi Helper used to test remote methods which raise exceptions which are L{pb.Error} subclasses. Nr rrrrr)src@s eZdZdS) JellyErrorNr rrrrrrr0src@s eZdZdS) SecurityErrorNrrrrrr4src@sheZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dddZ dddZ dS) SimpleRootcCsttdS)zD Fail asynchronously with a non-pb.Error exception. zremote asynchronous exception)rfailr selfrrrremote_asynchronousException?sz'SimpleRoot.remote_asynchronousExceptioncCs tddS)zC Fail synchronously with a non-pb.Error exception. zremote synchronous exceptionN)rrrrrremote_synchronousExceptionEsz&SimpleRoot.remote_synchronousExceptioncCsttdS)z@ Fail asynchronously with a pb.Error exception. zremote asynchronous error)rrrrrrrremote_asynchronousErrorKsz#SimpleRoot.remote_asynchronousErrorcCs tddS)z? Fail synchronously with a pb.Error exception. zremote synchronous errorN)rrrrrremote_synchronousErrorQsz"SimpleRoot.remote_synchronousErrorcCsGdddtj}|ddS)z> Fail with error that is not known to client. c@s eZdZdS)z4SimpleRoot.remote_unknownError..UnknownErrorNrrrrr UnknownError\sr!zI'm not known to client!N)rError)rr!rrrremote_unknownErrorWszSimpleRoot.remote_unknownErrorcCs |dSN) raiseJellyrrrr remote_jellyaszSimpleRoot.remote_jellycCs |dSr$) raiseSecurityrrrrremote_securitydszSimpleRoot.remote_securitycCs"t}||j|d|Sr$)rDeferred addCallbackr%callbackrdrrrremote_deferredJellygs  zSimpleRoot.remote_deferredJellycCs"t}||j|d|Sr$)rr)r*r'r+r,rrrremote_deferredSecurityms  z"SimpleRoot.remote_deferredSecurityNcCs tddS)NzI'm jellyable!)rrresultsrrrr%sszSimpleRoot.raiseJellycCs tddS)Nz I'm secure!)rr0rrrr'vszSimpleRoot.raiseSecurity)N)N)r rrrrrr r#r&r(r.r/r%r'rrrrr>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)rprotocolrrrclientConnectionMadesz.SaveProtocolServerFactory.clientConnectionMade)r rrrr3r5rrrrr2zsr2c@seZdZdZddZdS)PBConnTestCasercsJtt_jj_t_tfddfdd\__ _ dS)Ncs jdSr$) serverFactory buildProtocolrrrrz&PBConnTestCase.setUp..cs jdSr$) clientFactoryr8rrrrr9r:) r2rr7unsafeTracebacksrZPBClientFactoryr;r ZconnectedServerZconnectedClientpumprrrrsetUps     zPBConnTestCase.setUpN)r rrr<r>rrrrr6sr6c@seZdZejjZddZddZddZ ddZ d d Z d d Z d dZ d$ddZddZddZddZddZddZddZddZd d!Zd"d#ZdS)%PBFailureTestscs@fddj}fdd}||jdS)NcsF||jdr6}t|d|j|j|jfS)NTraceback unavailable )trapcompare tracebackflushLoggedErrors assertEquallentypevalue)errerrs) exceptionTypeflushrrrebs   z)PBFailureTests._exceptionTest..ebcs|}||Sr$) callRemote addErrback)rootr-)rNmethodrr gotRootObjects  z4PBFailureTests._exceptionTest..gotRootObjectr; getRootObjectr*r=rM)rrRrLrMr-rSr)rNrLrMrRrr_exceptionTests   zPBFailureTests._exceptionTestcCs|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. ZasynchronousExceptionT)rVr rrrrtest_asynchronousExceptionsz)PBFailureTests.test_asynchronousExceptioncCs|dtdS)zw Like L{test_asynchronousException}, but for a method which raises an exception synchronously. synchronousExceptionT)rVrrrrrtest_synchronousExceptionsz(PBFailureTests.test_synchronousExceptioncCs|dtdS)z Like L{test_asynchronousException}, but for a method which returns a Deferred failing with an L{pb.Error} subclass. ZasynchronousErrorF)rVrrrrrtest_asynchronousErrorsz%PBFailureTests.test_asynchronousErrorcCs|dtdS)z} Like L{test_asynchronousError}, but for a method which synchronously raises a L{pb.Error} subclass. synchronousErrorF)rVrrrrrtest_synchronousErrorsz$PBFailureTests.test_synchronousErrorcCs||||Sr$)rF)rresultexpectedResultrrr_successs zPBFailureTests._successcCs|j|j||fd|S)N)Z callbackArgs)Z addCallbacksr_)rZ remoteCallr^rNrrr_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|}dk r4fdd}|||S)Ncstd|S)NrA)rFrGrE)rJ)excrrr gotFailuresz@PBFailureTests._testImpl..gotRootObj..gotFailure)r`rOZaddBoth)objZfailureDeferredrbrNraexpectedrRrrr gotRootObjs z,PBFailureTests._testImpl..gotRootObjNrT)rrRrerNraZ rootDeferredrfrrdr _testImpls  zPBFailureTests._testImplcsfdd}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. cs,|t|jt|j|jdS)N+rBrassertNotIsInstancerHstrassertIsInstancerIrrrr failureJellys z6PBFailureTests.test_jellyFailure..failureJellyrrhrg)rrnrrrtest_jellyFailures z PBFailureTests.test_jellyFailurecsfdd}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}. cs,|t|jt|j|jdS)NrirmrrrfailureDeferredJellys zFPBFailureTests.test_deferredJellyFailure..failureDeferredJellyZ deferredJellyrqro)rrrrrrtest_deferredJellyFailures z(PBFailureTests.test_deferredJellyFailurecsfdd}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. cs|jddS)Ns3twisted.spread.test.test_pbfailure.SynchronousErrorrFrHrmrrrfailureUnjellyables zBPBFailureTests.test_unjellyableFailure..failureUnjellyabler[rtro)rrvrrrtest_unjellyableFailures z&PBFailureTests.test_unjellyableFailurecsfdd}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. cs|jddS)Ns/twisted.spread.test.test_pbfailure.UnknownErrorrurmrrrfailureUnknowns z:PBFailureTests.test_unknownFailure..failureUnknownZ unknownErrorrxro)rryrrrtest_unknownFailures z"PBFailureTests.test_unknownFailurecsfdd}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. cs,|t|jt|j|jdS)NrBrrjrHrkrlrIrmrrrfailureSecurity,s z.failureSecuritysecurityr{ro)rr}rrrtest_securityFailure%s z#PBFailureTests.test_securityFailurecsfdd}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. cs,|t|jt|j|jdS)Nr|rmrrrfailureDeferredSecurity;s zEPBFailureTests.test_deferredSecurity..failureDeferredSecurityZdeferredSecurityrro)rrrrrtest_deferredSecurity4s z$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)Nr@)rBrZ NoSuchMethodrCrDrmrrr failureNoSuchIs z>PBFailureTests.test_noSuchMethodFailure..failureNoSuchZnosuchr)rgAttributeError)rrrrrtest_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)NrX)rO)ZrootObjrrr connectedZsz;PBFailureTests.test_copiedFailureLogging..connectedcs(t|t}t|ddS)N)rrJrErrFrG)ZfailurerKrrr exception_s  z;PBFailureTests.test_copiedFailureLogging..exceptionN)r;rUr*rPr=rM)rr-rrrrrtest_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 3sHz dVWn.tjk r8}z|W5d}~XYn XddS)NzRemoteError not raised)rZ RemoteErrorappendr)rarrrr generatorFuncqs  zFPBFailureTests.test_throwExceptionIntoGenerator..generatorFuncNrArascii)rr@)rCopyableFailurerrunjelly DummyInvokersend assertRaises StopIterationZthrowExceptionIntoGeneratorrFrGZ remoteTyperencodeargsZremoteTraceback)roriginalcopyrgenrarrr test_throwExceptionIntoGeneratorgs z/PBFailureTests.test_throwExceptionIntoGenerator)N)r rrr TestCaserFrCrVrWrYrZr\r_r`rgrprsrwrzrrrrrrrrrr?s$  r?c@seZdZejjZdZdS)PBFailureUnsafeTestsrAN)r rrr rZ failIfEqualsrCr<rrrrrsrc@seZdZdZdZdS)rzZ A behaviorless object to be used as the invoker parameter to L{jelly.jelly}. N)r rrrZserializingPerspectiverrrrrsrc@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)rrcopiedrrrtest_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)rrZ 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 ExceptionrrrrZprintTracebackrformatrFgetvalue)rrroutputrZexpectedOutputrrr test_printTracebackIncludesValues z5FailureJellyingTests.test_printTracebackIncludesValueN)r rrrrrrrrrrrs rN)(riorZtwisted.internetrZtwisted.pythonrZtwisted.python.reflectrZtwisted.spreadrrrZtwisted.test.iosimr Z twisted.trialr rr rr"rrZ JellyableZ RemoteCopyrrZsetUnjellyableForClassZglobalSecurityZallowInstancesOfZRootrZPBServerFactoryr2rr6r?rrrrrrrs0         <j