U  W[^4@s6dZddlmZmZddgZddlmZddlmZm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZmZdd lmZmZdd lmZddlmZddlmZddlmZm Z m!Z!ddl"m#Z#eeGddde$Z%GdddZ&Gddde$Z'e edddddde'j(e'_(Gdddej)Z*dS)z2 Helpers related to HTTP requests, used by tests. )divisionabsolute_import DummyChannel DummyRequest)BytesIO) implementerverify) intToBytes) deprecated)Version)Deferred) IPv4Address IPv6Address) ISSLTransportIAddress)unittest)Headers)Resource) NOT_DONE_YETSessionSite)FOUNDc@seZdZdZdS) NullAddressz/ A null implementation of L{IAddress}. N)__name__ __module__ __qualname____doc__rr@/usr/lib/python3/dist-packages/twisted/web/test/requesthelper.pyr src@seZdZGdddZeeGdddeZeeZ dddZ dd Z d d Z d d Z ddZddZddZddZddZddZddZddZdS)rc@sVeZdZdZdZdddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)zDummyChannel.TCPPFNcCs,|dkrtddd}||_t|_g|_dS)NTCPz 192.168.1.1i80)r _peerrwritten producersselfZpeerrrr__init__-s  zDummyChannel.TCP.__init__cCs|jSN)r!r%rrrgetPeer4szDummyChannel.TCP.getPeercCs(t|tstd|f|j|dS)Nz+Can only write bytes to a transport, not %r) isinstancebytes TypeErrorr"writer%datarrrr-7s zDummyChannel.TCP.writecCs|D]}||qdSr')r-)r%iovecr/rrr writeSequence<szDummyChannel.TCP.writeSequencecCstdd|jS)Nr z10.0.0.1)r portr(rrrgetHost@szDummyChannel.TCP.getHostcCs|j||fdSr')r#appendr%ZproducerZ streamingrrrregisterProducerCsz!DummyChannel.TCP.registerProducercCsdSr'rr(rrrunregisterProducerFsz#DummyChannel.TCP.unregisterProducercCs d|_dS)NT) disconnectedr(rrrloseConnectionIszDummyChannel.TCP.loseConnection)N) rrrr2r8r&r)r-r1r3r6r7r9rrrrr )s r c@s eZdZdS)zDummyChannel.SSLN)rrrrrrrSSLMsr:NcCs|||_dSr')r transportr$rrrr&SszDummyChannel.__init__cCsdSr'r)r%requestrrr requestDoneWszDummyChannel.requestDonecCsL|d|d|d}|g}|dd|D|d|j|dS)N  css"|]\}}|d|dVqdS)s: r?Nr).0namevaluerrr ^sz,DummyChannel.writeHeaders..)extendr4r;r1)r%versioncodereasonheadersZ response_lineZheaderSequencerrr writeHeaders[s  zDummyChannel.writeHeaderscCs |jSr')r;r)r(rrrr)eszDummyChannel.getPeercCs |jSr')r;r3r(rrrr3iszDummyChannel.getHostcCs|j||dSr')r;r6r5rrrr6mszDummyChannel.registerProducercCs|jdSr')r;r7r(rrrr7qszDummyChannel.unregisterProducercCs|j|dSr')r;r-r.rrrr-uszDummyChannel.writecCs|j|dSr')r;r1)r%r0rrrr1yszDummyChannel.writeSequencecCs|jdSr')r;r9r(rrrr9}szDummyChannel.loseConnectioncCsdSr'rr(rrr endRequestszDummyChannel.endRequestcCst|j|jSr')r*r;r:r(rrrisSecureszDummyChannel.isSecure)N)rrrr rrr:rrZsiter&r=rIr)r3r6r7r-r1r9rJrKrrrrr(s $   c@seZdZdZdZdZdZddZddZd2d d Z d d Z d dZ ddZ ddZ ddZddZddZddZddZddZd3dd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd4d.d/Zd0d1ZdS)5ra 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/sGETNcCsd|_|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%Zprodsrrrr6s zDummyRequest.registerProducercCs d|_dS)Nr)rMr(rrrr7szDummyRequest.unregisterProducercCsfg|_g|_d|_||_g|_d|_|p0td||_i|_t |_ t |_ d|_ g|_ d|_d|_dS)NrsdummysHTTP/1.0)Zsitepathr"finishedpostpathZprepathsessionr protoSessionargsrrequestHeadersresponseHeaders responseCode_finishedDeferreds _serverNameZ clientproto)r%rPrQclientrrrr&szDummyRequest.__init__cCs,i}|jD]\}}|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}. )rTZgetAllRawHeaderslower)r%rHkvrrr getAllHeaderss zDummyRequest.getAllHeaderscCs|j|dgdS)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)rTZ getRawHeadersr[)r%rArrr getHeaders zDummyRequest.getHeadercCs|j||dS)zJTODO: make this assert on write() if the header is content-length N)rU addRawHeaderr%rArBrrr setHeaderszDummyRequest.setHeadercCs(|jr |jS|jrtd|j|_|jS)Nz8Session cannot be requested after data has been written.)rQr"AssertionErrorrRr(rrr getSessions 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)renderrr-finish)r%resourceresultrrrres  zDummyRequest.rendercCs"t|tstd|j|dS)Nzwrite() only accepts bytes)r*r+r,r"r4r.rrrr- s zDummyRequest.writecCst}|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. )r rWr4)r%rOrrr notifyFinishs zDummyRequest.notifyFinishcCs:|jd|_|jdk r6|j}d|_|D]}|dq&dS)z} Record that the request is finished and callback and L{Deferred}s waiting for notification of this. rLN)rOrWcallback)r% observersobsrrrrfs   zDummyRequest.finishcCs.|jdk r*|j}d|_|D]}||qdS)zK Errback and L{Deferreds} waiting for finish notification. N)rWZerrback)r%rGrkrlrrrprocessingFailed*s  zDummyRequest.processingFailedcCs|g|j|<dSr')rSrarrraddArg5szDummyRequest.addArgcCs*|jrtdd|j||_||_dS)z{ Set the HTTP status response code, but takes care that this is called before any data is written. zs8         b