a D g 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)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__\/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/test/utils/page_tests.pyrs zWagtailPageTestCase.setUpClasscCs ||vSr)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|vr.d|vr.t|d|d<|r:d|d<td|jj|jj|jgd}|jj ||dd}|j dkr| |d |jj|jj|j f}| ||j gkr4d |jvr| |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$r8clientpost status_coder!r%Zredirect_chaincontextr5r2sorteditemsr ZobjectsZorder_bylast) rparentrdatar'publishZadd_urlresponser/r5Z expected_urlrrrassertCanCreateHst             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``. rFN)rGrHZallowed_parent_page_models)rrZ parent_modelsr'rrrassertAllowedParentPageTypess z0WagtailPageTestCase.assertAllowedParentPageTypes/)page route_pathr'c Cs||j}|dkr,|dd|d}|}|durb||dt|j||f}||dd| dD}z|j j j |j|\}}}Wn<ty||d|t|j||d}||Yn0dS)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. rKNzKFailed to route to "%s" for %s "%s". The page does not belong to any sites.cSsg|] }|r|qSrr)r3 componentrrr z.zqFailed to route to "%(route_path)s" for %(page_type)s "%(page)s". A Http404 was raised for path: "%(full_path)s".)rM page_typerL full_path)get_urlrrstriplstripZget_siter!type__name__r%splitZ root_pageZ localizedZspecificZrouter) rrLrMr'pathsiteZpath_componentsr,kwargsrrrassertPageIsRoutables:     z(WagtailPageTestCase.assertPageIsRoutableF)rLrM query_data post_datauser accept_404accept_redirectr'c Cs|r|j|t||j} |dkr>| dd|d} i} |durfd|i} |rft|dd| d<zz2|dur|jj| |d} n|jj | fi| } WnNt y} z6| |d|t |j || d }||WYd} ~ n d} ~ 00W|r|jn|r |j0| jd ksL|r,| jd ksL|r>| jd vsLt| tjrPdS| |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. rKNrBT)doseq QUERYSTRINGrBzMFailed to render route "%(route_path)s" for %(page_type)s "%(page)s": %(exc)s)rMrQrLexcr.i)i-r9zFailed to render route "%(route_path)s" for %(page_type)s "%(page)s": A HTTP %(code)s response was received for path: "%(full_path)s".)rMrQrLcoderR)r: force_login AUTH_BACKENDrSrrTrUr getr; Exceptionr!rVrWr%logoutr< isinstancerZ MagicMock) rrLrMr]r^r_r`rar'rYZ post_kwargsresperrrassertPageIsRenderablesn  "    z*WagtailPageTestCase.assertPageIsRenderable)rLr^r_r'c Cs|r<||sX||dt|j||d||nt|dsR|d|_|j}|j |t t dd|j id}z|j |}WnVty}z>|j ||dt|j||d }||WYd }~n d }~00|jd kr |j ||d t|j||jd }|||d ur0|}nt|jdd}d|d<zxz|j ||WnNty}z4||dt|j||d }||WYd }~n d }~00W||j n||j 0d 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.)rQrLr__pageiseditable_superuserZassertpageiseditabler7page_idr[zFFailed to load edit view via GET for %(page_type)s "%(page)s": %(exc)s)rQrLreNr.zqFailed to load edit view via GET for %(page_type)s "%(page)s": Received response with HTTP status code: %(code)s.)rQrLrfpage-edit-formZform_idr+zGFailed to load edit view via POST for %(page_type)s "%(page)s": %(exc)s)Zpermissions_for_userZcan_editr!rVrWr%hasattrcreate_superuserrpr:rgrhr idrirjrkr<rcontentdecoder;save) rrLr^r_r'rYrDrnZ data_to_postrrrassertPageIsEditable)s|         " z(WagtailPageTestCase.assertPageIsEditableru)rLmoder^r_r'c Cs|s t|ds|d|_|j}|j|t|durftdd|jid}|j|j }t |dd}td d|jid}zB|jj ||d |d } | | jd || j d d dWnZty} z@|j||dt|j||| d}||WYd} ~ n d} ~ 00zxz|jj|d|idWnPty} z6||dt|j||| d}||WYd} ~ n d} ~ 00W|jn |j0dS)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_superuserZassertpageispreviewableNr7rqrrrsrtz"wagtailadmin_pages:preview_on_editzmode=)rBrcr.T)Zis_validZ is_availablezQFailed to load preview for %(page_type)s "%(page)s" with mode="%(mode)s": %(exc)s)rQrLr}rer}rd)rvrwr~r:rgrhr rxriryrzrr;rGr<ZassertJSONEqualrjrkr!rVrWr%) rrLr}r^r_r'Z edit_pathhtmlZ preview_pathrDrnrrrassertPageIsPreviewablezsf       "z+WagtailPageTestCase.assertPageIsPreviewable)N)N)NT)N)N)rKN)rKNNNFFN)NNN)ruNNN)rW __module__ __qualname____doc__ classmethodrrr(r)rErIrJr rstrr\dictrrboolror|r __classcell__rrrrrsj   L  - T TrcseZdZfddZZS)WagtailPageTestscst|dSr)rsetUplogin)rrrrrs zWagtailPageTests.setUp)rWrrrrrrrrrsrN)typingrrZunittestrZ django.confrZdjango.contrib.auth.base_userrZ django.httprZ django.testrZ django.urlsr Zdjango.utils.httpr Zdjango.utils.textr Zwagtail.coreutilsr Zwagtail.modelsr Z form_datarZ wagtail_testsrZAUTHENTICATION_BACKENDSrhrrrrrrs$             4