o Óó·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údC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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©ÚselfZpeerrrrÚ__init__3s zDummyChannel.TCP.__init__cCó|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/Údatarrrr8=s zDummyChannel.TCP.writecCs|D]}| |¡qdSr2)r8)r/Úiovecr:rrrÚ writeSequenceBsÿzDummyChannel.TCP.writeSequencecCstdd|jƒS)Nr*z10.0.0.1)rÚportr3rrrÚgetHostFózDummyChannel.TCP.getHostcCs|j ||f¡dSr2)r-Úappend©r/ZproducerZ streamingrrrÚregisterProducerIsz!DummyChannel.TCP.registerProducercCódSr2rr3rrrÚunregisterProducerLóz#DummyChannel.TCP.unregisterProducercCó d|_dS)NT)Údisconnectedr3rrrÚloseConnectionOó zDummyChannel.TCP.loseConnectionr2) r%r&r'r=rGr0r4r8r<r>rBrDrHrrrrr*/s r*c@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)zDummyChannel.SSLcCrCr2rr3rrrÚabortConnectionTóz DummyChannel.SSL.abortConnectioncCrCr2rr3rrrÚgetTcpKeepAliveXrKz DummyChannel.SSL.getTcpKeepAlivecCrCr2rr3rrrÚ getTcpNoDelay\rKzDummyChannel.SSL.getTcpNoDelaycCrCr2rr3rrrÚloseWriteConnection`rKz$DummyChannel.SSL.loseWriteConnectioncCrCr2r©r/ÚenabledrrrÚsetTcpKeepAlivedrKz DummyChannel.SSL.setTcpKeepAlivecCrCr2rrOrrrÚ setTcpNoDelayhrKzDummyChannel.SSL.setTcpNoDelaycCrCr2rr3rrrÚgetPeerCertificatelrKz#DummyChannel.SSL.getPeerCertificateN) r%r&r'rJrLrMrNrQrRrSrrrrÚSSLRsrTNcCs| |¡|_dSr2)r*Ú transportr.rrrr0rózDummyChannel.__init__cCrCr2r)r/ÚrequestrrrÚrequestDoneurEzDummyChannel.requestDonecCsht|tƒrdd„| ¡Dƒ}|d|d|d}|g}| dd„|Dƒ¡| d¡|j |¡dS)NcSs"g|] \}}|D]}||f‘qqSrr)rÚkÚvaluesÚvrrrrzs ÿÿz-DummyChannel.writeHeaders..ó ó css$|] \}}|d|dVqdS)s: r]Nr)rÚnameÚvaluerrrÚ s€"z,DummyChannel.writeHeaders..)r5rÚgetAllRawHeadersÚextendr@rUr<)r/ÚversionÚcodeÚreasonÚheadersZ response_lineZheaderSequencerrrÚwriteHeadersxs ÿ zDummyChannel.writeHeaderscCó |j ¡Sr2)rUr4r3rrrr4ƒrIzDummyChannel.getPeercCrhr2)rUr>r3rrrr>†rIzDummyChannel.getHostcCs|j ||¡dSr2)rUrBrArrrrB‰szDummyChannel.registerProducercCó|j ¡dSr2)rUrDr3rrrrDŒr?zDummyChannel.unregisterProducercCó|j |¡dSr2)rUr8r9rrrr8rVzDummyChannel.writecCrjr2)rUr<)r/r;rrrr<’rVzDummyChannel.writeSequencecCrir2)rUrHr3rrrrH•r?zDummyChannel.loseConnectioncCrCr2rr3rrrÚ endRequest˜rEzDummyChannel.endRequestcCst|j|jƒSr2)r5rUrTr3rrrÚisSecure›r?zDummyChannel.isSecurecCrCr2rr3rrrrJžrKzDummyChannel.abortConnectioncCrCr2rr3rrrrL¢rKzDummyChannel.getTcpKeepAlivecCrCr2rr3rrrrM¦rKzDummyChannel.getTcpNoDelaycCrCr2rr3rrrrNªrKz DummyChannel.loseWriteConnectioncCrCr2rr3rrrrQ®rKzDummyChannel.setTcpKeepAlivecCrCr2rr3rrrrR²rKzDummyChannel.setTcpNoDelaycCrCr2rr3rrrrS¶rKzDummyChannel.getPeerCertificater2)r%r&r'r*r rrTrrÚsiter0rXrgr4r>rBrDr8r<rHrkrlrJrLrMrNrQrRrSrrrrr.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)ÚgoZresumeProducing)r/ÚprodÚsrrrrBÜsÿzDummyRequest.registerProducercCrF)Nr)rur3rrrrDërIzDummyRequest.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)rmÚuidZreactorsdummysHTTP/1.0)rpr,ÚfinishedrxrqrzrrÚprotoSessionrrrÚrequestHeadersÚresponseHeadersÚresponseCodersÚ_serverNameZclientproto)r/rxrzrorrrr0î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}. éÿÿÿÿN)r‚raÚlower)r/rfrYr[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‚Z getRawHeadersr‡)r/r^rrrÚ getHeaderszDummyRequest.getHeadercCs|j ||¡dS)zATODO: make this assert on write() if the header is content-lengthN)rƒÚaddRawHeader©r/r^r_rrrÚ setHeader szDummyRequest.setHeadercCs(|jr|jS|jr Jdƒ‚|j|_|jS)Nz8Session cannot be requested after data has been written.)rzr,r)r/ZsessionInterfacerrrÚ 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)Úrenderrr8Úfinish)r/ÚresourceÚresultrrrrŽ-s zDummyRequest.rendercCs"t|tƒs tdƒ‚|j |¡dS)Nzwrite() only accepts bytes)r5r6r7r,r@r9rrrr8@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. N)rrsr@)r/r€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. rtN)r€rsÚcallback)r/Ú observersÚobsrrrrOs üzDummyRequest.finishcCs2|jdur|j}d|_|D] }| |¡q dSdS)zK Errback and L{Deferreds} waiting for finish notification. N)rsZerrback)r/rer•r–rrrÚprocessingFailed[s üzDummyRequest.processingFailedcCs|g|j|<dSr2)rrr‹rrrÚaddArgerVzDummyRequest.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„ZresponseMessage)r/rdÚmessagerrrÚsetResponseCodehsÿ ÿþ zDummyRequest.setResponseCodecCó |jrJd d |j¡¡ƒ‚dS)Nz¡szDummyRequest.getHostcCsD||_| ¡r d}nd}||kr|}nd||f}|j d|¡dS)a_ Change the host and port the request thinks it's using. @type host: C{bytes} @param host: The value to which to change the host header. @type ssl: C{bool} @param ssl: A flag which, if C{True}, indicates that the request is considered secure (if C{True}, L{isSecure} will return C{True}). i»r)s%b:%dshostN)Z _forceSSLrlr‚rŠ)r/r©r=ÚsslÚdefaultZ hostHeaderrrrÚsetHostªszDummyRequest.setHostcCs| t¡| d|¡dS)zu Utility function that does a redirect. The request should have finish() called after this. slocationN)rrrŒ)r/ÚurlrrrÚredirectÀs zDummyRequest.redirect)NN)rxryrzr{rornr|r}r2)r|r’)r) r%r&r'r(ÚuriÚmethodroÚ__annotations__rBrDr0rˆr‰rŒrrŽr8r“rr—r˜rr¡r¤rrrªr§r«r>r¯r±rrrrr»sF ü c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S)ÚDummyRequestTestsz$ Tests for L{DummyRequest}. cCs^tgƒ}| ¡|j|jgd}| dt|ƒ¡|\}| | d¡t¡| | d¡d¡dS)zp L{DummyRequest.getClientIP} is deprecated in favor of L{DummyRequest.getClientAddress} )ZoffendingFunctionsrtÚcategoryrœz}twisted.web.test.requesthelper.DummyRequest.getClientIP was deprecated in Twisted 18.4.0; please use getClientAddress insteadN)rrªZ flushWarningsÚtest_getClientIPDeprecatedÚassertEqualÚlenÚgetÚDeprecationWarning)r/rWÚwarningsÚwarningrrrr·Ïsÿýz,DummyRequestTests.test_getClientIPDeprecatedcCs.tgƒ}tdddƒ}||_| d| ¡¡dS)z L{DummyRequest.getClientIP} supports IPv6 addresses, just like L{twisted.web.http.Request.getClientIP}. r*z::1é90N)rr ror¸rª)r/rWrorrrÚtest_getClientIPSupportsIPv6èsz.DummyRequestTests.test_getClientIPSupportsIPv6cCs tgƒ}| ¡}t t|¡dS)zu L{DummyRequest.getClientAddress} returns an L{IAddress} provider no C{client} has been set. N)rr§r ZverifyObjectr)r/rWÚnullrrrÚ"test_getClientAddressWithoutClientósz4DummyRequestTests.test_getClientAddressWithoutClientcCs2tgƒ}tdddƒ}||_| ¡}| ||¡dS)zI L{DummyRequest.getClientAddress} returns the C{client}. r*r¬r¾N)rrror§ZassertIs)r/rWroÚaddressrrrÚtest_getClientAddressüs z'DummyRequestTests.test_getClientAddressN)r%r&r'r(r·r¿rÁrÃrrrrrµÊs rµN)1r(Ú __future__rÚ__all__ÚiorÚtypingrrrZzope.interfacer r ÚincrementalrZtwisted.internet.addressrr Ztwisted.internet.deferrZtwisted.internet.interfacesrrZtwisted.internet.taskrZtwisted.python.deprecaterZ twisted.trialrZtwisted.web._responsesrZtwisted.web.http_headersrZtwisted.web.resourcerZtwisted.web.serverrrrZtextLinearWhitespaceComponentsZ sanitizedTextZbytesLinearWhitespaceComponentsr"ZsanitizedBytesr$rrZSynchronousTestCaserµrrrrÚs>ÿ