"env_key": "PAPERLESS_TRAIN_TASK_CRON",
# Default hourly at 5 minutes past the hour
"env_default": "5 */1 * * *",
- "task": "documents.tasks.train_classifier",
+ "task": "paperless.tasks.train_classifier",
"options": {
# 1 minute before default schedule sends again
"expires": 59.0 * 60.0,
"env_key": "PAPERLESS_INDEX_TASK_CRON",
# Default daily at midnight
"env_default": "0 0 * * *",
- "task": "documents.tasks.index_optimize",
+ "task": "paperless.tasks.index_optimize",
"options": {
# 1 hour before default schedule sends again
"expires": 23.0 * 60.0 * 60.0,
"env_key": "PAPERLESS_SANITY_TASK_CRON",
# Default Sunday at 00:30
"env_default": "30 0 * * sun",
- "task": "documents.tasks.sanity_check",
+ "task": "paperless.tasks.sanity_check",
"options": {
# 1 hour before default schedule sends again
"expires": ((7.0 * 24.0) - 1.0) * 60.0 * 60.0,
"env_key": "PAPERLESS_EMPTY_TRASH_TASK_CRON",
# Default daily at 01:00
"env_default": "0 1 * * *",
- "task": "documents.tasks.empty_trash",
+ "task": "paperless.tasks.empty_trash",
"options": {
# 1 hour before default schedule sends again
"expires": 23.0 * 60.0 * 60.0,
"env_key": "PAPERLESS_WORKFLOW_SCHEDULED_TASK_CRON",
# Default hourly at 5 minutes past the hour
"env_default": "5 */1 * * *",
- "task": "documents.tasks.check_scheduled_workflows",
+ "task": "paperless.tasks.check_scheduled_workflows",
"options": {
# 1 minute before default schedule sends again
"expires": 59.0 * 60.0,
"django.template.context_processors.request",
"django.contrib.auth.context_processors.auth",
"django.contrib.messages.context_processors.messages",
- "documents.context_processors.settings",
+ "paperless.context_processors.settings",
],
},
},
https://docs.celeryq.dev/en/stable/internals/protocol.html#version-2
"""
- if "task" not in headers or headers["task"] != "documents.tasks.consume_file":
+ if "task" not in headers or headers["task"] != "paperless.tasks.consume_file":
# Assumption: this is only ever a v2 message
return
self.user = user
self.client.force_authenticate(user=user)
- patcher = mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ patcher = mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
self.async_task = patcher.start()
self.addCleanup(patcher.stop)
self.c1 = Correspondent.objects.create(name="c1")
m.return_value = return_value
m.__name__ = method_name
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_set_correspondent(self, bulk_update_task_mock):
self.assertNotEqual(self.doc1.correspondent, self.c1)
response = self.client.post(
self.assertEqual(self.doc1.correspondent, self.c1)
bulk_update_task_mock.assert_called_once_with(document_ids=[self.doc1.pk])
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_unset_correspondent(self, bulk_update_task_mock):
self.doc1.correspondent = self.c1
self.doc1.save()
self.doc1.refresh_from_db()
self.assertIsNone(self.doc1.correspondent)
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_set_type(self, bulk_update_task_mock):
self.assertNotEqual(self.doc1.document_type, self.dt1)
response = self.client.post(
self.assertEqual(self.doc1.document_type, self.dt1)
bulk_update_task_mock.assert_called_once_with(document_ids=[self.doc1.pk])
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_unset_type(self, bulk_update_task_mock):
self.doc1.document_type = self.dt1
self.doc1.save()
self.assertIsNone(self.doc1.document_type)
bulk_update_task_mock.assert_called_once_with(document_ids=[self.doc1.pk])
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_add_tag(self, bulk_update_task_mock):
self.assertFalse(self.doc1.tags.filter(pk=self.t1.pk).exists())
bulk_update_task_mock.assert_called_once_with(document_ids=[self.doc1.pk])
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_remove_tag(self, bulk_update_task_mock):
self.doc1.tags.add(self.t1)
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(StoragePath.objects.count(), 2)
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_update_storage_path(self, bulk_update_mock):
"""
GIVEN:
self.assertCountEqual([document.pk], args[0])
- @mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
def test_api_delete_storage_path(self, bulk_update_mock):
"""
GIVEN:
[d3.id, d2.id, d1.id],
)
- @mock.patch("documents.bulk_edit.bulk_update_documents")
+ @mock.patch("paperless.bulk_edit.bulk_update_documents")
def test_global_search(self, m):
"""
GIVEN:
self.assertEqual(response.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
mock_train_classifier.assert_called_once_with(scheduled=False)
- @mock.patch("documents.tasks.sanity_check")
+ @mock.patch("paperless.tasks.sanity_check")
def test_run_task_requires_superuser(self, mock_check_sanity):
"""
GIVEN:
overrides = DocumentMetadataOverrides(tag_ids=[1, 2, 9])
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
self.assertEqual(
tasks.consume_file(
ConsumableDocument(
dst = settings.SCRATCH_DIR / "barcode-39-asn-123.pdf"
shutil.copy(test_file, dst)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
tasks.consume_file(
ConsumableDocument(
source=DocumentSource.ConsumeFolder,
self.group1 = Group.objects.create(name="group1")
self.group2 = Group.objects.create(name="group2")
- patcher = mock.patch("documents.bulk_edit.bulk_update_documents.delay")
+ patcher = mock.patch("paperless.bulk_edit.bulk_update_documents.delay")
self.async_task = patcher.start()
self.addCleanup(patcher.stop)
self.c1 = Correspondent.objects.create(name="c1")
[self.doc3.id, self.doc4.id, self.doc5.id],
)
- @mock.patch("documents.tasks.bulk_update_documents.delay")
+ @mock.patch("paperless.tasks.bulk_update_documents.delay")
def test_set_permissions(self, m):
doc_ids = [self.doc1.id, self.doc2.id, self.doc3.id]
)
self.assertEqual(groups_with_perms.count(), 1)
- @mock.patch("documents.tasks.bulk_update_documents.delay")
+ @mock.patch("paperless.tasks.bulk_update_documents.delay")
def test_set_permissions_merge(self, m):
doc_ids = [self.doc1.id, self.doc2.id, self.doc3.id]
self.img_doc.archive_filename = img_doc_archive
self.img_doc.save()
- @mock.patch("documents.tasks.consume_file.s")
+ @mock.patch("paperless.tasks.consume_file.s")
def test_merge(self, mock_consume_file):
"""
GIVEN:
self.assertEqual(result, "OK")
- @mock.patch("documents.bulk_edit.delete.si")
- @mock.patch("documents.tasks.consume_file.s")
- @mock.patch("documents.bulk_edit.chain")
+ @mock.patch("paperless.bulk_edit.delete.si")
+ @mock.patch("paperless.tasks.consume_file.s")
+ @mock.patch("paperless.bulk_edit.chain")
def test_merge_and_delete_originals(
self,
mock_chain,
doc_ids,
)
- @mock.patch("documents.tasks.consume_file.s")
+ @mock.patch("paperless.tasks.consume_file.s")
def test_merge_with_archive_fallback(self, mock_consume_file):
"""
GIVEN:
expected_filename,
)
- @mock.patch("documents.tasks.consume_file.delay")
+ @mock.patch("paperless.tasks.consume_file.delay")
@mock.patch("pikepdf.open")
def test_merge_with_errors(self, mock_open_pdf, mock_consume_file):
"""
mock_consume_file.assert_not_called()
- @mock.patch("documents.tasks.consume_file.s")
+ @mock.patch("paperless.tasks.consume_file.s")
def test_split(self, mock_consume_file):
"""
GIVEN:
self.assertEqual(result, "OK")
- @mock.patch("documents.bulk_edit.delete.si")
- @mock.patch("documents.tasks.consume_file.s")
- @mock.patch("documents.bulk_edit.chord")
+ @mock.patch("paperless.bulk_edit.delete.si")
+ @mock.patch("paperless.tasks.consume_file.s")
+ @mock.patch("paperless.bulk_edit.chord")
def test_split_and_delete_originals(
self,
mock_chord,
doc_ids,
)
- @mock.patch("documents.tasks.consume_file.delay")
+ @mock.patch("paperless.tasks.consume_file.delay")
@mock.patch("pikepdf.Pdf.save")
def test_split_with_errors(self, mock_save_pdf, mock_consume_file):
"""
mock_consume_file.assert_not_called()
- @mock.patch("documents.tasks.bulk_update_documents.si")
- @mock.patch("documents.tasks.update_document_content_maybe_archive_file.s")
+ @mock.patch("paperless.tasks.bulk_update_documents.si")
+ @mock.patch("paperless.tasks.update_document_content_maybe_archive_file.s")
@mock.patch("celery.chord.delay")
def test_rotate(self, mock_chord, mock_update_document, mock_update_documents):
"""
mock_chord.assert_called_once()
self.assertEqual(result, "OK")
- @mock.patch("documents.tasks.bulk_update_documents.si")
- @mock.patch("documents.tasks.update_document_content_maybe_archive_file.s")
+ @mock.patch("paperless.tasks.bulk_update_documents.si")
+ @mock.patch("paperless.tasks.update_document_content_maybe_archive_file.s")
@mock.patch("pikepdf.Pdf.save")
def test_rotate_with_error(
self,
self.assertIn(expected_str, error_str)
mock_update_archive_file.assert_not_called()
- @mock.patch("documents.tasks.bulk_update_documents.si")
- @mock.patch("documents.tasks.update_document_content_maybe_archive_file.s")
+ @mock.patch("paperless.tasks.bulk_update_documents.si")
+ @mock.patch("paperless.tasks.update_document_content_maybe_archive_file.s")
@mock.patch("celery.chord.delay")
def test_rotate_non_pdf(
self,
mock_chord.assert_called_once()
self.assertEqual(result, "OK")
- @mock.patch("documents.tasks.update_document_content_maybe_archive_file.delay")
+ @mock.patch("paperless.tasks.update_document_content_maybe_archive_file.delay")
@mock.patch("pikepdf.Pdf.save")
def test_delete_pages(self, mock_pdf_save, mock_update_archive_file):
"""
self.doc2.refresh_from_db()
self.assertEqual(self.doc2.page_count, expected_page_count)
- @mock.patch("documents.tasks.update_document_content_maybe_archive_file.delay")
+ @mock.patch("paperless.tasks.update_document_content_maybe_archive_file.delay")
@mock.patch("pikepdf.Pdf.save")
def test_delete_pages_with_error(self, mock_pdf_save, mock_update_archive_file):
"""
from django.test import TestCase
from django.test import override_settings
-from documents.tests.factories import DocumentFactory
from paperless.checks import audit_log_check
from paperless.checks import binaries_check
from paperless.checks import changed_password_check
from paperless.checks import paths_check
from paperless.checks import settings_values_check
from paperless.models import Document
+from paperless.tests.factories import DocumentFactory
from paperless.tests.utils import DirectoriesMixin
from paperless.tests.utils import FileSystemAssertsMixin
dst.parent.mkdir(parents=True, exist_ok=True)
shutil.copy(src, dst)
with mock.patch(
- "documents.tasks.ProgressManager",
+ "paperless.tasks.ProgressManager",
DummyProgressManager,
):
msg = tasks.consume_file(
PASSPHRASE="test",
FILENAME_FORMAT=None,
)
- @mock.patch("documents.management.commands.decrypt_documents.input")
+ @mock.patch("paperless.management.commands.decrypt_documents.input")
def test_decrypt(self, m):
media_dir = tempfile.mkdtemp()
originals_dir = Path(media_dir) / "documents" / "originals"
class TestMakeIndex(TestCase):
- @mock.patch("documents.management.commands.document_index.index_reindex")
+ @mock.patch("paperless.management.commands.document_index.index_reindex")
def test_reindex(self, m):
call_command("document_index", "reindex")
m.assert_called_once()
- @mock.patch("documents.management.commands.document_index.index_optimize")
+ @mock.patch("paperless.management.commands.document_index.index_optimize")
def test_optimize(self, m):
call_command("document_index", "optimize")
m.assert_called_once()
class TestCreateClassifier(TestCase):
@mock.patch(
- "documents.management.commands.document_create_classifier.train_classifier",
+ "paperless.management.commands.document_create_classifier.train_classifier",
)
def test_create_classifier(self, m):
call_command("document_create_classifier")
from django.test import TransactionTestCase
from django.test import override_settings
-from documents.management.commands import document_consumer
from paperless.consumer import ConsumerError
from paperless.data_models import ConsumableDocument
+from paperless.management.commands import document_consumer
from paperless.models import Tag
from paperless.tests.utils import DirectoriesMixin
from paperless.tests.utils import DocumentConsumeDelayMixin
self.assertEqual(input_doc.original_file, f)
- @mock.patch("documents.management.commands.document_consumer.logger.error")
+ @mock.patch("paperless.management.commands.document_consumer.logger.error")
def test_slow_write_pdf(self, error_logger):
self.consume_file_mock.side_effect = self.bogus_task
self.assertEqual(input_doc.original_file, fname)
- @mock.patch("documents.management.commands.document_consumer.logger.error")
+ @mock.patch("paperless.management.commands.document_consumer.logger.error")
def test_slow_write_and_move(self, error_logger):
self.consume_file_mock.side_effect = self.bogus_task
error_logger.assert_not_called()
- @mock.patch("documents.management.commands.document_consumer.logger.error")
+ @mock.patch("paperless.management.commands.document_consumer.logger.error")
def test_slow_write_incomplete(self, error_logger):
self.consume_file_mock.side_effect = self.bogus_task
f'_is_ignored("{filepath}") != {expected_ignored_result}',
)
- @mock.patch("documents.management.commands.document_consumer.open")
+ @mock.patch("paperless.management.commands.document_consumer.open")
def test_consume_file_busy(self, open_mock):
# Calling this mock always raises this
open_mock.side_effect = OSError
from guardian.models import UserObjectPermission
from guardian.shortcuts import assign_perm
-from documents.management.commands import document_exporter
from documents.settings import EXPORTER_FILE_NAME
+from paperless.management.commands import document_exporter
from paperless.models import Correspondent
from paperless.models import CustomField
from paperless.models import CustomFieldInstance
st_mtime_1 = os.stat(os.path.join(self.target, "manifest.json")).st_mtime
with mock.patch(
- "documents.management.commands.document_exporter.copy_file_with_basic_stats",
+ "paperless.management.commands.document_exporter.copy_file_with_basic_stats",
) as m:
self._do_export()
m.assert_not_called()
Path(self.d1.source_path).touch()
with mock.patch(
- "documents.management.commands.document_exporter.copy_file_with_basic_stats",
+ "paperless.management.commands.document_exporter.copy_file_with_basic_stats",
) as m:
self._do_export()
self.assertEqual(m.call_count, 1)
self.assertIsFile(os.path.join(self.target, "manifest.json"))
with mock.patch(
- "documents.management.commands.document_exporter.copy_file_with_basic_stats",
+ "paperless.management.commands.document_exporter.copy_file_with_basic_stats",
) as m:
self._do_export()
m.assert_not_called()
self.d2.save()
with mock.patch(
- "documents.management.commands.document_exporter.copy_file_with_basic_stats",
+ "paperless.management.commands.document_exporter.copy_file_with_basic_stats",
) as m:
self._do_export(compare_checksums=True)
self.assertEqual(m.call_count, 1)
from django.core.management.base import CommandError
from django.test import TestCase
-from documents.management.commands.document_importer import Command
from documents.settings import EXPORTER_ARCHIVE_NAME
from documents.settings import EXPORTER_FILE_NAME
+from paperless.management.commands.document_importer import Command
from paperless.models import Document
from paperless.tests.utils import DirectoriesMixin
from paperless.tests.utils import FileSystemAssertsMixin
from django.core.management import call_command
from django.test import TestCase
-from documents.management.commands.document_thumbnails import _process_document
+from paperless.management.commands.document_thumbnails import _process_document
from paperless.models import Document
from paperless.parsers import get_default_thumbnail
from paperless.tests.utils import DirectoriesMixin
self.assertIsFile(get_default_thumbnail())
self.assertIsFile(self.d3.thumbnail_path)
- @mock.patch("documents.management.commands.document_thumbnails.shutil.move")
+ @mock.patch("paperless.management.commands.document_thumbnails.shutil.move")
def test_process_document_invalid_mime_type(self, m: mock.Mock):
self.d1.mime_type = "asdasdasd"
self.d1.save()
from django.test import TestCase
-from documents.tests.factories import CorrespondentFactory
-from documents.tests.factories import DocumentFactory
from paperless.models import Correspondent
from paperless.models import Document
+from paperless.tests.factories import CorrespondentFactory
+from paperless.tests.factories import DocumentFactory
class CorrespondentTestCase(TestCase):
"options": {"expires": self.MAIL_EXPIRE_TIME},
},
"Train the classifier": {
- "task": "documents.tasks.train_classifier",
+ "task": "paperless.tasks.train_classifier",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.CLASSIFIER_EXPIRE_TIME},
},
"Optimize the index": {
- "task": "documents.tasks.index_optimize",
+ "task": "paperless.tasks.index_optimize",
"schedule": crontab(minute=0, hour=0),
"options": {"expires": self.INDEX_EXPIRE_TIME},
},
"Perform sanity check": {
- "task": "documents.tasks.sanity_check",
+ "task": "paperless.tasks.sanity_check",
"schedule": crontab(minute=30, hour=0, day_of_week="sun"),
"options": {"expires": self.SANITY_EXPIRE_TIME},
},
"Empty trash": {
- "task": "documents.tasks.empty_trash",
+ "task": "paperless.tasks.empty_trash",
"schedule": crontab(minute=0, hour="1"),
"options": {"expires": self.EMPTY_TRASH_EXPIRE_TIME},
},
"Check and run scheduled workflows": {
- "task": "documents.tasks.check_scheduled_workflows",
+ "task": "paperless.tasks.check_scheduled_workflows",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.RUN_SCHEDULED_WORKFLOWS_EXPIRE_TIME},
},
"options": {"expires": self.MAIL_EXPIRE_TIME},
},
"Train the classifier": {
- "task": "documents.tasks.train_classifier",
+ "task": "paperless.tasks.train_classifier",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.CLASSIFIER_EXPIRE_TIME},
},
"Optimize the index": {
- "task": "documents.tasks.index_optimize",
+ "task": "paperless.tasks.index_optimize",
"schedule": crontab(minute=0, hour=0),
"options": {"expires": self.INDEX_EXPIRE_TIME},
},
"Perform sanity check": {
- "task": "documents.tasks.sanity_check",
+ "task": "paperless.tasks.sanity_check",
"schedule": crontab(minute=30, hour=0, day_of_week="sun"),
"options": {"expires": self.SANITY_EXPIRE_TIME},
},
"Empty trash": {
- "task": "documents.tasks.empty_trash",
+ "task": "paperless.tasks.empty_trash",
"schedule": crontab(minute=0, hour="1"),
"options": {"expires": self.EMPTY_TRASH_EXPIRE_TIME},
},
"Check and run scheduled workflows": {
- "task": "documents.tasks.check_scheduled_workflows",
+ "task": "paperless.tasks.check_scheduled_workflows",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.RUN_SCHEDULED_WORKFLOWS_EXPIRE_TIME},
},
"options": {"expires": self.MAIL_EXPIRE_TIME},
},
"Train the classifier": {
- "task": "documents.tasks.train_classifier",
+ "task": "paperless.tasks.train_classifier",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.CLASSIFIER_EXPIRE_TIME},
},
"Perform sanity check": {
- "task": "documents.tasks.sanity_check",
+ "task": "paperless.tasks.sanity_check",
"schedule": crontab(minute=30, hour=0, day_of_week="sun"),
"options": {"expires": self.SANITY_EXPIRE_TIME},
},
"Empty trash": {
- "task": "documents.tasks.empty_trash",
+ "task": "paperless.tasks.empty_trash",
"schedule": crontab(minute=0, hour="1"),
"options": {"expires": self.EMPTY_TRASH_EXPIRE_TIME},
},
"Check and run scheduled workflows": {
- "task": "documents.tasks.check_scheduled_workflows",
+ "task": "paperless.tasks.check_scheduled_workflows",
"schedule": crontab(minute="5", hour="*/1"),
"options": {"expires": self.RUN_SCHEDULED_WORKFLOWS_EXPIRE_TIME},
},
import celery
from django.test import TestCase
-from documents.tests.test_consumer import fake_magic_from_file
from paperless.data_models import ConsumableDocument
from paperless.data_models import DocumentMetadataOverrides
from paperless.data_models import DocumentSource
from paperless.signals.handlers import task_failure_handler
from paperless.signals.handlers import task_postrun_handler
from paperless.signals.handlers import task_prerun_handler
+from paperless.tests.test_consumer import fake_magic_from_file
from paperless.tests.utils import DirectoriesMixin
"""
headers = {
"id": str(uuid.uuid4()),
- "task": "documents.tasks.consume_file",
+ "task": "paperless.tasks.consume_file",
}
body = (
# args
headers = {
"id": str(uuid.uuid4()),
- "task": "documents.tasks.consume_file",
+ "task": "paperless.tasks.consume_file",
}
body = (
# args
"""
headers = {
"id": str(uuid.uuid4()),
- "task": "documents.tasks.consume_file",
+ "task": "paperless.tasks.consume_file",
}
body = (
# args
"""
headers = {
"id": str(uuid.uuid4()),
- "task": "documents.tasks.consume_file",
+ "task": "paperless.tasks.consume_file",
}
body = (
# args
from django.test import TestCase
from django.utils import timezone
-from documents.tests.test_classifier import dummy_preprocess
from paperless import tasks
from paperless.models import Correspondent
from paperless.models import Document
from paperless.models import Tag
from paperless.sanity_checker import SanityCheckFailedException
from paperless.sanity_checker import SanityCheckMessages
+from paperless.tests.test_classifier import dummy_preprocess
from paperless.tests.utils import DirectoriesMixin
from paperless.tests.utils import FileSystemAssertsMixin
class TestClassifier(DirectoriesMixin, FileSystemAssertsMixin, TestCase):
- @mock.patch("documents.tasks.load_classifier")
+ @mock.patch("paperless.tasks.load_classifier")
def test_train_classifier_no_auto_matching(self, load_classifier):
tasks.train_classifier()
load_classifier.assert_not_called()
- @mock.patch("documents.tasks.load_classifier")
+ @mock.patch("paperless.tasks.load_classifier")
def test_train_classifier_with_auto_tag(self, load_classifier):
load_classifier.return_value = None
Tag.objects.create(matching_algorithm=Tag.MATCH_AUTO, name="test")
load_classifier.assert_called_once()
self.assertIsNotFile(settings.MODEL_FILE)
- @mock.patch("documents.tasks.load_classifier")
+ @mock.patch("paperless.tasks.load_classifier")
def test_train_classifier_with_auto_type(self, load_classifier):
load_classifier.return_value = None
DocumentType.objects.create(matching_algorithm=Tag.MATCH_AUTO, name="test")
load_classifier.assert_called_once()
self.assertIsNotFile(settings.MODEL_FILE)
- @mock.patch("documents.tasks.load_classifier")
+ @mock.patch("paperless.tasks.load_classifier")
def test_train_classifier_with_auto_correspondent(self, load_classifier):
load_classifier.return_value = None
Correspondent.objects.create(matching_algorithm=Tag.MATCH_AUTO, name="test")
class TestSanityCheck(DirectoriesMixin, TestCase):
- @mock.patch("documents.tasks.sanity_checker.check_sanity")
+ @mock.patch("paperless.tasks.sanity_checker.check_sanity")
def test_sanity_check_success(self, m):
m.return_value = SanityCheckMessages()
self.assertEqual(tasks.sanity_check(), "No issues detected.")
m.assert_called_once()
- @mock.patch("documents.tasks.sanity_checker.check_sanity")
+ @mock.patch("paperless.tasks.sanity_checker.check_sanity")
def test_sanity_check_error(self, m):
messages = SanityCheckMessages()
messages.error(None, "Some error")
self.assertRaises(SanityCheckFailedException, tasks.sanity_check)
m.assert_called_once()
- @mock.patch("documents.tasks.sanity_checker.check_sanity")
+ @mock.patch("paperless.tasks.sanity_checker.check_sanity")
def test_sanity_check_error_no_raise(self, m):
messages = SanityCheckMessages()
messages.error(None, "Some error")
)
m.assert_called_once()
- @mock.patch("documents.tasks.sanity_checker.check_sanity")
+ @mock.patch("paperless.tasks.sanity_checker.check_sanity")
def test_sanity_check_warning(self, m):
messages = SanityCheckMessages()
messages.warning(None, "Some warning")
)
m.assert_called_once()
- @mock.patch("documents.tasks.sanity_checker.check_sanity")
+ @mock.patch("paperless.tasks.sanity_checker.check_sanity")
def test_sanity_check_info(self, m):
messages = SanityCheckMessages()
messages.info(None, "Some info")
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="DEBUG") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="DEBUG") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="DEBUG") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="DEBUG") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="DEBUG") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO") as cm:
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO"):
tasks.consume_file(
ConsumableDocument(
self.dirs.scratch_dir / "simple.pdf",
)
- with mock.patch("documents.tasks.ProgressManager", DummyProgressManager):
+ with mock.patch("paperless.tasks.ProgressManager", DummyProgressManager):
with self.assertLogs("paperless.matching", level="INFO"):
tasks.consume_file(
ConsumableDocument(
"""
def setUp(self) -> None:
- self.consume_file_patcher = mock.patch("documents.tasks.consume_file.delay")
+ self.consume_file_patcher = mock.patch("paperless.tasks.consume_file.delay")
self.consume_file_mock = self.consume_file_patcher.start()
super().setUp()
connect to Redis. Payloads are stored for test assertions if needed.
Use it with
- mock.patch("documents.tasks.ProgressManager", DummyProgressManager)
+ mock.patch("paperless.tasks.ProgressManager", DummyProgressManager)
"""
def __init__(self, filename: str, task_id: str | None = None) -> None: