mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-04-02 13:45:10 -05:00
24 lines
743 B
Python
24 lines
743 B
Python
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",
|
|
)
|
|
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)
|