paperless-ngx/src/documents/tests/test_document_model.py

from unittest import mock

from django.test import TestCase

from ..models import Document, Correspondent


class TestDocument(TestCase):

    def test_file_deletion(self):
        document = Document.objects.create(
            correspondent=Correspondent.objects.create(name="Test0"),
            title="Title",
            content="content",
            checksum="checksum",
            mime_type="application/pdf"
        )

        file_path = document.source_path
        thumb_path = document.thumbnail_path

        with mock.patch("documents.signals.handlers.os.unlink") as mock_unlink:
            document.delete()
            mock_unlink.assert_any_call(file_path)
            mock_unlink.assert_any_call(thumb_path)
            self.assertEqual(mock_unlink.call_count, 2)