a D gE@sRddlmZddlmZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZmZmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZGd ddeZedddgddGdddeZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%Gd d!d!eZ&Gd"d#d#eZ'd$S)%)patch)modelform_factory)TestCaseoverride_settings) translation) RichTextField)LocalePageSite)RichTextRichTextMaxLengthValidatorexpand_db_html)FeatureRegistry)PageLinkHandler) LinkRewriter extract_attrs) EventIndex EventPage) rich_textc@s*eZdZdgZddZddZddZdS) TestPageLinktypeHandler test.jsoncCs*tdtjjddji}||ddS)Nid/home/events/christmas/Zurl_pathz)rexpand_db_attributesr objectsgetr assertEqualselfresultr![/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/tests/test_rich_text.pytest_expand_db_attributessz1TestPageLinktypeHandler.test_expand_db_attributescCstddi}||ddS)Nrrrrrrr!r!r"-test_expand_db_attributes_page_does_not_existszETestPageLinktypeHandler.test_expand_db_attributes_page_does_not_existcCstddi}||ddS)Nrzr%rr!r!r"(test_expand_db_attributes_not_for_editorsz@TestPageLinktypeHandler.test_expand_db_attributes_not_for_editorN)__name__ __module__ __qualname__fixturesr#r&r(r!r!r!r"rsrT)enZEnglish)frZFrenchzwagtail.test.urls_multilang)ZWAGTAIL_I18N_ENABLEDZWAGTAIL_CONTENT_LANGUAGESZ ROOT_URLCONFc@s2eZdZdgZddZddZddZdd Zd S) TestPageLinktypeHandlerWithI18NrcCs^tjjdd|_tjjdd|_|jj|jdd|_d|j_ |jj dgd |j dS) Nr.)Z language_coderrT)Z copy_parentsZnoelslug)Z update_fields) rrcreateZ fr_localer r event_pageZcopy_for_translation fr_event_pager0save save_revisionpublishrr!r!r"setUp.sz%TestPageLinktypeHandlerWithI18N.setUpcCs"td|jji}||ddS)Nr )rrr2rrrr!r!r"r#8sz9TestPageLinktypeHandlerWithI18N.test_expand_db_attributescCsLtd.td|jji}||dWdn1s>0YdS)Nr.rz)roverriderrr2rrrr!r!r"'test_expand_db_attributes_autolocalizes<s zGTestPageLinktypeHandlerWithI18N.test_expand_db_attributes_autolocalizescCs`|j|jtd.td|jji}| |dWdn1sR0YdS)Nr.rr9) r3Z unpublishr4rr:rrr2rrrr!r!r">test_expand_db_attributes_doesnt_autolocalize_unpublished_pageCs    z^TestPageLinktypeHandlerWithI18N.test_expand_db_attributes_doesnt_autolocalize_unpublished_pageN)r)r*r+r,r8r#r;r<r!r!r!r"r/#s   r/c@seZdZddZdS)TestExtractAttrscCs"d}t|}||ddddS)Nz#snowmanbarZquux)ZfooZbaz)rrrhtmlr r!r!r"test_extract_attrNsz"TestExtractAttrs.test_extract_attrN)r)r*r+rAr!r!r!r"r=Msr=c@seZdZdgZddZddZddZdd Zd d Zd d Z e dddZ e dddiidddZ ddZe dddiidddZddZdS) TestExpandDbHtmlrcCsd}t|}||ddS)Nzfoor rr?r!r!r"test_expand_db_html_no_linktypeWsz0TestExpandDbHtml.test_expand_db_html_no_linktypecCsd}t|}||ddS)Nz$foo foorCr?r!r!r"'test_invalid_linktype_set_to_empty_link\sz8TestExpandDbHtml.test_invalid_linktype_set_to_empty_linkcCsd}t|}||ddS)Nz%fooz'foorCr?r!r!r"!test_valid_linktype_and_referenceasz2TestExpandDbHtml.test_valid_linktype_and_referencecCsd}t|}||ddS)Nz(foorErCr?r!r!r"7test_valid_linktype_invalid_reference_set_to_empty_linkfszHTestExpandDbHtml.test_valid_linktype_invalid_reference_set_to_empty_linkcCs|tdddS)Nzrr r7r!r!r"test_no_embedtype_remove_tagksz-TestExpandDbHtml.test_no_embedtype_remove_tagcCs|tdddS)Nz$rIrJr7r!r!r"!test_invalid_embedtype_remove_tagnsz2TestExpandDbHtml.test_invalid_embedtype_remove_tagzwagtail.embeds.embeds.get_embedcCs4ddlm}|dd|_d}t|}|d|dS)Nr)Embedz test htmlr@z>)Zwagtail.embeds.modelsrMZ return_valuer ZassertIn)rZ get_embedrMr@r r!r!r"test_expand_db_html_with_embedqs   z/TestExpandDbHtml.test_expand_db_html_with_embeddefaultZBACKENDz-django.core.cache.backends.locmem.LocMemCache)ZCACHEScCs>t|dtdWdn1s00YdS)Na} This rich text has 8 page links, and this test verifies that the code uses the minimal number of database queries (5) to expand them. All of these pages should be retrieved with 4 queries, one to do the base Page table lookup and then 1 each for the EventIndex, EventPage, and SimplePage tables. This links to an EventIndex page. This links to an EventPage page. This links to an EventPage page. This links to an EventPage page. This links to an EventPage page. This links to an EventPage page. This links to a SimplePage page. This links to a SimplePage page. Finally there's one additional query needed to do the Site root paths lookup. )r Zclear_site_root_paths_cacheassertNumQueriesr r7r!r!r"*test_expand_db_html_database_queries_pages|s  z;TestExpandDbHtml.test_expand_db_html_database_queries_pagescCs8|dtddWdn1s*0YdS)Nr'aE This rich text has 2 document links, and this test verifies that the code uses the minimal number of database queries (1) to expand them. Both of these documents should be retrieved with 1 query: This links to a document. This links to another document. rNrRr r7r!r!r".test_expand_db_html_database_queries_documentss z?TestExpandDbHtml.test_expand_db_html_database_queries_documentsZ renditionsz+django.core.cache.backends.dummy.DummyCachecCs6|dtdWdn1s(0YdS)Na This rich text has 2 image links, and this test verifies that the code uses the minimal number of database queries (3) to expand them. Both of these images should be retrieved with 3 queries, one to fetch the image objects in bulk and then one per image to fetch their renditions: This is an image: This is another image: rTr7r!r!r"+test_expand_db_html_database_queries_imagess zfoobarz@foobarznpagedocumentpagezbpagedocumentpagerJr7r!r!r"$test_expand_db_html_mixed_link_typessz5TestExpandDbHtml.test_expand_db_html_mixed_link_typesN)r)r*r+r,rDrFrGrHrKrLrrOrrSrUrWrXr!r!r!r"rBTs2   rBc@sBeZdZdgZddZddZddZdd Zd d Zd d Z dS)TestRichTextValuercCstd}||jddSNrIr rsourcervaluer!r!r"test_construct_with_nonesz*TestRichTextValue.test_construct_with_nonecCstd}||jddSrZr[r]r!r!r" test_construct_with_empty_stringsz2TestRichTextValue.test_construct_with_empty_stringcCstd}||jddS)Nz

