Fix unit tests

Ilhasoft · Aug 21, 2018 · 4462003 · 4462003
1 parent 580f436
commit 4462003
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 16 deletions.
diff --git a/temba/assets/models.py b/temba/assets/models.py
@@ -81,7 +81,7 @@ def resolve(self, user, pk):
         filename = '%s_%s.%s' % (self.key, pk, extension)
 
         # if our storage backend is S3
-        if settings.DEFAULT_FILE_STORAGE == 'storages.backends.s3boto.S3BotoStorage' or settings.DEFAULT_FILE_STORAGE == 'temba.utils.custom_storages.MediaStorage':
+        if settings.DEFAULT_FILE_STORAGE == "storages.backends.s3boto.S3BotoStorage":  # pragma: needs cover
             # generate our URL manually so that we can force the download name for the user
             url = default_storage.connection.generate_url(default_storage.querystring_expire,
                                                           method='GET', bucket=default_storage.bucket.name,

diff --git a/temba/utils/custom_storages.py b/temba/utils/custom_storages.py