U {h @srddlZddlZddlmZddlZddlmZmZddlm 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,mAZAmBZBmCZCzddlDZDdZEWneFk rdZEYnXGdddeZGGdd 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_filtersc CsP|tj||jdW5QRX|tj||jdW5QRXdS)Nfill-200x200|height-150zfill-800x600 alt"test")rQrTemplateSyntaxErrorrDr5r6r8r8r9,test_filter_specs_must_match_allowed_patternsz9TestImageTag.test_filter_specs_must_match_allowed_patternc Cs@|tj*td}td|ji}||W5QRXdS)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_argumentc Cs@|tj*td}td|ji}||W5QRXdS)Nz2{% load wagtailimages_tags %}{% image image_obj %}r<r^r_r8r8r9test_no_image_filter_provideds z*TestImageTag.test_no_image_filter_providedc Cs@|tj*td}td|ji}||W5QRXdS)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_asc Cs@|tj*td}td|ji}||W5QRXdS)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_viewc Cs>|t*td}td|ji}||W5QRXdS)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_formatc Cs.|t|jj}W5QRX||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_generationBsz1TestSignatureGeneration.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.setUpcCszt|jjd}|jtd||jjdfd}||jd||j ||ddt j d |j}||tdS) z6 Test a valid GET request to the view rreargsr Content-Type image/pngNr r5idrwrxr rO status_coder streamingwillowropenjoinstreaming_contentassertIsInstancerr7 signaturerzr5r8r8r9test_get_s  zTestFrontendServeView.test_getcCstjjdtd}t|jd}|jtd||jdfd}| |j d| |j | |ddt jtd |j}||tdS) NzTest SVGr/rrerrrz image/svg+xmlr)rr3r4r+r rrwrxr rOrrrrrrrrrrr7r5rrzr8r8r9 test_get_svgus  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_componentss z=TestFrontendServeView.test_get_with_too_many_extra_componentscCszt|jjd}|jtd||jjdfd}||jd||j ||ddt j d |j}||tdS)Nr wagtailimages_serve_action_serverrrrrrrr8r8r9test_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_actionc Cs&|ttjddW5QRXdS)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_keys  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_signatures z0TestFrontendServeView.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_spec+s  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_fileBs z8TestFrontendServeView.test_get_missing_source_image_filecCs@t|jjd}|jtd||jjdfd}||dddS)Nrrrz Cache-Controlzmax-age=3600, public)r r5rrwrxr rOrr8r8r9test_get_cache_controlWs z,TestFrontendServeView.test_get_cache_controlN)rprqrrr:rrr rrrrrrrrrrrrr8r8r8r9rWs   rc@sFeZdZddZeddee dddZedddd Z d S) TestFrontendSendfileViewcCstjjdtd|_dSr-r2r6r8r8r9r:cszTestFrontendSendfileView.setUpzsendfile.backends.development)ZSENDFILE_BACKENDzMissing django-sendfile app.cCsNt|jjd}|jtd||jjdfd}||jd||dddS)NrZwagtailimages_sendfilerrrrrrr8r8r9test_sendfile_nobackendis z0TestFrontendSendfileView.test_sendfile_nobackendcCsNt|jjd}|jtd||jjdfd}||jd|j|j dddS)NrZwagtailimages_sendfile_dummyrrzDummy backend response)msg) r r5rrwrxr rOrrcontentrr8r8r9test_sendfile_dummy_backendws z4TestFrontendSendfileView.test_sendfile_dummy_backendN) rprqrrr:r unittestZskipIf sendfile_modrrr8r8r8r9rbs   rc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)TestRectcCsJtdddd}||jd||jd||jd||jddS)Nrr)rrOlefttoprightbottomr7rectr8r8r9 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_reprs zTestRect.test_reprcCs*tdddd}||tdddddS) NrrrrK})rZ from_pointrOrr8r8r9test_from_pointszTestRect.test_from_pointN)rprqrrrrrrrrrrrrrrr8r8r8r9rs rc@s4eZdZddZddZddZddZd d Zd S) TestGetImageFormc Cs2tt}|t|jddddddddgdS) Nr0r1 collectiontags focal_point_x focal_point_yfocal_point_widthfocal_point_height)rrrOlist base_fieldskeysr7formr8r8r9 test_fieldss zTestGetImageForm.test_fieldsc Cs6tt}|t|jddddddddd d g dS) Nr0r1rrrr r r captionZ fancy_caption)rr#rOr r rrr8r8r9 test_admin_form_fields_attributes z1TestGetImageForm.test_admin_form_fields_attributecCs4tt}||jdt||jdjtjdS)Nr1)r WagtailImagerr rwidgetrZ FileInputrr8r8r9test_file_fieldsz TestGetImageForm.test_file_fieldcCs2tt}||jdt||jdjtdS)Nr)rrrr rrrrr8r8r9test_tags_fieldsz TestGetImageForm.test_tags_fieldcCstt}||jdtj||jdtj||jdtj||jdtj||jdjtj||jdjtj||jdjtj||jdjtjdS)Nrr r r )rrrr rZ IntegerFieldrZ HiddenInputrr8r8r9test_focal_point_fieldss0    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_filters  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/rrrz)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_dotted<s z8TestRenditionFilenames.test_filter_with_pipe_gets_dottedN)rprqrrrrr"r#r8r8r8r9rs  rc@seZdZddZdS)TestDifferentUploadcCs@tjjdtd}tjjdtddd}||jj|jjdS)Nr.r/z Test Imageblack)colour)r$r3r4r)rr1url)r7r5Z second_imager8r8r9test_upload_pathLsz$TestDifferentUpload.test_upload_pathN)rprqrrr(r8r8r8r9r$Ksr$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_model]sz-TestGetImageModel.test_custom_get_image_modelcCs|tddS)z5Test get_image_model_string with a custom image modelr*N)rOrr6r8r8r9"test_custom_get_image_model_stringbsz4TestGetImageModel.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_modelgs 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_stringosz6TestGetImageModel.test_standard_get_image_model_string)signalZreceiverztests.UnknownModelc Cs |t tW5QRXdS)z*Test get_image_model with an unknown modelNrQrrr6r8r8r9test_unknown_get_image_modelus z.TestGetImageModel.test_unknown_get_image_modelzinvalid-stringc Cs |t tW5QRXdS)z1Test get_image_model with an invalid model stringNr3r6r8r8r9test_invalid_get_image_model~s z.TestGetImageModel.test_invalid_get_image_modelN) rprqrrr r-r.r0r1r&r rr4r5r8r8r8r9r)\s(     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 rr5seekrrr9 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<rAr8r8r9,test_to_python_gets_content_type_from_willows  zBTestWagtailImageField.test_to_python_gets_content_type_from_willowc Csrt}tdd|jdR}|jd||j|d||}||jt | |j dW5QRXdS)N test_temp.pngrr) rr r9r5r=writereadr>rr?rOr@)r7rBtmp_filer>r8r8r9"test_to_python_with_temporary_files   z8TestWagtailImageField.test_to_python_with_temporary_filec Csnd}t}tdddd(}|t|||W5QRXW5QRX|td||jW5QRXdS)NzYUpload a valid image. The file you uploaded was either not an image or a corrupted image.rEr z;No file was submitted. Check the encoding type on the form.)rr ZassertRaisesMessagerr>r5)r7rrBrHr8r8r93test_to_python_raises_error_with_invalid_image_fileszITestWagtailImageField.test_to_python_raises_error_with_invalid_image_fileN) rprqrr classmethodr;rCrDrIrKr8r8r8r9r6s     r6)SrriorrdjangorrZ 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.formsrr/rrZwagtail.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)r6r8r8r8r9s\            'L #GI6,