def test_get_all_stores(nested_schema_spec: Dict) -> None:
nested_schema_spec['Store'] = {'Name': 'memstore', 'Type': Type.BLURR_STORE_MEMORY}
schema_loader = SchemaLoader()
schema_loader.add_schema_spec(nested_schema_spec)
# No store instantiated yet.
assert schema_loader.get_all_stores() == []
assert isinstance(schema_loader.get_store('test.memstore'), MemoryStore)
stores = schema_loader.get_all_stores()
assert len(stores) == 1
assert isinstance(stores[0], MemoryStore)
def _get_store(schema_loader: SchemaLoader) -> Store:
stores = schema_loader.get_all_stores()
if not stores:
fq_name_and_schema = schema_loader.get_schema_specs_of_type(
Type.BLURR_STORE_DYNAMO, Type.BLURR_STORE_MEMORY)
return schema_loader.get_store(next(iter(fq_name_and_schema)))
return stores[0]
def _get_store(schema_loader: SchemaLoader) -> Store:
return schema_loader.get_all_stores()[0]
