py_task/venv/lib/python3.9/site-packages/stevedore/tests/test_hook.py

#  Licensed under the Apache License, Version 2.0 (the "License"); you may
#  not use this file except in compliance with the License. You may obtain
#  a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#  License for the specific language governing permissions and limitations
#  under the License.

from stevedore import hook
from stevedore.tests import utils


class TestHook(utils.TestCase):
    def test_hook(self):
        em = hook.HookManager(
            'stevedore.test.extension',
            't1',
            invoke_on_load=True,
            invoke_args=('a',),
            invoke_kwds={'b': 'B'},
        )
        self.assertEqual(len(em.extensions), 1)
        self.assertEqual(em.names(), ['t1'])

    def test_get_by_name(self):
        em = hook.HookManager(
            'stevedore.test.extension',
            't1',
            invoke_on_load=True,
            invoke_args=('a',),
            invoke_kwds={'b': 'B'},
        )
        e_list = em['t1']
        self.assertEqual(len(e_list), 1)
        e = e_list[0]
        self.assertEqual(e.name, 't1')

    def test_get_by_name_missing(self):
        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'