U |hB@sddlZddlmZddlmZddlmZddlmZm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZmZddlmZmZdd lmZmZmZm Z Gd d d eeZ!Gd d d eZ"GdddeZ#GdddeeZ$dS)N)SimpleUploadedFile)TestCase)content_state_equal)PAGE_MODEL_CLASSESPageSite)DummyExternalStorage) BusinessChild BusinessIndexBusinessNowherePageBusinessSubIndex EventIndex EventPageSectionedRichTextPageSimpleChildPage SimplePageSimpleParentPage StreamPage)WagtailPageTestsWagtailTestUtils)inline_formsetnested_form_data rich_text streamfieldc@s$eZdZddZddZddZdS)TestAssertTagInHTMLc Csd}|d||jd|dd|t|d|W5QRX|t|jd|ddW5QRX|t|d|W5QRX|t|d|W5QRXdS) Nz'
  • z
  • countz(
    z
  • z9
  • ZassertTagInHTML assertRaisesAssertionErrorselfZhaystackr$`/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/tests/test_tests.pytest_assert_tag_in_html s$     z+TestAssertTagInHTML.test_assert_tag_in_htmlc Csxd}|jd|dd|jd|ddd|t|jd|ddW5QRX|t|jd|d ddW5QRXdS) Nrz
  • T)allow_extra_attrsz
  • )rr'z
  • rrr"r$r$r%(test_assert_tag_in_html_with_extra_attrs6s(  z

    not in a script tag

    z

    rrz

    )ZassertTagInTemplateScriptr r!r"r$r$r%"test_assert_tag_in_template_scriptMs  z6TestAssertTagInHTML.test_assert_tag_in_template_scriptN)__name__ __module__ __qualname__r&r)r*r$r$r$r%rsrcs\eZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ Z S)TestWagtailPageTestscs&ttjjdd}|jj|_dS)NT)Zis_default_site)supersetUprobjectsgetZ root_pageZspecificroot)r#site __class__r$r%r0bs zTestWagtailPageTests.setUpc Csl|tt|tt|tt|t|ttW5QRX|t|ttW5QRXdSN) ZassertCanCreateAtr rrZassertCanNotCreateAtrr r r!r#r$r$r%test_assert_can_create_atgs     z.TestWagtailPageTests.test_assert_can_create_atcCs|tj||jtddd|tj|tjj||jt dddddd d d dd d ddddd||jt dddddddddddd dS)N Event Indexz{"entityMap": {},"blocks": [ {"inlineStyleRanges": [], "text": "Event intro", "depth": 0, "type": "unstyled", "key": "00000", "entityRanges": []} ]}titleZintroFlierptextDit is onze mooie text0rz{"entityMap": {},"blocks": [ {"inlineStyleRanges": [], "text": "Dit is onze mooie text in een ferrari", "depth": 0, "type": "unstyled", "key": "00000", "entityRanges": []} ]}1productpegsa pound23)r<z body-0-typez body-0-valuez body-0-orderzbody-0-deletedz body-1-typez body-1-valuez body-1-orderzbody-1-deletedz body-2-typezbody-2-value-namezbody-2-value-pricez body-2-orderzbody-2-deletedz body-count Fight Club1000z{"entityMap": {},"blocks": [ {"inlineStyleRanges": [], "text": "Rule 1: You do not talk about Fight Club", "depth": 0, "type": "unstyled", "key": "00000", "entityRanges": []} ]}z{"entityMap": {},"blocks": [ {"inlineStyleRanges": [], "text": "Rule 2: You DO NOT talk about Fight Club", "depth": 0, "type": "unstyled", "key": "00000", "entityRanges": []} ]}) r<zsections-TOTAL_FORMSzsections-INITIAL_FORMSzsections-MIN_NUM_FORMSzsections-MAX_NUM_FORMSzsections-0-bodyzsections-0-ORDERzsections-0-DELETEzsections-1-bodyzsections-1-ORDERzsections-1-DELETE) assertFalser r1existsassertCanCreater3 assertTruer2liverrr8r$r$r%test_assert_can_createus\ z+TestWagtailPageTests.test_assert_can_createcCs8|j|jtdddddtjjdd}||jdS)NzSimple Lorem PagezLorem ipsum dolor sit amet)r<contentF)publish)r<)rLr3rrr1r2rJrN)r#Z created_pager$r$r%/test_assert_can_create_for_page_without_publishszDTestWagtailPageTests.test_assert_can_create_for_page_without_publishc Cs|tj||jttdtdd|tj||jt tdt ddtdfdd d d fgd ||jt td t dtdidtdigddS)Nr:z

    Event intro

    r;r=)r>r?rz,

    Dit is onze mooie text in een ferrari

    rCrDrE)nameZprice)r<bodyrHrTz/

    Rule 1: You do not talk about Fight Club

    z/

    Rule 2: You DO NOT talk about Fight Club

    )r<sections) rJr r1rKrLr3rrrMrrrrr8r$r$r%(test_assert_can_create_with_form_helperss`  z=TestWagtailPageTests.test_assert_can_create_with_form_helpersc CsFtdddd}|jj|d|td||tiW5QRXdS)Nz Simple PagesimpleZhello)r<ZslugrP)instancez=Can not create a tests.businesschild under a tests.simplepage)rr3Z add_childassertRaisesRegexr!rLr )r#Z simple_pager$r$r%$test_assert_can_create_subpage_rulessz9TestWagtailPageTests.test_assert_can_create_subpage_rulesc Cs,|td||jtiW5QRXdS)Nz\bslug:\n[\s\S]*\btitle:\n)rYr!rLr3rr8r$r$r%'test_assert_can_create_validation_errorsz Hello, world)r>Goodbye, worldcoffeeZlatteZsoya)typeZmilkrGr>rkr@rArlrBrF)z content-countzcontent-0-typezcontent-0-valuezcontent-0-orderzcontent-0-deletedzcontent-1-typezcontent-1-valuezcontent-1-orderzcontent-1-deletedzcontent-2-typezcontent-2-value-typezcontent-2-value-milkzcontent-2-orderzcontent-2-deleted)rrrgrhr$r$r%test_streamfield7s8  z$TestFormDataHelpers.test_streamfieldcCsBtdtddiddigi}||ddddddddd dd dS) Nlinesr>ZHelloZWorldrFr@rIrArB) zlines-TOTAL_FORMSzlines-INITIAL_FORMSzlines-MIN_NUM_FORMSzlines-MAX_NUM_FORMSz lines-0-textz lines-0-ORDERzlines-0-DELETEz lines-1-textz lines-1-ORDERzlines-1-DELETE)rrrgrhr$r$r%test_inline_formsetXs. z'TestFormDataHelpers.test_inline_formsetc CsHtd}|tt|igddddgdgddddgdgd dS) N

    title

    para

    r<rz header-two00000ZinlineStyleRangesr>depthrnkeyZ entityRangesparaunstyledZ entityMapblocksrrMrjsonloadsrhr$r$r%test_default_rich_textts. z*TestFormDataHelpers.test_default_rich_textc CsNtddgd}|tt|igddddgdgd dddgdgd dS) Nrrp)featuresr<rrxrsrtrwryr{rhr$r$r%#test_rich_text_with_custom_featuress. z7TestFormDataHelpers.test_rich_text_with_custom_featurescCstddd}||ddS)NrrZcustom)Zeditor)rrgrhr$r$r%&test_rich_text_with_alternative_editors z:TestFormDataHelpers.test_rich_text_with_alternative_editorN) r+r,r-rjrorqr~rrr$r$r$r%ra+s  !rac@seZdZddZdS)TestDummyExternalStoragec CsDd}td|dd}||tdtd|W5QRXdS)z Test that DummyExternalStorage correctly warns about attempts to write files that are not rewound to the start sFPNG  IHDRĉ IDATxc````E@IENDB`ztest.pngz image/png)rSrP content_typez4Content file pointer should be at 0 - got 70 insteadN)rreadZassertRaisesMessage ValueErrorrsave)r#ZpngZ simple_pngr$r$r%-test_save_with_incorrect_file_object_positionszFTestDummyExternalStorage.test_save_with_incorrect_file_object_positionN)r+r,r-rr$r$r$r%rsr)%r|Zdjango.core.files.uploadedfilerZ django.testrZ%wagtail.admin.tests.test_contentstaterZwagtail.modelsrrrZ#wagtail.test.dummy_external_storagerZwagtail.test.testapp.modelsr r r r r rrrrrrZwagtail.test.utilsrrZwagtail.test.utils.form_datarrrrrr.rarr$r$r$r%s    4 BK