U n5{iç<ã@sNdZddlmZddgZddlmZddlmZmZm Z ddl mZmZddl mZdd lmZmZdd lmZddlmZmZddlmZdd lmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&dd„dDƒZ'dZ(dd„e'DƒZ)e( *d¡Z+eeƒGdd„dƒƒZ,Gdd„dƒZ-Gdd„dƒZ.Gdd„dej/ƒZ0dS) z2 Helpers related to HTTP requests, used by tests. é)ÚannotationsÚDummyChannelÚDummyRequest)ÚBytesIO)ÚDictÚListÚOptional)ÚimplementerÚverify)ÚVersion)ÚIPv4AddressÚIPv6Address)ÚDeferred)ÚIAddressÚ ISSLTransport)ÚClock)Ú deprecated)Úunittest)ÚFOUND)ÚHeaders)ÚResource)ÚNOT_DONE_YETÚSessionÚSitecCsg|]}d|›d‘qS)ZFooÚbar©)Ú.0ZlwrrúB/tmp/pip-unpacked-wheel-mnhg7r0q/twisted/web/test/requesthelper.pyÚ sr)ú Ú z zFoo barcCsg|]}| d¡‘qS)Úascii)Úencode)rÚ componentrrrr!sr!c@seZdZdZdS)ÚNullAddressz/ A null implementation of L{IAddress}. N)Ú__name__Ú __module__Ú__qualname__Ú__doc__rrrrr$'sr$c@sÖeZdZGdd„dƒZeeƒGdd„deƒƒZeeƒƒZ d,dd„Z dd „Zd d„Zdd „Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„ZdS)-rc@sVeZdZdZdZddd„Zdd„Zdd „Zd d„Zdd „Z dd„Z dd„Zdd„ZdS)zDummyChannel.TCPéPFNcCs,|dkrtdddƒ}||_tƒ|_g|_dS)NÚTCPz192.168.1.1i80)rÚ_peerrÚwrittenÚ producers©ÚselfZpeerrrrÚ__init__3s zDummyChannel.TCP.__init__cCs|jS©N)r+©r/rrrÚgetPeer:szDummyChannel.TCP.getPeercCs(t|tƒstd|›ƒ‚|j |¡dS)Nz)Can only write bytes to a transport, not )Ú isinstanceÚbytesÚ TypeErrorr,Úwrite©r/Údatarrrr7=s zDummyChannel.TCP.writecCs|D]}| |¡qdSr1)r7)r/Úiovecr9rrrÚ writeSequenceBszDummyChannel.TCP.writeSequencecCstdd|jƒS)Nr*z10.0.0.1)rÚportr2rrrÚgetHostFszDummyChannel.TCP.getHostcCs|j ||f¡dSr1)r-Úappend©r/ZproducerZ streamingrrrÚregisterProducerIsz!DummyChannel.TCP.registerProducercCsdSr1rr2rrrÚunregisterProducerLsz#DummyChannel.TCP.unregisterProducercCs d|_dS)NT)Údisconnectedr2rrrÚloseConnectionOszDummyChannel.TCP.loseConnection)N) r%r&r'r<rBr0r3r7r;r=r@rArCrrrrr*/s r*c@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)zDummyChannel.SSLcCsdSr1rr2rrrÚabortConnectionTsz DummyChannel.SSL.abortConnectioncCsdSr1rr2rrrÚgetTcpKeepAliveXsz DummyChannel.SSL.getTcpKeepAlivecCsdSr1rr2rrrÚ getTcpNoDelay\szDummyChannel.SSL.getTcpNoDelaycCsdSr1rr2rrrÚloseWriteConnection`sz$DummyChannel.SSL.loseWriteConnectioncCsdSr1r©r/ZenabledrrrÚsetTcpKeepAlivedsz DummyChannel.SSL.setTcpKeepAlivecCsdSr1rrHrrrÚ setTcpNoDelayhszDummyChannel.SSL.setTcpNoDelaycCsdSr1rr2rrrÚgetPeerCertificatelsz#DummyChannel.SSL.getPeerCertificateN) r%r&r'rDrErFrGrIrJrKrrrrÚSSLRsrLNcCs| |¡|_dSr1)r*Ú transportr.rrrr0rszDummyChannel.__init__cCsdSr1r)r/ÚrequestrrrÚrequestDoneuszDummyChannel.requestDonecCsht|tƒrdd„| ¡Dƒ}|d|d|d}|g}| dd„|Dƒ¡| d¡|j |¡dS)NcSs"g|]\}}|D]}||f‘qqSrr)rÚkÚvaluesÚvrrrrzsz-DummyChannel.writeHeaders..ó ó css"|]\}}|d|dVqdS)s: rTNr)rÚnameÚvaluerrrÚ sz,DummyChannel.writeHeaders..)r4rÚgetAllRawHeadersÚextendr>rMr;)r/ÚversionÚcodeÚreasonÚheadersZ response_lineZheaderSequencerrrÚwriteHeadersxs ÿ zDummyChannel.writeHeaderscCs |j ¡Sr1)rMr3r2rrrr3ƒszDummyChannel.getPeercCs |j ¡Sr1)rMr=r2rrrr=†szDummyChannel.getHostcCs|j ||¡dSr1)rMr@r?rrrr@‰szDummyChannel.registerProducercCs|j ¡dSr1)rMrAr2rrrrAŒszDummyChannel.unregisterProducercCs|j |¡dSr1)rMr7r8rrrr7szDummyChannel.writecCs|j |¡dSr1)rMr;)r/r:rrrr;’szDummyChannel.writeSequencecCs|j ¡dSr1)rMrCr2rrrrC•szDummyChannel.loseConnectioncCsdSr1rr2rrrÚ endRequest˜szDummyChannel.endRequestcCst|j|jƒSr1)r4rMrLr2rrrÚisSecure›szDummyChannel.isSecurecCsdSr1rr2rrrrDžszDummyChannel.abortConnectioncCsdSr1rr2rrrrE¢szDummyChannel.getTcpKeepAlivecCsdSr1rr2rrrrF¦szDummyChannel.getTcpNoDelaycCsdSr1rr2rrrrGªsz DummyChannel.loseWriteConnectioncCsdSr1rr2rrrrI®szDummyChannel.setTcpKeepAlivecCsdSr1rr2rrrrJ²szDummyChannel.setTcpNoDelaycCsdSr1rr2rrrrK¶szDummyChannel.getPeerCertificate)N)r%r&r'r*r rrLrrÚsiter0rOr^r3r=r@rAr7r;rCr_r`rDrErFrGrIrJrKrrrrr.s.# c@s.eZdZUdZdZdZdZded<ded<ded <ded <ded<d ed<dd„Zdd„Z dGdddddœdd„Z dd„Zdd„Zdd„Z dHdd „Zd!d"„Zd#d$„Zd%d&œd'd(„Zd)d*„Zd+d,„Zd-d.„ZdId/d0„Zd1d2„Zd3d4„Zeed5d6d7d8ƒd9d:d;d<„ƒZd=d>„Zd?d@„ZdAdB„ZdJdCdD„ZdEdF„ZdS)Kra Represents a dummy or fake request. See L{twisted.web.server.Request}. @ivar _finishedDeferreds: L{None} or a C{list} of L{Deferreds} which will be called back with L{None} when C{finish} is called or which will be errbacked if C{processingFailed} is called. @type requestheaders: C{Headers} @ivar requestheaders: A Headers instance that stores values for all request headers. @type responseHeaders: C{Headers} @ivar responseHeaders: A Headers instance that stores values for all response headers. @type responseCode: C{int} @ivar responseCode: The response code which was passed to C{setResponseCode}. @type written: C{list} of C{bytes} @ivar written: The bytes which have been written to the request. s http://dummy/sGETNzOptional[IAddress]ÚclientzList[bytes]Úsitepathr,ÚprepathzDict[bytes, List[bytes]]ÚargszList[Deferred[None]]Ú_finishedDeferredscCsd|_|jr| ¡qdS)zû Call an L{IPullProducer}'s C{resumeProducing} method in a loop until it unregisters itself. @param prod: The producer. @type prod: L{IPullProducer} @param s: Whether or not the producer is streaming. éN)ÚgoZresumeProducing)r/ÚprodÚsrrrr@ÜszDummyRequest.registerProducercCs d|_dS)Nr)rhr2rrrrAëszDummyRequest.unregisterProducerzlist[bytes]zOptional[Session]ÚNone)ÚpostpathÚsessionrbÚreturncCslg|_g|_d|_||_g|_d|_|p6tddtƒd|_i|_ t ƒ|_t ƒ|_d|_ g|_d|_d|_dS)Nró0)raÚuidZreactorsdummysHTTP/1.0)rcr,ÚfinishedrlrdrmrrÚprotoSessionrerÚrequestHeadersÚresponseHeadersÚresponseCoderfÚ_serverNameZclientproto)r/rlrmrbrrrr0îszDummyRequest.__init__cCs,i}|j ¡D]\}}|d|| ¡<q|S)ah Return dictionary mapping the names of all received headers to the last value received for each. Since this method does not return all header information, C{self.requestHeaders.getAllRawHeaders()} may be preferred. NOTE: This function is a direct copy of C{twisted.web.http.Request.getAllRawHeaders}. éÿÿÿÿ)rsrXÚlower)r/r]rPrRrrrÚ getAllHeadersszDummyRequest.getAllHeaderscCs|j | ¡dg¡dS)aH Retrieve the value of a request header. @type name: C{bytes} @param name: The name of the request header for which to retrieve the value. Header names are compared case-insensitively. @rtype: C{bytes} or L{None} @return: The value of the specified request header. Nr)rsZ getRawHeadersrx)r/rUrrrÚ getHeaderszDummyRequest.getHeadercCs|j ||¡dS)zATODO: make this assert on write() if the header is content-lengthN)rtÚaddRawHeader©r/rUrVrrrÚ setHeader szDummyRequest.setHeadercCs(|jr|jS|jrtdƒ‚|j|_|jS)Nz8Session cannot be requested after data has been written.)rmr,ÚAssertionErrorrr)r/ZsessionInterfacerrrÚ getSession$sÿþzDummyRequest.getSessioncCs,| |¡}|tkrdS| |¡| ¡dS)a Render the given resource as a response to this request. This implementation only handles a few of the most common behaviors of resources. It can handle a render method that returns a string or C{NOT_DONE_YET}. It doesn't know anything about the semantics of request methods (eg HEAD) nor how to set any particular headers. Basically, it's largely broken, but sufficient for some tests at least. It should B{not} be expanded to do all the same stuff L{Request} does. Instead, L{DummyRequest} should be phased out and L{Request} (or some other real code factored in a different way) used. N)Úrenderrr7Úfinish)r/ÚresourceÚresultrrrr€-s zDummyRequest.rendercCs"t|tƒstdƒ‚|j |¡dS)Nzwrite() only accepts bytes)r4r5r6r,r>r8rrrr7@s zDummyRequest.writezDeferred[None])rncCstƒ}|j |¡|S)z¹ Return a L{Deferred} which is called back with L{None} when the request is finished. This will probably only work if you haven't called C{finish} yet. )rrfr>)r/rqrrrÚnotifyFinishEszDummyRequest.notifyFinishcCs:|jd|_|jdk r6|j}d|_|D]}| d¡q&dS)z} Record that the request is finished and callback and L{Deferred}s waiting for notification of this. rgN)rqrfÚcallback)r/Ú observersÚobsrrrrOs zDummyRequest.finishcCs.|jdk r*|j}d|_|D]}| |¡qdS)zK Errback and L{Deferreds} waiting for finish notification. N)rfZerrback)r/r\r†r‡rrrÚprocessingFailed[s zDummyRequest.processingFailedcCs|g|j|<dSr1)rer|rrrÚaddArgeszDummyRequest.addArgcCs,|jrtd d |j¡¡ƒ‚||_||_dS)z{ Set the HTTP status response code, but takes care that this is called before any data is written. z;Response code cannot be set after data hasbeen written: {}.ú@@@@N)r,r~ÚformatÚjoinruZresponseMessage)r/r[ÚmessagerrrÚsetResponseCodehsÿ ÿþzDummyRequest.setResponseCodecCs |jrtd d |j¡¡ƒ‚dS)Nzs<ÿ