1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180
|
""" Tests for the Egg plugin manager. """
# Standard library imports.
import unittest
# Enthought library imports.
from enthought.envisage.api import EggPluginManager
# Local imports.
from egg_based_test_case import EggBasedTestCase
class EggPluginManagerTestCase(EggBasedTestCase):
""" Tests for the Egg plugin manager. """
###########################################################################
# Tests.
###########################################################################
# fixme: Depending how many eggs are on sys.path, this test may take too
# long to be part of the TDD cycle.
def test_no_include_or_exclude(self):
""" no include or exclude """
# Add all of the eggs in the egg basket.
self._add_eggs_on_path([self.egg_dir])
# Make sure that the plugin manager only includes those plugins.
plugin_manager = EggPluginManager()
# We don't know how many plugins we will actually get - it depends on
# what eggs are on sys.path! What we *do* know however is the the 3
# 'acme' test eggs should be in there!
ids = [plugin.id for plugin in plugin_manager]
self.assert_('acme.foo' in ids)
self.assert_('acme.bar' in ids)
self.assert_('acme.baz' in ids)
return
def test_include_specific(self):
""" include specific """
# Add all of the eggs in the egg basket.
self._add_eggs_on_path([self.egg_dir])
# The Ids of the plugins that we expect the plugin manager to find.
expected = ['acme.foo', 'acme.bar']
# We explicitly limit the plugins to be just the 'acme' test plugins
# because otherwise the egg plugin manager will pick up *every* plugin
# in *every* egg on sys.path!
include = ['acme\.foo', 'acme\.bar']
# Make sure that the plugin manager only includes those plugins.
plugin_manager = EggPluginManager(include=include)
# Make sure the plugin manager found only the required plugins and that
# it starts and stops them correctly..
self._test_start_and_stop(plugin_manager, expected)
return
def test_include_multiple(self):
""" include multiple """
# Add all of the eggs in the egg basket.
self._add_eggs_on_path([self.egg_dir])
# The Ids of the plugins that we expect the plugin manager to find.
expected = ['acme.foo', 'acme.bar', 'acme.baz']
# We explicitly limit the plugins to be just the 'acme' test plugins
# because otherwise the egg plugin manager will pick up *every* plugin
# in *every* egg on sys.path!
include = ['acme.*']
# Make sure that the plugin manager only includes those plugins.
plugin_manager = EggPluginManager(include=include)
# Make sure the plugin manager found only the required plugins and that
# it starts and stops them correctly..
self._test_start_and_stop(plugin_manager, expected)
return
def test_exclude_specific(self):
""" exclude specific """
# Add all of the eggs in the egg basket.
self._add_eggs_on_path([self.egg_dir])
# The Ids of the plugins that we expect the plugin manager to find.
expected = ['acme.bar']
# We explicitly limit the plugins to be just the 'acme' test plugins
# because otherwise the egg plugin manager will pick up *every* plugin
# in *every* egg on sys.path!
include = ['acme.*']
# Now exclude all but 'acme.bar'...
exclude = ['acme\.foo', 'acme\.baz']
# Make sure that the plugin manager excludes the specified plugins.
plugin_manager = EggPluginManager(include=include, exclude=exclude)
# Make sure the plugin manager found only the required plugins and that
# it starts and stops them correctly..
self._test_start_and_stop(plugin_manager, expected)
return
def test_exclude_multiple(self):
""" exclude multiple """
# Add all of the eggs in the egg basket.
self._add_eggs_on_path([self.egg_dir])
# The Ids of the plugins that we expect the plugin manager to find.
expected = ['acme.foo']
# We explicitly limit the plugins to be just the 'acme' test plugins
# because otherwise the egg plugin manager will pick up *every* plugin
# in *every* egg on sys.path!
include = ['acme.*']
# Now exclude every plugin that starts with 'acme.b'.
exclude = ['acme\.b.*']
# Make sure that the plugin manager excludes the specified plugins.
plugin_manager = EggPluginManager(include=include, exclude=exclude)
# Make sure the plugin manager found only the required plugins and that
# it starts and stops them correctly..
self._test_start_and_stop(plugin_manager, expected)
return
###########################################################################
# Private interface.
###########################################################################
def _test_start_and_stop(self, plugin_manager, expected):
""" Make sure the plugin manager starts and stops the expected plugins.
"""
# Make sure the plugin manager found only the required plugins.
self.assertEqual(expected, [plugin.id for plugin in plugin_manager])
# Start the plugin manager. This starts all of the plugin manager's
# plugins.
plugin_manager.start()
# Make sure all of the the plugins were started.
for id in expected:
plugin = plugin_manager.get_plugin(id)
self.assertNotEqual(None, plugin)
self.assertEqual(True, plugin.started)
# Stop the plugin manager. This stops all of the plugin manager's
# plugins.
plugin_manager.stop()
# Make sure all of the the plugins were stopped.
for id in expected:
plugin = plugin_manager.get_plugin(id)
self.assertNotEqual(None, plugin)
self.assertEqual(True, plugin.stopped)
return
# Entry point for stand-alone testing.
if __name__ == '__main__':
unittest.main()
#### EOF ######################################################################
|