def test_chain_class_lookup_all(): reg1 = Registry() reg2 = Registry() reg1.register(target, (), 'reg1') reg2.register(target, (), 'reg2') lookup = ChainClassLookup(reg1, reg2) assert list(lookup.all(target, ())) == ['reg1', 'reg2']
def test_chain_class_lookup(): reg1 = Registry() reg2 = Registry() reg2.register(target, (), 'reg2 component') lookup = ChainClassLookup(reg1, reg2) assert lookup.get(target, ()) == 'reg2 component' reg1.register(target, (), 'reg1 component') assert lookup.get(target, ()) == 'reg1 component'