paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3315 from paperless-ngx/fix/__in-search-testing

Browse Source
This commit is contained in:
shamoon 2023-05-08 07:03:59 -07:00 committed by GitHub
commit f1c138eaed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 29 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/documents/tests/test_api.py
View File

@ -912,8 +912,11 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
t = Tag.objects.create(name="tag") t = Tag.objects.create(name="tag")
t2 = Tag.objects.create(name="tag2") t2 = Tag.objects.create(name="tag2")
c = Correspondent.objects.create(name="correspondent") c = Correspondent.objects.create(name="correspondent")
c2 = Correspondent.objects.create(name="correspondent2")
dt = DocumentType.objects.create(name="type") dt = DocumentType.objects.create(name="type")
dt2 = DocumentType.objects.create(name="type2")
sp = StoragePath.objects.create(name="path") sp = StoragePath.objects.create(name="path")
sp2 = StoragePath.objects.create(name="path2")
d1 = Document.objects.create(checksum="1", correspondent=c, content="test") d1 = Document.objects.create(checksum="1", correspondent=c, content="test")
d2 = Document.objects.create(checksum="2", document_type=dt, content="test") d2 = Document.objects.create(checksum="2", document_type=dt, content="test")
@ -934,6 +937,13 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
) )
Document.objects.create(checksum="6", content="test2") Document.objects.create(checksum="6", content="test2")
d7 = Document.objects.create(checksum="7", storage_path=sp, content="test") d7 = Document.objects.create(checksum="7", storage_path=sp, content="test")
d8 = Document.objects.create(
checksum="8",
correspondent=c2,
document_type=dt2,
storage_path=sp2,
content="test",
)
with AsyncWriter(index.open_index()) as writer: with AsyncWriter(index.open_index()) as writer:
for doc in Document.objects.all(): for doc in Document.objects.all():
@ -946,39 +956,39 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
self.assertCountEqual( self.assertCountEqual(
search_query(""), search_query(""),
[d1.id, d2.id, d3.id, d4.id, d5.id, d7.id], [d1.id, d2.id, d3.id, d4.id, d5.id, d7.id, d8.id],
) )
self.assertCountEqual(search_query("&is_tagged=true"), [d3.id, d4.id]) self.assertCountEqual(search_query("&is_tagged=true"), [d3.id, d4.id])
self.assertCountEqual( self.assertCountEqual(
search_query("&is_tagged=false"), search_query("&is_tagged=false"),
[d1.id, d2.id, d5.id, d7.id], [d1.id, d2.id, d5.id, d7.id, d8.id],
) )
self.assertCountEqual(search_query("&correspondent__id=" + str(c.id)), [d1.id]) self.assertCountEqual(search_query("&correspondent__id=" + str(c.id)), [d1.id])
self.assertCountEqual( self.assertCountEqual(
search_query("&correspondent__id__in=" + str(c.id)), search_query(f"&correspondent__id__in={c.id},{c2.id}"),
[d1.id], [d1.id, d8.id],
) )
self.assertCountEqual( self.assertCountEqual(
search_query("&correspondent__id__none=" + str(c.id)), search_query("&correspondent__id__none=" + str(c.id)),
[d2.id, d3.id, d4.id, d5.id, d7.id], [d2.id, d3.id, d4.id, d5.id, d7.id, d8.id],
) )
self.assertCountEqual(search_query("&document_type__id=" + str(dt.id)), [d2.id]) self.assertCountEqual(search_query("&document_type__id=" + str(dt.id)), [d2.id])
self.assertCountEqual( self.assertCountEqual(
search_query("&document_type__id__in=" + str(dt.id)), search_query(f"&document_type__id__in={dt.id},{dt2.id}"),
[d2.id], [d2.id, d8.id],
) )
self.assertCountEqual( self.assertCountEqual(
search_query("&document_type__id__none=" + str(dt.id)), search_query("&document_type__id__none=" + str(dt.id)),
[d1.id, d3.id, d4.id, d5.id, d7.id], [d1.id, d3.id, d4.id, d5.id, d7.id, d8.id],
) )
self.assertCountEqual(search_query("&storage_path__id=" + str(sp.id)), [d7.id]) self.assertCountEqual(search_query("&storage_path__id=" + str(sp.id)), [d7.id])
self.assertCountEqual( self.assertCountEqual(
search_query("&storage_path__id__in=" + str(sp.id)), search_query(f"&storage_path__id__in={sp.id},{sp2.id}"),
[d7.id], [d7.id, d8.id],
) )
self.assertCountEqual( self.assertCountEqual(
search_query("&storage_path__id__none=" + str(sp.id)), search_query("&storage_path__id__none=" + str(sp.id)),
[d1.id, d2.id, d3.id, d4.id, d5.id], [d1.id, d2.id, d3.id, d4.id, d5.id, d8.id],
) )
self.assertCountEqual( self.assertCountEqual(
@ -1002,6 +1012,14 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
search_query("&tags__id__all=" + str(t2.id)), search_query("&tags__id__all=" + str(t2.id)),
[d3.id, d4.id], [d3.id, d4.id],
) )
self.assertCountEqual(
search_query(f"&tags__id__in={t.id},{t2.id}"),
[d3.id, d4.id],
)
self.assertCountEqual(
search_query(f"&tags__id__none={t.id},{t2.id}"),
[d1.id, d2.id, d5.id, d7.id, d8.id],
)
self.assertIn( self.assertIn(
d4.id, d4.id,