31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from pyinfra.storage.adapters.adapter import StorageAdapter
|
|
from test.storage.client_mock import StorageClientMock
|
|
|
|
|
|
class StorageAdapterMock(StorageAdapter):
|
|
def __init__(self, client: StorageClientMock):
|
|
assert isinstance(client, StorageClientMock)
|
|
super().__init__(client=client)
|
|
self.__client = self._StorageAdapter__client
|
|
|
|
def make_bucket(self, bucket_name):
|
|
self.__client.make_bucket(bucket_name)
|
|
|
|
def has_bucket(self, bucket_name):
|
|
return self.__client.has_bucket(bucket_name)
|
|
|
|
def put_object(self, bucket_name, object_name, data):
|
|
return self.__client.put_object(bucket_name, object_name, data)
|
|
|
|
def get_object(self, bucket_name, object_name):
|
|
return self.__client.get_object(bucket_name, object_name)
|
|
|
|
def get_all_objects(self, bucket_name):
|
|
return self.__client.get_all_objects(bucket_name)
|
|
|
|
def clear_bucket(self, bucket_name):
|
|
return self.__client.clear_bucket(bucket_name)
|
|
|
|
def get_all_object_names(self, bucket_name):
|
|
return self.__client.get_all_object_names(bucket_name)
|