o hE@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/url_pathz)rexpand_db_attributesr objectsgetr assertEqualselfresultr"\/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/tests/test_rich_text.pytest_expand_db_attributessz1TestPageLinktypeHandler.test_expand_db_attributescCtddi}||ddS)Nrrrrrrr"r"r#-test_expand_db_attributes_page_does_not_existzETestPageLinktypeHandler.test_expand_db_attributes_page_does_not_existcCr%)Nrzr'rr"r"r#(test_expand_db_attributes_not_for_editorr)z@TestPageLinktypeHandler.test_expand_db_attributes_not_for_editorN)__name__ __module__ __qualname__fixturesr$r(r+r"r"r"r#rs  rT)enEnglish)frFrenchzwagtail.test.urls_multilang)WAGTAIL_I18N_ENABLEDWAGTAIL_CONTENT_LANGUAGES ROOT_URLCONFc@2eZdZdgZddZddZddZdd Zd S) TestPageLinktypeHandlerWithI18NrcCs^tjjdd|_tjjdd|_|jj|jdd|_d|j_ |jj dgd |j dS) Nr2) language_coderrT) copy_parentsnoelslug) update_fields) rrcreate fr_localer r event_pagecopy_for_translation fr_event_pager<save save_revisionpublishr r"r"r#setUp.sz%TestPageLinktypeHandlerWithI18N.setUpcCs"td|jji}||ddS)Nr )rrr@rrrr"r"r#r$8sz9TestPageLinktypeHandlerWithI18N.test_expand_db_attributescCsNtdtd|jji}||dWddS1s wYdS)Nr2rz)roverriderrr@rrrr"r"r#'test_expand_db_attributes_autolocalizes<s "zGTestPageLinktypeHandlerWithI18N.test_expand_db_attributes_autolocalizescCsb|j|jtdtd|jji}| |dWddS1s*wYdS)Nr2rrH) rB unpublishrCrrIrrr@rrrr"r"r#>test_expand_db_attributes_doesnt_autolocalize_unpublished_pageCs   "z^TestPageLinktypeHandlerWithI18N.test_expand_db_attributes_doesnt_autolocalize_unpublished_pageN)r,r-r.r/rGr$rJrLr"r"r"r#r8#s    r8c@eZdZddZdS)TestExtractAttrscCs"d}t|}||ddddS)Nz#snowmanbarquux)foobaz)rrr htmlr!r"r"r#test_extract_attrNsz"TestExtractAttrs.test_extract_attrN)r,r-r.rUr"r"r"r#rNM rNc@seZdZdgZddZddZddZdd Zd d Zd d Z e dddZ e dddiidddZ ddZe dddiidddZddZdS) TestExpandDbHtmlrcCsd}t|}||ddS)Nzfoor rrSr"r"r#test_expand_db_html_no_linktypeWz0TestExpandDbHtml.test_expand_db_html_no_linktypecCd}t|}||ddS)Nz$foo foorXrSr"r"r#'test_invalid_linktype_set_to_empty_link\rZz8TestExpandDbHtml.test_invalid_linktype_set_to_empty_linkcCr[)Nz%fooz'foorXrSr"r"r#!test_valid_linktype_and_referencearZz2TestExpandDbHtml.test_valid_linktype_and_referencecCr[)Nz(foor\rXrSr"r"r#7test_valid_linktype_invalid_reference_set_to_empty_linkfrZzHTestExpandDbHtml.test_valid_linktype_invalid_reference_set_to_empty_linkcC|tdddS)Nzrr rFr"r"r#test_no_embedtype_remove_tagkz-TestExpandDbHtml.test_no_embedtype_remove_tagcCr`)Nz$rarbrFr"r"r#!test_invalid_embedtype_remove_tagnrdz2TestExpandDbHtml.test_invalid_embedtype_remove_tagzwagtail.embeds.embeds.get_embedcCs4ddlm}|dd|_d}t|}|d|dS)Nr)Embedz test htmlrTz>)wagtail.embeds.modelsrf return_valuer assertIn)r get_embedrfrTr!r"r"r#test_expand_db_html_with_embedqs  z/TestExpandDbHtml.test_expand_db_html_with_embeddefaultBACKENDz-django.core.cache.backends.locmem.LocMemCache)CACHEScCs@t|d tdWddS1swYdS)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 clear_site_root_paths_cacheassertNumQueriesr rFr"r"r#*test_expand_db_html_database_queries_pages|s  "z;TestExpandDbHtml.test_expand_db_html_database_queries_pagescCs:|dtddWddS1swYdS)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. rgrrr rFr"r"r#.test_expand_db_html_database_queries_documentss "z?TestExpandDbHtml.test_expand_db_html_database_queries_documents renditionsz+django.core.cache.backends.dummy.DummyCachecCs8|d tdWddS1swYdS)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: rtrFr"r"r#+test_expand_db_html_database_queries_imagess "zfoobarz@foobarznpagedocumentpagezbpagedocumentpagerbrFr"r"r#$test_expand_db_html_mixed_link_typessz5TestExpandDbHtml.test_expand_db_html_mixed_link_typesN)r,r-r.r/rYr]r^r_rcrerrlrrsrurxryr"r"r"r#rWTs4     rWc@sBeZdZdgZddZddZddZdd Zd d Zd d Z dS)TestRichTextValuercCstd}||jddSNrar rsourcer valuer"r"r#test_construct_with_nonez*TestRichTextValue.test_construct_with_nonecCtd}||jddSr{r|r~r"r"r# test_construct_with_empty_stringrz2TestRichTextValue.test_construct_with_empty_stringcCr)Nz

