storage wrapper redesign WIP 2 (added test for dowloading all files from a bucket)

pyinfra/storage/storage.py

@@ -8,6 +8,9 @@ class Storage:
     def get(self, bucket_name, object_name):
         return self.__adapter.get(bucket_name, object_name)
 
+    def get_all(self, bucket_name):
+        return self.__adapter.get_all(bucket_name)
+
 
 
 

pyinfra/test/storage/adapter_mock.py

@@ -11,3 +11,6 @@ class StorageAdapterMock:
 
     def get(self, bucket_name, object_name):
         return self.__client.get(bucket_name, object_name)
+
+    def get_all(self, bucket_name):
+        return self.__client.get_all(bucket_name)

pyinfra/test/storage/client_mock.py

@@ -10,3 +10,6 @@ class StorageClientMock:
 
     def get(self, bucket_name, object_name):
         return self.__data[bucket_name][object_name]
+
+    def get_all(self, bucket_name):
+        return self.__data[bucket_name].values()

pyinfra/test/unit_tests/storage_test.py

@@ -25,6 +25,14 @@ def bucket_name():
 
 def test_getting_object_put_in_bucket_is_object(storage, bucket_name):
     with tempfile.SpooledTemporaryFile() as f:
-        storage.put(bucket_name, "testfile", f)
-        data_received = storage.get(bucket_name, "testfile")
+        storage.put(bucket_name, "file", f)
+        data_received = storage.get(bucket_name, "file")
         assert f == data_received
+
+
+def test_getting_objects_put_in_bucket_are_objects(storage, bucket_name):
+    with tempfile.SpooledTemporaryFile() as f1, tempfile.SpooledTemporaryFile() as f2:
+        storage.put(bucket_name, "file1", f1)
+        storage.put(bucket_name, "file2", f2)
+        data_received = storage.get_all(bucket_name)
+        assert {f1, f2} == {*data_received}