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]