hello world

r|r~r"r"r##test_construct_with_nonempty_stringrz5TestRichTextValue.test_construct_with_nonempty_stringcCs td}t|}||ddS)Nz5

Merry Christmas!

z8

Merry Christmas!

)r strr)r rr!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)Nrz

django

i90)r assertNotEqualr)r value1value2value3r"r"r#test_compare_values   z$TestRichTextValue.test_compare_valueN) r,r-r.r/rrrrrrr"r"r"r#rzs rzc@eZdZddZddZdS)TestFeatureRegistrycCs&t}|dd}||jdgdS)Ndraftail quotationz testapp/js/draftail-quotation.js)rget_editor_pluginrjs)r featuresrr"r"r#%test_register_rich_text_features_hooks z9TestFeatureRegistry.test_register_rich_text_features_hookcCs.t}||dd||dddS)Nmade_up_editor blockquotermade_up_feature)r assertIsNoner)r rr"r"r#'test_missing_editor_plugin_returns_nonesz;TestFeatureRegistry.test_missing_editor_plugin_returns_noneN)r,r-r.rrr"r"r"r#rs rc@r)TestLinkRewriterTagReplacingc Csdddi}t|}|d}||d|d}||d|d}||d|d}||d|d }||d |d }||d ||d ||d ddS)NpagecSd|dSNzrformatattrsr"r"r#zSTestLinkRewriterTagReplacing.test_should_follow_default_behaviour.."z#z0zr&z9z4)rrr) r rulesrewriterpage_type_linkexternal_type_linkemail_type_linkanchor_type_linklink_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)NcSrrrrr"r"r#r4rz\TestLinkRewriterTagReplacing.test_supported_type_should_follow_given_rules..cSr)Nzhrefrrr"r"r#r5cSr)Nzrrrr"r"r#r8rcSr)Nz rrrr"r"r#r;rcSr)Nzrrrr"r"r#r>r)rexternalemailanchorcustomrrrz.zz-rz4rz#zz/z/)rr) r rrrrexternal_type_link_httprrlink_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#r s #rc@r7) TestRichTextFieldrcCNtjjdd}d|_||jd}||}||}| |dgdS)NrrzX

Merry Christmas from Wagtail! & co.

bodyz#Merry Christmas from Wagtail! & co. rrrrrD_meta get_fieldvalue_from_objectget_searchable_contentrr christmas_page body_fieldrr!r"r"r#test_get_searchable_contenth   z-TestRichTextField.test_get_searchable_contentcCr)Nrrz0

buttery
mashed

potatoes

rzbuttery mashed potatoesrrr"r"r#&test_get_searchable_content_whitespacerrz8TestRichTextField.test_get_searchable_content_whitespacecCslttdgd}|dtdi}|||dtdi}|||dtdi}||dS)Nintro)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

)rrrris_validr)r EventIndexFormformr"r"r#test_max_length_validation|s& z,TestRichTextField.test_max_length_validationcCs&|ttdtdddfgdS)Nz6Link to an internal page1ra)rlistrextract_referencesr rFr"r"r#test_extract_referencess z)TestRichTextField.test_extract_referencesN)r,r-r.r/rrrrr"r"r"r#res   rc@rM)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

👨‍👨‍👧

rpN)r rclean)r validatorr"r"r#test_count_characterssz4TestRichTextMaxLengthValidator.test_count_charactersN)r,r-r.rr"r"r"r#rrVrN)( unittest.mockrdjango.forms.modelsr django.testrr django.utilsrwagtail.fieldsrwagtail.modelsrr r wagtail.rich_textr r r "wagtail.rich_text.feature_registryrwagtail.rich_text.pagesrwagtail.rich_text.rewritersrrwagtail.test.testapp.modelsrrwagtail.test.utils.form_datarrr8rNrWrzrrrrr"r"r"r#s:      "&X<