stevedore/stevedore/tests/test_hook.py
Doug Hellmann 69e17801a3 Add ExtensionManager.__getitem__
Allow extensions to be accessed directly by name.

Fixes issue #15

Signed-off-by: Doug Hellmann <doug.hellmann@dreamhost.com>
2013-06-03 10:54:49 -04:00

44 lines
902 B
Python

from stevedore import hook
def test_hook():
em = hook.HookManager(
'stevedore.test.extension',
't1',
invoke_on_load=True,
invoke_args=('a',),
invoke_kwds={'b': 'B'},
)
assert len(em.extensions) == 1
assert em.names() == ['t1']
def test_get_by_name():
em = hook.HookManager(
'stevedore.test.extension',
't1',
invoke_on_load=True,
invoke_args=('a',),
invoke_kwds={'b': 'B'},
)
e_list = em['t1']
assert len(e_list) == 1
e = e_list[0]
assert e.name == 't1'
def test_get_by_name_missing():
em = hook.HookManager(
'stevedore.test.extension',
't1',
invoke_on_load=True,
invoke_args=('a',),
invoke_kwds={'b': 'B'},
)
try:
em['t2']
except KeyError:
pass
else:
assert False, 'Failed to raise KeyError'