Fix for #131: delete files on document.delete

This commit is contained in:
Daniel Quinn
2016-08-16 19:13:37 +01:00
parent 58a299b671
commit f5daded930
4 changed files with 39 additions and 7 deletions

View File

@@ -1,6 +1,6 @@
from django.test import TestCase
from ..models import Document, FileInfo
from ..models import FileInfo
class TestAttachment(TestCase):

View File

@@ -0,0 +1,20 @@
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
with mock.patch("documents.signals.handlers.os.unlink") as mock_unlink:
document.delete()
mock_unlink.assert_called_with(file_path)