mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-07-28 18:24:38 -05:00
Merge branch 'fix/issue-401' into feature/any-all-filtering
This commit is contained in:
@@ -70,31 +70,6 @@ def _consume(filepath):
|
||||
"Error while consuming document: {}".format(e))
|
||||
|
||||
|
||||
def _test_inotify(directory):
|
||||
if not INotify:
|
||||
return False
|
||||
|
||||
test_file = os.path.join(directory, "__inotify_test_file__")
|
||||
inotify = INotify()
|
||||
descriptor = None
|
||||
try:
|
||||
inotify_flags = flags.CLOSE_WRITE | flags.MOVED_TO
|
||||
descriptor = inotify.add_watch(directory, inotify_flags)
|
||||
Path(test_file).touch()
|
||||
events = inotify.read(timeout=1000)
|
||||
return len(events) == 1
|
||||
except Exception as e:
|
||||
logger.warning(
|
||||
f"Error while checking inotify availability: {str(e)}")
|
||||
return False
|
||||
finally:
|
||||
if descriptor:
|
||||
inotify.rm_watch(descriptor)
|
||||
inotify.close()
|
||||
if os.path.isfile(test_file):
|
||||
os.unlink(test_file)
|
||||
|
||||
|
||||
def _consume_wait_unmodified(file, num_tries=20, wait_time=1):
|
||||
mtime = -1
|
||||
current_try = 0
|
||||
@@ -179,24 +154,16 @@ class Command(BaseCommand):
|
||||
return
|
||||
|
||||
if settings.CONSUMER_POLLING == 0:
|
||||
if _test_inotify(directory):
|
||||
self.handle_inotify(directory, recursive)
|
||||
else:
|
||||
logger.warning(
|
||||
f"Inotify notifications are not available on {directory}, "
|
||||
f"falling back to polling every 10 seconds")
|
||||
self.handle_polling(
|
||||
directory, recursive, 10)
|
||||
self.handle_inotify(directory, recursive)
|
||||
else:
|
||||
self.handle_polling(
|
||||
directory, recursive, settings.CONSUMER_POLLING)
|
||||
self.handle_polling(directory, recursive)
|
||||
|
||||
logger.debug("Consumer exiting.")
|
||||
|
||||
def handle_polling(self, directory, recursive, timeout):
|
||||
def handle_polling(self, directory, recursive):
|
||||
logging.getLogger(__name__).info(
|
||||
f"Polling directory for changes: {directory}")
|
||||
self.observer = PollingObserver(timeout=timeout)
|
||||
self.observer = PollingObserver(timeout=settings.CONSUMER_POLLING)
|
||||
self.observer.schedule(Handler(), directory, recursive=recursive)
|
||||
self.observer.start()
|
||||
try:
|
||||
|
@@ -63,12 +63,6 @@ class MatchingModel(models.Model):
|
||||
def __str__(self):
|
||||
return self.name
|
||||
|
||||
def save(self, *args, **kwargs):
|
||||
|
||||
self.match = self.match.lower()
|
||||
|
||||
models.Model.save(self, *args, **kwargs)
|
||||
|
||||
|
||||
class Correspondent(MatchingModel):
|
||||
|
||||
|
@@ -923,6 +923,14 @@ class TestBulkEdit(DirectoriesMixin, APITestCase):
|
||||
doc2 = Document.objects.get(id=self.doc2.id)
|
||||
self.assertEqual(doc2.correspondent, self.c1)
|
||||
|
||||
def test_api_no_correspondent(self):
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "set_correspondent",
|
||||
"parameters": {}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_invalid_document_type(self):
|
||||
self.assertEqual(self.doc2.document_type, self.dt1)
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
@@ -935,6 +943,14 @@ class TestBulkEdit(DirectoriesMixin, APITestCase):
|
||||
doc2 = Document.objects.get(id=self.doc2.id)
|
||||
self.assertEqual(doc2.document_type, self.dt1)
|
||||
|
||||
def test_api_no_document_type(self):
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "set_document_type",
|
||||
"parameters": {}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_add_invalid_tag(self):
|
||||
self.assertEqual(list(self.doc2.tags.all()), [self.t1])
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
@@ -946,6 +962,14 @@ class TestBulkEdit(DirectoriesMixin, APITestCase):
|
||||
|
||||
self.assertEqual(list(self.doc2.tags.all()), [self.t1])
|
||||
|
||||
def test_api_add_tag_no_tag(self):
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "add_tag",
|
||||
"parameters": {}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_delete_invalid_tag(self):
|
||||
self.assertEqual(list(self.doc2.tags.all()), [self.t1])
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
@@ -957,6 +981,14 @@ class TestBulkEdit(DirectoriesMixin, APITestCase):
|
||||
|
||||
self.assertEqual(list(self.doc2.tags.all()), [self.t1])
|
||||
|
||||
def test_api_delete_tag_no_tag(self):
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "remove_tag",
|
||||
"parameters": {}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_modify_invalid_tags(self):
|
||||
self.assertEqual(list(self.doc2.tags.all()), [self.t1])
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
@@ -966,6 +998,21 @@ class TestBulkEdit(DirectoriesMixin, APITestCase):
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_modify_tags_no_tags(self):
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "modify_tags",
|
||||
"parameters": {"remove_tags": [1123123]}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
response = self.client.post("/api/documents/bulk_edit/", json.dumps({
|
||||
"documents": [self.doc2.id],
|
||||
"method": "modify_tags",
|
||||
"parameters": {'add_tags': [self.t2.id, 1657]}
|
||||
}), content_type='application/json')
|
||||
self.assertEqual(response.status_code, 400)
|
||||
|
||||
def test_api_selection_data_empty(self):
|
||||
response = self.client.post("/api/documents/selection_data/", json.dumps({
|
||||
"documents": []
|
||||
|
@@ -7,9 +7,8 @@ from unittest import mock
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.management import call_command, CommandError
|
||||
from django.test import override_settings, TransactionTestCase, TestCase
|
||||
from django.test import override_settings, TransactionTestCase
|
||||
|
||||
from documents.management.commands.document_consumer import _test_inotify
|
||||
from documents.models import Tag
|
||||
from documents.consumer import ConsumerError
|
||||
from documents.management.commands import document_consumer
|
||||
@@ -261,27 +260,3 @@ class TestConsumerTags(DirectoriesMixin, ConsumerMixin, TransactionTestCase):
|
||||
@override_settings(CONSUMER_POLLING=1)
|
||||
def test_consume_file_with_path_tags_polling(self):
|
||||
self.test_consume_file_with_path_tags()
|
||||
|
||||
|
||||
class TestInotify(DirectoriesMixin, TestCase):
|
||||
|
||||
def test_inotify(self):
|
||||
self.assertTrue(_test_inotify(self.dirs.consumption_dir))
|
||||
|
||||
@mock.patch("documents.management.commands.document_consumer.Path.touch")
|
||||
def test_inotify_error(self, m):
|
||||
m.side_effect = OSError("Permission error")
|
||||
self.assertFalse(_test_inotify(self.dirs.consumption_dir))
|
||||
|
||||
@mock.patch("documents.management.commands.document_consumer.Command.handle_polling")
|
||||
@mock.patch("documents.management.commands.document_consumer.Command.handle_inotify")
|
||||
@mock.patch("documents.management.commands.document_consumer._test_inotify")
|
||||
def test_polling_fallback(self, test_inotify, handle_inotify, handle_polling):
|
||||
test_inotify.return_value = False
|
||||
|
||||
cmd = document_consumer.Command()
|
||||
cmd.handle(directory=settings.CONSUMPTION_DIR, oneshot=False)
|
||||
|
||||
test_inotify.assert_called_once()
|
||||
handle_polling.assert_called_once()
|
||||
handle_inotify.assert_not_called()
|
||||
|
@@ -120,3 +120,4 @@ class TestParserAvailability(TestCase):
|
||||
|
||||
self.assertTrue(is_file_ext_supported('.pdf'))
|
||||
self.assertFalse(is_file_ext_supported('.hsdfh'))
|
||||
self.assertFalse(is_file_ext_supported(''))
|
||||
|
Reference in New Issue
Block a user