Enhancement: support customfield filter for select type (#7199)

This commit is contained in:
shamoon
2024-07-09 10:54:34 -07:00
committed by GitHub
parent bb83c1eb0a
commit 61811a4bec
2 changed files with 43 additions and 0 deletions

View File

@@ -641,6 +641,38 @@ class TestDocumentApi(DirectoriesMixin, DocumentConsumeDelayMixin, APITestCase):
self.assertEqual(len(results), 1)
self.assertEqual(results[0]["id"], doc3.id)
def test_custom_field_select_filter(self):
"""
GIVEN:
- Documents with select custom field values
WHEN:
- API request with custom field filtering is made
THEN:
- Only docs with selected custom field values are returned
"""
doc1 = Document.objects.create(checksum="1", content="test 1")
Document.objects.create(checksum="2", content="test 2")
custom_field_select = CustomField.objects.create(
name="Test Custom Field Select",
data_type=CustomField.FieldDataType.SELECT,
extra_data={
"select_options": ["Option 1", "Choice 2"],
},
)
CustomFieldInstance.objects.create(
document=doc1,
field=custom_field_select,
value_select=1,
)
r = self.client.get("/api/documents/?custom_fields__icontains=choice")
self.assertEqual(r.status_code, status.HTTP_200_OK)
self.assertEqual(r.data["count"], 1)
r = self.client.get("/api/documents/?custom_fields__icontains=option")
self.assertEqual(r.status_code, status.HTTP_200_OK)
self.assertEqual(r.data["count"], 0)
def test_document_checksum_filter(self):
Document.objects.create(
title="none1",