From a83b9a3a4ddf4922991e82ee67e533fcbdf61154 Mon Sep 17 00:00:00 2001 From: shamoon <4887959+shamoon@users.noreply.github.com> Date: Mon, 28 Apr 2025 20:50:03 -0700 Subject: [PATCH] Cover matching --- src/paperless/tests/test_ai_matching.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/src/paperless/tests/test_ai_matching.py b/src/paperless/tests/test_ai_matching.py index e5eaf27d1..bcc545cba 100644 --- a/src/paperless/tests/test_ai_matching.py +++ b/src/paperless/tests/test_ai_matching.py @@ -68,3 +68,19 @@ class TestAIMatching(TestCase): matched_objects = [self.tag1] unmatched_names = extract_unmatched_names(llm_names, matched_objects) self.assertEqual(unmatched_names, ["Nonexistent Tag"]) + + @patch("paperless.ai.matching.get_objects_for_user_owner_aware") + def test_match_tags_by_name_with_empty_names(self, mock_get_objects): + mock_get_objects.return_value = Tag.objects.all() + names = [None, "", " "] + result = match_tags_by_name(names, user=None) + self.assertEqual(result, []) + + @patch("paperless.ai.matching.get_objects_for_user_owner_aware") + def test_match_tags_with_fuzzy_matching(self, mock_get_objects): + mock_get_objects.return_value = Tag.objects.all() + names = ["Test Taag 1", "Teest Tag 2"] + result = match_tags_by_name(names, user=None) + self.assertEqual(len(result), 2) + self.assertEqual(result[0].name, "Test Tag 1") + self.assertEqual(result[1].name, "Test Tag 2")