summaryrefslogtreecommitdiff
path: root/tests/model_fields
diff options
context:
space:
mode:
authorHisham Mahmood <hishammahmood41@gmail.com>2024-08-21 16:51:25 +0500
committerGitHub <noreply@github.com>2024-08-21 08:51:25 -0300
commit519087819ed6e8bfbe6be208df71a7df19f23a58 (patch)
treed6f37c583b170cf8876edef719ef46b3b8108f5d /tests/model_fields
parentba46b09f31f0f02eafae28ef8536e4560f98dadb (diff)
Fixed #35695 -- Ensured FileFields use a storage pointing to a temp directory in model_fields tests.
Diffstat (limited to 'tests/model_fields')
-rw-r--r--tests/model_fields/models.py14
1 files changed, 8 insertions, 6 deletions
diff --git a/tests/model_fields/models.py b/tests/model_fields/models.py
index 5dfed00329..599efafe7e 100644
--- a/tests/model_fields/models.py
+++ b/tests/model_fields/models.py
@@ -21,6 +21,10 @@ except ImportError:
Image = None
+# Set up a temp directory for file storage.
+temp_storage_dir = tempfile.mkdtemp()
+temp_storage = FileSystemStorage(temp_storage_dir)
+
test_collation = SimpleLazyObject(
lambda: connection.features.test_collations["virtual"]
)
@@ -206,7 +210,9 @@ class VerboseNameField(models.Model):
field5 = models.DateTimeField("verbose field5")
field6 = models.DecimalField("verbose field6", max_digits=6, decimal_places=1)
field7 = models.EmailField("verbose field7")
- field8 = models.FileField("verbose field8", upload_to="unused")
+ field8 = models.FileField(
+ "verbose field8", storage=temp_storage, upload_to="unused"
+ )
field9 = models.FilePathField("verbose field9")
field10 = models.FloatField("verbose field10")
# Don't want to depend on Pillow in this test
@@ -256,7 +262,7 @@ class DataModel(models.Model):
class Document(models.Model):
- myfile = models.FileField(upload_to="unused", unique=True)
+ myfile = models.FileField(storage=temp_storage, upload_to="unused", unique=True)
###############################################################################
@@ -282,10 +288,6 @@ if Image:
class TestImageField(models.ImageField):
attr_class = TestImageFieldFile
- # Set up a temp directory for file storage.
- temp_storage_dir = tempfile.mkdtemp()
- temp_storage = FileSystemStorage(temp_storage_dir)
-
class Person(models.Model):
"""
Model that defines an ImageField with no dimension fields.