3 (hxk@sddlmZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZmZmZdd lmZdd lmZmZmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6d-Z7dZ8de8Z9ej:dZ;Gddde<Z=Gdd d e>Z?d!d"Z@d#d$ZAGd%d&d&eZBd'd(ZCd)dZDd*dZEGd+dde>ZFGd,ddeFZGdS).)unicode_literalsN)copy) import_module)BytesIO)settings) BaseHandler) ISO_8859_1UTF_8 WSGIRequest)got_request_exceptionrequest_finishedrequest_started)close_old_connections) HttpRequest QueryDict SimpleCookie)TemplateDoesNotExist)signals) ContextList)resolve)six) force_bytes force_str uri_to_iri)SimpleLazyObjectcurry) urlencode) is_iterable)urljoinurlparseurlsplitClientRedirectCycleErrorRequestFactory encode_fileencode_multipartZBoUnDaRyStRiNgz multipart/form-data; boundary=%sz.*; charset=([\w\d-]+);?cs eZdZdZfddZZS)r"zC The test client has been asked to follow a redirect loop. cs"tt|j|||_|j|_dS)N)superr"__init__ last_responseredirect_chain)selfmessager() __class__6/tmp/pip-install-q3hcpn_q/Django/django/test/client.pyr'+szRedirectCycleError.__init__)__name__ __module__ __qualname____doc__r' __classcell__r-r-)r,r.r"'sc@s4eZdZdZd ddZddZd ddZd d ZdS) FakePayloada A wrapper around BytesIO that restricts what can be read since data from the network can't be seeked and cannot be read outside of its content length. This makes sure that views can't do anything under the test client that wouldn't work in Real Life. NcCs*t|_d|_d|_|dk r&|j|dS)NrF)r_FakePayload__content_FakePayload__len read_startedwrite)r*contentr-r-r.r'8s zFakePayload.__init__cCs|jS)N)r6)r*r-r-r.__len__?szFakePayload.__len__cCsZ|js|jjdd|_|dkr*|jp(d}|j|ks.to_bytescSst|dot|jS)Nr=)hasattrcallabler=)thingr-r-r.is_filesz!encode_multipart..is_filec3s|]}|VqdS)Nr-).0val)rgr-r. sz#encode_multipart..z--%sz)Content-Disposition: form-data; name="%s"c3s|]}|VqdS)Nr-)rlrm)rgr-r.rnsz--%s--rKs )itemsextendr$ isinstancer string_typesrjoin)boundarydatalinesrkkeyvaluerEr-)rgr.r%s0  cCsdd}t|dr tjj|jnd}t|dr6|j}n|rJtj|d}nd}|dkrZd}|sb|}|d||d ||f|d |d ||jgS) NcSs t|tjS)N)rrre)rfr-r-r.rgszencode_file..to_bytesnamero content_typerzapplication/octet-streamz--%sz8Content-Disposition: form-data; name="%s"; filename="%s"zContent-Type: %srK) rhospathbasenamerzr{ mimetypes guess_typer=)rurxfilergfilenamer{r-r-r.r$s"    c@seZdZdZddZddZddZdd Zd d Zd"ddZ d e d fddZ d#ddZ d$ddZ d%ddZd&ddZd'ddZd(ddZd)d d!Zd S)*r#aV Class that lets you create mock Request objects for use in testing. Usage: rf = RequestFactory() get_request = rf.get('/hello/') post_request = rf.post('/submit/', {'foo': 'bar'}) Once you have a request object you can pass it to any view function, just as if that view had been hooked up using a URLconf. cKs||_t|_t|_dS)N)defaultsrcookiesrerrors)r*rr-r-r.r'szRequestFactory.__init__cKsp|jjdddtdtdtdtdtdtdtd dtd td |jdddd}|j|j|j||S)z5 The base environment for a request. roz; )headersep/z 127.0.0.1GETZ testserver80zHTTP/1.1rhttprKTF)Z HTTP_COOKIE PATH_INFOZ REMOTE_ADDRREQUEST_METHODZ SCRIPT_NAME SERVER_NAME SERVER_PORTZSERVER_PROTOCOLz wsgi.versionzwsgi.url_schemez wsgi.inputz wsgi.errorszwsgi.multiprocesszwsgi.multithreadz wsgi.run_once)rr)routputstrr4rupdater)r*rRrZr-r-r. _base_environ s$   zRequestFactory._base_environcKst|jf|S)z#Construct a generic request object.)r r)r*rRr-r-r.rR*szRequestFactory.requestcCsB|tkrtt|Stj|}|r,|jd}ntj}t||dSdS)Nr)encoding) MULTIPART_CONTENTr%BOUNDARYCONTENT_TYPE_REmatchgrouprrer)r*rvr{rcharsetr-r-r. _encode_data.s   zRequestFactory._encode_datacCsNt|d}|dr,|tdt|d7}t|jt}tjrJ|jtS|S)N;) rrrencoder rPY3decoder)r*parsedr}r-r-r. _get_path:s  zRequestFactory._get_pathNFcKsB|dkr in|}dt|ddi}|j||jd|fd|i|S)zConstruct a GET request.N QUERY_STRINGT)doseqrsecure)rrgeneric)r*r}rvrextrarr-r-r.getEs zRequestFactory.getcKs8|dkr in|}|j||}|jd|||fd|i|S)zConstruct a POST request.NPOSTr)rr)r*r}rvr{rrZ post_datar-r-r.postOs zRequestFactory.postcKsB|dkr in|}dt|ddi}|j||jd|fd|i|S)zConstruct a HEAD request.NrT)rrMr)rrr)r*r}rvrrrr-r-r.headYs zRequestFactory.headcKs|jd|fd|i|S)zConstruct a TRACE request.TRACEr)r)r*r}rrr-r-r.tracecszRequestFactory.traceroapplication/octet-streamcKs|jd|||fd|i|S)zConstruct an OPTIONS request.OPTIONSr)r)r*r}rvr{rrr-r-r.optionsgszRequestFactory.optionscKs|jd|||fd|i|S)zConstruct a PUT request.PUTr)r)r*r}rvr{rrr-r-r.putmszRequestFactory.putcKs|jd|||fd|i|S)zConstruct a PATCH request.PATCHr)r)r*r}rvr{rrr-r-r.patchsszRequestFactory.patchcKs|jd|||fd|i|S)zConstruct a DELETE request.DELETEr)r)r*r}rvr{rrr-r-r.deleteyszRequestFactory.deletec Kstt|}t|tj}|j|t||r2tdntd|rDtdntdd}|rr|jt|t|t |d|j||j dst|d} t j r| j d } | |d<|jf|S) z%Constructs an arbitrary HTTP request.443rhttpsr)rrrzwsgi.url_scheme)CONTENT_LENGTH CONTENT_TYPEz wsgi.inputrz iso-8859-1)rrrrrerrrr@r4rrrrrR) r*rQr}rvr{rrrrZ query_stringr-r-r.rs$      zRequestFactory.generic)NF)NF)F)rorF)rorF)rorF)rorF)rorF)r/r0r1r2r'rrRrrrrrrrrrrrrr-r-r-r.r#s*        cseZdZdZd*fdd ZddZddZeeZd d Z d+fd d Z d e ddffdd Z d,fdd Z d-fdd Zd.fdd Zd/fdd Zd0fdd Zd1fdd ZddZd2d d!Zd3d"d#Zd$d%Zd&d'Zd(d)ZZS)4r!a A class that can act as a client for testing purposes. It allows the user to compose GET and POST requests, and obtain the response that the server gave to those requests. The server Response objects are annotated with the details of the contexts and templates that were rendered during the process of serving the request. Client objects are stateful - they will retain cookie (and thus session) details for the lifetime of the Client instance. This is not intended as a replacement for Twill/Selenium or the like - it is here to allow testing against the contexts and templates produced by a view, rather than the HTML rendered to the end-user. Fc s&tt|jf|t||_d|_dS)N)r&r!r'rUhandlerexc_info)r*rVr)r,r-r.r's zClient.__init__cKstj|_dS)zF Stores exceptions when they are generated by a view. N)sysr)r*rXr-r-r.store_exc_infoszClient.store_exc_infocCsJttj}|jjtj}|r(|j|jS|j}|j|j |jtj<|S)z8 Obtains the current session variables. ) rrSESSION_ENGINErrSESSION_COOKIE_NAME SessionStorerysave session_key)r*enginecookiesessionr-r-r._sessions  zClient._sessionc  sR|jf}i}tt|}dt}tjj||ddt}tj|j|dzy|j |}Wn.t k r}z|j d krWYdd}~XnX|j r|j } d|_ t j| ||_|_|jdg|_|jd|_t|j||_tfdd |_|jrt|jd kr|jd |_|jr.|jj|j|Stjj|dtj|dXdS) a The master request method. Composes the environment dictionary and passes to the handler, returning the result of the handler. Assumes defaults for the query environment, which can be overridden using the arguments to the request. ztemplate-render-%s)Z dispatch_uidzrequest-exception-%s500.htmlNr]r^cs tdS)Nr)rr-)rRr-r.sz Client.request..rr)r)rrrdidrZtemplate_renderedrBr rrrrWrrreraiseclientrRrr]r^ _parse_jsonjsonrZresolver_matchr@rrrA) r*rRrZrvZon_template_renderZ signal_uidZ exception_uidrSerr-)rRr.rRs<        zClient.requestNc s4tt|j|f||d|}|r0|j|f|}|S)z@ Requests a response from the server using GET. )rvr)r&r!r_handle_redirects)r*r}rvfollowrrrS)r,r-r.r s z Client.getc s6tt|j|f|||d|}|r2|j|f|}|S)zA Requests a response from the server using POST. )rvr{r)r&r!rr)r*r}rvr{rrrrS)r,r-r.rs z Client.postc s4tt|j|f||d|}|r0|j|f|}|S)z@ Request a response from the server using HEAD. )rvr)r&r!rr)r*r}rvrrrrS)r,r-r.r"s z Client.headroapplication/octet-streamc s6tt|j|f|||d|}|r2|j|f|}|S)zC Request a response from the server using OPTIONS. )rvr{r)r&r!rr)r*r}rvr{rrrrS)r,r-r.r,s zClient.optionsc s6tt|j|f|||d|}|r2|j|f|}|S)z: Send a resource to the server using PUT. )rvr{r)r&r!rr)r*r}rvr{rrrrS)r,r-r.r8s z Client.putc s6tt|j|f|||d|}|r2|j|f|}|S)z< Send a resource to the server using PATCH. )rvr{r)r&r!rr)r*r}rvr{rrrrS)r,r-r.rDs z Client.patchc s6tt|j|f|||d|}|r2|j|f|}|S)z6 Send a DELETE request to the server. )rvr{r)r&r!rr)r*r}rvr{rrrrS)r,r-r.rPs z Client.deletec s4tt|j|f||d|}|r0|j|f|}|S)z5 Send a TRACE request to the server. )rvr)r&r!rr)r*r}rvrrrrS)r,r-r.r\sz Client.tracecKs0ddlm}|f|}|r(|j|dSdSdS)z Sets the Factory to appear as if it has successfully logged into a site. Returns True if login is possible; False if the provided credentials are incorrect. r) authenticateTFN)django.contrib.authr_login)r* credentialsruserr-r-r.logines    z Client.logincCs(|dkrtjd}||_|j||dS)Nr)rZAUTHENTICATION_BACKENDSbackendr)r*rrr-r-r. force_logints zClient.force_logincCsddlm}ttj}t}|jr,|j|_n |j|_|||||jjtj }|jj |j |<ddtj tj ppddd}|j |j|dS)Nr)rr)zmax-ager}domainrexpires)rrrrrrrrrrrrZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREr)r*rrrrrRZsession_cookieZ cookie_datar-r-r.rzs       z Client._logincCsXddlm}m}t}ttj}|jr:|j|_|||_n |j |_||t |_ dS)z Removes the authenticated user's cookies and session object. Causes the authenticated user to be logged out. r)get_userlogoutN) rrrrrrrrrrrr)r*rrrRrr-r-r.rs   z Client.logoutcKs6d|jdkr"tdj|jdtj|jjf|S)Nzapplication/jsonz Content-Typez4Content-Type header is "{0}", not "application/json")rr?formatrloadsr9r)r*rSrr-r-r.rs zClient._parse_jsoncKsg|_x|jdkr|j}|j}|j||jft|}|jrF|j|d<|jrV|j|d<|jrjt|j|d<|j }|j dst |j d |}|j |t|jfd d i|}||_|d|d dkrtd|dt|dkrtd|dqW|S)zHFollows any redirects by requesting responses from the server using GET.-./3zwsgi.url_schemerrrrrFrNzRedirect loop detected.)r(zToo many redirects.)rrrrr)r)rNurlr`r schemehostnameportrr} startswithrrRrrqueryr"r@)r*rSrZ response_urlr)rr}r-r-r.rs,      zClient._handle_redirects)F)NFF)NFF)rorFF)rorFF)rorFF)rorFF)roFF)N)N)r/r0r1r2r'rrpropertyrrRrrrrrrrrrrrrrrrr3r-r-)r,r.r!s2 D          )r!r"r#r$r%)H __future__rrrr|rerr importlibriorZ django.confrZdjango.core.handlers.baserZdjango.core.handlers.wsgirr r Zdjango.core.signalsr r r Z django.dbrZ django.httprrrZdjango.templaterZ django.testrZdjango.test.utilsrZ django.urlsrZ django.utilsrZdjango.utils.encodingrrrZdjango.utils.functionalrrZdjango.utils.httprZdjango.utils.itercompatrZ#django.utils.six.moves.urllib.parserrr __all__rrcompiler Exceptionr"objectr4rFrTrUrdr%r$r#r!r-r-r-r.sN                $ 1 1"