def load(self): for path in getattr(settings, 'EXTENSIONS', []): logger.debug('Loading extension extension %s' % path) try: module = __import__(*self._make_import_args(path)) except ImportError: logger.warning('Cannot import extension %s' % path) continue self.load_module_members(module) # Core loading self.core = set([ ControlExtension(self, self.bot), HelpExtension(self, self.bot), IgnoreExtension(self.bot), ])
class HelpExtensionTestCase(TestCase): def setUp(self): self.help = HelpExtension(Mock(), mock_bot()) def test_help_all_single_extension(self): fake_ext = Mock(NAME='blah', usage='foo bar baz') self.help.registry.get_all_extensions.return_value = [fake_ext] assert 'blah: foo bar baz' in self.help.help_all() def test_help_all_multiple_extensions(self): fake_ext = Mock(NAME='blah', usage='foo bar baz') fake_ext2 = Mock(NAME='myext', usage='helga do something') self.help.registry.get_all_extensions.return_value = [fake_ext, fake_ext2] assert 'blah: foo bar baz' in self.help.help_all() assert 'myext: helga do something' in self.help.help_all() def test_help_unknown_ext(self): self.help.registry.get_all_extensions.return_value = [Mock()] assert "don't know that one" in self.help.help('foo') def test_help_misconfigured_ext(self): self.help.registry.is_extension_name.return_value = True self.help.registry.get_all_extensions.return_value = [Mock()] assert "don't know that one" in self.help.help('foo') def test_help_responds(self): fake_ext = Mock(NAME='blah', usage='foo bar baz') self.help.registry.is_extension_name.return_value = True self.help.registry.get_all_extensions.return_value = [fake_ext] assert 'USAGE: foo bar baz' == self.help.help('blah')
def setUp(self): self.help = HelpExtension(Mock(), mock_bot())