o b;@sjdZddlmZddlmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZGdd d eZGd d d eZGd d d e jZGddde jZGddde je je jZGddde je jZe eee eee jeGddde jZGddde jZ Gdddej!Z"Gddde"Z#Gddde#Z$GdddZ%Gd d!d!ej!Z&d"S)#z! Tests for error handling in PB. )StringIO)deferreactor)log)qual)flavorsjellypb)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__rrD/usr/lib/python3/dist-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) JellyErrorNrrrrrrrr/rc@r) SecurityErrorNrrrrrr3rrc@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_synchronousExceptionDz&SimpleRoot.remote_synchronousExceptioncCr)z@ Fail asynchronously with a pb.Error exception. zremote asynchronous error)rrrr rrrremote_asynchronousErrorJr#z#SimpleRoot.remote_asynchronousErrorcCr$)z? Fail synchronously with a pb.Error exception. zremote synchronous error)rr rrrremote_synchronousErrorPr&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!)r Error)r!r)rrrremote_unknownErrorVszSimpleRoot.remote_unknownErrorcC |dSN) raiseJellyr rrr remote_jelly` zSimpleRoot.remote_jellycCr,r-) raiseSecurityr rrrremote_securitycr0zSimpleRoot.remote_securitycC"t}||j|d|Sr-)rDeferred addCallbackr.callbackr!drrrremote_deferredJellyf  zSimpleRoot.remote_deferredJellycCr3r-)rr4r5r1r6r7rrrremote_deferredSecuritylr:z"SimpleRoot.remote_deferredSecurityNcCr$)NzI'm jellyable!)rr!resultsrrrr.rzSimpleRoot.raiseJellycCr$)Nz I'm secure!)rr<rrrr1ur>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?ys r?c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)PBConnTestCasercCs||dSr-) _setUpServer _setUpClientr rrrsetUps zPBConnTestCase.setUpcCs.tt|_|j|j_tjd|jdd|_dS)Nr 127.0.0.1) interface)r?r serverFactoryunsafeTracebacksr listenTCP serverPortr rrrrDs   zPBConnTestCase._setUpServercCs,|jj}t|_td||j|_dS)NrG) rLgetHostportr PBClientFactory clientFactoryr connectTCPclientConnector)r!portNorrrrEs   zPBConnTestCase._setUpClientcCs0|jjdur |jjjt||gSr-)rIr@ transportloseConnectionr gatherResults_tearDownServer_tearDownClientr rrrtearDowns zPBConnTestCase.tearDowncCst|jjSr-)r maybeDeferredrL stopListeningr rrrrWszPBConnTestCase._tearDownServercCs|jtdSr-)rR disconnectrsucceedr rrrrXs  zPBConnTestCase._tearDownClientN) rrrrJrFrDrErYrWrXrrrrrCs rCc@seZdZejjZddZddZddZ ddZ d d Z d d Z d dZ d$ddZddZddZddZddZddZddZddZd d!Zd"d#ZdS)%PBFailureTestscs6fddj}fdd}|||S)NcsF||jdr}t|d|j|j|jfS)NTraceback unavailable )trapcompare tracebackflushLoggedErrors assertEquallentypevalue)errerrs) exceptionTypeflushr!rrebs  z)PBFailureTests._exceptionTest..ebcs|}||Sr-) callRemote addErrback)rootr8)rmmethodrr gotRootObjects  z4PBFailureTests._exceptionTest..gotRootObjectrP getRootObjectr5)r!rqrkrlr8rrr)rmrkrlrqr!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)rur r rrrtest_asynchronousExceptionz)PBFailureTests.test_asynchronousExceptioncCrv)zw Like L{test_asynchronousException}, but for a method which raises an exception synchronously. synchronousExceptionT)rurr rrrtest_synchronousExceptionryz(PBFailureTests.test_synchronousExceptioncCrv)z Like L{test_asynchronousException}, but for a method which returns a Deferred failing with an L{pb.Error} subclass. asynchronousErrorF)rurr rrrtest_asynchronousErrorryz%PBFailureTests.test_asynchronousErrorcCrv)z} Like L{test_asynchronousError}, but for a method which synchronously raises a L{pb.Error} subclass. synchronousErrorF)rurr rrrtest_synchronousErrorryz$PBFailureTests.test_synchronousErrorcCs||||Sr-)re)r!resultexpectedResultrrr_successs zPBFailureTests._successcCs|j|j||fd|S)N) callbackArgs) addCallbacksr)r! remoteCallrrmrrr_addFailingCallbackssz#PBFailureTests._addFailingCallbacksNcs,j}fdd}|||S)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)Nr`)rerfrd)ri)excr!rr gotFailuresz@PBFailureTests._testImpl..gotRootObj..gotFailure)rrnaddBoth)objfailureDeferredrrmrexpectedrqr!rr gotRootObjs  z,PBFailureTests._testImpl..gotRootObjrs)r!rqrrmr rootDeferredrrrr _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+rarassertNotIsInstancergstrassertIsInstancerhrr 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_deferredJellyFailure rz(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.SynchronousErrorrergrr rrfailureUnjellyablezBPBFailureTests.test_unjellyableFailure..failureUnjellyabler~rr)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 rrfailureUnknown-rz:PBFailureTests.test_unknownFailure..failureUnknown unknownErrorrr)r!rrr rtest_unknownFailure'rz"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)Nrarrrgrrrhrr rrfailureSecurity<rz.failureSecuritysecurityrr)r!rrr rtest_securityFailure5rz#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 rrfailureDeferredSecurityKrzEPBFailureTests.test_deferredSecurity..failureDeferredSecuritydeferredSecurityrr)r!rrr rtest_deferredSecurityDrz$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_)rar NoSuchMethodrbrcrr rr failureNoSuchYs z>PBFailureTests.test_noSuchMethodFailure..failureNoSuchnosuchr)rAttributeError)r!rrr rtest_noSuchMethodFailureSs z'PBFailureTests.test_noSuchMethodFailurecs6j}dd}||fdd}|||S)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)Nrz)rn)rootObjrrr connectedjs z;PBFailureTests.test_copiedFailureLogging..connectedcs(t|t}t|ddS)N)rrirdrrerf)failurerjr rr exceptionos  z;PBFailureTests.test_copiedFailureLogging..exception)rPrtr5ro)r!r8rrrr rtest_copiedFailureLogging`s    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)rrr!rr generatorFuncs zFPBFailureTests.test_throwExceptionIntoGenerator..generatorFuncNr`rascii)rr_)r CopyableFailurerrunjelly DummyInvokersend assertRaises StopIterationthrowExceptionIntoGeneratorrerf remoteTyperencodeargsremoteTraceback)r!originalcopyrgenrrrr test_throwExceptionIntoGeneratorxs z/PBFailureTests.test_throwExceptionIntoGeneratorr-)rrrr TestCasererbrurxr{r}rrrrrrrrrrrrrrrrrr^s&  r^c@seZdZejjZdZdS)PBFailureUnsafeTestsr`N)rrrr r failIfEqualsrbrJrrrrrsrc@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 r rZeroDivisionErrorassertIscheckArithmeticErrorrrr)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) r r ExceptionrrrrprintTracebackrformatregetvalue)r!rroutputrexpectedOutputrrr test_printTracebackIncludesValues z5FailureJellyingTests.test_printTracebackIncludesValueN)rrrrrrrrrrrrs  rN)'riortwisted.internetrrtwisted.pythonrtwisted.python.reflectrtwisted.spreadrrr twisted.trialr rr rr*rr Jellyable RemoteCopyrrsetUnjellyableForClassglobalSecurityallowInstancesOfRootrPBServerFactoryr?rrCr^rrrrrrrs0       <"k