fixed up a test case

This commit is contained in:
Jonas Winkler 2020-11-25 17:28:49 +01:00
parent 81aaadb2a3
commit 64180b5668

View File

@ -1,12 +1,29 @@
import os
import shutil
import tempfile
from pathlib import Path
from unittest import mock from unittest import mock
from django.test import TestCase from django.test import TestCase, override_settings
from ..models import Document, Correspondent from ..models import Document, Correspondent
class TestDocument(TestCase): class TestDocument(TestCase):
def setUp(self) -> None:
self.originals_dir = tempfile.mkdtemp()
self.thumb_dir = tempfile.mkdtemp()
override_settings(
ORIGINALS_DIR=self.originals_dir,
THUMBNAIL_DIR=self.thumb_dir,
).enable()
def tearDown(self) -> None:
shutil.rmtree(self.originals_dir)
shutil.rmtree(self.thumb_dir)
def test_file_deletion(self): def test_file_deletion(self):
document = Document.objects.create( document = Document.objects.create(
correspondent=Correspondent.objects.create(name="Test0"), correspondent=Correspondent.objects.create(name="Test0"),
@ -19,6 +36,9 @@ class TestDocument(TestCase):
file_path = document.source_path file_path = document.source_path
thumb_path = document.thumbnail_path thumb_path = document.thumbnail_path
Path(file_path).touch()
Path(thumb_path).touch()
with mock.patch("documents.signals.handlers.os.unlink") as mock_unlink: with mock.patch("documents.signals.handlers.os.unlink") as mock_unlink:
document.delete() document.delete()
mock_unlink.assert_any_call(file_path) mock_unlink.assert_any_call(file_path)