pyinfra/pyinfra/test/storage/adapter_mock.py

20 lines
618 B
Python

from pyinfra.test.storage.client_mock import StorageClientMock
class StorageAdapterMock:
def __init__(self, client: StorageClientMock):
assert isinstance(client, StorageClientMock)
self.__client = client
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 purge(self, bucket_name):
return self.__client.purge(bucket_name)