##// END OF EJS Templates
upstream » ipython
RSS

Clone from https://github.com/ipython/ipython.git

Fixing code to assume msg_type and msg_id are top-level....
Fixing code to assume msg_type and msg_id are top-level. * I have gone through and looked for instances of ['msg_type'] and ['msg_id'] and tried to make sure that I added ['header'] so pull the values out of the header. * But there are many cases where I can't tell if the dict is the full message or the header already. This is especially true of the msg_id in the parallel db parts of the code. * Tests pass, but this is scary.
Brian Granger - - Load All Authors

File last commit:

r2738:95a4f176
r4230:36064ac1
Show More
test_plugin.py
46 lines | 1.3 KiB | text/x-python | PythonLexer
/ IPython / core / tests / test_plugin.py
History | Annotation | Raw |Copy content |Copy permalink
"""Tests for plugin.py"""
#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------
from unittest import TestCase
from IPython.core.plugin import Plugin, PluginManager
#-----------------------------------------------------------------------------
# Tests
#-----------------------------------------------------------------------------
class FooPlugin(Plugin):
pass
class BarPlugin(Plugin):
pass
class BadPlugin(object):
pass
class PluginTest(TestCase):
def setUp(self):
self.manager = PluginManager()
def test_register_get(self):
self.assertEquals(None, self.manager.get_plugin('foo'))
foo = FooPlugin()
self.manager.register_plugin('foo', foo)
self.assertEquals(foo, self.manager.get_plugin('foo'))
bar = BarPlugin()
self.assertRaises(KeyError, self.manager.register_plugin, 'foo', bar)
bad = BadPlugin()
self.assertRaises(TypeError, self.manager.register_plugin, 'bad')
def test_unregister(self):
foo = FooPlugin()
self.manager.register_plugin('foo', foo)
self.manager.unregister_plugin('foo')
self.assertEquals(None, self.manager.get_plugin('foo'))