lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor: tests coverage added for delete pages

Browse Source
This commit is contained in:
s0llvan 2024-09-22 13:06:50 +00:00
parent be6a42b0e5
commit 08af50e919
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/documents/tests/test_bulk_edit.py
View File

@ -389,6 +389,7 @@ class TestPDFActions(DirectoriesMixin, TestCase):
title="B", title="B",
filename=sample2, filename=sample2,
mime_type="application/pdf", mime_type="application/pdf",
pages_count=8,
) )
self.doc2.archive_filename = sample2_archive self.doc2.archive_filename = sample2_archive
self.doc2.save() self.doc2.save()
@ -681,14 +682,20 @@ class TestPDFActions(DirectoriesMixin, TestCase):
THEN: THEN:
- Save should be called once - Save should be called once
- Archive file should be updated 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] doc_ids = [self.doc2.id]
initial_pages_count = self.doc2.pages_count
pages = [1, 3] pages = [1, 3]
result = bulk_edit.delete_pages(doc_ids, pages) result = bulk_edit.delete_pages(doc_ids, pages)
mock_pdf_save.assert_called_once() mock_pdf_save.assert_called_once()
mock_update_archive_file.assert_called_once() mock_update_archive_file.assert_called_once()
self.assertEqual(result, "OK") 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("documents.tasks.update_document_archive_file.delay")
@mock.patch("pikepdf.Pdf.save") @mock.patch("pikepdf.Pdf.save")
def test_delete_pages_with_error(self, mock_pdf_save, mock_update_archive_file): def test_delete_pages_with_error(self, mock_pdf_save, mock_update_archive_file):