o blw@s|dZddlZddlmZmZddlmZddlmZm Z ddl m Z ddl m Z mZmZmZddlmZdd lmZmZdd lmZdd lmZmZdd lmZdd lmZmZmZm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*zedWn e+ydZ,YnwdZ,Gdddej-Z.Gddde/Z0Gddde1Z2Gddde%Z3Gddde%Z4Gddde%Z5Gdd d e3Z6Gd!d"d"e!j7Z8Gd#d$d$e!j&Z9Gd%d&d&e!j&Z:Gd'd(d(ej-Z;Gd)d*d*e;ZGd/d0d0e>ej-Z?Gd1d2d2e>ej-Z@Gd3d4d4e;ZAGd5d6d6e;ZBGd7d8d8ej-ZCGd9d:d:ej-ZDGd;d<d.AsyncResourcecsSNr$selfresultr$>/usr/lib/python3/dist-packages/twisted/web/test/test_xmlrpc.py xmlrpc_async=z:AsyncXMLRPCTests.setUp..AsyncResource.xmlrpc_asyncN)__name__ __module__ __qualname__r+r$r(r$r* AsyncResource<sr0) rrequestmethodrrrdumpscontentrDeferredr)rresource)r'r0r$r(r*setUp3s  zAsyncXMLRPCTests.setUpcCs^|j|j||jjg|jdtd |jj}||d||jj ddS)z If an L{XMLRPC} C{xmlrpc_*} method returns a L{defer.Deferred}, the response to the request is the result of that L{defer.Deferred}. r))r(NN) r6renderr1 assertEqualwrittenr)callbackrloadsjoinfinished)r'respr$r$r*test_deferredResponseBs   z&AsyncXMLRPCTests.test_deferredResponsecCsT|j|j|jttd|jd| |jj g| |jj ddS)a While waiting for the L{Deferred} returned by an L{XMLRPC} C{xmlrpc_*} method to fire, the connection the request was issued over may close. If this happens, neither C{write} nor C{finish} is called on the request. Simulatedr)rN) r6r:r1processingFailedr Failurerr)r=r;r<r@r&r$r$r* test_interruptedDeferredResponsePs  z1AsyncXMLRPCTests.test_interruptedDeferredResponseN)r-r.r/__doc__r7rBrFr$r$r$r*r .s  r c@ eZdZdS)TestRuntimeErrorNr-r.r/r$r$r$r*rI^rIc@rH)TestValueErrorNrJr$r$r$r*rLbrKrLc@seZdZdZdZdZddZddZgdgd ge_d d Z gd ge _d dZ ddZ ddZ ddZ ddZddZddZddZde_eddZd d!Zd"S)#Test*cC|Sr%r$)r'argr$r$r* xmlrpc_echoor,zTest.xmlrpc_echocCs||S)z0 This function add two numbers. r$)r'abr$r$r* xmlrpc_addszTest.xmlrpc_addintrYrYdoubler[r[cCs||gS)zC This function puts the two arguments in an array. r$)r'stringnumr$r$r* xmlrpc_pairrWzTest.xmlrpc_pairarrayr\rYcCs t|S)Help for defer.)rsucceedr'xr$r$r* xmlrpc_defers zTest.xmlrpc_defercCs ttSr%)rfailrLr&r$r$r*xmlrpc_deferFail zTest.xmlrpc_deferFailcCstr%)rIr&r$r$r* xmlrpc_failr,zTest.xmlrpc_failcCs tddS)N hellorFaultr&r$r$r* xmlrpc_faultrhzTest.xmlrpc_faultcCsttddS)Nhi)rrfrrmr&r$r$r*xmlrpc_deferFaultzTest.xmlrpc_deferFaultcCsd}||kr tddS|S)z8 Used to test that we can pass Unicode. ☃ zPayload not unicode snowmanrl)r'payloadsnowmanr$r$r*xmlrpc_snowmans zTest.xmlrpc_snowmancCsdddggddS)NrUcrjfoorTDr$r&r$r$r*xmlrpc_complexrrzTest.xmlrpc_complexcCs||Sr%r$)r'mapkeyr$r$r* xmlrpc_dictzTest.xmlrpc_dictHelp for dict.cCs|jd|S)z A method decorated with L{withRequest} which can be called by a test to verify that the request object really is passed as an argument.  )r2)r'r1otherr$r$r*xmlrpc_withRequests  zTest.xmlrpc_withRequestcCs:zt||WStjy|drt|jdw)NSESSIONzSession non-existent/expired.)rlookupProcedurerNoSuchFunction startswithrmSESSION_EXPIRED)r' procedurePathr$r$r*rs zTest.lookupProcedureN)r-r.r/FAILURE NOT_FOUNDrrSrV signaturer^rergrirnrqrwr|rhelprrrr$r$r$r*rMfs.    rMc@s eZdZdZddZddZdS)TestLookupProcedurez} This is a resource which customizes procedure lookup to be used by the tests of support for this customization. cCrQr%r$rcr$r$r*echor,zTestLookupProcedure.echocCs$|dkr|jSt|jd|d)zs Lookup a procedure from a fixed set of choices, either I{echo} or I{system.listeMethods}. rz procedure z not found)rrrr)r' procedureNamer$r$r*rs z#TestLookupProcedure.lookupProcedureN)r-r.r/rGrrr$r$r$r*rs rc@eZdZdZddZdS)TestListProceduresz This is a resource which customizes procedure enumeration to be used by the tests of support for this customization. cCsdgS)zW Return a list of a single method this resource will claim to support. ryr$r&r$r$r*listProceduressz!TestListProcedures.listProceduresN)r-r.r/rGrr$r$r$r*r rc@r) TestAuthHeaderzU This is used to get the header info so that we can test authentication. cCst|d|_dSr%)rM__init__r1r&r$r$r*rs  zTestAuthHeader.__init__cCs||_t||Sr%)r1rMr:)r'r1r$r$r*r:s zTestAuthHeader.rendercCs|j|jfSr%)r1getUser getPasswordr&r$r$r*xmlrpc_authinfozTestAuthHeader.xmlrpc_authinfoN)r-r.r/rGrr:rr$r$r$r*rs  rc@r) TestQueryProtocolz_ QueryProtocol for tests that saves headers received and sent, inside the factory. cCs|j|j_tj|dSr%) transportfactoryr QueryProtocolconnectionMader&r$r$r*rs z TestQueryProtocol.connectionMadecCs||jj|<dSr%)rheaderslowerr'r~valr$r$r* handleHeaderrzTestQueryProtocol.handleHeadercCs$||jj|<tj|||dS)zA Keep sent headers so we can inspect them later. N)r sent_headersrrr sendHeaderrr$r$r*rszTestQueryProtocol.sendHeaderN)r-r.r/rGrrrr$r$r$r*rs  rc@seZdZdZeZddZdS)TestQueryFactoryzE QueryFactory using L{TestQueryProtocol} for saving headers. cOs*i|_i|_tjj|g|Ri|dSr%)rrrrr)r'argskwargsr$r$r*rszTestQueryFactory.__init__N)r-r.r/rGrprotocolrr$r$r$r*rs rc@r)TestQueryFactoryCancelz{ QueryFactory that saves a reference to the L{twisted.internet.interfaces.IConnector} to test connection lost. cCs ||_dSr%) connector)r'rr$r$r*startedConnecting" z(TestQueryFactoryCancel.startedConnectingN)r-r.r/rGrr$r$r$r*rrrc@seZdZddZddZddZd!dd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZeeddd ZdS)" XMLRPCTestscC2tjdttdd|_|jj|_g|_dSNr 127.0.0.1 interface) r listenTCPrSiterMpgetHostport factoriesr&r$r$r*r7's zXMLRPCTests.setUpcCsg|_|jSr%)rr stopListeningr&r$r$r*tearDown,s zXMLRPCTests.tearDowncOst|i|}|j||S)zw Specific queryFactory for proxy that uses our custom L{TestQueryFactory}, and save factories. )rrappend)r'rrrr$r$r* queryFactory0s zXMLRPCTests.queryFactoryNcCs2ttd|j}|dur|j|_|S||_|S)z Return a new xmlrpc.Proxy for the test site created in setUp(), using the given factory as the queryFactory, or self.queryFactory if no factory is provided. http://127.0.0.1:%d/NrProxyrrrr'rrr$r$r*proxy9s zXMLRPCTests.proxyc Csdddddidfdfddddgfdd d d d d ggddfg}g}|D]\}}}|j|g|R}||j|||q"tj|ddS)N)add))r)rTrTdictrTr9pair)rTr9)rvrsrscomplexr$rUrxrjryrzTfireOnOneErrbackr callRemote addCallbackr;rr DeferredList)r' inputOutputdlmethroutpdr$r$r* test_resultsFs   zXMLRPCTests.test_resultscs@dd}fdd}fdd}|||||S)z Verify that headers sent from the client side and the ones we get back from the server side are correct. rvrscs4jdjddjdjdddS)Nr content-typetext/xml; charset=utf-8content-lengths129)r;rr)ingr&r$r*check_server_headers_sz6XMLRPCTests.test_headers..check_server_headerscsLjdjddjdjddjdjdddS)Nrs user-agentsTwisted/XMLRPClibrrrs155)r;rr)ignr&r$r*check_client_headersesz6XMLRPCTests.test_headers..check_client_headersrrr)r'rrrr$r&r* test_headersWs    zXMLRPCTests.test_headerscsttttjdgtfddg}dD]"\}} |} |t j }| |ffdd ||qtj|dd}fdd }| ||S) z Verify that for each way a method exposed via XML-RPC can fail, the correct 'Content-type' header is set in the response and that the client-side Deferred is errbacked with an appropriate C{Fault} instance. defaultLogLevelc tSr%r removeObserverr$filteredr$r* z)XMLRPCTests.test_errors..))rNrf)rN deferFail)rjfault)rO noSuchMethod)ro deferFault)rP SESSION_TESTcs|j|Sr%)r; faultCode)exccoder&r$r*rsTrcsjD] }|jddqdtddj}ddj}t|tr1|t n |t |t t tdS)Nrrrr log_failurer9) rr;r assertEqualslenvalue isinstancerLassertIsInstancerIflushLoggedErrors)rrf1f2 logObserverr'r$r*cbs     z#XMLRPCTests.test_errors..cb)rrr r criticalr addObserver addCleanuprr assertFailurerrmrrrr)r'rr methodNamerrr$rrr'r* test_errorsss     zXMLRPCTests.test_errorscs^fdd|ddd}|jjjd||jjjd||t j }|S)z A deferred from the Proxy can be cancelled, disconnecting the L{twisted.internet.interfaces.IConnector}. cst|i|_jSr%)rf)rkwrr$r*rsz(XMLRPCTests.test_cancel..factoryrrr disconnected) rrassertNotEqualrrstatecancelr;rrCancelledErrorr'rr$rr* test_cancels zXMLRPCTests.test_cancelcs<tt}|dtdjf}fdd}|||S)zN A classic GET on the xml server should return a NOT_ALLOWED. sGETrcs|jtjdSr%)r;rr NOT_ALLOWED)responser&r$r* checkResponserz0XMLRPCTests.test_errorGet..checkResponse)rAgentrr1rrr)r'agentrrr$r&r* test_errorGets   zXMLRPCTests.test_errorGetcsVtt}|jtdjfdttdd}|tj fdd}|||S)zL Test that an invalid XML input returns an L{xmlrpc.Fault}. rsPOSTsfoo)urir2 bodyProducercstjtj|dSr%) assertRaisesrrmrr>r(r&r$r*rsz,XMLRPCTests.test_errorXMLContent..cb) rrrr1rrFileBodyProducerrrreadBody)r'rrrr$r&r*test_errorXMLContents     z XMLRPCTests.test_errorXMLContentcCs*t}|d|}||j||S)z If an L{xmlrpclib.DateTime} is passed as an argument to an XML-RPC call and then returned by the server unmodified, the result should be equal to the original object. r)rDateTimerrrr;)r'whenrr$r$r*test_datetimeRoundtripsz"XMLRPCTests.test_datetimeRoundtripcsttttjdgtfdd dd}dd} t d| |t }fd d }|||S) a. If it is not possible to encode a response to the request (for example, because L{xmlrpclib.dumps} raises an exception when encoding a L{Fault}) the exception which prevents the response from being generated is logged and the request object is finished anyway. rcrr%rr$rr$r*rrz6XMLRPCTests.test_doubleEncodingError..rr!c_std)NzCannot encode anything at all!) RuntimeError)rrr$r$r* fakeDumpsrz7XMLRPCTests.test_doubleEncodingError..fakeDumpsr3cs@dtddjtttddS)Nr9rr)rrrrrr;r)ignoredrr$r*cbFailedsz6XMLRPCTests.test_doubleEncodingError..cbFailed)rrr r rr rrrrpatchrr Exceptionr)r'rrr!r$rr*test_doubleEncodingErrors   z$XMLRPCTests.test_doubleEncodingErrorcs&dd}fdd}||S)zV The connection to the web server is closed when the request is done. rr!cs(j\}|jj|jjdSr%)r assertFalser connected assertTruer)r rr&r$r* responseDoneszBXMLRPCTests.test_closeConnectionAfterRequest..responseDoner)r'rr(r$r&r* test_closeConnectionAfterRequests  z,XMLRPCTests.test_closeConnectionAfterRequestcC:t}tjdd|d}|d||jddddS)z For I{HTTP} URIs, L{xmlrpc.Proxy.callRemote} passes the value it received for the C{connectTimeout} parameter as the C{timeout} argument to the underlying connectTCP call. shttp://127.0.0.1:69g@connectTimeoutr someMethodrrN)rrrrr; tcpClientsr'rrr$r$r*test_tcpTimeouts  zXMLRPCTests.test_tcpTimeoutzOpenSSL not presentcCr*)z For I{HTTPS} URIs, L{xmlrpc.Proxy.callRemote} passes the value it received for the C{connectTimeout} parameter as the C{timeout} argument to the underlying connectSSL call. shttps://127.0.0.1:69g@r+r-rN)rrrrr; sslClientsr/r$r$r*test_sslTimeouts  zXMLRPCTests.test_sslTimeoutr%)r-r.r/r7rrrrrrrrrrr$r)r0rsslSkipr3r$r$r$r*r&s"  0  #  rc@seZdZdZdddZdS)XMLRPCProxyWithoutSlashTestsz3 Test with proxy that doesn't add a slash. NcCs2ttd|j}|dur|j|_|S||_|S)Nhttp://127.0.0.1:%drrr$r$r*r0s z"XMLRPCProxyWithoutSlashTests.proxyr%)r-r.r/rGrr$r$r$r*r5+sr5c@s0eZdZdZddZddZddZdd Zd S) XMLRPCPublicLookupProcedureTestszr Tests for L{XMLRPC}'s support of subclasses which override C{lookupProcedure} and C{listProcedures}. cCsNtjdt|dd|_||jj|jj|_t t d|j|_ dS)Nrrrr6) rrrrrrrrrrrrr)r'r6r$r$r* createServer?sz-XMLRPCPublicLookupProcedureTests.createServercCs0|td}|jd|}||j||S)z A subclass of L{XMLRPC} can override C{lookupProcedure} to find procedures that are not defined using a C{xmlrpc_}-prefixed method name. rkr)r8rrrrr;)r'whatrr$r$r*test_lookupProcedureEs z5XMLRPCPublicLookupProcedureTests.test_lookupProcedurecCs,|t|jdd}||tj}|S)z A subclass of L{XMLRPC} can override C{lookupProcedure} to raise L{NoSuchFunction} to indicate that a requested method is not available to be called, signalling a fault to the XML-RPC client. xxxxrk)r8rrrrrrmr r$r$r*rPs z,XMLRPCPublicLookupProcedureTests.test_errorscs>t}t||jd}fdd}|||S)z A subclass of L{XMLRPC} can override C{listProcedures} to define Overriding listProcedures should prevent introspection from being broken. system.listMethodscsd|dS)Nry)assertIn) proceduresr&r$r*listedfszAXMLRPCPublicLookupProcedureTests.test_listMethods..listed)rrr8rrr)r'r6rr?r$r&r*test_listMethods[s    z1XMLRPCPublicLookupProcedureTests.test_listMethodsN)r-r.r/rGr8r:rr@r$r$r$r*r79s  r7c@r) SerializationConfigMixinas Mixin which defines a couple tests which should pass when a particular flag is passed to L{XMLRPC}. These are not meant to be exhaustive serialization tests, since L{xmlrpclib} does all of the actual serialization work. They are just meant to exercise a few codepaths to make sure we are calling into xmlrpclib correctly. @ivar flagName: A C{str} giving the name of the flag which must be passed to L{XMLRPC} to allow the tests to pass. Subclasses should set this. @ivar value: A value which the specified flag will allow the serialization of. Subclasses should set this. cCsl|jdi}tjdttdi|dd|_||jj|j j |_ t j t d|j ffi||_dS)zO Create a new XML-RPC server with C{allowNone} set to C{True}. TrrrrNr$)flagNamerrrrrMrrrrrrrrr)r'rr$r$r*r7s  zSerializationConfigMixin.setUpcCs$|jd|j}||j|j|S)zY C{self.value} can be round-tripped over an XMLRPC method call/response. rrrrrr;r r$r$r*test_roundtripValuesz,SerializationConfigMixin.test_roundtripValuecCs,|jdd|ji}||jd|ji|S)zz A C{dict} which contains C{self.value} can be round-tripped over an XMLRPC method call/response. rrTrCr r$r$r*test_roundtripNestedValuesz2SerializationConfigMixin.test_roundtripNestedValueN)r-r.r/rGr7rDrEr$r$r$r*rAps  rAc@seZdZdZdZdZdS)XMLRPCAllowNoneTestszF Tests for passing L{None} when the C{allowNone} flag is set. allowNoneN)r-r.r/rGrBrr$r$r$r*rFsrFc@s(eZdZdZdZeddddddZd S) XMLRPCUseDateTimeTestszd Tests for passing a C{datetime.datetime} instance when the C{useDateTime} flag is set. useDateTimeirjr-;N)r-r.r/rGrBdatetimerr$r$r$r*rHsrHc@s@eZdZdZdZdZddZddZdd Zd d Z d d Z dS)XMLRPCAuthenticatedTestsz] Test with authenticated proxy. We run this with the same input/output as above. susernamesasecretcCrr) rrrrrrrrrr&r$r$r*r7s  zXMLRPCAuthenticatedTests.setUpcCsNdt|jt|j|jf}tt|}|d}||j |j|jg|S)Nzhttp://%s:%s@127.0.0.1:%d/authinfo) r userpasswordrrrrrrr;)r'urlrrr$r$r*test_authInfoInURLs z+XMLRPCAuthenticatedTests.test_authInfoInURLcCBttd|jf|j|j}|d}||j|j|jg|S)NrrO rrrrrPrQrrr;r'rrr$r$r*test_explicitAuthInfo z.XMLRPCAuthenticatedTests.test_explicitAuthInfocCsH|jd}ttd|jf|j|}|d}||j|j|g|S)z C{QueryProtocol} uses the C{base64.b64encode} function to encode user name and password in the I{Authorization} header, so that it doesn't embed new lines when using long inputs. (rrO) rQrrrrrPrrr;)r' longPasswordrrr$r$r*test_longPasswords  z*XMLRPCAuthenticatedTests.test_longPasswordcCrT)Nzhttp://wrong:info@127.0.0.1:%d/rOrUrVr$r$r*test_explicitAuthInfoOverriderXz6XMLRPCAuthenticatedTests.test_explicitAuthInfoOverrideN) r-r.r/rGrPrQr7rSrWr[r\r$r$r$r*rNs  rNc@s,eZdZddZddZddZddZd S) XMLRPCIntrospectionTestscCs>t}t|tjdt|dd|_|jj|_g|_ dSr) rMrrrrrrrrr)r'rr$r$r*r7s  zXMLRPCIntrospectionTests.setUpcs(fdd}d}|||S)Ncs||gddS)N)rrrrrrrrfrrrvr<system.methodHelpsystem.methodSignaturer)sortr;)methsr&r$r* cbMethodss z.cbMethodsr<r)r'rbrr$r&r*r@s  z)XMLRPCIntrospectionTests.test_listMethodscCsPgd}g}|D]\}}|d|}||j|||qtj|ddS)N))rra)rfr!)rrr^Trrr' inputOutputsrrexpectedrr$r$r*test_methodHelps  z(XMLRPCIntrospectionTests.test_methodHelpcCslddgdgdgfdgdgfg}g}|D]\}}|d|}||j|||qtj|dd S) N)rr!rrXrZrr_r_Trrrcr$r$r*test_methodSignature%s   z-XMLRPCIntrospectionTests.test_methodSignatureN)r-r.r/r7r@rfrgr$r$r$r*r]s  r]c@r) XMLRPCClientErrorHandlingTestsz3 Test error handling on the xmlrpc client. cCs4tdd|_d|j_tjdt|jdd|_dS)Ns*This text is not a valid XML-RPC response.s text/plainTrrr) rDatar6isLeafrrrrrr&r$r$r*r79s z$XMLRPCClientErrorHandlingTests.setUpcCs |jSr%)rrr&r$r$r*rCrz'XMLRPCClientErrorHandlingTests.tearDowncCs.ttd|jjf}||dtS)zj Test that calling the xmlrpc client on a static http server raises an exception. rr-)rrrrrrr ValueError)r'rr$r$r*test_erroneousResponseFsz5XMLRPCClientErrorHandlingTests.test_erroneousResponseN)r-r.r/rGr7rrlr$r$r$r*rh4s  rhc@s8eZdZdZddZddZddZdd Zd d Zd S) QueryFactoryParseResponseTestsz> Test the behaviour of L{QueryFactory.parseResponse}. c Cs<tdddddddd|_td|_d|_tt|_ dS)Nr"Fr$)pathhostr2rPrQrGr)r!z invalid xml) rrrr3 goodContents badContentsr rErreasonr&r$r$r*r7Vs z$QueryFactoryParseResponseTests.setUpcC,|jj}||jj|j|j|j|S)z We can safely call L{QueryFactory.clientConnectionLost} as a callback of L{QueryFactory.parseResponse}. )rdeferredrclientConnectionLostrr parseResponserpr r$r$r* test_parseResponseCallbackSafetyjz?QueryFactoryParseResponseTests.test_parseResponseCallbackSafetycCrs)z We can safely call L{QueryFactory.clientConnectionLost} as an errback of L{QueryFactory.parseResponse}. )rrt addErrbackrurrrvrqr r$r$r*test_parseResponseErrbackSafetywrxz>QueryFactoryParseResponseTests.test_parseResponseErrbackSafetycCs,|jj}||jj|j|jdd|S)z} We can safely call L{QueryFactory.clientConnectionLost} as an errback of L{QueryFactory.badStatus}. statusmessage)rrtryrurr badStatusr r$r$r*test_badStatusErrbackSafetyrxz:QueryFactoryParseResponseTests.test_badStatusErrbackSafetycCs$d}|jj}|j|||tS)z Some server can send a response without any data: L{QueryFactory.parseResponse} should catch the error and call the result errback. zM )rrtrvr IndexError)r'r4rr$r$r*test_parseResponseWithoutDatas  zr<r;)nr1datar&r$r*validsz6XMLRPCWithRequestTests.test_withRequest..valid) rr2rrr3r4 notifyFinishrr6 render_POST)r'r1rrr$r&r*test_withRequests  z'XMLRPCWithRequestTests.test_withRequestN)r-r.r/r7rr$r$r$r*rs r)FrGrMiorrrrtwisted.internetrrtwisted.internet.errorrtwisted.loggerrr r r twisted.pythonr twisted.python.compatr rtwisted.python.reflectrtwisted.test.proto_helpersrr twisted.trial twisted.webrrrrrtwisted.web.test.test_webrtwisted.web.xmlrpcrrrrrr ImportErrorr4TestCaser rrIrkrLrMrrrrrrrrr5r7rArFrHrNr]rhrmrr$r$r$r*sX         0`   70  @AR