U |hF@sddlmZmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZdd lmZd d lmZd dlmZejdZGdddee ZGdddeZdS))AnyDictOptional)mock)settings)AbstractBaseUser)Http404)TestCase)reverse) urlencode)slugify)get_dummy_request)Page)querydict_from_html)WagtailTestUtilsc seZdZdZefddZddZd!ddZd"d d Zd#d d Z d$ddZ d%ddZ d&e e ee edddZd'e e ee eeefe eeefe ee ee ee edddZd(e e eeefe ee edddZd)e e ee eeefe ee eddd ZZS)*WagtailPageTestCasezO A set of assertions to help write tests for custom Wagtail page types cstt|_dSN)super setUpClassr dummy_request)cls __class__e/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/test/utils/page_tests.pyrs zWagtailPageTestCase.setUpClasscCs ||kSr)allowed_subpage_models)self parent_model child_modelrrr_testCanCreateAtsz$WagtailPageTestCase._testCanCreateAtNcCsB|||s>||d|jj|jj|jj|jjf}||dS)z Assert a particular child Page type can be created under a parent Page type. ``parent_model`` and ``child_model`` should be the Page classes being tested. z$Can not create a %s.%s under a %s.%sNr _formatMessage_meta app_label model_namefailureExceptionrrrmsgrrrassertCanCreateAt"s  z%WagtailPageTestCase.assertCanCreateAtcCsB|||r>||d|jj|jj|jj|jjf}||dS)z Assert a particular child Page type can not be created under a parent Page type. ``parent_model`` and ``child_model`` should be the Page classes being tested. z Can create a %s.%s under a %s.%sNr!r'rrrassertCanNotCreateAt5s  z(WagtailPageTestCase.assertCanNotCreateAtTc Cs||j|d|kr.d|kr.t|d|d<|r:d|d<td|jj|jj|jgd}|jj ||dd}|j dkr| |d |jj|jj|j f}| ||j gkr4d |jkr| |d }| ||jd }|js| |d }| |d ddt|jD} | |d|jj|jj| f}| ||rLtd|jgd} ntdtjdjgd} |j | dfgkr| |d|jj|jj| |j f}| |dS)aJ Assert that a child of the given Page type can be created under the parent, using the supplied POST data. ``parent`` should be a Page instance, and ``child_model`` should be a Page subclass. ``data`` should be a dict that will be POSTed at the Wagtail admin Page creation method. Zslugtitleaction-publishzwagtailadmin_pages:add)argsT)followzCreating a %s.%s returned a %dformz Creating a page failed unusuallyz,Creating a page failed for an unknown reason css$|]\}}d|d|VqdS)z {}: {}z N)formatjoin).0fielderrorsrrr usz6WagtailPageTestCase.assertCanCreate..z1Validation errors found when creating a %s.%s: %sZwagtailadmin_explorewagtailadmin_pages:editpk.zQCreating a page %s.%s didn't redirect the user to the expected page %s, but to %sN)r)Zspecific_classr r r#r$r%r9clientpost status_coder"r&Zredirect_chaincontextr6r3sorteditemsrobjectsZorder_bylast) rparentrdatar(publishZadd_urlresponser0r6Z expected_urlrrrassertCanCreateHsx             z#WagtailPageTestCase.assertCanCreatecCs |jt|t||ddS)a  Test that the only page types that can be created under ``parent_model`` are ``child_models``. The list of allowed child models may differ from those set in ``Page.subpage_types``, if the child models have set ``Page.parent_page_types``. r(N) assertEqualsetr)rrZ child_modelsr(rrrassertAllowedSubpageTypess  z-WagtailPageTestCase.assertAllowedSubpageTypescCs |jt|t||ddS)a" Test that the only page types that ``child_model`` can be created under are ``parent_models``. The list of allowed parent models may differ from those set in ``Page.parent_page_types``, if the parent models have set ``Page.subpage_types``. rHN)rIrJZallowed_parent_page_models)rrZ parent_modelsr(rrrassertAllowedParentPageTypess  z0WagtailPageTestCase.assertAllowedParentPageTypes/)page route_pathr(c Cs||j}|dkr,|dd|d}|}|dkrb||dt|j||f}||dd| dD}z|j j j |j|\}}}Wn>tk r||d|t|j||d}||YnXdS)z Asserts that ``page`` can be routed to without raising a ``Http404`` error. For page types with multiple routes, you can use ``route_path`` to specify an alternate route to test. rMNzKFailed to route to "%s" for %s "%s". The page does not belong to any sites.cSsg|] }|r|qSrr)r4 componentrrr sz.zqFailed to route to "%(route_path)s" for %(page_type)s "%(page)s". A Http404 was raised for path: "%(full_path)s".)rO page_typerN full_path)get_urlrrstriplstripZget_siter"type__name__r&splitZ root_pageZ localizedZspecificZrouter) rrNrOr(pathsiteZpath_componentsr-kwargsrrrassertPageIsRoutables<    z(WagtailPageTestCase.assertPageIsRoutableF)rNrO query_data post_datauser accept_404accept_redirectr(c Csl|r|j|t||j} |dkr>| dd|d} i} |dk rfd|i} |rft|dd| d<zz.|dkr|jj | |d} n|jj | f| } WnLt k r} z.| |d|t |j|| d }||W5d} ~ XYnXW5|r|jX| jd ks6|r| jd ks6|r(| jd ks6t| tjr:dS| |d |t |j|| j| d}||dS)aE Asserts that ``page`` can be rendered without raising a fatal error. For page types with multiple routes, you can use ``route_path`` to specify an alternate route to test. When ``post_data`` is provided, the test makes a ``POST`` request with ``post_data`` in the request body. Otherwise, a ``GET`` request is made. When supplied, ``query_data`` is converted to a querystring and added to the request URL (regardless of whether ``post_data`` is provided). When ``user`` is provided, the test is conducted with them as the active user. By default, the assertion will fail if the request to the page URL results in a 301, 302 or 404 HTTP response. If you are testing a page/route where a 404 response is expected, you can use ``accept_404=True`` to indicate this, and the assertion will pass when encountering a 404. Likewise, if you are testing a page/route where a redirect response is expected, you can use `accept_redirect=True` to indicate this, and the assertion will pass when encountering 301 or 302. rMNrDT)doseq QUERYSTRINGrDzMFailed to render route "%(route_path)s" for %(page_type)s "%(page)s": %(exc)s)rOrRrNexcr/i)i-r:zFailed to render route "%(route_path)s" for %(page_type)s "%(page)s": A HTTP %(code)s response was received for path: "%(full_path)s".)rOrRrNcoderS)r; force_login AUTH_BACKENDrTrrUrVr logoutgetr< Exceptionr"rWrXr&r= isinstancerZ MagicMock) rrNrOr^r_r`rarbr(rZZ post_kwargsresperrrassertPageIsRenderablesj     z*WagtailPageTestCase.assertPageIsRenderable)rNr_r`r(c Cs|r<||sX||dt|j||d||nt|dsR|d|_|j}|j |t t dd|j id}z|j |}WnTtk r}z6|j ||dt|j||d }||W5d }~XYnX|jd kr|j ||d t|j||jd }|||d k r.|}nt|jdd}d|d<zdz|j ||WnLtk r}z,||dt|j||d }||W5d }~XYnXW5||j Xd S)a Asserts that the page edit view works for ``page`` without raising a fatal error. When ``user`` is provided, the test is conducted with them as the active user. Otherwise, a superuser is created and used for the test. After a successful ``GET`` request, a ``POST`` request is made with field data in the request body. If ``post_data`` is provided, that will be used for this purpose. If not, this data will be extracted from the ``GET`` response HTML. zeFailed to load edit view for %(page_type)s "%(page)s": User "%(user)s" have insufficient permissions.)rRrNr`_pageiseditable_superuserZassertpageiseditabler8page_idr\zFFailed to load edit view via GET for %(page_type)s "%(page)s": %(exc)s)rRrNrfNr/zqFailed to load edit view via GET for %(page_type)s "%(page)s": Received response with HTTP status code: %(code)s.)rRrNrgpage-edit-formZform_idr,zGFailed to load edit view via POST for %(page_type)s "%(page)s": %(exc)s)Zpermissions_for_userZcan_editr"rWrXr&hasattrcreate_superuserrqr;rhrir idrkrlrjr=rcontentdecodesaver<) rrNr_r`r(rZrFroZ data_to_postrrrassertPageIsEditable)sz        z(WagtailPageTestCase.assertPageIsEditablerv)rNmoder_r`r(c Cs|s t|ds|d|_|j}|j|t|dkrftdd|jid}|j|j }t |dd}td d|jid}zB|jj ||d |d } | | jd || j d d dWnXtk r} z8|j||dt|j||| d}||W5d} ~ XYnXzlz|jj|d|idWnNtk r|} z.||dt|j||| d}||W5d} ~ XYnXW5|jXdS)a Asserts that the page preview view can be loaded for ``page`` without raising a fatal error. For page types that support multiple preview modes, ``mode`` can be used to specify the preview mode to be tested. When ``user`` is provided, the test is conducted with them as the active user. Otherwise, a superuser is created and used for the test. To load the preview, the test client needs to make a ``POST`` request including all required field data in the request body. If ``post_data`` is provided, that will be used for this purpose. If not, the method will attempt to extract this data from the page edit view. _pageispreviewable_superuserZassertpageispreviewableNr8rrrsrtruz"wagtailadmin_pages:preview_on_editzmode=)rDrdr/T)Zis_validZ is_availablezQFailed to load preview for %(page_type)s "%(page)s" with mode="%(mode)s": %(exc)s)rRrNr~rfr~re)rwrxrr;rhrir ryrkrzr{rr<rIr=ZassertJSONEqualrlrjr"rWrXr&) rrNr~r_r`r(Z edit_pathhtmlZ preview_pathrFrorrrassertPageIsPreviewablezsl     z+WagtailPageTestCase.assertPageIsPreviewable)N)N)NT)N)N)rMN)rMNNNFFN)NNN)rvNNN)rX __module__ __qualname____doc__ classmethodrr r)r*rGrKrLrrstrr]rrrboolrpr}r __classcell__rrrrrsj   L  - T TrcseZdZfddZZS)WagtailPageTestscst|dSr)rsetUplogin)rrrrrs zWagtailPageTests.setUp)rXrrrrrrrrrsrN) typingrrrZunittestrZ django.confrZdjango.contrib.auth.base_userrZ django.httprZ django.testr Z django.urlsr Zdjango.utils.httpr Zdjango.utils.textr Zwagtail.coreutilsr Zwagtail.modelsrZ form_datarZ wagtail_testsrZAUTHENTICATION_BACKENDSrirrrrrrs$             4