B `r@s@dZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlZddgZGdddeZ Gdddej!Z"Gdddej#Z$Gddde%Z&dS)zF Routines for testing WSGI applications. Most interesting is TestApp )unicode_literalsN) b64encode)StringIO)BytesIO) string_types) binary_type) text_type)http_cookiejar)urlparse)to_bytes)escape_cookie_value) TestResponse)forms)lint)utilsTestApp TestRequestc@seZdZddZdS)AppErrorc Gst|tr|d}d}x|D]~}t|tjr\|j}t|tr|jrR||j}qt|}n6t|try|d}Wntk rt|}YnX||f7}qW||}t ||dS)Nutf8) isinstancerdecodewebobResponsebodycharsetreprUnicodeDecodeError Exception__init__)selfmessageargsZstr_argsargrrrS/home/kop/projects/devel/pgwui/test_venv/lib/python3.7/site-packages/webtest/app.pyr,s$       zAppError.__init__N)__name__ __module__ __qualname__rrrrr$r*src@s eZdZdZddZddZdS) CookiePolicyzOA subclass of DefaultCookiePolicy to allow cookie set for Domain=localhost.cCs|jdkrdStj|||S)Nz .localhostT)domainr DefaultCookiePolicyreturn_ok_domain)r cookierequestrrr$r+Fs zCookiePolicy.return_ok_domaincCs|jdkrdStj|||S)Nz .localhostT)r)r r* set_ok_domain)r r,r-rrr$r.Ls zCookiePolicy.set_ok_domainN)r%r&r'__doc__r+r.rrrr$r(Bsr(c@seZdZdZeZdS)rzA subclass of webob.RequestN)r%r&r'r/r ResponseClassrrrr$rSsc@s6eZdZdZeZd:ddZddZdd Ze eeZ e d d Z d d Z ddZ ddZd;ddZdddZd?ddZd@ddZdAd d!Zed"Zed#Zed$Zed%Zd&d'ZdBd(d)ZdCd*d+Zd,d-Zd.d/Z dDd0d1Z!d2d3Z"ej#ddddddfd4d5Z$d6d7Z%e&d8d9Z'dS)EraX Wraps a WSGI application in a more convenient interface for testing. It uses extended version of :class:`webob.BaseRequest` and :class:`webob.Response`. :param app: May be an WSGI application or Paste Deploy app, like ``'config:filename.ini#test'``. .. versionadded:: 2.0 It can also be an actual full URL to an http server and webtest will proxy requests with `WSGIProxy2 `_. :type app: WSGI application :param extra_environ: A dictionary of values that should go into the environment for each request. These can provide a communication channel with the application. :type extra_environ: dict :param relative_to: A directory used for file uploads are calculated relative to this. Also ``config:`` URIs that aren't absolute. :type relative_to: string :param cookiejar: :class:`cookielib.CookieJar` alike API that keeps cookies across requests. :type cookiejar: CookieJar instance .. attribute:: cookies A convenient shortcut for a dict of all cookies in ``cookiejar``. :param parser_features: Passed to BeautifulSoup when parsing responses. :type parser_features: string or list :param json_encoder: Passed to json.dumps when encoding json :type json_encoder: A subclass of json.JSONEncoder :param lint: If True (default) then check that the application is WSGI compliant :type lint: A boolean NTc Csdtjkrtjd}t|tr|dryddlm} Wntk rTtdYnXd|krf|d7}|dd\} } | | | d }ndd l m } | ||d }||_ ||_ ||_ |dkri}||_||_|dkrtjtd }||_|dkrd }||jj_|dkrtj}||_dS)NZWEBTEST_TARGET_URLhttpr) HostProxyzaUsing webtest with a real url requires WSGIProxy2. Please install it with: pip install WSGIProxy2#z#httplib)client)loadapp) relative_to)policyz html.parser)osenvironrr startswithZ wsgiproxyr2 ImportErrorsplitZ paste.deployr6apprr7 extra_environ use_unicoder CookieJarr( cookiejar RequestClassr0parser_featuresjson JSONEncoder) r r>r?r7r@rBrDZ json_encoderrr2urlr5r6rrr$rs>         zTestApp.__init__cCs|jS)a Allow to set the HTTP_AUTHORIZATION environ key. Value should look like one of the following: * ``('Basic', ('user', 'password'))`` * ``('Bearer', 'mytoken')`` * ``('JWT', 'myjwt')`` If value is None the the HTTP_AUTHORIZATION is removed )authorization_value)r rrr$get_authorizations zTestApp.get_authorizationcCs||_|dk rd}t|ttfrt|dkr|\}}|dkrx|rxt|ttfrxdt|}tt|}| d}n,|dkr|rt|t t fr|}nt |t d||f}nt ||j d|ind|j kr|j d=dS) NzfYou should use a value like ('Basic', ('user', 'password')) OR ('Bearer', 'token') OR ('JWT', 'token')ZBasic:latin1)ZBearerZJWTz%s %sZHTTP_AUTHORIZATION)rHrlisttuplelenjoinrr striprstrr ValueErrorr?update)r valueZ invalid_valueZauthtypevalrrr$set_authorizations(      zTestApp.set_authorizationcCstdd|jDS)NcSsg|]}|j|jfqSr)namerU).0r,rrr$ sz#TestApp.cookies..)dictrB)r rrr$cookiesszTestApp.cookiescCsp|jdd}|ddd}d|kr.d|}t|}tjd||dd |d d d d d dd dddd }|j|dS) zD Sets a cookie to be passed through with requests. Z HTTP_HOSTz .localhostrKr4r.z%s.localNFT/)versionrXrUportport_specifiedr)domain_specifieddomain_initial_dotpathpath_specifiedsecureexpiresdiscardcomment comment_urlrest)r?getr=r r CookierB set_cookie)r rXrUZ cookie_domainr,rrr$rns.zTestApp.set_cookiecCs|jdS)zc Resets the state of the application; currently just clears saved cookies. N)rBclear)r rrr$resetsz TestApp.resetcCs||jj_dS)zx Changes the parser used by BeautifulSoup. See its documentation to know the supported parsers. N)rCr0rD)r rDrrr$set_parser_features szTestApp.set_parser_featuresFc Cs||}t|}||}|r,t||}td|krR|tdd\}|d<n td|d<|j||} |rz||}|r| j ||j | ||dS)a Do a GET request given the url path. :param params: A query string, or a dictionary that will be encoded into a query string. You may also include a URL query string on the ``url``. :param headers: Extra headers to send. :type headers: dictionary :param extra_environ: Environmental variables that should be added to the request. :type extra_environ: dictionary :param status: The HTTP status code you expect in response (if not 200 or 3xx). You can also use a wildcard, like ``'3*'`` or ``'*'``. :type status: integer or string :param expect_errors: If this is False, then if anything is written to environ ``wsgi.errors`` it will be an error. If it is True, then non-200/3xx responses are also okay. :type expect_errors: boolean :param xhr: If this is true, then marks response as ajax. The same as headers={'X-REQUESTED-WITH': 'XMLHttpRequest', } :type xhr: boolean :returns: :class:`webtest.TestResponse` instance. ?r4 QUERY_STRING)status expect_errors) _make_environrR_remove_fragmentr build_paramsr=rCblank_add_xhr_headerheadersrT do_request) r rGparamsr|r?rurvxhrr:reqrrr$rls%       z TestApp.getrtc Cs*| r||}|jd||||||||d S)a( Do a POST request. Similar to :meth:`~webtest.TestApp.get`. :param params: Are put in the body of the request. If params is an iterator, it will be urlencoded. If it is a string, it will not be encoded, but placed in the body directly. Can be a :class:`python:collections.OrderedDict` with :class:`webtest.forms.Upload` fields included:: app.post('/myurl', collections.OrderedDict([ ('textfield1', 'value1'), ('uploadfield', webapp.Upload('filename.txt', 'contents'), ('textfield2', 'value2')]))) :param upload_files: It should be a list of ``(fieldname, filename, file_content)``. You can also use just ``(fieldname, filename)`` and the file contents will be read from disk. :type upload_files: list :param content_type: HTTP content type, for example `application/json`. :type content_type: string :param xhr: If this is true, then marks response as ajax. The same as headers={'X-REQUESTED-WITH': 'XMLHttpRequest', } :type xhr: boolean :returns: :class:`webtest.TestResponse` instance. POST)r~r|r?ru upload_filesrv content_type)r{ _gen_request) r rGr~r|r?rurrvrrrrr$postJs'  z TestApp.postc Cs*| r||}|jd||||||||d S)z Do a PUT request. Similar to :meth:`~webtest.TestApp.post`. :returns: :class:`webtest.TestResponse` instance. PUT)r~r|r?rurrvr)r{r) r rGr~r|r?rurrvrrrrr$putys   z TestApp.putc Cs*| r||}|jd||||||||d S)z Do a PATCH request. Similar to :meth:`~webtest.TestApp.post`. :returns: :class:`webtest.TestResponse` instance. PATCH)r~r|r?rurrvr)r{r) r rGr~r|r?rurrvrrrrr$patchs   z TestApp.patchc Cs*|r||}|jd|||||d||d S)z Do a DELETE request. Similar to :meth:`~webtest.TestApp.get`. :returns: :class:`webtest.TestResponse` instance. DELETEN)r~r|r?rurrvr)r{r) r rGr~r|r?rurvrrrrr$deletes   zTestApp.deletec Cs&|r||}|jd||||d|dS)z Do a OPTIONS request. Similar to :meth:`~webtest.TestApp.get`. :returns: :class:`webtest.TestResponse` instance. OPTIONSN)r|r?rurrv)r{r)r rGr|r?rurvrrrr$optionss   zTestApp.optionsc Cs6|rt||}|r||}|jd||||d|dS)z Do a HEAD request. Similar to :meth:`~webtest.TestApp.get`. :returns: :class:`webtest.TestResponse` instance. HEADN)r|r?rurrv)rryr{r)r rGr~r|r?rurvrrrr$heads   z TestApp.headrrrrc stttddddgfdd}x6|D],\}}t|trry|d}WnYnXt|tjr|jr||gt |jn||ddgq@t|tj r||j g}|j dk r| |j |jdk r| |j||q@t|trt|d }n>t|tr*|d }n&t|ttfsPtd |t||d d |d d|gq@Wx|D]}||qxWd d dgd}dd}||fS)z Encodes a set of parameters (typically a name/value list) and a set of files (a list of (name, filename, file_body, mimetype)) into a typical POST body, returning the (content_type, body). rJNs----------a_BoUnDaRy$cs|\}}}}t|tr:y|d}WnYnXt|trby|d}WnYnX|szt|dd}t|}|pd}dd|d|dd |d |gdS) Nasciirrsapplication/octet-streams--s&Content-Disposition: form-data; name="s "; filename=""sContent-Type: ) _get_file_inforrencode mimetypes guess_typerr extend) file_infokeyfilenamerUZfcontent)boundarylinesr rr$ _append_files&  z.TestApp.encode_multipart.._append_filerrrzCValue for field {0} is a {1} ({2}). It must be str, bytes or an ints--s&Content-Disposition: form-data; name="rs z multipart/form-data; boundary=%s)r rRrandomrrrrFilerUrMUploadrcontentappendrintbytesrSformattyperrPr) r r~filesrrrUrrrr)rrr r$encode_multipartsP                 zTestApp.encode_multipartc Kst|trt|}x*|D]\}}t|trt|||<qWt|trZ|jj|f|}n*|}x |D]\}} t||| qlWd|j d<x$|j D]\}} |j || qW|j |||dS)a Creates and executes a request. You may either pass in an instantiated :class:`TestRequest` object, or you may pass in a URL and keyword arguments to be passed to :meth:`TestRequest.blank`. You can use this to run a request without the intermediary functioning of :meth:`TestApp.get` etc. For instance, to test a WebDAV method:: resp = app.request('/new-col', method='MKCOL') Note that the request won't have a body unless you specify it, like:: resp = app.request('/test.txt', method='PUT', body='test') You can use :class:`webtest.TestRequest`:: req = webtest.TestRequest.blank('/url/', method='GET') resp = app.do_request(req) Tzpaste.throw_errors)rurv) rrrRitemsrrCrzcopysetattrr:r? setdefaultr}) r Z url_or_reqrurvZ req_paramskvrrXrUrrr$r-s     zTestApp.requestc CsTt}||jd<|jdd}|rB|j|rB|jt|d|_d|jd<i|jd<|jt ||j rzt |j n|j }|j |dd}||j|_||_||_ ||_y |jWntk rYnX||_x>|jdD],\}} t||r td |t||| qW|s6||||||jt|t ||S) a Executes the given webob Request (``req``), with the expected ``status``. Generally :meth:`~webtest.TestApp.get` and :meth:`~webtest.TestApp.post` are used instead. To use this:: req = webtest.TestRequest.blank('url', ...args...) resp = app.do_request(req) .. note:: You can pass any keyword arguments to ``TestRequest.blank()``, which will be set on the request. These can be arguments like ``content_type``, ``accept``, etc. z wsgi.errorsZ SCRIPT_NAMErtNTz paste.testingzpaste.testing_variables)Zcatch_exc_infozopaste.testing_variables contains the variable %r, but the response object already has an attribute by that name)rr:rlZ path_infor;rOrBadd_cookie_headerrZ_RequestCookieAdapterrZ middlewarer>Z get_responsedecode_contentr@Z _use_unicoder-Ztest_appr TypeErrorgetvalueerrorsrhasattrrSr _check_status _check_errorsextract_cookiesZ_ResponseCookieAdapter) r rrurvr script_namer>resrXrUrrr$r}Ks@         zTestApp.do_requestcCs|dkr dS|j}t|tr>d|kr>tt||tjr>dSt|trT||krTdSt|tt fr|j |krt d|d t t||jj|dS|dkr|j dkr|j dkrdSt d||jj|||j krt d|||dS)N*z*Bad response: %s (not one of %s for %s) %sz, iz7Bad response: %s (not 200 OK or 3xx redirect for %s) %szBad response: %s (not %s) %s)rurrrematchfnmatch translateIrMrNZ status_intrrPmaprRr-rG)r rurZ res_statusrrr$rs4    zTestApp._check_statuscCs|j}|rtd|dS)Nz!Application had errors logged: %s)rr)r rrrrr$rszTestApp._check_errorscCs$|j}d|d<|r |||S)NTzpaste.throw_errors)r?rrT)r r?r:rrr$rws   zTestApp._make_environcCs(t|\}}}}}t||||dfS)Nrt)r urlsplit urlunsplit)r rGschemenetlocrdqueryfragmentrrr$rxszTestApp._remove_fragmentc Csj||} g} t|ts"t|dr.t|}t|ttfrJdd|D} t| dkrt|||pbd\} }| | d<nbt || }|s| rt |  drt j|dd }|||pd\} }| | d<n|r| dtd | d k r| | d<t|| d <t|}||}|j|| } t|tr0|| jp,d }t|| jd<t|| _|rZ| j||j| ||dS)z' Do a generic request. rcSs&g|]\}}t|tjtjfr|qSr)rrrr)rYrrrrr$rZsz(TestApp._gen_request..rr CONTENT_TYPEs multipartT)keep_blank_valuesz!application/x-www-form-urlencodedNREQUEST_METHODrz wsgi.input)rurv)rwrr[rrMrrNrOrrZ encode_paramsr r;r parse_qslrrRrxrCrzrrrrr:content_lengthr|rTr}) r methodrGr~r|r?rurrvrr:Zinline_uploadsrrrr$rsF            zTestApp._gen_requestcCst|dkrT|d}|jr*tj|j|}t|d}|}||d||dfSdt|krldkrnnD|d}t|t st dt t |ft|dkrt |dS|Snt d t |dd dS) NrJr4rbrzFile content must be %s not %s)Nzupload_files need to be a list of tuples of (fieldname, filename, filecontent, mimetype) or (fieldname, filename, filecontent) or (fieldname, filename); you gave: %rd)rOr7r9rdrPopenreadcloserrrSrrNr)r rrfrrrr$rs&     zTestApp._get_file_infocCs|pi}|dtdi|S)NzX-REQUESTED-WITHZXMLHttpRequest)rTrR)r|rrr$r{szTestApp._add_xhr_header)NNTNNNT)NNNNFF)rtNNNNFNF)rtNNNNFNF)rtNNNNFNF)rtNNNFNF)NNNFF)NNNNFF)NF)NN)N)(r%r&r'r/rrCrrIrWproperty authorizationr\rnrprqrlrrrrrrrZ json_methodZ post_jsonZput_jsonZ patch_jsonZ delete_jsonrr-r}rrrwrxZ NoDefaultrr staticmethodr{rrrr$rXs^4 '    5 -         M , G  1)'r/ __future__rr9rrErrrbase64rsixrrrrrZ six.movesr Zwebtest.compatr r r Zwebtest.responser Zwebtestrrrr__all__rrr*r(Z BaseRequestrobjectrrrrr$ s6