Uploads are already 'lazily' executed

This commit is contained in:
2020-05-21 18:37:41 +02:00
parent 8c4ec83993
commit 019adcba56
2 changed files with 11 additions and 3 deletions

View File

@@ -26,9 +26,7 @@ def create(model: Type[T], **kwargs) -> T:
return _create_with_defaults(models.Transcoding, kwargs, **defaults)
if model is models.Upload:
file = SimpleUploadedFile('some_file.txt', b'some contents') \
if 'file' not in kwargs else None
return _create_with_defaults(models.Upload, kwargs, file=file)
return _create_with_defaults(models.Upload, kwargs, file=SimpleUploadedFile('some_file.txt', b'some contents'))
def _create_with_defaults(model: Type[T], kwargs: dict, **defaults) -> T:

View File

@@ -1,3 +1,6 @@
import os
from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import TestCase
from tests.videodinges import factories, UploadMixin
@@ -7,3 +10,10 @@ class UploadTestCase(UploadMixin, TestCase):
def test_model_is_created(self):
upload = factories.create(Upload)
self.assertEqual(upload.file.name, 'some_file.txt')
self.assertTrue(os.path.exists(os.path.join(self.media_root.name, 'some_file.txt')))
def test_upload_does_not_create_file_when_providing_upload(self):
upload = factories.create(Upload, file=SimpleUploadedFile('my_file.txt', b'some contents'))
self.assertEqual(upload.file.name, 'my_file.txt')
self.assertFalse(os.path.exists(os.path.join(self.media_root.name, 'some_file.txt')))
self.assertTrue(os.path.exists(os.path.join(self.media_root.name, 'my_file.txt')))