o h F@sddlmZmZddlmZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZd d lmZd dlmZejdZGdddee ZGdddeZdS))AnyOptional)mock)settings)AbstractBaseUser)Http404)TestCase)reverse) urlencode)slugify)get_dummy_request)Page)querydict_from_html)WagtailTestUtilscs^eZdZdZefddZddZd&ddZd&d d Zd'd d Z d&ddZ d&ddZ  d(de de ede efddZ       d)de de ede eeefde eeefde ede ede ede efddZ   d*de de eeefde ede efd d!Z "   d+de d#e ede eeefde ede ef d$d%ZZS),WagtailPageTestCasezO A set of assertions to help write tests for custom Wagtail page types cstt|_dSN)super setUpClassr dummy_request)cls __class__]/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/test/utils/page_tests.pyrs  zWagtailPageTestCase.setUpClasscCs ||vSr)allowed_subpage_models)self parent_model child_modelrrr_testCanCreateAts z$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"  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&rrrassertCanNotCreateAt5r)z(WagtailPageTestCase.assertCanNotCreateAtTc Cs||j|d|vrd|vrt|d|d<|rd|d<td|jj|jj|jgd}|jj ||dd}|j dkrN| |d |jj|jj|j f}| ||j gkrd |jvrc| |d }| ||jd }|jsv| |d }| |d ddt|jD} | |d|jj|jj| f}| ||rtd|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. slugtitleaction-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 us  z6WagtailPageTestCase.assertCanCreate..z1Validation errors found when creating a %s.%s: %swagtailadmin_explorewagtailadmin_pages:editpk.zQCreating a page %s.%s didn't redirect the user to the expected page %s, but to %sN)r(specific_classr r r"r#r$r;clientpost status_coder!r%redirect_chaincontextr7r4sorteditemsr objectsorder_bylast) rparentrdatar'publishadd_urlresponser1r7 expected_urlrrrassertCanCreateHsv            z#WagtailPageTestCase.assertCanCreatecC |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)rr child_modelsr'rrrassertAllowedSubpageTypes  z-WagtailPageTestCase.assertAllowedSubpageTypescCrO)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``. rPN)rQrRallowed_parent_page_models)rr parent_modelsr'rrrassertAllowedParentPageTypesrUz0WagtailPageTestCase.assertAllowedParentPageTypes/page route_pathr'c Cs||j}|dkr|dd|d}|}|dur1||dt|j||f}||dd| dD}z|j j j |j|\}}}WdStyg||d|t|j||d}||w)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. rYNzKFailed to route to "%s" for %s "%s". The page does not belong to any sites.cSsg|]}|r|qSrr)r5 componentrrr sz.zqFailed to route to "%(route_path)s" for %(page_type)s "%(page)s". A Http404 was raised for path: "%(full_path)s".)r[ page_typerZ full_path)get_urlrrstriplstripget_siter!type__name__r%split root_page localizedspecificrouter) rrZr[r'pathsitepath_componentsr.kwargsrrrassertPageIsRoutables<     z(WagtailPageTestCase.assertPageIsRoutableF query_data post_datauser accept_404accept_redirectc Csf|r |j|t||j} |dkr| dd|d} i} |dur3d|i} |r3t|dd| d<zDz|durB|jj| |d} n |jj | fi| } Wn!t yn} z| |d|t |j || d }||d} ~ wwW|rw|jn |r|jww| jd ks|r| jd ks|r| jd vst| tjrdS| |d |t |j || j| d}||)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. rYNrIT)doseq QUERYSTRINGrIzMFailed to render route "%(route_path)s" for %(page_type)s "%(page)s": %(exc)s)r[r^rZexcr0i)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".)r[r^rZcoder_)r> force_login AUTH_BACKENDr`rrarbr getr? Exceptionr!rdrer%logoutr@ isinstancer MagicMock) rrZr[rprqrrrsrtr'rk post_kwargsresperrrassertPageIsRenderablest         z*WagtailPageTestCase.assertPageIsRenderablec Cs|r||s||dt|j||d||nt|ds)|d|_|j}|j |t t dd|j id}z|j |}Wn%tyi}z|j ||dt|j||d }||d }~ww|jd kr|j ||d t|j||jd }|||d ur|}n t|jdd}d|d<z6z |j ||Wn ty}z||dt|j||d }||d }~wwW||j d S||j w)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.)r^rZrr_pageiseditable_superuserassertpageiseditabler:page_idrnzFFailed to load edit view via GET for %(page_type)s "%(page)s": %(exc)s)r^rZrxNr0zqFailed to load edit view via GET for %(page_type)s "%(page)s": Received response with HTTP status code: %(code)s.)r^rZrypage-edit-formform_idr-zGFailed to load edit view via POST for %(page_type)s "%(page)s": %(exc)s)permissions_for_usercan_editr!rdrer%hasattrcreate_superuserrr>rzr{r idr|r}r~r@rcontentdecoder?save) rrZrqrrr'rkrLr data_to_postrrrassertPageIsEditable)s           z(WagtailPageTestCase.assertPageIsEditablermodec Cs|st|ds |d|_|j}|j|t|dur3tdd|jid}|j|j }t |dd}td d|jid}z!|jj ||d |d } | | jd || j d d dWn&ty} z|j||dt|j||| d}||d} ~ wwz6z |jj|d|idWn!ty} z||dt|j||| d}||d} ~ wwW|jdS|jw)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_superuserassertpageispreviewableNr:rrrrz"wagtailadmin_pages:preview_on_editzmode=)rIrvr0T)is_valid is_availablezQFailed to load preview for %(page_type)s "%(page)s" with mode="%(mode)s": %(exc)s)r^rZrrxrrw)rrrr>rzr{r rr|rrrr?rQr@assertJSONEqualr}r~r!rdrer%) rrZrrqrrr' edit_pathhtml preview_pathrLrrrrassertPageIsPreviewablezsp       z+WagtailPageTestCase.assertPageIsPreviewabler)NT)rYN)rYNNNFFN)NNN)rNNN)re __module__ __qualname____doc__ classmethodrrr(r*rNrTrXr rstrrodictrrboolrrr __classcell__rrrrrs    L  - T TrcseZdZfddZZS)WagtailPageTestscst|dSr)rsetUplogin)rrrrrs  zWagtailPageTests.setUp)rerrrrrrrrrsrN)typingrrunittestr django.confrdjango.contrib.auth.base_userr django.httpr django.testr django.urlsr django.utils.httpr django.utils.textr wagtail.coreutilsr wagtail.modelsr form_datar wagtail_testsrAUTHENTICATION_BACKENDSr{rrrrrrs&             4