mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-05-01 11:19:32 -05:00
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from django.test import TestCase
|
|
|
|
from documents import index
|
|
from documents.index import JsonFormatter
|
|
from documents.models import Document
|
|
from documents.tests.utils import DirectoriesMixin
|
|
|
|
|
|
class JsonFormatterTest(TestCase):
|
|
|
|
def setUp(self) -> None:
|
|
self.formatter = JsonFormatter()
|
|
|
|
def test_empty_fragments(self):
|
|
self.assertListEqual(self.formatter.format([]), [])
|
|
|
|
|
|
class TestAutoComplete(DirectoriesMixin, TestCase):
|
|
|
|
def test_auto_complete(self):
|
|
|
|
doc1 = Document.objects.create(title="doc1", checksum="A", content="test test2 test3")
|
|
doc2 = Document.objects.create(title="doc2", checksum="B", content="test test2")
|
|
doc3 = Document.objects.create(title="doc3", checksum="C", content="test2")
|
|
|
|
index.add_or_update_document(doc1)
|
|
index.add_or_update_document(doc2)
|
|
index.add_or_update_document(doc3)
|
|
|
|
ix = index.open_index()
|
|
|
|
self.assertListEqual(index.autocomplete(ix, "tes"), [b"test3", b"test", b"test2"])
|
|
self.assertListEqual(index.autocomplete(ix, "tes", limit=3), [b"test3", b"test", b"test2"])
|
|
self.assertListEqual(index.autocomplete(ix, "tes", limit=1), [b"test3"])
|
|
self.assertListEqual(index.autocomplete(ix, "tes", limit=0), [])
|