From 897f7d219986a94fca631b6afe37ff0dfa6981e0 Mon Sep 17 00:00:00 2001 From: Trenton H <797416+stumpylog@users.noreply.github.com> Date: Mon, 30 Mar 2026 14:11:07 -0700 Subject: [PATCH] Tests: cover document_index reindex --if-needed flag Two cases: skips when _needs_rebuild returns False; runs when True. Co-Authored-By: Claude Sonnet 4.6 --- src/documents/tests/test_management.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/documents/tests/test_management.py b/src/documents/tests/test_management.py index 7719d21dd..afc589705 100644 --- a/src/documents/tests/test_management.py +++ b/src/documents/tests/test_management.py @@ -114,6 +114,26 @@ class TestMakeIndex(TestCase): call_command("document_index", "optimize", skip_checks=True) m.assert_called_once() + @mock.patch("documents.management.commands.document_index.index_reindex") + @mock.patch("documents.search._schema._needs_rebuild", return_value=False) + def test_reindex_if_needed_skips_when_up_to_date( + self, + _needs_rebuild, + reindex, + ) -> None: + call_command("document_index", "reindex", if_needed=True, skip_checks=True) + reindex.assert_not_called() + + @mock.patch("documents.management.commands.document_index.index_reindex") + @mock.patch("documents.search._schema._needs_rebuild", return_value=True) + def test_reindex_if_needed_runs_when_rebuild_needed( + self, + _needs_rebuild, + reindex, + ) -> None: + call_command("document_index", "reindex", if_needed=True, skip_checks=True) + reindex.assert_called_once() + @pytest.mark.management class TestRenamer(DirectoriesMixin, FileSystemAssertsMixin, TestCase):