o ¼>hç<ã@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)ÚFooÚbar©)Ú.0Úlwrrú~/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/web/test/requesthelper.pyÚ sr )ú Ú z zFoo barcCsg|]}| d¡‘qS)Úascii)Úencode)rÚ componentrrrr !s ÿr#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,|dur tdddƒ}||_tƒ|_g|_dS)NÚTCPz192.168.1.1i80)rÚ_peerrÚwrittenÚ producers©ÚselfÚpeerrrrÚ__init__3s zDummyChannel.TCP.__init__cCó|jS©N)r-©r1rrrÚgetPeer:szDummyChannel.TCP.getPeercCs(t|tƒstd|›ƒ‚|j |¡dS)Nz)Can only write bytes to a transport, not )Ú isinstanceÚbytesÚ TypeErrorr.Úwrite©r1Údatarrrr;=s zDummyChannel.TCP.writecCs|D]}| |¡qdSr5)r;)r1Úiovecr=rrrÚ writeSequenceBsÿzDummyChannel.TCP.writeSequencecCstdd|jƒS)Nr,z10.0.0.1)rÚportr6rrrÚgetHostFózDummyChannel.TCP.getHostcCs|j ||f¡dSr5)r/Úappend©r1ÚproducerÚ streamingrrrÚregisterProducerIsz!DummyChannel.TCP.registerProducercCódSr5rr6rrrÚunregisterProducerLóz#DummyChannel.TCP.unregisterProducercCó d|_dS)NT)Údisconnectedr6rrrÚloseConnectionOó zDummyChannel.TCP.loseConnectionr5) r'r(r)r@rLr3r7r;r?rArGrIrMrrrrr,/s r,c@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)zDummyChannel.SSLcCrHr5rr6rrrÚabortConnectionTóz DummyChannel.SSL.abortConnectioncCrHr5rr6rrrÚgetTcpKeepAliveXrPz DummyChannel.SSL.getTcpKeepAlivecCrHr5rr6rrrÚ getTcpNoDelay\rPzDummyChannel.SSL.getTcpNoDelaycCrHr5rr6rrrÚloseWriteConnection`rPz$DummyChannel.SSL.loseWriteConnectioncCrHr5r©r1ÚenabledrrrÚsetTcpKeepAlivedrPz DummyChannel.SSL.setTcpKeepAlivecCrHr5rrTrrrÚ setTcpNoDelayhrPzDummyChannel.SSL.setTcpNoDelaycCrHr5rr6rrrÚgetPeerCertificatelrPz#DummyChannel.SSL.getPeerCertificateN) r'r(r)rOrQrRrSrVrWrXrrrrÚSSLRsrYNcCs| |¡|_dSr5)r,Ú transportr0rrrr3rózDummyChannel.__init__cCrHr5r)r1ÚrequestrrrÚrequestDoneurJzDummyChannel.requestDonecCsht|tƒrdd„| ¡Dƒ}|d|d|d}|g}| dd„|Dƒ¡| d¡|j |¡dS)NcSs"g|] \}}|D]}||f‘qqSrr)rÚkÚvaluesÚvrrrr zs ÿÿz-DummyChannel.writeHeaders..ó ó css$|] \}}|d|dVqdS)s: rbNr)rÚnameÚvaluerrrÚ s€"z,DummyChannel.writeHeaders..)r8rÚgetAllRawHeadersÚextendrCrZr?)r1ÚversionÚcodeÚreasonÚheadersÚ response_lineÚheaderSequencerrrÚwriteHeadersxs ÿ zDummyChannel.writeHeaderscCó |j ¡Sr5)rZr7r6rrrr7ƒrNzDummyChannel.getPeercCror5)rZrAr6rrrrA†rNzDummyChannel.getHostcCs|j ||¡dSr5)rZrGrDrrrrG‰szDummyChannel.registerProducercCó|j ¡dSr5)rZrIr6rrrrIŒrBzDummyChannel.unregisterProducercCó|j |¡dSr5)rZr;r<rrrr;r[zDummyChannel.writecCrqr5)rZr?)r1r>rrrr?’r[zDummyChannel.writeSequencecCrpr5)rZrMr6rrrrM•rBzDummyChannel.loseConnectioncCrHr5rr6rrrÚ endRequest˜rJzDummyChannel.endRequestcCst|j|jƒSr5)r8rZrYr6rrrÚisSecure›rBzDummyChannel.isSecurecCrHr5rr6rrrrOžrPzDummyChannel.abortConnectioncCrHr5rr6rrrrQ¢rPzDummyChannel.getTcpKeepAlivecCrHr5rr6rrrrR¦rPzDummyChannel.getTcpNoDelaycCrHr5rr6rrrrSªrPz DummyChannel.loseWriteConnectioncCrHr5rr6rrrrV®rPzDummyChannel.setTcpKeepAlivecCrHr5rr6rrrrW²rPzDummyChannel.setTcpNoDelaycCrHr5rr6rrrrX¶rPzDummyChannel.getPeerCertificater5)r'r(r)r,r rrYrrÚsiter3r]rnr7rArGrIr;r?rMrrrsrOrQrRrSrVrWrXrrrrr.s0# c@s$eZdZUdZdZdZdZded<ded<ded <ded <ded<d ed<dd„Zdd„Z dHdIdd„Z dd„Zdd„Zdd „Z dJd!d"„Zd#d$„Zd%d&„ZdKd(d)„Zd*d+„Zd,d-„Zd.d/„ZdJd0d1„Zd2d3„Zd4d5„Zeed6d7d8d9ƒd:d;dd?„Zd@dA„ZdBdC„ZdLdDdE„ZdFdG„ZdS)Mra 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/sGETNúOptional[IAddress]ÚclientzList[bytes]Úsitepathr.ÚprepathzDict[bytes, List[bytes]]ÚargszList[Deferred[None]]Ú_finishedDeferredscCs"d|_|jr| ¡|jsdSdS)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)ÚgoÚresumeProducing)r1ÚprodÚsrrrrGÜsÿzDummyRequest.registerProducercCrK)Nr)r|r6rrrrIërNzDummyRequest.unregisterProducerÚpostpathúlist[bytes]ÚsessionúOptional[Session]ÚreturnÚNonecCslg|_g|_d|_||_g|_d|_|ptddtƒd|_i|_ t ƒ|_t ƒ|_d|_ g|_d|_d|_dS)Nró0)rtÚuidÚreactorsdummysHTTP/1.0)rwr.Úfinishedr€rxr‚rrÚprotoSessionryrÚrequestHeadersÚresponseHeadersÚresponseCoderzÚ_serverNameÚclientproto)r1r€r‚rvrrrr3îs zDummyRequest.__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}. éÿÿÿÿ)r‹rfÚlower)r1rkr^r`rrrÚ 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)r‹Ú getRawHeadersr‘)r1rcrrrÚ getHeaderszDummyRequest.getHeadercCs|j ||¡dS)zATODO: make this assert on write() if the header is content-lengthN)rŒÚaddRawHeader©r1rcrdrrrÚ setHeader szDummyRequest.setHeadercCs(|jr|jS|jr Jdƒ‚|j|_|jS)Nz8Session cannot be requested after data has been written.)r‚r.rŠ)r1ÚsessionInterfacerrrÚ getSession$sÿþzDummyRequest.getSessioncCs,| |¡}|turdS| |¡| ¡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)r1ÚresourceÚresultrrrrš-s zDummyRequest.rendercCs"t|tƒs tdƒ‚|j |¡dS)Nzwrite() only accepts bytes)r8r9r:r.rCr<rrrr;@s zDummyRequest.writeúDeferred[None]cCstƒ}|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. )rrzrC)r1r‰rrrÚnotifyFinishEszDummyRequest.notifyFinishcCs>|jd|_|jdur|j}d|_|D] }| d¡qdSdS)z} Record that the request is finished and callback and L{Deferred}s waiting for notification of this. r{N)r‰rzÚcallback)r1Ú observersÚobsrrrr›Os üzDummyRequest.finishcCs2|jdur|j}d|_|D] }| |¡q dSdS)zK Errback and L{Deferreds} waiting for finish notification. N)rzÚerrback)r1rjr¡r¢rrrÚprocessingFailed[s üzDummyRequest.processingFailedcCs|g|j|<dSr5)ryr–rrrÚaddArger[zDummyRequest.addArgcCs,|jrJd 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.ÚformatÚjoinrÚresponseMessage)r1riÚmessagerrrÚsetResponseCodehsÿ ÿþ zDummyRequest.setResponseCodecCó |jrJd d |j¡¡ƒ‚dS)Nzs>ÿ