28 lines
957 B
Python
28 lines
957 B
Python
from pyinfra.storage.adapters.adapter import StorageAdapter
|
|
from pyinfra.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 bucket_exists(self, bucket_name):
|
|
return self.__client.bucket_exists(bucket_name)
|
|
|
|
def put(self, bucket_name, object_name, data):
|
|
return self.__client.put(bucket_name, object_name, data)
|
|
|
|
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)
|
|
|
|
def clear_bucket(self, bucket_name):
|
|
return self.__client.clear_bucket(bucket_name)
|