From 08af50e919ece23f679964d7f713351c87b89568 Mon Sep 17 00:00:00 2001
From: s0llvan <178677095+s0llvan@users.noreply.github.com>
Date: Sun, 22 Sep 2024 13:06:50 +0000
Subject: [PATCH] Refactor: tests coverage added for delete pages

---
 src/documents/tests/test_bulk_edit.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/documents/tests/test_bulk_edit.py b/src/documents/tests/test_bulk_edit.py
index fed93cd01..1f91619a6 100644
--- a/src/documents/tests/test_bulk_edit.py
+++ b/src/documents/tests/test_bulk_edit.py
@@ -389,6 +389,7 @@ class TestPDFActions(DirectoriesMixin, TestCase):
             title="B",
             filename=sample2,
             mime_type="application/pdf",
+            pages_count=8,
         )
         self.doc2.archive_filename = sample2_archive
         self.doc2.save()
@@ -681,14 +682,20 @@ class TestPDFActions(DirectoriesMixin, TestCase):
         THEN:
             - Save should be called once
             - Archive file should be updated once
+            - The document's pages_count should be reduced by the number of deleted pages
         """
         doc_ids = [self.doc2.id]
+        initial_pages_count = self.doc2.pages_count
         pages = [1, 3]
         result = bulk_edit.delete_pages(doc_ids, pages)
         mock_pdf_save.assert_called_once()
         mock_update_archive_file.assert_called_once()
         self.assertEqual(result, "OK")
 
+        expected_pages_count = initial_pages_count - len(pages)
+        self.doc2.refresh_from_db()
+        self.assertEqual(self.doc2.pages_count, expected_pages_count)
+
     @mock.patch("documents.tasks.update_document_archive_file.delay")
     @mock.patch("pikepdf.Pdf.save")
     def test_delete_pages_with_error(self, mock_pdf_save, mock_update_archive_file):