a D g)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%b/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/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&(z5TestDocumentQuerySet.test_operators..)key)rrrrrr r!sorted)r" aaa_document zzz_documentr$Zsorted_resultsr%r%r&test_operators sz#TestDocumentQuerySet.test_operatorscCs|tjjjdd}tjjjdd}tjjdjddd}|t|||gtjjdjddd}|t|||gdS) Nr(rr)rrF)Zorder_by_relevancez-title)rrrrZorder_byrr r!)r"r3r4r$r%r%r&test_custom_ordering+sz)TestDocumentQuerySet.test_custom_orderingN)__name__ __module__ __qualname__fixturesr'r5r6r%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.comZEditors)name administratorzadministrator@email.comzDocument addersZ add_document)codename)groupZ collectionZ permissionr)rZuploaded_by_user)Z create_userr<r@rAgroupsaddrrgetZcreate_superuserrCrZ adders_grouprrZget_first_root_noderrrr#r"r%r%r&setUp:s2 zTestDocumentPermissions.setUpcCs||j|jdSr-) assertTruer#is_editable_by_userrCrIr%r%r&test_administrator_can_editZsz3TestDocumentPermissions.test_administrator_can_editcCs||j|jdSr-)rKr#rLrArIr%r%r&test_editor_can_edit]sz,TestDocumentPermissions.test_editor_can_editcCs||j|jdSr-)rKr#rLr@rIr%r%r&test_owner_can_edit`sz+TestDocumentPermissions.test_owner_can_editcCs||j|jdSr-) assertFalser#rLr<rIr%r%r&test_user_cant_editcsz+TestDocumentPermissions.test_user_cant_editN)r7r8r9rJrMrNrOrQr%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_documentrIr%r%r&rJhsz$TestDocumentFilenameProperties.setUpcCs4|d|jj|d|jj|d|jjdS)NrSrTrU)r r#filenamerXrYrIr%r%r& test_filename{sz,TestDocumentFilenameProperties.test_filenamecCs4|d|jj|d|jj|d|jjdS)Nr.pdf)r r#Zfile_extensionrXrYrIr%r%r&test_file_extensionsz2TestDocumentFilenameProperties.test_file_extensioncCs4|d|jj|d|jj|d|jjdS)Nzapplication/mswordzapplication/pdfzapplication/octet-stream)r r# content_typerXrYrIr%r%r&test_content_types z0TestDocumentFilenameProperties.test_content_typecCs:||jd||jd||jddS)NZ(7d8c4778b182e4f3bd442408c64a6e22a4b0ed85)r r#Z get_file_hashrXrYrIr%r%r&test_file_hashs z-TestDocumentFilenameProperties.test_file_hashcCs4|d|jj|d|jj|d|jjdS)NzFattachment; filename=sample_name.doc; filename*=UTF-8''sample_name.docinlinez>attachment; filename=sample_name; filename*=UTF-8''sample_name)r r#content_dispositionrXrYrIr%r%r&test_content_dispositionsz7TestDocumentFilenameProperties.test_content_dispositioncCs(|jj|jj|jjdSr-)r#rVdeleterXrYrIr%r%r&tearDowns  z'TestDocumentFilenameProperties.tearDownN) r7r8r9rJr[r^r`rardrfr%r%r%r&rRgs  rRc@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 |Wdn1sj0Y| |jj |dS)Nz Test ImagerrV) rZatomicr rrrrVrBrKZstorageexistsrerP)r"r#rZr%r%r&#test_document_file_deleted_oncommits 2zDTestFilesDeletedForDefaultModels.test_document_file_deleted_oncommitN)r7r8r9__doc__r:rjr%r%r%r&rgs rgr\)ZWAGTAILDOCS_EXTENSIONSc@s,eZdZddZddZddZddZd S) TestDocumentValidateExtensionscCs,tjjjddd|_tjjjddd|_dS)Nrztest.docrhztest.pdf)rrrrdocument_invaliddocument_validrIr%r%r&rJs z$TestDocumentValidateExtensions.setUpcCsT|t}|jWdn1s*0Yd}||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.rVr) assertRaisesrrm full_cleanr exceptionZ message_dict)r"eZexpected_messager%r%r&!test_create_doc_invalid_extensions (z@TestDocumentValidateExtensions.test_create_doc_invalid_extensioncCs0z|jWnty*|dYn0dS)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)rnrprZfailrIr%r%r&test_create_doc_valid_extensions  z>TestDocumentValidateExtensions.test_create_doc_valid_extensioncCs|jj|jjdSr-)rmrVrernrIr%r%r&rfs z'TestDocumentValidateExtensions.tearDownN)r7r8r9rJrsrtrfr%r%r%r&rlsrltests.CustomDocumentWAGTAILDOCS_DOCUMENT_MODELc@seZdZddZddZdS)TestFilesDeletedForCustomModelscCs tjjdddddtdS)NZRootZ0001r)rBpathdepthZnumchild)rrZ get_or_createrZregister_signal_handlersrIr%r%r&rJs z%TestFilesDeletedForCustomModels.setUpcCs&t}||jjd|jddS)N.ru)r r Z_metaZ app_labelr7)r"clsr%r%r&test_document_model sz3TestFilesDeletedForCustomModels.test_document_modelN)r7r8r9rJr~r%r%r%r&rxsrxc@sxeZdZeddddZddZeddddZed d Zed d Zed dddZ eddddZ dS)TestGetDocumentModelrurvcCs|ttdS)z4Test get_document_model with a custom document modelN)assertIsr rrIr%r%r&test_custom_get_document_modelsz3TestGetDocumentModel.test_custom_get_document_modelcCs|ttjdSr-)r rrrrIr%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 modelruN)r r rIr%r%r&%test_custom_get_document_model_stringsz:TestGetDocumentModel.test_custom_get_document_model_stringcCs"t`ddlm}|t|dS)z:Test get_document_model with no WAGTAILDOCS_DOCUMENT_MODELr)rN)rrwZwagtail.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)rrwr r rIr%r%r&'test_standard_get_document_model_string&szs(      $.E ,