Fix failing tests

Author: codingjoe

s3file/middleware.py

@@ -29,7 +29,11 @@ def get_files_from_storage(paths):
         for path in paths:
             path = pathlib.PurePosixPath(path)
             try:
-                f = storage.open(str(path.relative_to(storage.location)))
+                location = storage.aws_location
+            except AttributeError:
+                location = storage.location
+            try:
+                f = storage.open(str(path.relative_to(location)))
                 f.name = path.name
                 yield f
             except (OSError, ValueError):

s3file/storages.py

@@ -11,11 +11,12 @@
 
 class S3MockStorage(FileSystemStorage):
     @property
-    def location(self):
+    def aws_location(self):
         return getattr(settings, "AWS_LOCATION", "")
 
-    def path(self, name):
-        return safe_join(os.path.abspath(self.base_location), self.location, name)
+    @property
+    def location(self):
+        return safe_join(os.path.abspath(self.base_location), self.aws_location)
 
     class connection:
         class meta:

tests/test_forms.py

@@ -36,10 +36,11 @@ def freeze(self, monkeypatch):
         """Freeze datetime and UUID."""
         monkeypatch.setattr(
             "s3file.forms.S3FileInputMixin.upload_folder",
-            os.path.join(storage.location, "tmp"),
+            os.path.join(storage.aws_location, "tmp"),
         )
 
     def test_value_from_datadict(self, client, upload_file):
+        print(storage.location)
         with open(upload_file) as f:
             uploaded_file = storage.save("test.jpg", f)
         response = client.post(
@@ -227,7 +228,5 @@ def test_media(self):
         assert ClearableFileInput().media._js == ["s3file/js/s3file.js"]
 
     def test_upload_folder(self):
-        assert ClearableFileInput().upload_folder.startswith(
-            "custom/location/tmp/s3file/"
-        )
-        assert len(ClearableFileInput().upload_folder) == 49
+        assert "custom/location/tmp/s3file/" in ClearableFileInput().upload_folder
+        assert len(os.path.basename(ClearableFileInput().upload_folder)) == 22

tests/test_middleware.py

@@ -14,7 +14,7 @@ def test_get_files_from_storage(self):
             "tmp/s3file/test_get_files_from_storage", ContentFile(content)
         )
         files = S3FileMiddleware.get_files_from_storage(
-            [os.path.join(storage.location, name)]
+            [os.path.join(storage.aws_location, name)]
         )
         file = next(files)
         assert file.read() == content

tests/testapp/settings.py

@@ -44,7 +44,7 @@
     },
 ]
 
-USE_L10N = True
+USE_TZ = True
 
 AWS_ACCESS_KEY_ID = "testaccessid"
 AWS_SECRET_ACCESS_KEY = "supersecretkey"