Refs #29983 -- Added test for FILE_UPLOAD_TEMP_DIR pathlib support.

This commit is contained in:
Jon Dufresne 2019-11-05 23:20:23 -08:00 committed by Carlton Gibson
parent 92e5abd7a3
commit 1c4c68dfed
1 changed files with 8 additions and 0 deletions

View File

@ -5,6 +5,7 @@ import struct
import tempfile import tempfile
import unittest import unittest
from io import BytesIO, StringIO, TextIOWrapper from io import BytesIO, StringIO, TextIOWrapper
from pathlib import Path
from unittest import mock from unittest import mock
from django.core.files import File from django.core.files import File
@ -15,6 +16,7 @@ from django.core.files.uploadedfile import (
InMemoryUploadedFile, SimpleUploadedFile, TemporaryUploadedFile, InMemoryUploadedFile, SimpleUploadedFile, TemporaryUploadedFile,
UploadedFile, UploadedFile,
) )
from django.test import override_settings
try: try:
from PIL import Image from PIL import Image
@ -231,6 +233,12 @@ class TemporaryUploadedFileTests(unittest.TestCase):
with TemporaryUploadedFile('test.txt', 'text/plain', 1, 'utf8') as temp_file: with TemporaryUploadedFile('test.txt', 'text/plain', 1, 'utf8') as temp_file:
self.assertTrue(temp_file.file.name.endswith('.upload.txt')) self.assertTrue(temp_file.file.name.endswith('.upload.txt'))
def test_file_upload_temp_dir_pathlib(self):
with tempfile.TemporaryDirectory() as tmp_dir:
with override_settings(FILE_UPLOAD_TEMP_DIR=Path(tmp_dir)):
with TemporaryUploadedFile('test.txt', 'text/plain', 1, 'utf-8') as temp_file:
self.assertTrue(os.path.exists(temp_file.file.name))
class DimensionClosingBug(unittest.TestCase): class DimensionClosingBug(unittest.TestCase):
""" """