hello world

r[r]r!r!r"#test_construct_with_nonempty_stringsz5TestRichTextValue.test_construct_with_nonempty_stringcCs td}t|}||ddS)Nz5

Merry Christmas!

z8

Merry Christmas!

)r strr)rr^r r!r!r" test_renders zTestRichTextValue.test_rendercCs(td}||td}||dS)N

wagtail

)r assertFalse assertTruer]r!r!r"test_evaluate_values z%TestRichTextValue.test_evaluate_valuecCs@td}td}td}|||||d|||dS)Nrdz

django

i90)r assertNotEqualr)rZvalue1Zvalue2Zvalue3r!r!r"test_compare_values   z$TestRichTextValue.test_compare_valueN) r)r*r+r,r_r`rarcrgrir!r!r!r"rYsrYc@seZdZddZddZdS)TestFeatureRegistrycCs&t}|dd}||jdgdS)Ndraftail quotationz testapp/js/draftail-quotation.js)rget_editor_pluginrZjs)rfeaturesrlr!r!r"%test_register_rich_text_features_hooks z9TestFeatureRegistry.test_register_rich_text_features_hookcCs.t}||dd||dddS)NZmade_up_editor blockquoterkZmade_up_feature)rZ assertIsNonerm)rrnr!r!r"'test_missing_editor_plugin_returns_nonesz;TestFeatureRegistry.test_missing_editor_plugin_returns_noneN)r)r*r+rorqr!r!r!r"rjsrjc@seZdZddZddZdS)TestLinkRewriterTagReplacingc Csdddi}t|}|d}||d|d}||d|d}||d|d}||d|d }||d |d }||d ||d ||d ddS)NpagecSsd|dSNzrformatattrsr!r!r"zSTestLinkRewriterTagReplacing.test_should_follow_default_behaviour.."z#z0zr$z9z4)rrrh) rrulesrewriterpage_type_linkexternal_type_linkemail_type_linkanchor_type_linkZlink_without_linktypelink_with_custom_linktyper!r!r"$test_should_follow_default_behaviours*        zATestLinkRewriterTagReplacing.test_should_follow_default_behaviourc Csddddddddddd}t|}|d}||d |d }||d |d }||d |d}||d|d}||d|d}||d|d} || ddS)NcSsd|dSrtrurwr!r!r"ry4rzz\TestLinkRewriterTagReplacing.test_supported_type_should_follow_given_rules..cSsd|dS)Nzhrefrurwr!r!r"ry5scSsd|dS)Nzrrurwr!r!r"ry8scSsd|dS)Nz rrurwr!r!r"ry;scSsd|dS)Nzrrurwr!r!r"ry>s)rsZexternalemailanchorZcustomr{r|r}z.zz-r~z4rz#zz/z/)rr) rrrrrZexternal_type_link_httprrZlink_with_no_linktyperr!r!r"-test_supported_type_should_follow_given_rules1s>   zJTestLinkRewriterTagReplacing.test_supported_type_should_follow_given_rulesN)r)r*r+rrr!r!r!r"rr s#rrc@s2eZdZdgZddZddZddZdd Zd S) TestRichTextFieldrcCsNtjjdd}d|_||jd}||}||}| |dgdS)NrrzX

