diff options
| author | Hisham Mahmood <hishammahmood41@gmail.com> | 2024-08-21 16:51:25 +0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-08-21 08:51:25 -0300 |
| commit | 519087819ed6e8bfbe6be208df71a7df19f23a58 (patch) | |
| tree | d6f37c583b170cf8876edef719ef46b3b8108f5d /tests/model_fields | |
| parent | ba46b09f31f0f02eafae28ef8536e4560f98dadb (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.py | 14 |
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. |
