storage wrapper redesign WIP 3 (added test purging a bucket)

pyinfra/storage/storage.py

@@ -11,6 +11,9 @@ class Storage:
     def get_all(self, bucket_name):
         return self.__adapter.get_all(bucket_name)
 
+    def purge(self, bucket_name):
+        return self.__adapter.purge(bucket_name)
+
 
 
 

pyinfra/test/storage/adapter_mock.py

@@ -14,3 +14,6 @@ class StorageAdapterMock:
 
     def get_all(self, bucket_name):
         return self.__client.get_all(bucket_name)
+
+    def purge(self, bucket_name):
+        return self.__client.purge(bucket_name)

pyinfra/test/storage/client_mock.py

@@ -13,3 +13,7 @@ class StorageClientMock:
 
     def get_all(self, bucket_name):
         return self.__data[bucket_name].values()
+
+    def purge(self, bucket_name):
+        self.__data[bucket_name] = {}
+

pyinfra/test/unit_tests/storage_adapter_test.py

@@ -1,4 +1,2 @@
 from pyinfra.storage.adapter import StorageAdapter
 from pyinfra.test.storage.client_mock import StorageClientMock
-
-

pyinfra/test/unit_tests/storage_test.py

@@ -18,9 +18,10 @@ def bucket_name():
     return "test-bucket"
 
 
-# def test_purging_bucket_yields_empty_bucket(storage, bucket):
-#     storage.purge(bucket)
-#     assert storage.list_objects(bucket)
+def test_purging_bucket_yields_empty_bucket(storage, bucket_name):
+    storage.purge(bucket_name)
+    data_received = storage.get_all(bucket_name)
+    assert not {*data_received}
 
 
 def test_getting_object_put_in_bucket_is_object(storage, bucket_name):