]j*FddlZddlZddlZddlmZddlmZm Z ddlm Z ddlm Z m Z ddl mZddlmZddlmZegd Zegd Zd ZeZGd d e ZdZGddeZdZGddej:ZdZGddeZ GddeZ!y)N) HTTPAdapter) MockRequest MockResponse)RequestsCookieJar) merge_cookiescookiejar_from_dict)get_encoding_from_headers) HTTPResponse) exceptions)rawbodycontenttextjson) status_codereasonheaderscookies json_encoderceZdZfdZxZS) CookieJarc .tt| ||fi|S)azAdd a cookie to the Jar. :param str name: cookie name/key. :param str value: cookie value. :param int version: Integer or None. Netscape cookies have version 0. RFC 2965 and RFC 2109 cookies have a version cookie-attribute of 1. However, note that cookielib may 'downgrade' RFC 2109 cookies to Netscape cookies, in which case version is 0. :param str port: String representing a port or a set of ports (eg. '80', or '80,8080'), :param str domain: The domain the cookie should apply to. :param str path: Cookie path (a string, eg. '/acme/rocket_launchers'). :param bool secure: True if cookie should only be returned over a secure connection. :param int expires: Integer expiry date in seconds since epoch or None. :param bool discard: True if this is a session cookie. :param str comment: String comment from the server explaining the function of this cookie. :param str comment_url: URL linking to a comment from the server explaining the function of this cookie. )superrset)selfnamevaluekwargs __class__s @/root/env/lib/python3.12/site-packages/requests_mock/response.pyrz CookieJar.set)s.Y)$@@@)__name__ __module__ __qualname__r __classcell__)r s@r!rr'sAAr"rc (tDcgc]}|j|d|}}t|dkDrtddj |z|Dcgc] }|t vs |}}|rt ddj |zycc}wcc}w)Nz5You may only supply one body element. You supplied %sz, z5Too many arguments provided. Unexpected arguments %s.) _BODY_ARGSpoplen RuntimeErrorjoin _HTTP_ARGS TypeError)rxprovidedextras r!_check_body_argumentsr3Cs%IaAt)<)HIHI 8}q)+/99X+>?@ @ 61!:"5Q 6E 6 (*.))E*:;< < J 7sB B  B$BceZdZdZdZdZy)_FakeConnectionzBAn object that can mock the necessary parts of a socket interface.c 0d}tj|)NzwThis response was created without a connection. You are therefore unable to make a request directly on that connection.)r InvalidRequest)rrequestrmsgs r!sendz_FakeConnection.sendUsP'',,r"cyN)rs r!closez_FakeConnection.closeZs r"N)r#r$r%__doc__r:r>r=r"r!r5r5RsL-  r"r5c|jjt|jjt ||rt |j|yy)zAdd cookies to the response. Cookies in requests are extracted from the headers in the original_response httplib.HTTPMessage which we don't create so we have to do this step manually. N)rextract_cookiesrr rrr)r8responsers r!_extract_cookiesrC^sK $$\(,,2F2F%G%0%9; h&&0r"ceZdZdZdZy) _IOReaderaA reader that makes a BytesIO look like a HTTPResponse. A HTTPResponse will return an empty string when you read from it after the socket has been closed. A BytesIO will raise a ValueError. For compatibility we want to do the same thing a HTTPResponse does. c|jryt|dkDr |ddk(rytjj|g|i|}|dk(r|j |S)Nr"r)closedr+ioBytesIOreadr>)rargsrresults r!rJz_IOReader.readxsY ;; t9q=T!W\777 S= JJL r"N)r#r$r%r?rJr=r"r!rErEps r"rEc |jdt}tdi||jdd}|jdd}|jdd}|jdd}|jdd}|jdi}d} |t|ts t d |t|t s t d |;|jd dxstj} tj|| }| t|xsd } |j| }| t|}|so|jdt} |jdtj j"j| } t%| | ||xs tddddd}t&j)||} || _| r| j,s| | _t/|| |jd| S)a :param int status_code: The status code to return upon a successful match. Defaults to 200. :param HTTPResponse raw: A HTTPResponse object to return upon a successful match. :param io.IOBase body: An IO object with a read() method that can return a body on successful match. :param bytes content: A byte string to return upon a successful match. :param unicode text: A text string to return upon a successful match. :param object json: A python object to be converted to a JSON string and returned upon a successful match. :param class json_encoder: Encoder object to use for JOSON. :param dict headers: A dictionary object containing headers that are returned upon a successful match. :param CookieJar cookies: A cookie jar with cookies to set on the response. :returns requests.Response: A response object that can be returned to requests. connectionr Nr rrrrzContent should be binary datazText should be string datar)clszutf-8rrr"F)statusrrr decode_contentenforce_content_lengthpreload_contentoriginal_responserr=)r*r5r3 isinstancebytesr/str jsonutils JSONEncoderdumpsr encoderEget_DEFAULT_STATUShttpclient responsesr _http_adapterbuild_responserNencodingrC)r8rrNr r rrrrrcencoderrPrrBs r!create_responseres*L/*; Or"ceZdZdZdZy)_Contextz:Stores the data being used to process a current URL match.c<||_||_||_||_yr<)rrrr)rrrrrs r!__init__z_Context.__init__s &  r"N)r#r$r%r?rir=r"r!rgrgs Dr"rgceZdZdZdZy)_MatcherResponsec |jdd|_|jr |r tdtdi|||_|jj d}|jj d}|&t |st|ts td|(t |st|ts tdyyy)Nexcz(Cannot provide other arguments with exc.rrz+Content should be a callback or binary dataz(Text should be a callback or string datar=) r*_excr/r3_paramsr\callablerUrVrW)rrrrs r!riz_MatcherResponse.__init__sJJud+ 99FG G'' ,,""9-||'  (9(27E(BIJ J  Xd^%/c%:FG G&;&4 r"c |jr |j|jjdt}t |t rt |t}t|jjdij|jjdt|jjd|fd}t||jjd||jjd||jjd||jjd ||jjd |jjd jjjj S) Nrrrrc8t|r |g|i|S|Sr<)rp)frKrcontextr8s r!_callz,_MatcherResponse.get_response.._call s%;CA;1Wg777 MA Mr"rrrr r r) rrrr r rrrrr)rnror\rrUdictrrgcopyr]rerrrr)rr8rrurts ` @r! get_responsez_MatcherResponse.get_responsesN 99))O,,""9ik: gt $)'9;?G4<<++Ir:??A<<++M?K<<++H5"$  Nw$)$,,*:*:6*B$C$)$,,*:*:6*B$C',T\\-=-=i-H'I$)$,,*:*:6*B$C#()9)9%)@#A,0LL,<,<^,L+2+>+>&-nn'.'. 8 8r"N)r#r$r%rirxr=r"r!rkrks H48r"rk)"rH http.clientr^rrXrequests.adaptersrrequests.cookiesrrrrrrequests.utilsr urllib3.responser requests_mockr frozensetr)r.r]rarr3objectr5rCrIrErergrkr=r"r!rs )6.?4)$ A B   A!A8 <  f  1$ 8CLv:8v:8r"