կidZdZddlmZddlZddlmZddlmZGddeZ d Z d Z d Z Gd d ejjejjZGddeZy)z$Fake urllib HTTP connection objects.zMark Roach (mrroach@google.com))pformatN)BytesIO)urllibceZdZdZy)Errorz$Base exception type for this module.N)__name__ __module__ __qualname____doc__U/var/www/html/venv/lib/python3.12/site-packages/pyactiveresource/testing/http_fake.pyrrs.r rctjjt}tjj |y)zAInstall TestHandler as the only active handler for http requests.N)rrequest build_opener TestHandlerinstall_opener)openers r initializers( ^^ ( ( 5F NN!!&)r c tjj|}tjj|j}tjj t |jDcgc] }|||f c}}t|tjj|j|j|j||jft|fScc}w)aCreate the response key for a request. Args: method: The http method (e.g. 'get', 'put', etc.) url: The path being requested including site. request_headers: Dictionary of headers passed along with the request. Returns: The key as a string. )rparseurlsplitparse_qsquery urlencodesortedkeysstr urlunsplitschemenetlocpathfragmentdictionary_to_canonical_str)methodurlrequest_headersparsedqskrs rcreate_response_keyr+s\\ " "3 'F   v|| ,B LL " "rwwy8I#J8I1Q1J8I#J KE   MM6==&++ufoo!O P#O4 6 77$Ks=C3ctt|jDcgc]}|j||fc}Scc}w)zCreate canonical string from a dictionary. Args: dictionary: The dictionary to convert. Returns: A string of the dictionary in canonical form. )rrr capitalize) dictionaryr*s rr$r$,sO ::AA/: sAcReZdZdZdZiZdZdZddZe dZ e d dZ dZ y) rz*>sxx*N"13t-.&+#c(#r c R||j_|jrt|j |j |j }t||jvr)|jt|\}}}t|||Std|j d|j dt|dtt|jjt|jtr |j|jS)arReturn the response object for the given request. Overrides the HTTPHandler method of the same name to return a FakeResponse instead of creating any network connections. Args: http_class: The http protocol being used. request: A urllib.request.Request object. Returns: A FakeResponse object. zUnknown request  z request:z response_map:) __class__rr=r+ get_method get_full_urlheadersr FakeResponserrlistr isinstancer> Exception)r8 http_classrhttp_conn_argsrHrFrErGs rdo_openzTestHandler.do_opencs")   %""$g&:&:&> !r )r)N) rr r r r>r=rrDr; classmethodrArJrWr r rrr8sNFIMG D< !!EI,,&"r rc*eZdZdZddZdZdZdZy)rQz'A fake HTTPResponse object for testing.Nc|_t|_|i}|_fd_t |t jr|jd}t|_ y)NcjSN)rPr8srz'FakeResponse.__init__..s DLLr zutf-8) rFrmsgrPinforSsix text_typeencoder body_file)r8rFrErPs` rr;zFakeResponse.__init__sT t9 ?G ( dCMM *;;w'D r c6|jjS)zRead the entire response body.)rereadr^s rrgzFakeResponse.reads~~""$$r c6|jjS)z*Read a single line from the response body.)rereadliner^s rrizFakeResponse.readlines~~&&((r cy)zClose the connection.Nr r^s rclosezFakeResponse.closes r r])rr r r r;rgrirkr r rrQrQs1 '%) r rQ)r __author__pprintrrbr six.movesrrTrrr+r$r HTTPHandler HTTPSHandlerrobjectrQr r rrrsi+ .  /I/* 7( F"&..,,fnn.I.IF"R 6 r