28 lines
755 B
Python
28 lines
755 B
Python
from itertools import repeat
|
|
|
|
|
|
class StorageClientMock:
|
|
def __init__(self):
|
|
self.__data = {}
|
|
|
|
def make_bucket(self, bucket_name):
|
|
self.__data[bucket_name] = {}
|
|
|
|
def bucket_exists(self, bucket_name):
|
|
return bucket_name in self.__data
|
|
|
|
def put(self, bucket_name, object_name, data):
|
|
self.__data[bucket_name][object_name] = data
|
|
|
|
def get(self, bucket_name, object_name):
|
|
return self.__data[bucket_name][object_name]
|
|
|
|
def get_all(self, bucket_name):
|
|
return self.__data[bucket_name].values()
|
|
|
|
def clear_bucket(self, bucket_name):
|
|
self.__data[bucket_name] = {}
|
|
|
|
def get_all_object_names(self, bucket_name):
|
|
return zip(repeat(bucket_name), self.__data[bucket_name])
|