pyinfra/pyinfra/test/storage/client_mock.py
2022-02-23 15:46:59 +01:00

13 lines
331 B
Python

from collections import defaultdict
class StorageClientMock:
def __init__(self):
self.__data = defaultdict(dict)
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]