a D g@spddlZddlZddlmZddlZddlmZmZddlm Z ddl m Z m Z ddl mZmZddlmZmZddlmZdd lmZdd lmZmZdd lmZmZmZdd lmZdd l m!Z!m"Z"ddl#m$Z$ddl%m&Z&m'Z'm(Z(ddl)m*Z*ddl+m,Z-ddl.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9m:Z:ddl;mm?Z?ddl@m,Z,mAZAmBZBmCZCzddlDZDdZEWneFydZEYn0GdddeZGGdd d eZHGd!d"d"er<r=rAr8r8r9render_image_tag_asPsz TestImageTag.render_image_tag_ascCs6||jd}|d||d||d|dSrE)rTr5rIrJr8r8r9test_image_tag_attrsYs  z!TestImageTag.test_image_tag_attrscCs$td}t||d}||S)Nzx{% load wagtailimages_tags %}{% image image_obj width-400 class="photo" title=title|lower alt="Alternate" %})r<r0r=)r7r5r0rBrCr8r8r9&render_image_tag_with_extra_attributesas z3TestImageTag.render_image_tag_with_extra_attributescCsN||jd}|d||d||d||d||d|dS)NzMy Wonderful TitlerFrGz class="photo"zalt="Alternate"ztitle="my wonderful title")rVr5rIrJr8r8r9$test_image_tag_with_extra_attributesis    z1TestImageTag.test_image_tag_with_extra_attributescCs$td}td|d}||S)NzX{% load wagtailimages_tags %}{% image image_primary|default:image_alternate width-400 %})Z image_primaryZimage_alternater=)r7r5rBrCr8r8r9render_image_tag_with_filtersus z*TestImageTag.render_image_tag_with_filterscCs(||j}|d||d|dS)NrFrG)rXr5rIrJr8r8r9test_image_tag_with_filters|s  z(TestImageTag.test_image_tag_with_filterscCs*||jd}|d||d|dS)Nzfill-200x200 height-150z width="150"z height="150"rHrJr8r8r9#test_image_tag_with_chained_filterss z0TestImageTag.test_image_tag_with_chained_filterscCsx|tj||jdWdn1s00Y|tj||jdWdn1sj0YdS)Nfill-200x200|height-150zfill-800x600 alt"test")rQrTemplateSyntaxErrorrDr5r6r8r8r9,test_filter_specs_must_match_allowed_patterns,z9TestImageTag.test_filter_specs_must_match_allowed_patterncCsT|tj4td}td|ji}||Wdn1sF0YdS)Nz{% load wagtailimages_tags %}{% image image_obj fill-200x200 as test_img this_one_should_not_be_there %}r<rQrr\r>r?r5r@r7rBrCr8r8r9*test_context_may_only_contain_one_arguments z7TestImageTag.test_context_may_only_contain_one_argumentcCsT|tj4td}td|ji}||Wdn1sF0YdS)Nz2{% load wagtailimages_tags %}{% image image_obj %}r<r^r_r8r8r9test_no_image_filter_provideds z*TestImageTag.test_no_image_filter_providedcCsT|tj4td}td|ji}||Wdn1sF0YdS)Nz9{% load wagtailimages_tags %}{% image image_obj as foo %}r<r^r_r8r8r9+test_no_image_filter_provided_when_using_ass z8TestImageTag.test_no_image_filter_provided_when_using_ascCsT|tj4td}td|ji}||Wdn1sF0YdS)NzE{% load wagtailimages_tags %}{% image image_obj class="cover-image"%}r<r^r_r8r8r95test_no_image_filter_provided_but_attributes_provideds zBTestImageTag.test_no_image_filter_provided_but_attributes_providedcCs*td|d}td|i}||S)NzA{% load wagtailimages_tags %}{% image_url image_obj "width-400" "z" %}r<r=)r7r5Z view_namerBrCr8r8r9render_image_url_tagsz!TestImageTag.render_image_url_tagcCs4||jd}||d|jjjdddS)Nwagtailimages_servez/images/.*/width-400/{}/rdr5 assertRegexformatr1namesplitrJr8r8r9test_image_urls zTestImageTag.test_image_urlcCs4||jd}||d|jjjdddS)NZwagtailimages_serve_custom_viewz'/testimages/custom_view/.*/width-400/{}rfrgrhrJr8r8r9test_image_url_custom_viewsz'TestImageTag.test_image_url_custom_viewcCsR|t4td}td|ji}||Wdn1sD0YdS)Nza{% load wagtailimages_tags %}{% image_url image_obj "width-400" "mynonexistingimageserve_view" %}r<)rQrrr>r?r5r@r_r8r8r9'test_image_url_no_imageserve_view_addeds  z4TestImageTag.test_image_url_no_imageserve_view_addedN)__name__ __module__ __qualname__r:rDrLrPrSrTrUrVrWrXrYrZr]r`rarbrcrdrmrnror8r8r8r9r,1s*        r,c@s&eZdZdZdgZddZddZdS)TestMissingImagea- Missing image files in media/original_images should be handled gracefully, to cope with pulling live databases to a development instance without copying the corresponding image files. In this case, it's acceptable to render broken images, but not to fail rendering the page outright. z test.jsoncCs |jd}|j|ddddS)Nz/events/christmas/zfA missing imageThtmlclientgetZassertContainsr7responser8r8r9!test_image_tag_with_missing_images  z2TestMissingImage.test_image_tag_with_missing_imagecCs |jd}|j|ddddS)Nz/events/final-event/z|where did my image go?Trtrvryr8r8r9!test_rich_text_with_missing_images  z2TestMissingImage.test_rich_text_with_missing_imageN)rprqrr__doc__Zfixturesr{r|r8r8r8r9rss rsc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS) TestFormatcCs(tdddd|_tjjdtd|_dS)N test namez test labeltest is-primaryoriginalr.r/)rrjrr3r4r)r5r6r8r8r9r:s zTestFormat.setUpcCs.|j|jd}||ddd|jjddS)N test alt textr5r)zdata-altzdata-embedtypez data-formatzdata-id)rjZeditor_attributesr5rOpkrJr8r8r9test_editor_attributessz!TestFormat.test_editor_attributescCs,|j|jd}|jd|jj|dddS)Nrztest alt textTZallow_extra_attrs)rjimage_to_editor_htmlr5assertTagInHTMLrrJr8r8r9test_image_to_editor_html sz$TestFormat.test_image_to_editor_htmlcCs0|j|jd}d|jj}|j||dddS)NArthur "two sheds" JacksonzArthur "two sheds" JacksonTr)rjrr5rr)r7rKZ expected_htmlr8r8r9&test_image_to_editor_html_with_quotingsz1TestFormat.test_image_to_editor_html_with_quotingcCs4d|j_|j|jd}|jd|ddd|j_dS)Nrz2test alt textTrr)rj classname image_to_htmlr5rrJr8r8r9 test_image_to_html_no_classnames!sz+TestFormat.test_image_to_html_no_classnamescCs$|j|jd}|jd|dddS)NrzaArthur "two sheds" JacksonTr)rjrr5rrJr8r8r9test_image_to_html_with_quoting-s z*TestFormat.test_image_to_html_with_quotingcCs$t|jtd}|||jdS)Nr)rrjrrOrJr8r8r9test_get_image_format6s z TestFormat.test_get_image_formatcCsB|t|jj}Wdn1s(0Y||ddS)Nr)Z assertWarnsr'rjZ classnamesrO)r7rr8r8r9*test_deprecated_classnames_property_access;s &z5TestFormat.test_deprecated_classnames_property_accessN) rprqrrr:rrrrrrrr8r8r8r9r~s      r~c@s,eZdZddZddZddZddZd S) TestSignatureGenerationcCs|tddddS)Nd fill-800x600xnZOzQyUg6pkfciqcfRJRosOrGg=)rOr r6r8r8r9test_signature_generationBs z1TestSignatureGeneration.test_signature_generationcCs|tddddS)Nrrr) assertTruer!r6r8r8r9test_signature_verificationGs z3TestSignatureGeneration.test_signature_verificationcCs|tddddS)NrrZ assertFalser!r6r8r8r9"test_signature_changes_on_image_idLs z:TestSignatureGeneration.test_signature_changes_on_image_idcCs|tddddS)Nrrz fill-800x700rr6r8r8r9%test_signature_changes_on_filter_specQs z=TestSignatureGeneration.test_signature_changes_on_filter_specN)rprqrrrrrrr8r8r8r9rAsrc@seZdZddZddZddZeddidd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFrontendServeViewcCstjjdtd|_dSr-r2r6r8r8r9r:XszTestFrontendServeView.setUpcCst|jjd}|jtd||jjdfd}||jd||j ||dd||dd||d d t j d |j}||td S) z6 Test a valid GET request to the view rreargsr Content-Type image/pngContent-Security-Policydefault-src 'none'X-Content-Type-OptionsnosniffNr r5idrwrxr rO status_coder streamingwillowropenjoinstreaming_contentassertIsInstancerr7 signaturerzr5r8r8r9test_get_s zTestFrontendServeView.test_getcCstjjdtd}t|jd}|jtd||jdfd}| |j d| |j | |dd| |d d | |d d t jtd |j}||tdS)NzTest SVGr/rrerrrz image/svg+xmlrrrrr)rr3r4r+r rrwrxr rOrrrrrrrrrrr7r5rrzr8r8r9 test_get_svgws  z"TestFrontendServeView.test_get_svgZavif)Z WAGTAILIMAGES_FORMAT_CONVERSIONScCstjjdtd}t|jd}|jtd||jdfd}| |j d| |j | |ddt jd |j}||tdS) Nz Test AVIFr/rrerrrz image/avifr)rr3r4r*r rrwrxr rOrrrrrrrrrrr8r8r9 test_get_avifs  z#TestFrontendServeView.test_get_avifcCs~t|jjd}|jtd||jjdfdd}||jd||j ||ddt j d |j}||td S) zU Test that a filename can be optionally added to the end of the URL. rrertest.pngrrrrNrrr8r8r9test_get_with_extra_components z3TestFrontendServeView.test_get_with_extra_componentcCsBt|jjd}|jtd||jjdfdd}||jddS)za A filename can be appended to the end of the URL, but it must not contain a '/' rrerz test/test.pngiNr r5rrwrxr rOrr7rrzr8r8r9'test_get_with_too_many_extra_componentssz=TestFrontendServeView.test_get_with_too_many_extra_componentscCst|jjd}|jtd||jjdfd}||jd||j ||dd||dd||d d t j d |j}||tdS) Nr wagtailimages_serve_action_serverrrrrrrrrrrr8r8r9test_get_with_serve_actions  z0TestFrontendServeView.test_get_with_serve_actioncCsdt|jjd}|jtd||jjdfd}djtj tj |jj jd}|j ||ddddS) NrZ#wagtailimages_serve_action_redirectrz>/media/images/{filename[0]}.2e16d0ba.fill-800x600{filename[1]}filenamei.F)rZfetch_redirect_response) r r5rrwrxr rjospathsplitextbasenamer1ZassertRedirects)r7rrzZexpected_redirect_urlr8r8r9test_get_with_redirect_actions" z3TestFrontendServeView.test_get_with_redirect_actioncCs:|ttjddWdn1s,0YdS)Nunknown)action)rQrr"Zas_viewr6r8r8r9*test_init_with_unknown_action_raises_errors z@TestFrontendServeView.test_init_with_unknown_action_raises_errorcCsft|jjddd}|jtd||jjdfdd}||jdtj d |j }| |td S) z> Test that the key can be changed on the view rZcustom)keywagtailimages_serve_custom_keyrrrrN)r r5rrwrxr rOrrrrrrrrrr8r8r9test_get_with_custom_keys  z.TestFrontendServeView.test_get_with_custom_keycCsBt|jjd}|jtd||jjdfdd}||jddS)z Test that the key can be changed on the view This tests that the default key no longer works when the key is changed on the view rrrrNrrr8r8r9*test_get_with_custom_key_using_default_key s  z@TestFrontendServeView.test_get_with_custom_key_using_default_keycCsBt|jjdd}|jtd||jjdfd}||jddS)zG Test that an invalid signature returns a 403 response r(rrerrNrrr8r8r9test_get_invalid_signature sz0TestFrontendServeView.test_get_invalid_signaturecCs>t|jjd}|jtd||jjdfd}||jddS)a3 Test that an invalid filter spec returns a 400 response This is very unlikely to happen in reality. A user would have to create signature for the invalid filter spec which can't be done with Wagtails built in URL generator. We should test it anyway though. zbad-filter-specrerNrrr8r8r9test_get_invalid_filter_spec1s  z2TestFrontendServeView.test_get_invalid_filter_speccCsNt|jjjt|jjd}|jt d||jjdfd}| |j ddS)z Test that a missing image file gives a 410 response When the source image file is missing, it is presumed deleted so we return a 410 "Gone" response. rreriN) rremover5r1rr rrwrxr rOrrr8r8r9"test_get_missing_source_image_fileHsz8TestFrontendServeView.test_get_missing_source_image_filecCs@t|jjd}|jtd||jjdfd}||dddS)Nrrrz Cache-Controlzmax-age=3600, public)r r5rrwrxr rOrr8r8r9test_get_cache_control]s z,TestFrontendServeView.test_get_cache_controlN)rprqrrr:rrr rrrrrrrrrrrrr8r8r8r9rWs   rc@sFeZdZddZeddee dddZedddd Z d S) TestFrontendSendfileViewcCstjjdtd|_dSr-r2r6r8r8r9r:iszTestFrontendSendfileView.setUpzsendfile.backends.development)ZSENDFILE_BACKENDzMissing django-sendfile app.cCsnt|jjd}|jtd||jjdfd}||jd||dd||dd||d d dS) NrZwagtailimages_sendfilerrrrrrrrrrr8r8r9test_sendfile_nobackendos z0TestFrontendSendfileView.test_sendfile_nobackendcCsnt|jjd}|jtd||jjdfd}||jd|j|j dd||dd||d d dS) NrZwagtailimages_sendfile_dummyrrzDummy backend response)msgrrrr) r r5rrwrxr rOrrcontentrr8r8r9test_sendfile_dummy_backends z4TestFrontendSendfileView.test_sendfile_dummy_backendN) rprqrrr:r unittestZskipIf sendfile_modrrr8r8r8r9rhs   rc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)TestRectcCsJtdddd}||jd||jd||jd||jddS)Nrr)rrOlefttoprightZbottomr7Zrectr8r8r9 test_inits zTestRect.test_initcCsD|tddddtdddd|tddddtdddddS) Nrrrr )rOrassertNotEqualr6r8r8r9 test_equalitys zTestRect.test_equalitycCsbtdddd}||dd||dd||dd||dd|t|jd dS) Nrrrrrr()rrOrQ IndexError __getitem__rr8r8r9 test_getitems zTestRect.test_getitemcCs"tdddd}||ddS)Nrrrr)rrrr)rrOas_tuplerr8r8r9 test_as_tupleszTestRect.test_as_tuplecCsJtdddd}||jt||jd||jd||jddS)Nrrr^)rr)rrsizerrOwidthheightrr8r8r9 test_sizes zTestRect.test_sizecCs$tdddd}d|_||ddS)Nrrrr)rr2rr)rrrOrr8r8r9test_set_size_with_tuplesz!TestRect.test_set_size_with_tuplecCs*tdddd}tdd|_||ddS)Nrrrrrr)rrrrOrr8r8r9test_set_size_with_vectors z"TestRect.test_set_size_with_vectorcCsftdddd}||jt||jd||jd||jd||jd||jddS)Nrrrr)rrr) rrcentroidrrOxyZ centroid_xZ centroid_yrr8r8r9 test_centroidszTestRect.test_centroidcCs$tdddd}d|_||ddS)Nrrrr)rrri&iX)rrrOrr8r8r9test_set_centroid_with_tuplesz%TestRect.test_set_centroid_with_tuplecCs*tdddd}tdd|_||ddS)Nrrrrrr)rrrrOrr8r8r9test_set_centroid_with_vectors z&TestRect.test_set_centroid_with_vectorcCs"tdddd}|t|ddS)Nrrrrz2Rect(left: 100, top: 150, right: 200, bottom: 250))rrOreprrr8r8r9 test_reprszTestRect.test_reprcCs*tdddd}||tdddddS) NrrrrK})rZ from_pointrOrr8r8r9test_from_pointszTestRect.test_from_pointN)rprqrrrrrrrrrrrrrrr8r8r8r9rs rc@s4eZdZddZddZddZddZd d Zd S) TestGetImageFormcCs&tt}|t|jgddS)N) r0r1 description collectiontags focal_point_x focal_point_yfocal_point_widthfocal_point_height)rrrOlist base_fieldskeysr7formr8r8r9 test_fieldss  zTestGetImageForm.test_fieldscCs&tt}|t|jgddS)N) r0r1rr r r r r rcaptionZ fancy_caption)rr#rOrrrrr8r8r9 test_admin_form_fields_attributes  z1TestGetImageForm.test_admin_form_fields_attributecCs4tt}||jdt||jdjtjdS)Nr1)r WagtailImagerrrwidgetrZ FileInputrr8r8r9test_file_fieldsz TestGetImageForm.test_file_fieldcCs2tt}||jdt||jdjtdS)Nr )rrrrrrrrr8r8r9test_tags_fieldsz TestGetImageForm.test_tags_fieldcCstt}||jdtj||jdtj||jdtj||jdtj||jdjtj||jdjtj||jdjtj||jdjtjdS)Nr r r r)rrrrrZ IntegerFieldrZ HiddenInputrr8r8r9test_focal_point_fields s& z(TestGetImageForm.test_focal_point_fieldsN)rprqrrrrrrrr8r8r8r9rs rc@s,eZdZddZddZddZddZd S) TestRenditionFilenamescCs4tjjdtddd}|d}||jjddS)Nr.z test_rf1.pngrr/z width-100zimages/test_rf1.width-100.pngrr3r4r) get_renditionrOr1rkr7r5Z renditionr8r8r9test_normal_filter&s  z)TestRenditionFilenames.test_normal_filtercCs4tjjdtddd}|d}||jjddS)Nr.z test_rf2.pngrr/ fill-100x100z)images/test_rf2.2e16d0ba.fill-100x100.pngrrr8r8r9test_fill_filter/s z'TestRenditionFilenames.test_fill_filtercCsPtjjdtddd}|tdddd||d}||j j ddS) Nr.z test_rf3.pngrr/rrr!z)images/test_rf3.15ee4958.fill-100x100.png rr3r4r)Zset_focal_pointrsaverrOr1rkrr8r8r9!test_fill_filter_with_focal_point:s z8TestRenditionFilenames.test_fill_filter_with_focal_pointcCsPtjjdtddd}|tdddd||d}||j j ddS) Nr.z test_rf4.pngrr/rrr[z4images/test_rf4.15ee4958.fill-200x200.height-150.pngr#rr8r8r9!test_filter_with_pipe_gets_dottedHs z8TestRenditionFilenames.test_filter_with_pipe_gets_dottedN)rprqrrr r"r%r&r8r8r8r9r!s  rc@seZdZddZdS)TestDifferentUploadcCsDtjjddtd}tjjddtddd}||jj|jjdS)Nr.zA test description)r0rr1z Test ImageZblack)Zcolour)r$r3r4r)rr1url)r7r5Z second_imager8r8r9test_upload_pathXsz$TestDifferentUpload.test_upload_pathN)rprqrrr)r8r8r8r9r'Wsr'c@seZdZeddddZeddddZeddZed d Zee e d ed dd dZ ee e d eddddZ dS)TestGetImageModeltests.CustomImage)WAGTAILIMAGES_IMAGE_MODELcCs|ttdS)z.Test get_image_model with a custom image modelN)assertIsrr#r6r8r8r9test_custom_get_image_modelksz-TestGetImageModel.test_custom_get_image_modelcCs|tddS)z5Test get_image_model_string with a custom image modelr+N)rOrr6r8r8r9"test_custom_get_image_model_stringpsz4TestGetImageModel.test_custom_get_image_model_stringcCs"t`ddlm}|t|dS)z6Test get_image_model with no WAGTAILIMAGES_IMAGE_MODELrrN)rr,wagtail.images.modelsrr-r)r7rr8r8r9test_standard_get_image_modelus z/TestGetImageModel.test_standard_get_image_modelcCst`|tddS)z=Test get_image_model_STRING with no WAGTAILIMAGES_IMAGE_MODELzwagtailimages.ImageN)rr,rOrr6r8r8r9$test_standard_get_image_model_string}sz6TestGetImageModel.test_standard_get_image_model_string)signalZreceiverztests.UnknownModelcCs4|ttWdn1s&0YdS)z*Test get_image_model with an unknown modelNrQrrr6r8r8r9test_unknown_get_image_models z.TestGetImageModel.test_unknown_get_image_modelzinvalid-stringcCs4|ttWdn1s&0YdS)z1Test get_image_model with an invalid model stringNr4r6r8r8r9test_invalid_get_image_models z.TestGetImageModel.test_invalid_get_image_modelN) rprqrrr r.r/r1r2r&r rr5r6r8r8r8r9r*js$     r*c@s8eZdZeddZddZddZddZd d Zd S) TestWagtailImageFieldcCs(d|_t|jdj|_|jj|_dS)Nzwagtailimagefield.pngr)rr)r1r5 getbuffernbytes image_size)clsr8r8r9 setUpTestDatasz#TestWagtailImageField.setUpTestDatacCsTt}|jdt|jd|jd|jd}||}||jt| |j ddS)NrrNr rr5seekrrr: to_pythonrWillowImageFilerO content_typer7fr1r?r8r8r9 test_to_python_with_inmemoryfiles  z6TestWagtailImageField.test_to_python_with_inmemoryfilecCsTt}|jdt|jd|jd|jd}||}||jt| |j ddS)NrrNz image/jpegrr=rBr8r8r9,test_to_python_gets_content_type_from_willows  zBTestWagtailImageField.test_to_python_gets_content_type_from_willowcCst}tdd|jd\}|jd||j|d||}||jt | |j dWdn1sx0YdS)N test_temp.pngrr) rr r:r5r>writereadr?rr@rOrA)r7rCtmp_filer?r8r8r9"test_to_python_with_temporary_files    z8TestWagtailImageField.test_to_python_with_temporary_filec Csd}t}tddddF}|t|||Wdn1sF0YWdn1sd0Y|td||jWdn1s0YdS)NzYUpload a valid image. The file you uploaded was either not an image or a corrupted image.rFr z;No file was submitted. Check the encoding type on the form.)rr ZassertRaisesMessagerr?r5)r7rrCrIr8r8r93test_to_python_raises_error_with_invalid_image_filesFzITestWagtailImageField.test_to_python_raises_error_with_invalid_image_fileN) rprqrr classmethodr<rDrErJrLr8r8r8r9r7s     r7)SrriorrZdjangorrZ django.confrZdjango.core.exceptionsrrZdjango.core.files.uploadedfilerr Z django.testr r Zdjango.test.signalsr Z django.urlsr Z taggit.formsrrZ willow.imagerrrrr@Zwagtail.imagesrrZwagtail.images.fieldsrZwagtail.images.formatsrrrZwagtail.images.formsrr0rrZwagtail.images.permissionsrZwagtail.images.rectrrZwagtail.images.utilsr r!Zwagtail.images.views.server"Zwagtail.test.testapp.modelsr#r$Zwagtail.test.utilsr%r&Zwagtail.utils.deprecationr'utilsr)r*r+sendfiler ImportErrorr,rsr~rrrrrrr'r*r7r8r8r8r9s\            'L'GK6,