mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
prevent date suggestion search if disabled
This commit is contained in:
parent
3ab2892066
commit
c6900c5d51
@ -1948,6 +1948,29 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@mock.patch("documents.parsers.parse_date_generator")
|
||||||
|
@override_settings(NUMBER_OF_SUGGESTED_DATES=0)
|
||||||
|
def test_get_suggestions_dates_disabled(
|
||||||
|
self,
|
||||||
|
parse_date_generator,
|
||||||
|
):
|
||||||
|
"""
|
||||||
|
GIVEN:
|
||||||
|
- NUMBER_OF_SUGGESTED_DATES = 0 (disables feature)
|
||||||
|
WHEN:
|
||||||
|
- API reuqest for document suggestions
|
||||||
|
THEN:
|
||||||
|
- Dont check for suggested dates at all
|
||||||
|
"""
|
||||||
|
doc = Document.objects.create(
|
||||||
|
title="test",
|
||||||
|
mime_type="application/pdf",
|
||||||
|
content="this is an invoice from 12.04.2022!",
|
||||||
|
)
|
||||||
|
|
||||||
|
self.client.get(f"/api/documents/{doc.pk}/suggestions/")
|
||||||
|
self.assertFalse(parse_date_generator.called)
|
||||||
|
|
||||||
def test_saved_views(self):
|
def test_saved_views(self):
|
||||||
u1 = User.objects.create_superuser("user1")
|
u1 = User.objects.create_superuser("user1")
|
||||||
u2 = User.objects.create_superuser("user2")
|
u2 = User.objects.create_superuser("user2")
|
||||||
|
@ -422,10 +422,12 @@ class DocumentViewSet(
|
|||||||
|
|
||||||
classifier = load_classifier()
|
classifier = load_classifier()
|
||||||
|
|
||||||
gen = parse_date_generator(doc.filename, doc.content)
|
dates = []
|
||||||
dates = sorted(
|
if settings.NUMBER_OF_SUGGESTED_DATES > 0:
|
||||||
{i for i in itertools.islice(gen, settings.NUMBER_OF_SUGGESTED_DATES)},
|
gen = parse_date_generator(doc.filename, doc.content)
|
||||||
)
|
dates = sorted(
|
||||||
|
{i for i in itertools.islice(gen, settings.NUMBER_OF_SUGGESTED_DATES)},
|
||||||
|
)
|
||||||
|
|
||||||
return Response(
|
return Response(
|
||||||
{
|
{
|
||||||
|
Loading…
x
Reference in New Issue
Block a user