[XMc@sdZddlmZddlmZmZmZddlmZm Z ddl m Z de fdYZ de fd YZd ejfd YZd ejfd YZdejejejfdYZdejejfdYZejeeejeeejjedejfdYZdejfdYZdejfdYZdefdYZdefdYZ de!fdYZ"dejfdYZ#d S(!s! Tests for error handling in PB. i(tunittest(tpbtflavorstjelly(treactortdefer(tlogtAsynchronousExceptioncBseZdZRS(s Helper used to test remote methods which return Deferreds which fail with exceptions which are not L{pb.Error} subclasses. (t__name__t __module__t__doc__(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRstSynchronousExceptioncBseZdZRS(sm Helper used to test remote methods which raise exceptions which are not L{pb.Error} subclasses. (RR R (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR stAsynchronousErrorcBseZdZRS(s Helper used to test remote methods which return Deferreds which fail with exceptions which are L{pb.Error} subclasses. (RR R (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR stSynchronousErrorcBseZdZRS(si Helper used to test remote methods which raise exceptions which are L{pb.Error} subclasses. (RR R (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR &st JellyErrorcBseZRS((RR (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR.st SecurityErrorcBseZRS((RR (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR2st SimpleRootcBsqeZdZdZdZdZdZdZdZdZ dZ d d Z d d Z RS( cCstjtdS(sD Fail asynchronously with a non-pb.Error exception. sremote asynchronous exception(RtfailR(tself((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_asynchronousException>scCstddS(sC Fail synchronously with a non-pb.Error exception. sremote synchronous exceptionN(R (R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_synchronousExceptionDscCstjtdS(s@ Fail asynchronously with a pb.Error exception. sremote asynchronous error(RRR (R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_asynchronousErrorJscCstddS(s? Fail synchronously with a pb.Error exception. sremote synchronous errorN(R (R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_synchronousErrorPscCs)dtjfdY}|ddS(s> Fail with error that is not known to client. t UnknownErrorcBseZRS((RR (((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRZssI'm not known to client!N(RtError(RR((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_unknownErrorVscCs|jdS(N(t raiseJelly(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt remote_jelly^scCs|jdS(N(t raiseSecurity(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_securityascCs-tj}|j|j|jd|S(N(RtDeferredt addCallbackRtcallbacktNone(Rtd((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_deferredJellyds  cCs-tj}|j|j|jd|S(N(RRRRR R!(RR"((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytremote_deferredSecurityjs  cCstddS(NsI'm jellyable!(R(Rtresults((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRpscCstddS(Ns I'm secure!(R(RR%((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRssN(RR RRRRRRRR#R$R!RR(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR=s          tSaveProtocolServerFactorycBseZdZdZdZRS(sd A L{pb.PBServerFactory} that saves the latest connected client in C{protocolInstance}. cCs ||_dS(s3 Keep track of the given protocol. N(tprotocolInstance(Rtprotocol((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytclientConnectionMadesN(RR R R!R'R)(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR&xstPBConnTestCasecBsDeZdZdZdZdZdZdZdZRS(icCs|j|jdS(N(t _setUpServert _setUpClient(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytsetUps cCsCtt|_|j|j_tjd|jdd|_dS(Nit interfaces 127.0.0.1(R&Rt serverFactorytunsafeTracebacksRt listenTCPt serverPort(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR+scCs@|jjj}tj|_tjd||j|_dS(Ns 127.0.0.1( R2tgetHosttportRtPBClientFactoryt clientFactoryRt connectTCPtclientConnector(RtportNo((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR,scCsG|jjdk r(|jjjjntj|j|jgS(N( R/R'R!t transporttloseConnectionRt gatherResultst_tearDownServert_tearDownClient(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttearDowns  cCstj|jjS(N(Rt maybeDeferredR2t stopListening(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR=scCs|jjtjdS(N(R8t disconnectRtsucceedR!(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR>s ( RR R0R-R+R,R?R=R>(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyR*s     t PBFailureTestcBseZejjZdZdZdZdZ dZ dZ dZ ddZdZd Zd Zd Zd Zd ZdZdZRS(csGfdjj}fd}|j||S(Ncsd|jj|jdrNj}jt|dn|j|j|jfS(NsTraceback unavailable i(ttraptcomparet tracebacktflushLoggedErrorst assertEqualtlenttypetvalue(terrterrs(Rt exceptionTypetflush(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytebs  cs |j}|j|S(N(t callRemotet addErrback(trootR"(RQtmethod(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt gotRootObjects (R6t getRootObjectR(RRURORPR"RV((RRORQRPRUs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt_exceptionTests  cCs|jdttS(s Test that a Deferred returned by a remote method which already has a Failure correctly has that error passed back to the calling side. tasynchronousException(RXRtTrue(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_asynchronousExceptionscCs|jdttS(sw Like L{test_asynchronousException}, but for a method which raises an exception synchronously. tsynchronousException(RXR RZ(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_synchronousExceptionscCs|jdttS(s Like L{test_asynchronousException}, but for a method which returns a Deferred failing with an L{pb.Error} subclass. tasynchronousError(RXR tFalse(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_asynchronousErrorscCs|jdttS(s} Like L{test_asynchronousError}, but for a method which synchronously raises a L{pb.Error} subclass. tsynchronousError(RXR R_(R((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_synchronousErrorscCs|j|||S(N(t assertEquals(RtresulttexpectedResult((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt_successscCs |j|j|d|f|S(Nt callbackArgs(t addCallbacksRf(Rt remoteCallReRQ((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt_addFailingCallbackss cs;jj}fd}|j||S(s 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. csPj|j}dk rLfd}|j|n|S(Ncs#jtjd|S(Ni(RcRJRH(RM(Rtexc(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt gotFailures(RjRRR!taddBoth(tobjtfailureDeferredRl(RkRRQtexpectedRU(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt gotRootObjs  (R6RWR(RRURpRQRkt rootDeferredRq((RkRRURpRQs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt _testImpls cs"fd}jdd|S(s Test that an exception which is a subclass of L{pb.Error} has more information passed across the network to the calling side. csF|jtjt|jtjt|j|jdS(Ni+(RERtfailIft isinstanceRKtstrt failUnlessRL(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt failureJellys Ri+(Rs(RRx((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_jellyFailurescs"fd}jdd|S(s Test that a Deferred which fails with a L{pb.Error} is treated in the same way as a synchronously raised L{pb.Error}. csF|jtjt|jtjt|j|jdS(Ni(RERRtRuRKRvRwRL(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytfailureDeferredJellys t deferredJellyi(Rs(RRz((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_deferredJellyFailure scs"fd}jdd|S(s An 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. csj|jddS(Ns,twisted.test.test_pbfailure.SynchronousErrori(RIRK(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytfailureUnjellyables Rai(Rs(RR}((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_unjellyableFailurescs"fd}jdd|S(s Test that an exception which is a subclass of L{pb.Error} but not known on the client side has its type set properly. csj|jddS(Ns(twisted.test.test_pbfailure.UnknownErrori(RIRK(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytfailureUnknown+s t unknownErrori(Rs(RR((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_unknownFailure&scs"fd}jdd|S(s 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. csF|jtjt|jtjt|j|jdS(Ni(RERRtRuRKRvRwRL(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytfailureSecurity8s tsecurityi(Rs(RR((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_securityFailure2scs"fd}jdd|S(s 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. csF|jtjt|jtjt|j|jdS(Ni(RERRtRuRKRvRwRL(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pytfailureDeferredSecurityFs tdeferredSecurityi(Rs(RR((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_deferredSecurity@scs%fd}jdd|tS(s Test that attempting to call a method which is not defined correctly results in an AttributeError on the calling side. cs'|jtjj|jddS(NsTraceback unavailable i(RERt NoSuchMethodRFRG(R(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt failureNoSuchSstnosuchi(RstAttributeError(RR((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_noSuchMethodFailureNscsEjj}d}|j|fd}|j||S(s 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 |jdS(NR\(RR(trootObj((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt connecteddscs6tj|jt}jt|ddS(Ni(RRMRHR RcRJ(tfailureRN(R(s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyt exceptionhs (R6RWRRS(RR"RR((Rs?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_copiedFailureLoggingZs    N(RR RtTestCaseRcRFRXR[R]R`RbRfRjR!RsRyR|R~RRRRR(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRDs"        tPBFailureTestUnsafecBseZejjZdZRS(i(RR RRt failIfEqualsRFR0(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRrs t DummyInvokercBseZdZdZRS(sZ A behaviorless object to be used as the invoker parameter to L{jelly.jelly}. N(RR R R!tserializingPerspective(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRxstFailureJellyingTestscBs eZdZdZdZRS(s: Tests for the interaction of jelly and failures. cCstjt}|j|jtt|j|jtttjtj|dt}|j|jtt|j|jttdS(s An unjellied L{CopyableFailure} has a check method which behaves the same way as the original L{CopyableFailure}'s check method. tinvokerN( RtCopyableFailuretZeroDivisionErrortassertIdenticaltchecktArithmeticErrorRtunjellyR(Rtoriginaltcopied((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_unjelliedFailureChecks!cCstjt}|j|jtt|j|jtttjtj|dt}tj|}tjtj|dt}|j|jtt|j|jttdS(sX 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. RN( RRRRRRRRR(RRt copiedOncet derivativet copiedTwice((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyttest_twiceUnjelliedFailureChecks(RR R RR(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyRs N($R t twisted.trialRttwisted.spreadRRRttwisted.internetRRttwisted.pythonRt ExceptionRR RR R t Jellyablet RemoteCopyRRtsetUnjellyableForClasstglobalSecuritytallowInstancesOftRootRtPBServerFactoryR&RR*RDRtobjectRR(((s?/usr/lib/python2.7/dist-packages/twisted/test/test_pbfailure.pyts(%;!