U {h@sddlmZddlmZmZddlmZddlmZddl m Z ddl m Z ddlm ZddlmZddlmZdd lmZGd d d eeZGd d d eZGdddeZdS))patch)TestCaseoverride_settings) reverse_lazy)EmbedNotFoundException)Embed)MediaEmbedHandler)expand_db_html)FeatureRegistry)WagtailTestUtilsc@s4eZdZddZedddZedddZdS) TestEditorHtmlMediaEmbedHandlercCs.|d}|j}t|}||ddidS)Nzfoourlztest-url)Zget_soupbEditorHtmlMediaEmbedHandlerZget_db_attributes assertEqual)selfZsouptagresultrk/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/embeds/tests/test_rich_text.pytest_get_db_attributess  z6TestEditorHtmlMediaEmbedHandler.test_get_db_attributeswagtail.embeds.embeds.get_embedc Csxtddddddddddd |_td di}|d ||d ||d ||d||d||d|dS)Nhttp://www.youtube.com/watch/video test html test titletest author nametest provider namehttp://test/thumbnail.url r max_widthtypehtmltitleZ author_nameZ provider_nameZ thumbnail_urlwidthheightr zw
z

test title

z)

URL: http://www.youtube.com/watch/

z#

Provider: test provider name

z

Author: test author name

z6test title)r return_valuerexpand_db_attributesassertInrZ get_embedrrrr$test_expand_db_attributes_for_editors4     zDTestEditorHtmlMediaEmbedHandler.test_expand_db_attributes_for_editorcCs$t|_tddi}||ddSNr r)r side_effectrr(rr*rrrtest_expand_db_attributes_for_frontend_catches_embed_not_foundtsz\TestFrontendMediaEmbedHandler.test_expand_db_attributes_for_frontend_catches_embed_not_foundc CsDtddddddddddd |_td }|d||d dddS) Nrrrrrrrrr ze

1 2 3 4

z0https://www.youtube.com/watch?v=O7D-1RG-VRk&t=25)rr'r r)Zassert_called_withr*rrr$test_expand_html_escaping_end_to_end~s*  zBTestFrontendMediaEmbedHandler.test_expand_html_escaping_end_to_endN) r0r1r2rrr6r7r8r9rrrrr3Ds   r3c@seZdZddZdS)TestEntityFeatureChooserUrlscCsBt}|dd}||jd||jddtddS)NZdraftailembedZ chooserUrlsZ embedsChooserzwagtailembeds:chooser)r Zget_editor_pluginZassertIsNotNonedatagetrr)rfeaturesr;rrrtest_chooser_urls_exists  z4TestEntityFeatureChooserUrls.test_chooser_urls_existN)r0r1r2r?rrrrr:sr:N)Z unittest.mockrZ django.testrrZ django.urlsrZwagtail.embeds.exceptionsrZwagtail.embeds.modelsrZwagtail.embeds.rich_textrr5Z$wagtail.embeds.rich_text.editor_htmlrZwagtail.rich_textr Z"wagtail.rich_text.feature_registryr Zwagtail.test.utilsr r r3r:rrrrs         3R