U a(n@sddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lmZmZmZdd lm Z ddl!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2dZ3dZ4de4Z5e6dZ7e6dZ8Gddde9Z:GdddZ;dd Zd%d&Z?d'd(Z@d)d*ZAGd+d,d,ZBGd-d.d.eBZCdS)/N)copy)partial) HTTPStatus) import_module)BytesIO)unquote_to_bytesurljoinurlparseurlsplit)settings) BaseHandler) WSGIRequest)DjangoJSONEncoder)got_request_exceptionrequest_finishedrequest_started)close_old_connections) HttpRequest QueryDict SimpleCookie)signals) ContextListresolve) force_bytes)SimpleLazyObject) urlencode) is_iterable)ClientRedirectCycleErrorRequestFactory encode_fileencode_multipartZBoUnDaRyStRiNgz multipart/form-data; boundary=%sz.*; charset=([\w\d-]+);?z^application\/(.+\+)?jsoncs eZdZdZfddZZS)rz9The test client has been asked to follow a redirect loop.cst|||_|j|_dSN)super__init__ last_responseredirect_chain)selfmessager& __class__/home/adriano.carvalho/ftp/files/BrinquedotecaVirtual/brinquedotecavirtual/venv/lib/python3.8/site-packages/django/test/client.pyr%*s zRedirectCycleError.__init__)__name__ __module__ __qualname____doc__r% __classcell__r,r,r*r-r(src@s4eZdZdZd ddZddZd ddZd d ZdS) FakePayloada A wrapper around BytesIO that restricts what can be read since data from the network can't be sought 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&||dS)NrF)r_FakePayload__content_FakePayload__len read_startedwriter(contentr,r,r-r%7s zFakePayload.__init__cCs|jSr#)r5)r(r,r,r-__len__>szFakePayload.__len__cCsZ|js|jdd|_|dkr*|jp(d}|j|ks.to_bytescSst|dot|jS)Nr=)hasattrcallabler=)thingr,r,r-is_filesz!encode_multipart..is_fileNzeCannot encode None for key '%s' as POST data. Did you mean to pass an empty string or omit the value?c3s|]}|VqdSr#r,.0valrfr,r- sz#encode_multipart..--%sz)Content-Disposition: form-data; name="%s"c3s|]}|VqdSr#r,rkrnr,r-rosz--%s--rIs )items TypeErrorextendr! isinstancestrrjoin)boundarydatalinesrjkeyvalueitemr,rnr-r"sB  r"cCsdd}t|dot|jt}|r0tj|jnd}t|drF|j}n|rZt |d}nd}|dkrjd}|pp|}|d||d ||f|d |d || gS) NcSs t|tjSr#rbrdr,r,r-rfszencode_file..to_bytesnamerq content_typerapplication/octet-streamrpz8Content-Disposition: form-data; name="%s"; filename="%s"zContent-Type: %srI) rgrur~rvospathbasenamer mimetypes guess_typer=)rxr{filerfZfile_has_string_namefilenamerr,r,r-r!s&    r!c@seZdZdZedddZddZddZd d Zd d Z d dZ d%ddZ de dfddZ d&ddZd'ddZd(ddZd)ddZd*dd Zd+d!d"Zd,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% szRequestFactory.__init__cKsLdtdd|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)rlmorselr,r,r-rosz/RequestFactory._base_environ../z 127.0.0.1GETrqZ testserver80zHTTP/1.1)rhttprITF)Z HTTP_COOKIE PATH_INFOZ REMOTE_ADDRREQUEST_METHODZ SCRIPT_NAME SERVER_NAME SERVER_PORTZSERVER_PROTOCOLz wsgi.versionwsgi.url_scheme wsgi.inputz wsgi.errorszwsgi.multiprocesszwsgi.multithreadz wsgi.run_once)rwsortedrvaluesr3rrr(rOr,r,r- _base_environs* zRequestFactory._base_environcKst|jf|S)z#Construct a generic request object.)r rrr,r,r-rO2szRequestFactory.requestcCsB|tkrtt|St|}|r,|d}ntj}t||dSdS)Nr)encoding) MULTIPART_CONTENTr"BOUNDARYCONTENT_TYPE_REmatchgroupr rcr)r(ryrrcharsetr,r,r- _encode_data6s   zRequestFactory._encode_datacCs2t|ot|tttf}|r.tj||jdS|S)zu Return encoded JSON if data is a dict, list, or tuple and content_type is application/json. )cls) JSON_CONTENT_TYPE_RErrudictlisttuplejsondumpsr)r(ryrZ should_encoder,r,r- _encode_jsonBszRequestFactory._encode_jsoncCs,|j}|jr|d|j7}t|}|dS)N; iso-8859-1)rparamsrdecode)r(parsedrr,r,r- _get_pathJs zRequestFactory._get_pathNFcKs8|dkr in|}|jd|fd|idt|ddi|S)zConstruct a GET request.Nrsecure QUERY_STRINGTdoseqgenericrr(rryrextrar,r,r-getUs zRequestFactory.getcKs@||dkrin||}|||}|jd|||fd|i|S)zConstruct a POST request.NPOSTr)rrr)r(rryrrrZ post_datar,r,r-post]s zRequestFactory.postcKs8|dkr in|}|jd|fd|idt|ddi|S)zConstruct a HEAD request.NrJrrTrrrr,r,r-headfs zRequestFactory.headcKs|jd|fd|i|S)zConstruct a TRACE request.TRACErr)r(rrrr,r,r-tracenszRequestFactory.tracerqrcKs|jd|||fd|i|S)zConstruct an OPTIONS request.OPTIONSrrr(rryrrrr,r,r-optionsrs zRequestFactory.optionscKs(|||}|jd|||fd|i|S)zConstruct a PUT request.PUTrrrrr,r,r-putxs  zRequestFactory.putcKs(|||}|jd|||fd|i|S)zConstruct a PATCH request.PATCHrrrr,r,r-patchs  zRequestFactory.patchcKs(|||}|jd|||fd|i|S)zConstruct a DELETE request.DELETErrrr,r,r-deletes  zRequestFactory.deletec Kstt|}t|tj}||||r*dnd|r4dndd}|r^|tt||t|d||| ds|d d } | |d<|j f|S) z$Construct an arbitrary HTTP request.443rhttpsr)rrrr)CONTENT_LENGTH CONTENT_TYPErrr) r rvrr rcrupdater@r3rencoderrO) r(rNrryrrrrrZ query_stringr,r,r-rs$       zRequestFactory.generic)NF)NF)F)rqrF)rqrF)rqrF)rqrF)rqrF)r.r/r0r1rr%rrOrrrrrrrrrrrrrr,r,r,r-r s:         r cseZdZdZd+fdd ZddZedd Zd d Zd,fd d Z d e ddffdd Z d-fdd Z d.fdd Z d/fdd Zd0fdd Zd1fdd Zd2fdd Zdd Zd3d!d"Zd4d#d$Zd%d&Zd'd(Zd5d)d*ZZS)6ra 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. FTc s(tjf|t||_||_d|_dSr#)r$r%rRhandlerraise_request_exceptionexc_info)r(rSrrr*r,r-r%s zClient.__init__cKst|_dS)z3Store exceptions when they are generated by a view.N)sysr)r(rUr,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 SessionStorer|save session_key)r(enginecookiesessionr,r,r-rs  zClient.sessionc s,|jf}i}tt|}dt}tjj||ddt}tj|j|dz| |}W5tjj |dtj |dX|j |_ |j r|j \}} }d|_ |j r| ||_ |_|dg|_|d|_t|j||_tfdd|_|jrt|jd kr|jd |_|jr(|j|j|S) 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)Z dispatch_uidzrequest-exception-%sNrZr[cs tdS)Nrrr,rOr,r-rIz Client.request..rr)rrraidrZtemplate_renderedrBrrrArrrclientrOrrZr[ _parse_jsonrrZresolver_matchr@rr) r(rOrWryZon_template_renderZ signal_uidZ exception_uidrP_ exc_valuer,rr-rOs8       zClient.requestNc s8tj|f||d|}|r4|j|fd|i|}|S)z-Request a response from the server using GET.ryrry)r$r_handle_redirectsr(rryfollowrrrPr*r,r-rsz Client.getc s<tj|f|||d|}|r8|j|f||d|}|S)z.Request a response from the server using POST.ryrrryr)r$rrr(rryrrrrrPr*r,r-r sz Client.postc s8tj|f||d|}|r4|j|fd|i|}|S)z.Request a response from the server using HEAD.rry)r$rrrr*r,r-rsz Client.headrqrc s<tj|f|||d|}|r8|j|f||d|}|S)z1Request a response from the server using OPTIONS.rr)r$rrrr*r,r-rszClient.optionsc s<tj|f|||d|}|r8|j|f||d|}|S)z(Send a resource to the server using PUT.rr)r$rrrr*r,r-r"sz Client.putc s<tj|f|||d|}|r8|j|f||d|}|S)z*Send a resource to the server using PATCH.rr)r$rrrr*r,r-r*sz Client.patchc s<tj|f|||d|}|r8|j|f||d|}|S)z$Send a DELETE request to the server.rr)r$rrrr*r,r-r2sz Client.deletec s8tj|f||d|}|r4|j|fd|i|}|S)z#Send a TRACE request to the server.rry)r$rrrr*r,r-r:sz Client.tracecKs0ddlm}|f|}|r(||dSdSdS)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) authenticateTFN)django.contrib.authr_login)r( credentialsruserr,r,r-loginAs    z Client.logincCs,dd}|dkr|}||_|||dS)NcSs6ddlm}tjD]}||}t|dr|SqdS)Nr) load_backendget_user)rrr ZAUTHENTICATION_BACKENDSrg)r backend_pathbackendr,r,r- get_backendQs    z'Client.force_login..get_backend)rr)r(rrrr,r,r- force_loginPs zClient.force_logincCsddlm}ttj}t}|jr,|j|_n ||_|||||jtj }|jj |j |<ddtj tj ppddd}|j ||dS)Nr)rr)zmax-agerdomainrexpires)rrrr rrrrrrrrZSESSION_COOKIE_DOMAINZSESSION_COOKIE_SECUREr)r(rrrrrOZsession_cookieZ cookie_datar,r,r-r\s"      z Client._logincCsXddlm}m}t}ttj}|jr:|j|_|||_n | |_||t |_ dS)zsP                   $ 0 6)