o h&@s~ddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZdd lmZmZmZmZdd lmZdd lmZdd lmZm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5m6Z6m7Z7m8Z8m9Z9ddl:m;Z;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFmGZGmHZHe"dZIGdd d e@eZJGd!d"d"e@eZKGd#d$d$e@eZLGd%d&d&eBe@eZMGd'd(d(e@eZNGd)d*d*eBe@eZOed+d,Gd-d.d.e@eZPGd/d0d0e@eZQGd1d2d2e@eZRGd3d4d4e@eZSGd5d6d6e@eZTGd7d8d8e@eZUGd9d:d:e@eZVGd;d<dd>e@eZXGd?d@d@e@eZYGdAdBdBeBe@eZZed+d,GdCdDdDe@eZ[edEd,GdFdGdGe@eZ\GdHdIdIeBe@eZ]GdJdKdKe@eZ^GdLdMdMe@eZ_GdNdOdOe@eZ`GdPdQdQe@eZadS)RN)patch)settings)Group Permission) ContentType)SimpleUploadedFileTemporaryUploadedFile)In)filesizeformat)render_to_string)RequestFactoryTestCaseTransactionTestCaseoverride_settings)reverse) force_str)escapeescapejs)RFC3986_SUBDELIMS urlencode) mark_safe)capfirst) OptimizerBase)registry)AdminURLFinder)get_image_model)generate_signature)ImagesFilterSet) CollectionGroupCollectionPermissionPage UploadedFileget_root_collection_id) CustomImageCustomImageWithAuthor EventPageVariousOnDeleteModel)WagtailTestUtils)AdminTemplateTestUtils)local_datetime)Imageget_test_image_fileget_test_image_file_svgz/~:@c@seZdZddZifddZddZddZd d Zd d Zd dZ ddZ ddZ e ddddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Ze d'd(d)iid*d+d,Zd-d.Zd/S)0TestImageIndexViewcCs>||_tjjdtddd|_tjjdtddd|_dS)N a cute kittenr*r*sizetitlefile a cute puppy)loginuserr+objectscreater, kitten_image puppy_imageselfr?e/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/images/tests/test_admin_views.pysetUp3s  zTestImageIndexView.setUpcC|jtd|SNwagtailimages:indexclientgetrr>paramsr?r?r@rG>zTestImageIndexView.getcCsV|}||jd||d||d||d||d||ddS)Nwagtailimages/images/index.html Add an imageThis field is required.r/r6)rG assertEqual status_codeassertTemplateUsedassertContainsassertNotContainsr>responser?r?r@ test_simpleAs    zTestImageIndexView.test_simplecCsb|ddi}||jd||jdd||d||d||d||ddS) NqrK query_stringrMrNr/r6rGrOrPcontextrRrSrTr?r?r@ test_empty_qLs   zTestImageIndexView.test_empty_qcCs2gd}|D]}|d|i}||jdqdS)N)01z-19999z Not a pageprKrGrOrP)r>pagespagerUr?r?r@test_paginationWs z"TestImageIndexView.test_paginationcCst}|jdd}tddD]}tjjd|tdd|d|_q| |j d d }| |j d |j d }|d |j |vpFd|j |v|d|j |vpWd|j |vdS)N Evil plansnamer*B Test image %ir0r1r4r5 collection) collection_idr`rKutf8z?p=1&collection_id=%iz?collection_id=%i&p=1z?p=3&collection_id=%iz?collection_id=%i&p=3)rget_first_root_node add_childranger+r9r:r,imagerGidrOrPcontentdecode assertTrue)r>root_collectionevil_plans_collectionirU response_bodyr?r?r@&test_pagination_preserves_other_params]s&     z9TestImageIndexView.test_pagination_preserves_other_paramscCsV|ddi}||jd|j}||djd|j||djd|jdS)Norderingr4rKpage_objrr*)rGrOrPr[ object_listr;r<)r>rUr[r?r?r@test_order_by_titlexs z&TestImageIndexView.test_order_by_titlecCsHgd}|D]}|d|i}||jd|j}||d|qdS)N)r4z-title created_at -created_at file_sizez -file_sizer|rKcurrent_orderingrGrOrPr[)r> orderingsr|rUr[r?r?r@test_valid_orderingssz'TestImageIndexView.test_valid_orderingscCsj|ddi}||jd|j}d}||d|||djd|j||djd|jdS) Nr|bogusrKrrr}rr*)rGrOrPr[r~r<r;)r>rUr[default_orderingr?r?r@7test_default_ordering_used_if_invalid_ordering_providedszJTestImageIndexView.test_default_ordering_used_if_invalid_ordering_provided)WAGTAILIMAGES_INDEX_PAGE_SIZEcCsVddtddD}tj||}||jd|jdj}|t |ddS)NcS"g|] }td|tdddqSrir0r1r3r+r,.0ryr?r?r@ zDTestImageIndexView.test_default_entries_per_page..r*!rKr}r rqr+r9 bulk_createrGrOrPr[r~len)r>imagesrUr~r?r?r@test_default_entries_per_pages  z0TestImageIndexView.test_default_entries_per_pagecCsZddtddD}tj|d}|}||jd|jdj}|t ||dS)NcSrrrrr?r?r@rrzQTestImageIndexView.test_default_entries_per_page_uses_default..r*rrKr}r)r>rdefault_num_entries_per_pagerUr~r?r?r@*test_default_entries_per_page_uses_defaults  z=TestImageIndexView.test_default_entries_per_page_uses_defaultcCs\t}|jdd|jdd|}||j}|d}|dd|DgddS)Nrerf Good plansz@select[name="collection_id"] option[value]:not(option[value=""])cSsg|] }|jdddqS)Tstripu↳ )get_textlstrip)rrkr?r?r@rsz.)Rootrer)rrorprGget_souprtselectrO)r>rwrUsoupcollection_optionsr?r?r@test_collection_orders   z(TestImageIndexView.test_collection_ordercC8t}|jdd}|jdd|}||ddSNrerf Eviler plans,    ↳ Eviler plansrrorprGrRr>rw evil_plansrUr?r?r@test_collection_nesting   z*TestImageIndexView.test_collection_nestingcCst}|jdd}tjjdtdd|d}|d|ji}| |j dt d |jfd }t j |j}|||d |dS) Nrerf Test imager0r1rjrmrKwagtailimages:editargsz?next=)rrorpr+r9r:r,rGrsrOrPrurllibparsequote_request get_full_pathrR)r>rwrxrrrUedit_urlnext_urlr?r?r@&test_edit_image_link_contains_next_urls z9TestImageIndexView.test_edit_image_link_contains_next_urlcCs~tjjdtd}|jdd|}||jd| |j }| d}| || d}| dd |DddgdS) NTest image with two tagsr3onetworKzinput[name=tag][checked] #id_tag labelcSsg|]}|jddqS)Trr)rtagsr?r?r@rsz0TestImageIndexView.test_tags..)r+r9r:r,raddrGrOrPrrtr assertFalseassertCountEqual)r>image_two_tagsrUr current_tagsrr?r?r@ test_tagss     zTestImageIndexView.test_tagscCsTtjjdtdtjjdtd}|jdtjjdtd}|jddtjjdtd}|jd|}||jd j j d |d di}||jd j j d |d di}||jd j j d |d ddgi}||jd j j d| |j }| d}|dd|Dddg| d}|dd|DddgdS)NTest image with no tagsr3Test image with one tagrrrzTest image with a different tag unrelatedr}tagrlr*rcSs&g|]}|ddur|jddqS)zinput[checked]NTr) select_oner)rrr?r?r@r+s  z9TestImageIndexView.test_tag_filtering..z"[data-w-active-filter-id="id_tag"]cSsg|] }|jdddqS) T) separatorrr)rfilterr?r?r@r6szTag: onezTag: unrelated)r+r9r:r,rrrGrOr[ paginatorcountrrtrr)r> image_one_tagrimage_unrelated_tagrUrractive_filtersr?r?r@test_tag_filteringsP      z%TestImageIndexView.test_tag_filteringcCstddD]}tjjd|tddd}|ddkr$|jd |q|d dd }| |j d |j d }| d |vpCd|v| d|vpNd|vdS)Nr*rir0r1r3rlreven)rr`rKrnz?p=1&tag=evenz?tag=even&p=1z?p=3&tag=evenz?tag=even&p=3)rqr+r9r:r,rrsaverGrOrPrtrurv)r>ryrrrUrzr?r?r@)test_tag_filtering_preserves_other_params:s$   zr*T)rallow_extra_attrs)rGrtrurassertTagInHTML)r>rUhtml search_urlr?r?r@test_search_form_renderedRs   z,TestImageIndexView.test_search_form_rendereddefaultBACKEND+django.core.cache.backends.dummy.DummyCacheCACHEScCs||d |Wdn1swYtdD]}tjjd|tddd|_q!|d |Wdn1sFwY|d |WddS1s`wYdS)N rir0r1r3 )rGassertNumQueriesrqr+r9r:r,rrr>ryr?r?r@test_num_queries_s       "z#TestImageIndexView.test_num_queriescCsvtjjdtd}|jdttd}|j |_ t itj |dd}dd|j j jjD}|t|d dS) Nrr3rrDT)dataquerysetrequest is_searchingcSsg|] }t|tr|qSr?) isinstancer )rclauser?r?r@rs zZTestImageIndexView.test_empty_tag_filter_does_not_perform_id_filtering..r)r+r9r:r,rrr rGrr8rallqsquerywherechildrenrOr)r>rr filterset in_clausesr?r?r@3test_empty_tag_filter_does_not_perform_id_filtering{s  zFTestImageIndexView.test_empty_tag_filter_does_not_perform_id_filteringN)__name__ __module__ __qualname__rArGrVr\rdr{rrrrrrrrrrrrrrrr?r?r?r@r.2s2     =   r.c@sNeZdZdgZddZifddZddZdd Zd d Zd d Z ddZ dS)TestImageIndexViewSearchtest_empty.jsoncCsp|tjjdtddd|_tjjdtddd|_tddd|j_|j tdd d |j_|j dS) Nr/r0r1r3r6ir*irl) r7r+r9r:r,r;r<r)rrr=r?r?r@rAs zTestImageIndexViewSearch.setUpcCrBrCrErHr?r?r@rGrJzTestImageIndexViewSearch.getcCJ|ddi}||jd||jdd||d||ddS)NrWkittenrKrYr/r6rZrTr?r?r@ test_search  z$TestImageIndexViewSearch.test_searchcCr)NrWkitrKrYr/r6rZrTr?r?r@test_search_partial_matchrz2TestImageIndexViewSearch.test_search_partial_matchc Cs6t}|jdd|jddg}ddg}g}tdD](}tjj||dd|td d ||dd |_|dd krB| |jq| d|d j d }|j }|j d}|j dj} |jd} ||d||d||| td ddD] }|d|| qztd} ||d| d|d jddS)NzBaker CollectionrfzOther CollectionBakerOther rlrr0r1rjr)rWrmrYr}utf-8rKzBaker %iwagtailimages:add_multiplez rwchild_collection title_list answer_listryrUrPrY response_listrzurlr?r?r@test_collection_query_searchs<            z5TestImageIndexViewSearch.test_collection_query_searchcCstjjdtddd}tddd|_||ddd }||j d ||j d d|t |j d j ||j |jg||j}|d }||||dd||jddddS)Nzdecades old cute tortoiser0r1r3ir*cuter)rWr|rKrYr}z(select[name="ordering"] option[selected]valueTrOldest)r+r9r:r,r)rrrGrOrPr[listr~r;r<rrtrassertIsNotNoner)r> old_imagerUroptionr?r?r@#test_search_and_order_by_created_ats$    zrrrUr?r?r@#test_tag_filtering_with_search_terms  zr(rUr?r?r@rsz'TestImageListingResultsView.test_searchN)rrrr!rArGrr?r?r?r@r#   r#c@seZdZddZifddZifddZddZd d Zd d Ze d dddZ ddZ ddZ e dgdddZ ddZe iejdddiiddd Zd!d"Ze d#d$d%d&Ze d#d'd(d)Zd*d+Zd,d-Ze d dd.d/Zd0S)1TestImageAddViewcCr$r%r&r=r?r?r@rAr'zTestImageAddView.setUpcCrBNwagtailimages:addrErHr?r?r@rG"rJzTestImageAddView.getcCrBr+rFpostrr> post_datar?r?r@r.%rJzTestImageAddView.postcCsl|}||jd||d||d||d||d|tdddd d dg|jdS) NrKwagtailimages/images/add.htmlKTry this.rr[rrrFrGrOrPrQrrtr;rr r8 context_datacopyrr r r>r template_namerUrresponse_contentrr[rr?r?r@$test_add_error_max_file_size_escaped (       z>TestMultipleImageUploader.test_add_error_max_file_size_escapedc Cr)Nr rrKz1Invalid image type. Get help.error_accepted_file_typesrrrr?r?r@*test_add_error_accepted_file_types_escaped rzDTestMultipleImageUploader.test_add_error_accepted_file_types_escapedcCs|jtddtdtjd}||jd||dd| |d| d |j ||j d j d| |j d j| |j d jtjjdd }|d |j| d |j| d |j ||j dd|j d j||j dd|j d j||j d jd dt|j}| d || d|| d|| d|||d|j d j| |d||ddS)i This tests that a POST request to the add view saves the image and returns an edit form r  test titlerAr4files[]rKrr3wagtailadmin/generic/multiple_upload/edit_form.htmlrrrBr4.pngrP edit_action/admin/images/multiple/%d/ delete_action!/admin/images/multiple/%d/delete/image_idsuccess duplicateN)rFr.rrr,r5rErOrPrQrr[r4rvrrJrr9rGr;rrsinitialr9r:rtrurr>rUrrr=r?r?r@ test_add_post sL      z'TestMultipleImageUploader.test_add_postcCs|jjtddtdtjidd}||jd||dd| d |j | |j d j | |j d j tjjdd }|d|j| d |j| d |j ||j d jd dt|j}| d || d|dS)zv A POST request to the add view without the title value saves the image and uses file title if needed r rz no-title.pngXMLHttpRequest)HTTP_X_REQUESTED_WITHrKrrrrrBrrPr4rN)rFr.rrr,r5rErOrPrr[rvrrJrr9rGrrr9r:rtrurr?r?r@test_add_post_no_title s*   z0TestMultipleImageUploader.test_add_post_no_titlecCs$|jtdi}||jddS)zX This tests that the add view checks for a file when a user POSTs to it r N)rFr.rrOrPrTr?r?r@test_add_post_nofile7 sz.TestMultipleImageUploader.test_add_post_nofilecC|jtddtddi}||jd||ddt|j }| d|| d || d || d || |d ||d d d SzQ The add view must check that the uploaded file is a valid image r rrAsThis is not an image!rKrrrrPr error_messagerN rFr.rrrOrPr9r:rtrur;rrrSr?r?r@test_add_post_badfile@ "     z/TestMultipleImageUploader.test_add_post_badfiler?r>rUcCsD|jtddtdtji}|jtddtdtji}||jd||ddt |j }| d|| d ||d ||d |||d ||d d ||jd||ddt |j }| d|| d ||d ||d |||d ||d d d S)zU The add view must check that the uploaded file extension is a valid r rztest.txtrArKrrrrPrrz:Not a supported image format. Supported formats: JPG, GIF.N)rFr.rrr,r5rErOrPr9r:rtrur;rr)r>rUpost_with_invalid_extensionr=r?r?r@test_add_post_bad_extension[ sP            z5TestMultipleImageUploader.test_add_post_bad_extensioncsdfdd }||dd}|jd|d|jdjdt|j} d | d | |d  |d d S)z When a duplicate image is saved, the add view shows that it's a duplicate and prompts user to confirm the upload. rcrNr rArrrBr=r?r@r rzETestMultipleImageUploader.test_add_post_duplicate..post_imagetest title duplicaterBrK2wagtailimages/images/confirm_duplicate_upload.htmlrrrPconfirm_duplicate_uploadrrNr) rOrPrQr[r4r9r:rtrurrvr>rrUr=r?r=r@test_add_post_duplicate s   z1TestMultipleImageUploader.test_add_post_duplicatecs tjjdd}tjjddd}|j|tjjtd}|j tdddt jj|tjjd d dd dfd d }|dj _ j j |j |dd}|jd|d|jdjdt|j}|d|dd|d|dS)z When a duplicate image is added but the user doesn't have permission to choose the original image, the add views lets the user upload it as if it weren't a duplicate. rFrfr|r}ryrrGrrwrxr~rcs*jtd|tdtjjdS)Nr rA)r4rrk)rFr.rrr,r5rErsrBrNr>r?r@r s zWTestMultipleImageUploader.test_add_post_duplicate_choose_permission..post_imageFrrBrKrrrrrrPrNr)rr9r:rrGrrrr"rprr8rrrrOrPassertTemplateNotUsedr[r4r9r:rtrurvrrr;)r>rKrLrMrrUr=r?rr@)test_add_post_duplicate_choose_permission s<    zCTestMultipleImageUploader.test_add_post_duplicate_choose_permissioncC,|jtd|jjfd}||jddS)zl This tests that a GET request to the edit view returns a 405 "METHOD NOT ALLOWED" response wagtailimages:edit_multiplerNrFrGrrrrsrOrPrTr?r?r@ test_edit_get z'TestMultipleImageUploader.test_edit_getcCs|jtd|jjfdd|jjdd|jjdi}||jd||dd t|j }| d || d || d |||d |jj| |d tjj|jjd }||jd| d|jdS)Q This tests that a POST request to the edit view edits the image rrimage-%d-title New title! image-%d-tagszcromarty, finisterrerKrrrrPrrcromartyN)rFr.rrrrsrOrPr9r:rtrurr;rvr+r9rGr4rnames)r>rUr=rrr?r?r@test_edit_post s"      z(TestMultipleImageUploader.test_edit_postcCs|jtd|jjfdd|jjdd|jjdi}||jd||dd||d ||j d d d t |j }|d ||d ||d|||d |jj||ddS)z This tests that a POST request to the edit page returns a json document with "success=False" and a form with the validation error indicated rrr rXrrKrrrrPr4rNrrN)rFr.rrrrsrOrPrQrRr[r9r:rtrurrrSr?r?r@test_edit_post_validation_error& s(       z9TestMultipleImageUploader.test_edit_post_validation_errorcCr)zn This tests that a GET request to the delete view returns a 405 "METHOD NOT ALLOWED" response wagtailimages:delete_multiplerrNrrTr?r?r@test_delete_getH r z)TestMultipleImageUploader.test_delete_getcCs|jtd|jjfd}||jd||dd|tj j |jjd t |j}|d||d|||d|jj||dd S) U This tests that a POST request to the delete view deletes the image rrrKrrrrrN)rFr.rrrrsrOrPrr+r9rrcr9r:rtrurrvrSr?r?r@test_delete_postT s  z*TestMultipleImageUploader.test_delete_postN)rrr__doc__rArKrrrrrrrrrrrr rrrrr?r?r?r@ry s*   6%  7'B " rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)-TestMultipleImageUploaderWithCustomImageModel} This tests the multiple image upload views located in wagtailimages/views/multiple.py with a custom image model cCr)Ntest-image.pngr3)r7r#r9r:r,rrr=r?r?r@rAs rz3TestMultipleImageUploaderWithCustomImageModel.setUpcCs:|jtd}||jd||d||ddS)rr rKrr4N)rFrGrrOrPrQrRrTr?r?r@rK| s z6TestMultipleImageUploaderWithCustomImageModel.test_addcCsr|jtddtdtji}||jd||dd| |d| d|j ||j dj d| |j dj| |j dj| d |j ||j d jd d| d |j d j|d |j d j||j d d|j dj||j dd|j djt|j}| d || d|| d|| |d||ddS)rr rrArKrrrrrrPr4captionnot_editable_fieldrrrrrrN)rFr.rrr,r5rErOrPrQrr[r4rvrrJrfieldsr;rsr9r:rtrurrSr?r?r@r sD     z;TestMultipleImageUploaderWithCustomImageModel.test_add_postcCrrrrSr?r?r@r rzCTestMultipleImageUploaderWithCustomImageModel.test_add_post_badfilecsdfdd }||dd}|jd|d|jdjdd |jd jd |jd jt |j }d |d | |d  |ddS)NrcrrrrBr=r?r@r rzYTestMultipleImageUploaderWithCustomImageModel.test_add_post_duplicate..post_imagerrBrKrrrrrPrrrrr) rOrPrQr[r4rrr;r9r:rtrurvrr?r=r@r s   zETestMultipleImageUploaderWithCustomImageModel.test_add_post_duplicatecCst}|jdd}||j_|jtj}t j}|j t dt dtj|jd}tj}t j}||||||d||jd||dd ||d d S) z~ If unique_together validation fails, create an UploadedFile and return a form so the user can fix it holiday snapsrfr r)rrkr*rKrrrN)rrorprrrkrr#r9rr!rFr.rrr,r5rErsrOrPrQ)r>rwnew_collectionimage_count_beforeuploaded_image_count_beforerUimage_count_afteruploaded_image_count_afterr?r?r@rs s.       zSTestMultipleImageUploaderWithCustomImageModel.test_unique_together_validation_errorc Cs|jtd|jjfdd|jjdd|jjdd|jjdi}||jd ||d d t|j }| d || d || d|||d |jj| |dtjj|jjd}||jd||jd| d|jdS)r rrr r rzfootwear, dystopiaimage-%d-captionz(a boot stamping on a human face, foreverrKrrrrPrrfootwearN)rFr.rrrrsrOrPr9r:rtrurr;rvr#r9rGr4rrr)r>rUr=rr?r?r@r s(       zrwrrUr=r?r?r@*test_edit_fails_unique_together_validation@ s6     zXTestMultipleImageUploaderWithCustomImageModel.test_edit_fails_unique_together_validationcCs|jtd|jjfd}||jd||dd|tj j |jjd t |j}|d||d|||d|jj||d|tj j |jjdd d S) rrrrKrrrrrrN)rFr.rrrrsrOrPrr+r9rrcr9r:rtrurrvr#rrSr?r?r@rj s  "z>TestMultipleImageUploaderWithCustomImageModel.test_delete_postN) rrrrrArKrrrrsrr&rr?r?r?r@rl s 0$%" *rztests.CustomImageWithAuthorc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)1TestMultipleImageUploaderWithCustomRequiredFieldsrcCs0||_tjjtjtt|jd|_ dS)N)for_content_typer5r^) r7r8r!r9r:r get_for_modelrr,uploaded_imager=r?r?r@rA s   z7TestMultipleImageUploaderWithCustomRequiredFields.setUpcCs.|jtd}||jd||ddS)rr rKrN)rFrGrrOrPrQrTr?r?r@rK sz:TestMultipleImageUploaderWithCustomRequiredFields.test_addcCsdtj}tj}|jtddtdtj i}tj}tj}| ||| ||d| |j d| |dd| |d|d |j||jd j j|d |j| |jd jd d|d |jd j| |jd d|jd j| |jdd|jd jt|j}|d ||d|||ddS)z A POST request to the add view should create an UploadedFile rather than an image, as we do not have enough data to pass CustomImageWithAuthor's validation yet r rrAr*rKrrrr*rPr4authorr5/admin/images/multiple/create_from_uploaded_image/%d/r(/admin/images/multiple/delete_upload/%d/rN)r$r9rr!rFr.rrr,r5rErOrPrQrr[rvrgrrrsr9r:rtrur>r r!rUr"r#r=r?r?r@r sN         z?TestMultipleImageUploaderWithCustomRequiredFields.test_add_postcCrrrrSr?r?r@r rzGTestMultipleImageUploaderWithCustomRequiredFields.test_add_post_badfilec CsNtj}tj}|jtd|jjfdd|jjdd|jjdd|jjdi}tj}tj}| ||| ||| |j d| |d d | d |j | d |j d j | |j d d|j dj| |j dd|j dj||j d d dt|j}| d || d|d ||ddS)z Posting an invalid form to the create_from_uploaded_image view throws a validation error and leaves the UploadedFile intact 1wagtailimages:create_multiple_from_uploaded_imageruploaded-image-%d-titler uploaded-image-%d-tagsrXuploaded-image-%d-authorrKrrrPr+rr,r*rr-rNrN)r$r9rr!rFr.rr*rsrOrPrr[rrRr9r:rtrurr.r?r?r@$test_create_from_upload_invalid_post sN            zVTestMultipleImageUploaderWithCustomRequiredFields.test_create_from_upload_invalid_postc CsVtj}tj}|jtd|jjfdd|jjdd|jjdd|jjdi}tj}tj}| |j d | |d d t |j }|d |||d | ||d| ||dtjj|d d}| |jd| |jd||jj||j||j| |jd| |jd|d|jdS)zc Posting a valid form to the create_from_uploaded_image view will create the image r/rr0r r1zabstract, squaresr2z Piet MondrianrKrrrrr*rrCrDabstractN)r$r9rr!rFr.rr*rsrOrPr9r:rtrurrvrGr4r+r5rgrJrrHrIrr)r>r r!rUr"r#r=rrr?r?r@test_create_from_upload+ s>          zITestMultipleImageUploaderWithCustomRequiredFields.test_create_from_uploadcCsv|jtd|jjfd}||jd||dd|tj j |jjd t |j}||ddS) z\ This tests that a POST request to the delete view deletes the UploadedFile z$wagtailimages:delete_upload_multiplerrKrrrrN)rFr.rr*rsrOrPrr!r9rrcr9r:rtrurvrSr?r?r@test_delete_uploaded_image[ s zLTestMultipleImageUploaderWithCustomRequiredFields.test_delete_uploaded_imageN) rrrrrArKrrr3r5r6r?r?r?r@r' s  87 0r'c@s$eZdZddZddZddZdS)TestURLGeneratorViewcC"tjjdtd|_||_dSrr+r9r:r,rrr7r8r=r?r?r@rAu zTestURLGeneratorView.setUpcCs~|jtd|jjfd}||jd||d||d|tdddtd |jjfdd dd d d d g|j dS)l This tests that the view responds correctly for a user with edit permissions on this image wagtailimages:url_generatorrrKz'wagtailimages/images/url_generator.htmlzwagtailadmin/generic/base.htmlrDr5r6rrrXz Generate URL)rr7sublabelN) rFrGrrrrsrOrPrQr8rtrTr?r?r@r9 s    zTestURLGeneratorView.test_getcCsZd|j_|jjtjjddd|j|jt d|j j fd}| |t ddS) z This tests that the view returns a "permission denied" redirect if a user without correct permissions attempts to access it Fr|r}ryr<rrN) r8rrrrr9rGrrFrrrrsrFrTr?r?r@test_get_bad_permissions s z-TestURLGeneratorView.test_get_bad_permissionsN)rrrrAr9r>r?r?r?r@r7t s r7c@s4eZdZddZddZddZddZd d Zd S) TestGenerateURLViewcCr8rr9r=r?r?r@rA r:zTestGenerateURLView.setUpcCs|jtd|jjdfd}||jd||ddt|j }|t | ddhd t jjt|jjdtd |jjd }||d|td |jjdfd}||d|d S)r;wagtailimages:generate_url fill-800x600rrKrrr preview_urlz@http://localhost/images/%(signature)s/%(image_id)d/fill-800x600/)safe) signaturerwagtailimages:previewN)rFrGrrrrsrOrPr9r:rtrurxryrrrrurlquote_safechars)r>rU content_jsonrexpected_preview_urlr?r?r@r9 s*   zTestGenerateURLView.test_getcCsd|j_|jjtjjddd|j|jt d|j j dfd}| |j d| |d d ||jtd d id S)zr This tests that the view gives a 403 if a user without correct permissions attempts to access it Fr|r}ryr@rArirrerrorz s& z,TestGenerateURLView.test_get_bad_permissionscCs^|jtd|jjddfd}||jd||dd||j t dd id S) z~ This tests that the view gives a 404 response if a user attempts to use it with an image which doesn't exist r@r*rArrWrrrIzCannot find image.N rFrGrrrrsrOrPrJrtrur9rKrTr?r?r@test_get_bad_image sz&TestGenerateURLView.test_get_bad_imagecCsZ|jtd|jjdfd}||jd||dd||j t ddid S) zz This tests that the view gives a 400 response if the user attempts to use it with an invalid filter spec r@bad-filter-specrrrrrIzInvalid filter spec.NrLrTr?r?r@test_get_bad_filter_specs z,TestGenerateURLView.test_get_bad_filter_specN)rrrrAr9r>rMrOr?r?r?r@r? s  # r?c@r) TestPreviewViewcCr8rr9r=r?r?r@rA2r:zTestPreviewView.setUpcCs>|jtd|jjdfd}||jd||dddS)z6 Test a valid GET request to the view rErArrKrrYNrrTr?r?r@r9<s zTestPreviewView.test_getcCs~Gdddt}ttd|g|jtd|jjdfd}Wdn1s)wY| |j d| |d d dS) z\ Test that preview works with optimizers Willow optimizers require c@s2eZdZdZdZeddZedefddZdS) zCTestPreviewView.test_preview_with_optimizer..DummyOptimizerdummyrAcSsdS)NTr?)clsr?r?r@ check_libraryTzQTestPreviewView.test_preview_with_optimizer..DummyOptimizer.check_library file_pathcSsdSr%r?)rRrUr?r?r@processXrTzKTestPreviewView.test_preview_with_optimizer..DummyOptimizer.processN) rrr library_name image_format classmethodrSrrVr?r?r?r@DummyOptimizerPs rZ_registered_optimizersrErArNrKrrY) rrobjectrrFrGrrrrsrOrP)r>rZrUr?r?r@test_preview_with_optimizerIs z+TestPreviewView.test_preview_with_optimizercCs.|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. rErNrrNrrTr?r?r@test_get_invalid_filter_specfs z,TestPreviewView.test_get_invalid_filter_specN)rrrrAr9r]r^r?r?r?r@rP1s   rPc@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestEditOnlyPermissionscCstjjdtd|_|jdddd}tjjddd }tjjd d d }tjjd d }|j |t jj|t |d|j ||jddddS)Nrr3r"r#rrrwr$ryr|r}r%rfr~r)r+r9r:r,rrrrrGrrrrrrorr7)r>r8r'rimage_changers_groupr?r?r@rAys,  zTestEditOnlyPermissions.setUpcCsF|jtd}||jd||d||d||ddS)NrDrKrLrMr)rFrGrrOrPrQrSrRrTr?r?r@test_get_indexs   z&TestEditOnlyPermissions.test_get_indexcCs:|jtdddi}||jd||jdddS)NrDrWrUrKrY)rFrGrrOrPr[rTr?r?r@rsz#TestEditOnlyPermissions.test_searchcC$|jtd}||tddS)Nr,rrFrGrrFrTr?r?r@ test_get_addz$TestEditOnlyPermissions.test_get_addcC8|jtd|jjfd}||jd||ddS)NrrrKrrFrGrrrrsrOrPrQrTr?r?r@ test_get_editsz%TestEditOnlyPermissions.test_get_editcCrf)NrrrKrrgrTr?r?r@test_get_deletes z'TestEditOnlyPermissions.test_get_deletecCrb)Nr rrcrTr?r?r@test_get_add_multiplerez-TestEditOnlyPermissions.test_get_add_multipleN) rrrrArarrdrhrirjr?r?r?r@r_xs  r_c@seZdZddZddZdS)TestImageAddMultipleViewcCs6||jtd}||jd||ddS)Nr rKr)r7rFrGrrOrPrQrTr?r?r@test_as_superusersz*TestImageAddMultipleView.test_as_superusercCs|jddd}tjjddd}tjjddd}tjjd d }|j|tjj|t |d |j ||j ddd|j td }||jd ||ddS)Neditorrrrwrxryr|r}r*rfr~r rKr)rrr9rGrr:rrrrrorr7rFrrOrPrQ)r>r8r,rimage_adders_grouprUr?r?r@test_as_ordinary_editors&  z0TestImageAddMultipleView.test_as_ordinary_editorN)rrrrlror?r?r?r@rks rk)brr9r unittest.mockr django.confrdjango.contrib.auth.modelsrr"django.contrib.contenttypes.modelsrdjango.core.files.uploadedfilerrdjango.db.models.lookupsr django.template.defaultfiltersr django.template.loaderr django.testr r rr django.urlsrdjango.utils.encodingrdjango.utils.htmlrrdjango.utils.httprrdjango.utils.safestringrdjango.utils.textrwillow.optimizers.baserwillow.registryrwagtail.admin.admin_url_finderrwagtail.imagesrwagtail.images.utilsrwagtail.images.views.imagesrwagtail.modelsrrr r!r"wagtail.test.testapp.modelsr#r$r%r&wagtail.test.utilsr'!wagtail.test.utils.template_testsr(wagtail.test.utils.timestampsr)utilsr+r,r-rFr.rr#r*rtrrrrr.rmror}rrrrrr'r7r?rPr_rkr?r?r?r@s                   btVP f}e0-_!Ovo<GE