o uvXjd"d#Z?Gd$d%d%eZ@d&d'ZAd(d)ZBd*d+ZCGd,d-d-ZDGd.d/d/eDZEdS)0N)copy)partial) HTTPStatus) import_module)BytesIO)unquote_to_bytesurljoinurlparseurlsplit)settings) BaseHandler) WSGIRequest)DjangoJSONEncoder)got_request_exceptionrequest_finishedrequest_started)close_old_connections) HttpRequest QueryDict SimpleCookie)TemplateDoesNotExist)signals) ContextListresolve) force_bytes)SimpleLazyObject) urlencode) is_iterable)ClientRedirectCycleErrorRequestFactory encode_fileencode_multipartBoUnDaRyStRiNgz multipart/form-data; boundary=%sz.*; charset=([\w\d-]+);?z^application\/(.+\+)?jsoncs eZdZdZfddZZS)r z9The test client has been asked to follow a redirect loop.cst|||_|j|_dSN)super__init__ last_responseredirect_chain)selfmessager( __class__O/var/www/html/myproject/venv/lib/python3.10/site-packages/django/test/client.pyr'+s  zRedirectCycleError.__init__)__name__ __module__ __qualname____doc__r' __classcell__r.r.r,r/r )sr c@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|_|dur||dSdS)NrF)r_FakePayload__content_FakePayload__len read_startedwriter*contentr.r.r/r'8s zFakePayload.__init__cCs|jSr%)r7)r*r.r.r/__len__?szFakePayload.__len__cCsZ|js |jdd|_|dur|jpd}|j|ksJd|j|}|j|8_|S)NrTzFCannot read more than the available bytes from the HTTP incoming data.)r8r6seekr7read)r* num_bytesr;r.r.r/r>Bs   zFakePayload.readcCs8|jrtdt|}|j||jt|7_dS)Nz.Unable to write a payload after he's been read)r8 ValueErrorrr6r9r7lenr:r.r.r/r9Ms  zFakePayload.writer%)r0r1r2r3r'r<r>r9r.r.r.r/r51s    r5c csJz|EdHWtt|ttdStt|ttwr%)r disconnectrconnect)iterablecloser.r.r/closing_iterator_wrapperUs    rFcCsZd|jkr dksn|jdvr|jrg|_nd|_|jdkr+|jr(g|_|Sd|_|S)z Simulate the behavior of most Web servers by removing the content of responses for HEAD requests, 1xx, 204, and 304 responses. Ensure compliance with RFC 7230, section 3.3.3. d)i0HEAD) status_code streamingstreaming_contentr;method)requestresponser.r.r/conditional_content_removal^s" rRcs*eZdZdZdfdd ZddZZS) ClientHandlerz A HTTP Handler that can be used for testing purposes. Use the WSGI interface to compose requests, but return the raw HttpResponse object with the originating WSGIRequest attached to its ``wsgi_request`` attribute. Tcs||_tj|i|dSr%)enforce_csrf_checksr&r')r*rTargskwargsr,r.r/r'wszClientHandler.__init__cCs|jdur |tttj|j|dttt|}|j |_ | |}t ||||_ |jr>t|j|j|_|Stt|tt|S)N)senderenviron)_middleware_chainload_middlewarerrBrsendr-rCr rT_dont_enforce_csrf_checks get_responserR wsgi_requestrMrFrNrEr)r*rXrPrQr.r.r/__call__{s&        zClientHandler.__call__)T)r0r1r2r3r'r_r4r.r.r,r/rSqsrScKs:|dg|d|vrt|d<|dt|dS)z Store templates and contexts that are rendered. The context is copied so that it is an accurate representation at the time of rendering. templatescontextN) setdefaultappendrr)storesignalrWtemplaterarVr.r.r/store_rendered_templatess rgc sg}dddd}|D]_\}}|durtd||r(|t|||qt|tsYt|rY|D]$}||rC|t|||q3|fddd |d |d |fDq3q|fd dd |d |d |fDq|d |dgd|S)a+ Encode multipart POST data from a dictionary of form values. The key will be used as the form data name; the value will be transmitted as content. If the value is a file, the contents of the file will be sent as an application/octet-stream; otherwise, str(value) will be sent. cS t|tjSr%rr DEFAULT_CHARSETsr.r.r/to_bytes z"encode_multipart..to_bytescSst|do t|jS)Nr>)hasattrcallabler>)thingr.r.r/is_filesz!encode_multipart..is_fileNzXCannot encode None as POST data. Did you mean to pass an empty string or omit the value?c3|]}|VqdSr%r..0valrmr.r/ z#encode_multipart..--%sz)Content-Disposition: form-data; name="%s"c3rsr%r.rtrwr.r/rxryz--%s--rJs )items TypeErrorextendr" isinstancestrrjoin)boundarydatalinesrrkeyvalueitemr.rwr/r#s@     r#cCsdd}t|dot|jt}|rtj|jnd}t|dr#|j}n |r-t |d}nd}|dur5d}|p8|}|d||d ||f|d |d || gS) NcSrhr%rirkr.r.r/rmrnzencode_file..to_bytesnamer{ content_typerapplication/octet-streamrzz8Content-Disposition: form-data; name="%s"; filename="%s"zContent-Type: %srJ) rorrrospathbasenamer mimetypes guess_typer>)rrfilermfile_has_string_namefilenamerr.r.r/r"s&    r"c@seZdZdZedddZddZddZd d Zd d Z d dZ d%ddZ de dfddZ d%ddZd&ddZ  d'ddZ  d'ddZ  d'dd Z  d'd!d"Z  d'd#d$ZdS)(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. ) json_encodercKs ||_||_t|_t|_dSr%)rdefaultsrcookiesrerrors)r*rrr.r.r/r's zRequestFactory.__init__cKsNdtdd|jDdddddd d d d td |jdddd|j|S)z5 The base environment for a request. z; css |] }d|j|jfVqdS)z%s=%sN)r coded_value)rumorselr.r.r/rxs  z/RequestFactory._base_environ../z 127.0.0.1GETr{ testserver80zHTTP/1.1)rhttprJTF) HTTP_COOKIE PATH_INFO REMOTE_ADDRREQUEST_METHOD SCRIPT_NAME SERVER_NAME SERVER_PORTSERVER_PROTOCOLz wsgi.versionwsgi.url_scheme wsgi.inputz wsgi.errorszwsgi.multiprocesszwsgi.multithreadz wsgi.run_once)rsortedrvaluesr5rrr*rPr.r.r/ _base_environs, zRequestFactory._base_environcKst|jdi|S)z#Construct a generic request object.Nr.)r rrr.r.r/rP3szRequestFactory.requestcCs>|tur tt|St|}|r|d}ntj}t||dS)Nr)encoding) MULTIPART_CONTENTr#BOUNDARYCONTENT_TYPE_REmatchgroupr rjr)r*rrrcharsetr.r.r/ _encode_data7s    zRequestFactory._encode_datacCs2t|o t|tttf}|rtj||jdS|S)zu Return encoded JSON if data is a dict, list, or tuple and content_type is application/json. )cls) JSON_CONTENT_TYPE_RErrdictlisttuplejsondumpsr)r*rr should_encoder.r.r/ _encode_jsonCszRequestFactory._encode_jsoncCs,|j}|jr |d|j7}t|}|dS)N; iso-8859-1)rparamsrdecode)r*parsedrr.r.r/ _get_pathKs  zRequestFactory._get_pathNFcK8|durin|}|jd|fd|idt|ddi|S)zConstruct a GET request.Nrsecure QUERY_STRINGTdoseqgenericrr*rrrextrar.r.r/getV  zRequestFactory.getcKs@||durin||}|||}|jd|||fd|i|S)zConstruct a POST request.NPOSTr)rrr)r*rrrrr post_datar.r.r/post^s zRequestFactory.postcKr)zConstruct a HEAD request.NrKrrTrrrr.r.r/headgrzRequestFactory.headcKs|jd|fd|i|S)zConstruct a TRACE request.TRACErr)r*rrrr.r.r/traceoszRequestFactory.tracer{rcKs|jd|||fd|i|S)zConstruct an OPTIONS request.OPTIONSrrr*rrrrrr.r.r/optionsss zRequestFactory.optionscK(|||}|jd|||fd|i|S)zConstruct a PUT request.PUTrrrrr.r.r/puty zRequestFactory.putcKr)zConstruct a PATCH request.PATCHrrrr.r.r/patchrzRequestFactory.patchcKr)zConstruct a DELETE request.DELETErrrr.r.r/deleterzRequestFactory.deletec Kstt|}t|tj}||||rdnd|rdndd}|r/|tt||t|d||| dsF|d d } | |d<|j d i|S) z$Construct an arbitrary HTTP request.443rhttpsr)rrrr)CONTENT_LENGTH CONTENT_TYPErrrNr.) r rrr rjrupdaterAr5rencoderrP) r*rOrrrrrrr query_stringr.r.r/rs$       zRequestFactory.generic)NFF)r{rF)r0r1r2r3rr'rrPrrrrrrrrrrrrrr.r.r.r/r!s:        r!cseZdZdZd*fdd ZddZeddZd d Zd+fd d Z d e ddffdd Z d+fdd Z  d,fdd Z  d,fdd Z  d,fdd Z  d,fdd Zd-fdd ZddZd.d d!Zd.d"d#Zd$d%Zd&d'Zd/d(d)ZZS)0ra 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&tjdi|t||_d|_dS)Nr.)r&r'rShandlerexc_info)r*rTrr,r.r/r's  zClient.__init__cKst|_dS)z3Store exceptions when they are generated by a view.N)sysr)r*rVr.r.r/store_exc_infoszClient.store_exc_infocCsJttj}|jtj}|r||jS|}||j |jtj<|S)z%Return the current session variables.) rr SESSION_ENGINErrSESSION_COOKIE_NAME SessionStorersave session_key)r*enginecookiesessionr.r.r/rs  zClient.sessionc sj|jd i}i}tt|}dt}tjj||ddt}tj|j|dz{z| |}Wnt yK}z |j dkrAWYd}~nd}~ww|j rZ|j \} } } d|_ | ||_ |_|dg|_|d|_t|j||_tfdd |_|jrt|jd kr|jd |_|jr|j|j|Wtjj|dtj|dStjj|dtj|dw) a  The master request method. Compose the environment dictionary and pass to the handler, return the result of the handler. Assume defaults for the query environment, which can be overridden using the arguments to the request. ztemplate-render-%s) dispatch_uidzrequest-exception-%s)z500.htmlNr`racs tdS)Nrrr.rPr.r/s z Client.request..rrr.)rrrgidrtemplate_renderedrCrrrrrUrclientrPrr`ra _parse_jsonrrresolver_matchrArrrB) r*rPrXron_template_render signal_uid exception_uidrQe_ exc_valuer.rr/rPsF        zClient.requestNc 8tj|f||d|}|r|j|fd|i|}|S)z-Request a response from the server using GET.rrr)r&r_handle_redirectsr*rrfollowrrrQr,r.r/rz Client.getc <tj|f|||d|}|r|j|f||d|}|S)z.Request a response from the server using POST.rrrrr)r&rrr*rrrr rrrQr,r.r/rz Client.postc r)z.Request a response from the server using HEAD.rr)r&rrr r,r.r/r$r z Client.headr{rc r )z1Request a response from the server using OPTIONS.r r)r&rrrr,r.r/r+rzClient.optionsc r )z(Send a resource to the server using PUT.r r)r&rrrr,r.r/r3rz Client.putc r )z*Send a resource to the server using PATCH.r r)r&rrrr,r.r/r;rz Client.patchc r )z$Send a DELETE request to the server.r r)r&rrrr,r.r/rCrz Client.deletec r)z#Send a TRACE request to the server.rr)r&rrr r,r.r/rKr z Client.tracecKs0ddlm}|di|}|r||dSdS)z Set the Factory to appear as if it has successfully logged into a site. Return True if login is possible; False if the provided credentials are incorrect. r) authenticateTFNr.)django.contrib.authr_login)r* credentialsruserr.r.r/loginRs  z Client.logincCs,dd}|dur |}||_|||dS)NcSs6ddlm}tjD]}||}t|dr|Sq dS)Nr) load_backendget_user)rrr AUTHENTICATION_BACKENDSro)r backend_pathbackendr.r.r/ get_backendbs   z'Client.force_login..get_backend)rr)r*rrrr.r.r/ force_loginas zClient.force_logincCsddlm}ttj}t}|jr|j|_n||_|||||jtj }|jj |j |<ddtj tj p8ddd}|j ||dS)Nr)rr)zmax-agerdomainrexpires)rrrr rrrrrrrrSESSION_COOKIE_DOMAINSESSION_COOKIE_SECUREr)r*rrrrrPsession_cookie cookie_datar.r.r/rms"      z Client._logincCsXddlm}m}t}ttj}|jr|j|_|||_n| |_||t |_ dS)z  D   r)Frrrrerr functoolsrrr importlibrior urllib.parserrr r django.confr django.core.handlers.baser django.core.handlers.wsgir django.core.serializers.jsonrdjango.core.signalsrrr django.dbr django.httprrrdjango.templater django.testrdjango.test.utilsr django.urlsrdjango.utils.encodingrdjango.utils.functionalrdjango.utils.httprdjango.utils.itercompatr__all__rrcompilerr Exceptionr r5rFrRrSrgr#r"r!rr.r.r.r/sT                    $ 0 6)