Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Browse files

[1.5.x] Prevented file_upload tests to leave files behind

Refs #19206.
Backport of 73245b3 from master.
  • Loading branch information...
commit 2f9f211da867d872a0d3576c3ac515d74f5798cd 1 parent a24ffa5
@claudep claudep authored
View
9 tests/regressiontests/file_uploads/models.py
@@ -1,12 +1,5 @@
-import tempfile
-import os
-
-from django.core.files.storage import FileSystemStorage
from django.db import models
-temp_storage = FileSystemStorage(tempfile.mkdtemp())
-UPLOAD_TO = os.path.join(temp_storage.location, 'test_upload')
-
class FileModel(models.Model):
- testfile = models.FileField(storage=temp_storage, upload_to='test_upload')
+ testfile = models.FileField(upload_to='test_upload')
View
49 tests/regressiontests/file_uploads/tests.py
@@ -7,22 +7,36 @@
import json
import os
import shutil
+import tempfile as sys_tempfile
from django.core.files import temp as tempfile
from django.core.files.uploadedfile import SimpleUploadedFile
from django.http.multipartparser import MultiPartParser
from django.test import TestCase, client
+from django.test.utils import override_settings
from django.utils.encoding import force_bytes
from django.utils.six import StringIO
from django.utils import unittest
from . import uploadhandler
-from .models import FileModel, temp_storage, UPLOAD_TO
+from .models import FileModel
UNICODE_FILENAME = 'test-0123456789_中文_Orléans.jpg'
+MEDIA_ROOT = sys_tempfile.mkdtemp()
+UPLOAD_TO = os.path.join(MEDIA_ROOT, 'test_upload')
+@override_settings(MEDIA_ROOT=MEDIA_ROOT)
class FileUploadTests(TestCase):
+ @classmethod
+ def setUpClass(cls):
+ if not os.path.isdir(MEDIA_ROOT):
+ os.makedirs(MEDIA_ROOT)
+
+ @classmethod
+ def tearDownClass(cls):
+ shutil.rmtree(MEDIA_ROOT)
+
def test_simple_upload(self):
with open(__file__, 'rb') as fp:
post_data = {
@@ -83,7 +97,8 @@ def test_base64_upload(self):
self.assertEqual(received['file'], test_string)
def test_unicode_file_name(self):
- tdir = tempfile.gettempdir()
+ tdir = sys_tempfile.mkdtemp()
+ self.addCleanup(shutil.rmtree, tdir, True)
# This file contains chinese symbols and an accented char in the name.
with open(os.path.join(tdir, UNICODE_FILENAME), 'w+b') as file1:
@@ -96,11 +111,6 @@ def test_unicode_file_name(self):
response = self.client.post('/file_uploads/unicode_name/', post_data)
- try:
- os.unlink(file1.name)
- except OSError:
- pass
-
self.assertEqual(response.status_code, 200)
def test_dangerous_file_names(self):
@@ -347,26 +357,28 @@ def test_filename_case_preservation(self):
# shouldn't differ.
self.assertEqual(os.path.basename(obj.testfile.path), 'MiXeD_cAsE.txt')
-class DirectoryCreationTests(unittest.TestCase):
+@override_settings(MEDIA_ROOT=MEDIA_ROOT)
+class DirectoryCreationTests(TestCase):
"""
Tests for error handling during directory creation
via _save_FIELD_file (ticket #6450)
"""
+ @classmethod
+ def setUpClass(cls):
+ if not os.path.isdir(MEDIA_ROOT):
+ os.makedirs(MEDIA_ROOT)
+
+ @classmethod
+ def tearDownClass(cls):
+ shutil.rmtree(MEDIA_ROOT)
+
def setUp(self):
self.obj = FileModel()
- if not os.path.isdir(temp_storage.location):
- os.makedirs(temp_storage.location)
- if os.path.isdir(UPLOAD_TO):
- os.chmod(UPLOAD_TO, 0o700)
- shutil.rmtree(UPLOAD_TO)
-
- def tearDown(self):
- os.chmod(temp_storage.location, 0o700)
- shutil.rmtree(temp_storage.location)
def test_readonly_root(self):
"""Permission errors are not swallowed"""
- os.chmod(temp_storage.location, 0o500)
+ os.chmod(MEDIA_ROOT, 0o500)
+ self.addCleanup(os.chmod, MEDIA_ROOT, 0o700)
try:
self.obj.testfile.save('foo.txt', SimpleUploadedFile('foo.txt', b'x'))
except OSError as err:
@@ -378,6 +390,7 @@ def test_not_a_directory(self):
"""The correct IOError is raised when the upload directory name exists but isn't a directory"""
# Create a file with the upload directory name
open(UPLOAD_TO, 'wb').close()
+ self.addCleanup(os.remove, UPLOAD_TO)
with self.assertRaises(IOError) as exc_info:
self.obj.testfile.save('foo.txt', SimpleUploadedFile('foo.txt', b'x'))
# The test needs to be done on a specific string as IOError
View
4 tests/regressiontests/file_uploads/views.py
@@ -9,8 +9,8 @@
from django.utils import six
from django.utils.encoding import force_bytes
-from .models import FileModel, UPLOAD_TO
-from .tests import UNICODE_FILENAME
+from .models import FileModel
+from .tests import UNICODE_FILENAME, UPLOAD_TO
from .uploadhandler import QuotaUploadHandler, ErroringUploadHandler
Please sign in to comment.
Something went wrong with that request. Please try again.