o h@snddlZddlZddlmZddlZddlmZmZddlm 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,mAZAmBZBmCZCzddlDZDdZEWn eFydZEYnwGdddeZGGdd d eZHGd!d"d"ez+ as test_img %}r?r@rDr9r9r:render_image_tag_asP z TestImageTag.render_image_tag_ascCrIrJ)r\r6rNrOr9r9r:test_image_tag_attrsYrRz!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?r1r@)r8r6r1rFrGr9r9r:&render_image_tag_with_extra_attributesas  z3TestImageTag.render_image_tag_with_extra_attributescCsN||jd}|d||d||d||d||d|dS)NzMy Wonderful TitlerKrLz class="photo"zalt="Alternate"ztitle="my wonderful title")r_r6rNrOr9r9r:$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 %}) image_primaryimage_alternater@)r8r6rFrGr9r9r:render_image_tag_with_filtersus  z*TestImageTag.render_image_tag_with_filterscCs(||j}|d||d|dS)NrKrL)rcr6rNrOr9r9r:test_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"rMrOr9r9r:#test_image_tag_with_chained_filters z0TestImageTag.test_image_tag_with_chained_filterscCsz|tj||jdWdn1swY|tj||jdWddS1s6wYdS)Nfill-200x200|height-150zfill-800x600 alt"test")rXrTemplateSyntaxErrorrHr6r7r9r9r:,test_filter_specs_must_match_allowed_patterns "z9TestImageTag.test_filter_specs_must_match_allowed_patterncCV|tjtd}td|ji}||WddS1s$wYdS)Nz{% load wagtailimages_tags %}{% image image_obj fill-200x200 as test_img this_one_should_not_be_there %}r?rXrrhrArBr6rCr8rFrGr9r9r:*test_context_may_only_contain_one_arguments "z7TestImageTag.test_context_may_only_contain_one_argumentcCrj)Nz2{% load wagtailimages_tags %}{% image image_obj %}r?rkrlr9r9r:test_no_image_filter_provided "z*TestImageTag.test_no_image_filter_providedcCrj)Nz9{% load wagtailimages_tags %}{% image image_obj as foo %}r?rkrlr9r9r:+test_no_image_filter_provided_when_using_asroz8TestImageTag.test_no_image_filter_provided_when_using_ascCrj)NzE{% load wagtailimages_tags %}{% image image_obj class="cover-image"%}r?rkrlr9r9r:5test_no_image_filter_provided_but_attributes_providedrozBTestImageTag.test_no_image_filter_provided_but_attributes_providedcCr=)NzA{% load wagtailimages_tags %}{% image_url image_obj "width-400" "z" %}r?r@)r8r6 view_namerFrGr9r9r:render_image_url_tagr]z!TestImageTag.render_image_url_tagcC4||jd}||d|jjjdddS)Nwagtailimages_servez/images/.*/width-400/{}/rsr6 assertRegexformatr2namesplitrOr9r9r:test_image_urls zTestImageTag.test_image_urlcCrt)Nwagtailimages_serve_custom_viewz'/testimages/custom_view/.*/width-400/{}rvrwrxrOr9r9r:test_image_url_custom_viewsz'TestImageTag.test_image_url_custom_viewcCsT|ttd}td|ji}||WddS1s#wYdS)Nza{% load wagtailimages_tags %}{% image_url image_obj "width-400" "mynonexistingimageserve_view" %}r?)rXrrrArBr6rCrlr9r9r:'test_image_url_no_imageserve_view_addeds  "z4TestImageTag.test_image_url_no_imageserve_view_addedN)__name__ __module__ __qualname__r;rHrQrVrZr\r^r_r`rcrdrerirmrnrprqrsr}rrr9r9r9r:r,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.jsoncC |jd}|j|ddddS)Nz/events/christmas/zfA missing imageThtmlclientgetassertContainsr8responser9r9r:!test_image_tag_with_missing_image  z2TestMissingImage.test_image_tag_with_missing_imagecCr)Nz/events/final-event/z|where did my image go?Trrrr9r9r:!test_rich_text_with_missing_imagerz2TestMissingImage.test_rich_text_with_missing_imageN)rrr__doc__fixturesrrr9r9r9r:rs  rc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS) TestFormatcCs(tdddd|_tjjdtd|_dS)N test namez test labeltest is-primaryoriginalr/r0)rrzrr4r5r)r6r7r9r9r:r;s  zTestFormat.setUpcCs.|j|jd}||ddd|jjddS)N test alt textr6r)zdata-altzdata-embedtypez data-formatzdata-id)rzeditor_attributesr6rUpkrOr9r9r:test_editor_attributessz!TestFormat.test_editor_attributescCs,|j|jd}|jd|jj|dddS)Nrztest alt textTallow_extra_attrs)rzimage_to_editor_htmlr6assertTagInHTMLrrOr9r9r:test_image_to_editor_html s z$TestFormat.test_image_to_editor_htmlcCs0|j|jd}d|jj}|j||dddS)NArthur "two sheds" JacksonzArthur "two sheds" JacksonTr)rzrr6rr)r8rP expected_htmlr9r9r:&test_image_to_editor_html_with_quotingsz1TestFormat.test_image_to_editor_html_with_quotingcCs4d|j_|j|jd}|jd|ddd|j_dS)Nrz2test alt textTrr)rz classname image_to_htmlr6rrOr9r9r: test_image_to_html_no_classnames!s z+TestFormat.test_image_to_html_no_classnamescCs$|j|jd}|jd|dddS)NrzaArthur "two sheds" JacksonTr)rzrr6rrOr9r9r:test_image_to_html_with_quoting-s  z*TestFormat.test_image_to_html_with_quotingcCs$t|jtd}|||jdS)Nr)rrzrrUrOr9r9r:test_get_image_format6s z TestFormat.test_get_image_formatcCsB|t |jj}Wdn1swY||ddS)Nr) assertWarnsr'rz classnamesrU)r8rr9r9r:*test_deprecated_classnames_property_access;s  z5TestFormat.test_deprecated_classnames_property_accessN) rrrr;rrrrrrrr9r9r9r:rs      rc@,eZdZddZddZddZddZd S) TestSignatureGenerationcCs|tddddS)Nd fill-800x600xnZOzQyUg6pkfciqcfRJRosOrGg=)rUr r7r9r9r:test_signature_generationB z1TestSignatureGeneration.test_signature_generationcC|tddddS)Nrrr) assertTruer!r7r9r9r:test_signature_verificationGrz3TestSignatureGeneration.test_signature_verificationcCr)Nrr assertFalser!r7r9r9r:"test_signature_changes_on_image_idLrz:TestSignatureGeneration.test_signature_changes_on_image_idcCr)Nrrz fill-800x700rr7r9r9r:%test_signature_changes_on_filter_specQrz=TestSignatureGeneration.test_signature_changes_on_filter_specN)rrrrrrrr9r9r9r:rAs  rc@seZdZddZddZddZeddidd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!S)"TestFrontendServeViewcCr-r.r3r7r9r9r:r;Xr<zTestFrontendServeView.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 rruargsr Content-Type image/pngContent-Security-Policydefault-src 'none'X-Content-Type-OptionsnosniffNr r6idrrr rU status_coder streamingwillowropenjoinstreaming_contentassertIsInstancerr8 signaturerr6r9r9r:test_get_s zTestFrontendServeView.test_getcCstjjdtd}t|jd}|jtd||jdfd}| |j d| |j | |dd| |d d | |d d t jtd |j}||tdS)NzTest SVGr0rrurrrz image/svg+xmlrrrrr)rr4r5r+r rrrr rUrrrrrrrrrrr8r6rrr9r9r: test_get_svgws  z"TestFrontendServeView.test_get_svgavif) WAGTAILIMAGES_FORMAT_CONVERSIONScCstjjdtd}t|jd}|jtd||jdfd}| |j d| |j | |ddt jd |j}||tdS) Nz Test AVIFr0rrurrrz image/avifr)rr4r5r*r rrrr rUrrrrrrrrrrr9r9r: 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. rrurtest.pngrrrrNrrr9r9r:test_get_with_extra_components z3TestFrontendServeView.test_get_with_extra_componentcCBt|jjd}|jtd||jjdfdd}||jddS)za A filename can be appended to the end of the URL, but it must not contain a '/' rrurz test/test.pngiNr r6rrrr rUrr8rrr9r9r:'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_serverrrrrrrrrrrr9r9r:test_get_with_serve_actions  z0TestFrontendServeView.test_get_with_serve_actioncCsdt|jjd}|jtd||jjdfd}djtj tj |jj jd}|j ||ddddS) Nr#wagtailimages_serve_action_redirectrz>/media/images/{filename[0]}.2e16d0ba.fill-800x600{filename[1]}filenamei.F)rfetch_redirect_response) r r6rrrr rzospathsplitextbasenamer2assertRedirects)r8rrexpected_redirect_urlr9r9r:test_get_with_redirect_actions"  z3TestFrontendServeView.test_get_with_redirect_actioncCs<|ttjddWddS1swYdS)Nunknown)action)rXrr"as_viewr7r9r9r:*test_init_with_unknown_action_raises_errorr[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 rcustom)keywagtailimages_serve_custom_keyrrrrN)r r6rrrr rUrrrrrrrrrr9r9r:test_get_with_custom_keys  z.TestFrontendServeView.test_get_with_custom_keycCr)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 rrrrNrrr9r9r:*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(rrurrNrrr9r9r:test_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-specrurNrrr9r9r:test_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. rruriN) rremover6r2rr rrrr rUrrr9r9r:"test_get_missing_source_image_fileHsz8TestFrontendServeView.test_get_missing_source_image_filecCs@t|jjd}|jtd||jjdfd}||dddS)Nrrrz Cache-Controlzmax-age=3600, public)r r6rrrr rUrr9r9r:test_get_cache_control]s z,TestFrontendServeView.test_get_cache_controlN)rrrr;rrr rrrrrrrrrrrrr9r9r9r:rWs"   rc@sFeZdZddZeddee dddZedddd Z d S) TestFrontendSendfileViewcCr-r.r3r7r9r9r:r;is zTestFrontendSendfileView.setUpzsendfile.backends.development)SENDFILE_BACKENDzMissing django-sendfile app.cCsnt|jjd}|jtd||jjdfd}||jd||dd||dd||d d dS) Nrwagtailimages_sendfilerrrrrrrrrrr9r9r:test_sendfile_nobackendos z0TestFrontendSendfileView.test_sendfile_nobackendcCsnt|jjd}|jtd||jjdfd}||jd|j|j dd||dd||d d dS) Nrwagtailimages_sendfile_dummyrrzDummy backend response)msgrrrr) r r6rrrr rUrrcontentrr9r9r:test_sendfile_dummy_backends z4TestFrontendSendfileView.test_sendfile_dummy_backendN) rrrr;r unittestskipIf sendfile_modrrr9r9r9r:rhs  rc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZdS)TestRectcCsJtdddd}||jd||jd||jd||jddS)Nrr)rrUlefttoprightbottomr8rectr9r9r: test_init zTestRect.test_initcCsD|tddddtdddd|tddddtdddddS) Nrrrr )rUrassertNotEqualr7r9r9r: test_equalitys $zTestRect.test_equalitycCsbtdddd}||dd||dd||dd||dd|t|jd dS) Nrrrrrr()rrUrX IndexError __getitem__r r9r9r: test_getitems zTestRect.test_getitemcCs"tdddd}||ddS)Nrrrr)rrrr)rrUas_tupler r9r9r: test_as_tupleszTestRect.test_as_tuplecCsJtdddd}||jt||jd||jd||jddS)Nrrr^)rr)rrsizerrUwidthheightr r9r9r: test_sizer zTestRect.test_sizecC$tdddd}d|_||ddS)Nrrrr)rr2r#r)rrrUr r9r9r:test_set_size_with_tuplez!TestRect.test_set_size_with_tuplecCs*tdddd}tdd|_||ddS)Nrrrrrr")rrrrUr r9r9r:test_set_size_with_vectorrfz"TestRect.test_set_size_with_vectorcCsftdddd}||jt||jd||jd||jd||jd||jddS)Nrrrr)rrr) rrcentroidrrUxy centroid_x centroid_yr r9r9r: test_centroidszTestRect.test_centroidcCr!)Nrrrr)r.r$ri&iX)rr(rUr r9r9r:test_set_centroid_with_tupler&z%TestRect.test_set_centroid_with_tuplecCs*tdddd}tdd|_||ddS)Nrrrrr.r/)rrr(rUr r9r9r:test_set_centroid_with_vectorrfz&TestRect.test_set_centroid_with_vectorcCs"tdddd}|t|ddS)Nrrrrz2Rect(left: 100, top: 150, right: 200, bottom: 250))rrUreprr r9r9r: test_reprszTestRect.test_reprcCs*tdddd}||tdddddS) Nrrr#rK})r from_pointrUr r9r9r:test_from_pointszTestRect.test_from_pointN)rrrr rrrr r%r'r-r0r1r3r9r9r9r9r:rs  rc@s4eZdZddZddZddZddZd d Zd S) TestGetImageFormcC&tt}|t|jgddS)N) r1r2 description collectiontags focal_point_x focal_point_yfocal_point_widthfocal_point_height)rrrUlist base_fieldskeysr8formr9r9r: test_fields  zTestGetImageForm.test_fieldscCr;)N) r1r2r<r=r>r?r@rArBcaption fancy_caption)rr#rUrCrDrErFr9r9r: test_admin_form_fields_attributerIz1TestGetImageForm.test_admin_form_fields_attributecCs4tt}||jdt||jdjtjdS)Nr2)r WagtailImagerrDrwidgetr FileInputrFr9r9r:test_file_fieldsz TestGetImageForm.test_file_fieldcCs2tt}||jdt||jdjtdS)Nr>)rrMrrDrrNrrFr9r9r:test_tags_fieldsz TestGetImageForm.test_tags_fieldcCstt}||jdtj||jdtj||jdtj||jdtj||jdjtj||jdjtj||jdjtj||jdjtjdS)Nr?r@rArB)rrMrrDr IntegerFieldrN HiddenInputrFr9r9r:test_focal_point_fields s& z(TestGetImageForm.test_focal_point_fieldsN)rrrrHrLrPrQrTr9r9r9r:r:s  r:c@r) TestRenditionFilenamescC4tjjdtddd}|d}||jjddS)Nr/z test_rf1.pngrr0z width-100zimages/test_rf1.width-100.pngrr4r5r) get_renditionrUr2r{r8r6 renditionr9r9r:test_normal_filter&s  z)TestRenditionFilenames.test_normal_filtercCrV)Nr/z test_rf2.pngrr0 fill-100x100z)images/test_rf2.2e16d0ba.fill-100x100.pngrWrYr9r9r:test_fill_filter/s z'TestRenditionFilenames.test_fill_filtercCPtjjdtddd}|tdddd||d}||j j ddS) Nr/z test_rf3.pngrr0rrr\z)images/test_rf3.15ee4958.fill-100x100.png rr4r5r)set_focal_pointrsaverXrUr2r{rYr9r9r:!test_fill_filter_with_focal_point: z8TestRenditionFilenames.test_fill_filter_with_focal_pointcCr^) Nr/z test_rf4.pngrr0rrrgz4images/test_rf4.15ee4958.fill-200x200.height-150.pngr_rYr9r9r:!test_filter_with_pipe_gets_dottedHrcz8TestRenditionFilenames.test_filter_with_pipe_gets_dottedN)rrrr[r]rbrdr9r9r9r:rU!s   rUc@seZdZddZdS)TestDifferentUploadcCsDtjjddtd}tjjddtddd}||jj|jjdS)Nr/zA test description)r1r<r2z Test Imageblack)colour)r$r4r5r)rr2url)r8r6 second_imager9r9r:test_upload_pathXsz$TestDifferentUpload.test_upload_pathN)rrrrjr9r9r9r:reWs rec@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#r7r9r9r:test_custom_get_image_modelkz-TestGetImageModel.test_custom_get_image_modelcCs|tddS)z5Test get_image_model_string with a custom image modelrlN)rUrr7r9r9r:"test_custom_get_image_model_stringprpz4TestGetImageModel.test_custom_get_image_model_stringcCs"t`ddlm}|t|dS)z6Test get_image_model with no WAGTAILIMAGES_IMAGE_MODELrrN)rrmwagtail.images.modelsrrnr)r8rr9r9r:test_standard_get_image_modelus z/TestGetImageModel.test_standard_get_image_modelcCst`|tddS)z=Test get_image_model_STRING with no WAGTAILIMAGES_IMAGE_MODELzwagtailimages.ImageN)rrmrUrr7r9r9r:$test_standard_get_image_model_string}sz6TestGetImageModel.test_standard_get_image_model_string)signalreceiverztests.UnknownModelcC6|t tWddS1swYdS)z*Test get_image_model with an unknown modelNrXrrr7r9r9r:test_unknown_get_image_model "z.TestGetImageModel.test_unknown_get_image_modelzinvalid-stringcCrw)z1Test get_image_model with an invalid model stringNrxr7r9r9r:test_invalid_get_image_modelrzz.TestGetImageModel.test_invalid_get_image_modelN) rrrr rorqrsrtr&r rryr{r9r9r9r:rkjs&     rkc@s8eZdZeddZddZddZddZd d Zd S) TestWagtailImageFieldcCs(d|_t|jdj|_|jj|_dS)Nzwagtailimagefield.pngr)rr)r2r6 getbuffernbytes image_size)clsr9r9r: setUpTestDatasz#TestWagtailImageField.setUpTestDatacCsTt}|jdt|jd|jd|jd}||}||jt| |j ddS)NrrTr rr6seekrrr to_pythonrWillowImageFilerU content_typer8fr2rr9r9r: test_to_python_with_inmemoryfile  z6TestWagtailImageField.test_to_python_with_inmemoryfilecCsTt}|jdt|jd|jd|jd}||}||jt| |j ddS)NrrTz image/jpegrrrr9r9r:,test_to_python_gets_content_type_from_willowrzBTestWagtailImageField.test_to_python_gets_content_type_from_willowcCst}tdd|jd/}|jd||j|d||}||jt | |j dWddS1s=wYdS)N test_temp.pngrr) rr rr6rwritereadrrrrUr)r8rtmp_filerr9r9r:"test_to_python_with_temporary_files    "z8TestWagtailImageField.test_to_python_with_temporary_filec Csd}t}tdddd#}|t| ||Wdn1s#wYWdn1s2wY|td||jWddS1sOwYdS)NzYUpload a valid image. The file you uploaded was either not an image or a corrupted image.rr z;No file was submitted. Check the encoding type on the form.)rr assertRaisesMessagerrr6)r8rrrr9r9r:3test_to_python_raises_error_with_invalid_image_files "zITestWagtailImageField.test_to_python_raises_error_with_invalid_image_fileN) rrr classmethodrrrrrr9r9r9r:r|s   r|)Srriorrdjangorr django.confrdjango.core.exceptionsrrdjango.core.files.uploadedfilerr django.testr r django.test.signalsr django.urlsr taggit.formsrr willow.imagerrrrrwagtail.imagesrrwagtail.images.fieldsrwagtail.images.formatsrrrwagtail.images.formsrrrrrMwagtail.images.permissionsrwagtail.images.rectrrwagtail.images.utilsr r!wagtail.images.views.server"wagtail.test.testapp.modelsr#r$wagtail.test.utilsr%r&wagtail.utils.deprecationr'utilsr)r*r+sendfiler ImportErrorr,rrrrrrr:rUrerkr|r9r9r9r:s`            'L'GK6,