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)