plugin_test.py 文件源码

def setUp(self):
        self.plugin_a = Mock(
            name='snmptrapd plugin a')
        self.plugin_a.strip.return_value = 'nav.snmptrapd.handlers.foo'

        self.plugin_b = Mock(
            name='snmptrapd plguin b')
        self.plugin_b.strip.return_value = 'nav.snmptrapd.handlers.bar'
        del self.plugin_b.initialize

        def raise_exception():
            raise Exception('boom')

        self.bad_plugin = Mock(name='snmptrapd plugin which is bad')
        self.bad_plugin.strip.return_value = 'nav.snmptrapd.handlers.bad_plugin'
        self.bad_plugin.initialize=raise_exception

        self.patcher = patch.dict(sys.modules, {
            'nav.snmptrapd.handlers.foo': self.plugin_a,
            'nav.snmptrapd.handlers.bar': self.plugin_b,
            'nav.snmptrapd.handlers.bad_plugin': self.bad_plugin,
            'nav.snmptrapd.handlers.non_existent': None
        })
        self.patcher.start()