def parametrised_storage(request, tmpdir_factory) -> KeyValueStorage: if request.param == 'memory': db = KeyValueStorageInMemory() elif request.param == 'leveldb': db = KeyValueStorageLeveldb( tmpdir_factory.mktemp('').strpath, 'some_db') else: raise ValueError('Unsupported storage') yield db db.close()
def parametrised_storage(request, tmpdir_factory) -> KeyValueStorage: if request.param == 'memory': db = KeyValueStorageInMemory() elif request.param == 'leveldb': db = KeyValueStorageLeveldb(tmpdir_factory.mktemp('').strpath, 'some_db') else: raise ValueError('Unsupported storage') yield db db.close()
def kv() -> KeyValueStorageLeveldb: kv = KeyValueStorageInMemory() yield kv kv.close()