Merry Christmas from Wagtail! & co.

bodyz#Merry Christmas from Wagtail! & co. rrrrr5Z_meta get_fieldZvalue_from_objectZget_searchable_contentrrZchristmas_pageZ body_fieldr^r r!r!r"test_get_searchable_contenths   z-TestRichTextField.test_get_searchable_contentcCsNtjjdd}d|_||jd}||}||}| |dgdS)Nrrz0

buttery
mashed

potatoes

rzbuttery mashed potatoesrrr!r!r"&test_get_searchable_content_whitespacers   z8TestRichTextField.test_get_searchable_content_whitespacecCslttdgd}|dtdi}|||dtdi}|||dtdi}||dS)NZintro)modelfieldsz%

less than 50 characters

z]

a piece of text that is considerably longer than the limit of fifty characters of text

z

less than 50 characters

)rrrrfZis_validre)rZEventIndexFormformr!r!r"test_max_length_validation|s& z,TestRichTextField.test_max_length_validationcCs&|ttdtdddfgdS)Nz6Link to an internal page1rI)rlistrZextract_referencesr r7r!r!r"test_extract_referencess z)TestRichTextField.test_extract_referencesN)r)r*r+r,rrrrr!r!r!r"res   rc@seZdZddZdS)TestRichTextMaxLengthValidatorcCstd}||dd||dd||dd||dd ||d d ||d d ||d d||dd||dddS)z=Keep those tests up-to-date with MaxLength tests client-side.2z

Plain text

z

There's quote

z

Line
break

z

Multi

blocks

z

Empty

blocks

z

With

HR

z

With

embed

u

U+2764 U+FE0F ❤️

u

👨‍👨‍👧

rQN)r rclean)rZ validatorr!r!r"test_count_characterssz4TestRichTextMaxLengthValidator.test_count_charactersN)r)r*r+rr!r!r!r"rsrN)(Z unittest.mockrZdjango.forms.modelsrZ django.testrrZ django.utilsrZwagtail.fieldsrZwagtail.modelsrr r Zwagtail.rich_textr r r Z"wagtail.rich_text.feature_registryrZwagtail.rich_text.pagesrZwagtail.rich_text.rewritersrrZwagtail.test.testapp.modelsrrZwagtail.test.utils.form_datarrr/r=rBrYrjrrrrr!r!r!r"s8       "&X<