o h)1@s>ddlmZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z mZddlmZddlmZmZmZmZdd lmZdd lmZmZdd lmZmZdd lmZGd ddeZ Gdddee Z!Gddde Z"GdddeZ#edgdGddde Z$eddGddde#Z%Gdddee Z&dS) )settings)Group Permission)ImproperlyConfiguredValidationError) ContentFile) transaction)TestCaseTransactionTestCase)override_settings)get_document_modelget_document_model_stringmodelssignal_handlers)get_test_image_file) CollectionGroupCollectionPermission)CustomDocumentReimportedDocumentModel)WagtailTestUtilsc@s*eZdZdgZddZddZddZdS) TestDocumentQuerySettest_empty.jsoncCs4tjjjdd}tjjd}|t||gdS)N Test documenttitleTest)rDocumentobjectscreatesearch assertEquallist)selfdocumentresultsr%c/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/documents/tests/test_models.pytest_search_methodsz'TestDocumentQuerySet.test_search_methodcCsztjjjdd}tjjjdd}tjjjddd}|t||gtjjjddd}t|dd d }||||gdS) NAAA Test documentrZZZ Test documentzaaa testand)operatororcSs|jSNr)docr%r%r&(sz5TestDocumentQuerySet.test_operators..)key)rrrrrr r!sorted)r" aaa_document zzz_documentr$sorted_resultsr%r%r&test_operators sz#TestDocumentQuerySet.test_operatorscCs|tjjjdd}tjjjdd}tjjdjddd}|t|||gtjjdjddd}|t|||gdS) Nr(rr)rrF)order_by_relevancez-title)rrrrorder_byrr r!)r"r2r3r$r%r%r&test_custom_ordering+sz)TestDocumentQuerySet.test_custom_orderingN)__name__ __module__ __qualname__fixturesr'r5r8r%r%r%r&rs  rc@s4eZdZddZddZddZddZd d Zd S) TestDocumentPermissionscCs|jdddd|_|jdddd|_|jdddd|_|jjtjjd d |j d d dd|_ tjj d d |_ t jj |j ttjjddd|jj|j tjjj d|jd|_dS)Nuserzuser@email.compassword)usernameemailr?ownerzowner@email.comeditorzeditor@email.comEditors)name administratorzadministrator@email.comzDocument adders add_document)codename)group collection permissionr)ruploaded_by_user) create_userr>rBrCgroupsaddrrgetcreate_superuserrFr adders_grouprrget_first_root_noderrrr#r"r%r%r&setUp:s2  zTestDocumentPermissions.setUpcC||j|jdSr-) assertTruer#is_editable_by_userrFrTr%r%r&test_administrator_can_editZz3TestDocumentPermissions.test_administrator_can_editcCrVr-)rWr#rXrCrTr%r%r&test_editor_can_edit]rZz,TestDocumentPermissions.test_editor_can_editcCrVr-)rWr#rXrBrTr%r%r&test_owner_can_edit`rZz+TestDocumentPermissions.test_owner_can_editcCrVr-) assertFalser#rXr>rTr%r%r&test_user_cant_editcrZz+TestDocumentPermissions.test_user_cant_editN)r9r:r;rUrYr[r\r^r%r%r%r&r=9s   r=c@sDeZdZddZddZddZddZd d Zd d Zd dZ dS)TestDocumentFilenamePropertiescCsjtjdd|_|jjdtdtjdd|_|jjdtdtjdd|_|jjdtddS)Nrrsample_name.doczA boring example documentsample_name.pdf sample_name)rrr#filesaver pdf_documentextensionless_documentrTr%r%r&rUhsz$TestDocumentFilenameProperties.setUpcC4|d|jj|d|jj|d|jjdS)Nr`rarb)r r#filenamererfrTr%r%r& test_filename{z,TestDocumentFilenameProperties.test_filenamecCrg)Nr.pdf)r r#file_extensionrerfrTr%r%r&test_file_extensionrjz2TestDocumentFilenameProperties.test_file_extensioncCrg)Nzapplication/mswordzapplication/pdfzapplication/octet-stream)r r# content_typererfrTr%r%r&test_content_types z0TestDocumentFilenameProperties.test_content_typecCs:||jd||jd||jddS)N(7d8c4778b182e4f3bd442408c64a6e22a4b0ed85)r r# get_file_hashrerfrTr%r%r&test_file_hashs z-TestDocumentFilenameProperties.test_file_hashcCrg)NzFattachment; filename=sample_name.doc; filename*=UTF-8''sample_name.docinlinez>attachment; filename=sample_name; filename*=UTF-8''sample_name)r r#content_dispositionrerfrTr%r%r&test_content_dispositionsz7TestDocumentFilenameProperties.test_content_dispositioncCs(|jj|jj|jjdSr-)r#rcdeletererfrTr%r%r&tearDowns  z'TestDocumentFilenameProperties.tearDownN) r9r:r;rUrirnrprsrvrxr%r%r%r&r_gs r_c@seZdZdZdgZddZdS) TestFilesDeletedForDefaultModelsa Because we expect file deletion to only happen once a transaction is successfully committed, we must run these tests using TransactionTestCase per the following documentation: Django's TestCase class wraps each test in a transaction and rolls back that transaction after each test, in order to provide test isolation. This means that no transaction is ever actually committed, thus your on_commit() callbacks will never be run. If you need to test the results of an on_commit() callback, use a TransactionTestCase instead. https://docs.djangoproject.com/en/1.10/topics/db/transactions/#use-in-tests rcCst.tjjdtd}|jj}||jj || ||jj |Wdn1s5wY| |jj |dS)Nz Test Imagerrc) ratomicr rrrrcrErWstorageexistsrwr])r"r#rhr%r%r&#test_document_file_deleted_oncommits  zDTestFilesDeletedForDefaultModels.test_document_file_deleted_oncommitN)r9r:r;__doc__r<r~r%r%r%r&rys ryrk)WAGTAILDOCS_EXTENSIONSc@s,eZdZddZddZddZddZd S) TestDocumentValidateExtensionscCs,tjjjddd|_tjjjddd|_dS)Nrztest.docrzztest.pdf)rrrrdocument_invaliddocument_validrTr%r%r&rUs  z$TestDocumentValidateExtensions.setUpcCsT|t }|jWdn1swYd}||jjdd|dS)a3 Checks if the uploaded document has the expected extensions mentioned in settings.WAGTAILDOCS_EXTENSIONS This is caught in form.error and should be raised during model creation when called full_clean. This specific testcase invalid file extension is passed NuEFile extension “doc” is not allowed. Allowed extensions are: pdf.rcr) assertRaisesrr full_cleanr exception message_dict)r"eexpected_messager%r%r&!test_create_doc_invalid_extensions z@TestDocumentValidateExtensions.test_create_doc_invalid_extensioncCs0z|jWdSty|dYdSw)a7 Checks if the uploaded document has the expected extensions mentioned in settings.WAGTAILDOCS_EXTENSIONS This is caught in form.error and should be raised during model creation when called full_clean. In this specific testcase invalid file extension is passed. z>Validation error is raised even when valid file name is passedN)rrrfailrTr%r%r&test_create_doc_valid_extensions   z>TestDocumentValidateExtensions.test_create_doc_valid_extensioncCs|jj|jjdSr-)rrcrwrrTr%r%r&rxs z'TestDocumentValidateExtensions.tearDownN)r9r:r;rUrrrxr%r%r%r&rs  rtests.CustomDocumentWAGTAILDOCS_DOCUMENT_MODELc@seZdZddZddZdS)TestFilesDeletedForCustomModelscCs tjjdddddtdS)NRoot0001r)rEpathdepthnumchild)rr get_or_createrregister_signal_handlersrTr%r%r&rUs z%TestFilesDeletedForCustomModels.setUpcCs&t}||jjd|jddS)N.r)r r _meta app_labelr9)r"clsr%r%r&test_document_model sz3TestFilesDeletedForCustomModels.test_document_modelN)r9r:r;rUrr%r%r%r&rs rc@sxeZdZeddddZddZeddddZed d Zed d Zed dddZ eddddZ dS)TestGetDocumentModelrrcCs|ttdS)z4Test get_document_model with a custom document modelN)assertIsr rrTr%r%r&test_custom_get_document_modelz3TestGetDocumentModel.test_custom_get_document_modelcCs|ttjdSr-)r rrrrTr%r%r&&test_get_document_model_at_import_timesz;TestGetDocumentModel.test_get_document_model_at_import_timecCs|tddS)z;Test get_document_model_string with a custom document modelrN)r r rTr%r%r&%test_custom_get_document_model_stringrz:TestGetDocumentModel.test_custom_get_document_model_stringcCs"t`ddlm}|t|dS)z:Test get_document_model with no WAGTAILDOCS_DOCUMENT_MODELr)rN)rrwagtail.documents.modelsrrr )r"rr%r%r& test_standard_get_document_models z5TestGetDocumentModel.test_standard_get_document_modelcCst`|tddS)zATest get_document_model_string with no WAGTAILDOCS_DOCUMENT_MODELzwagtaildocs.DocumentN)rrr r rTr%r%r&'test_standard_get_document_model_string&szs*      $.E ,