← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_plugins.py

[merge] robert

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_plugins.py
23
23
 
24
24
import os
25
25
 
 
26
import bzrlib.plugin
 
27
import bzrlib.plugins
26
28
from bzrlib.tests import TestCaseInTempDir
27
29
 
28
30
class PluginTest(TestCaseInTempDir):
77
79
"""
78
80
 
79
81
# TODO: Write a test for plugin decoration of commands.
 
82
 
 
83
class TestOneNamedPluginOnly(TestCaseInTempDir):
 
84
 
 
85
    activeattributes = {}
 
86
 
 
87
    def test_plugins_with_the_same_name_are_not_loaded(self):
 
88
        # This test tests that having two plugins in different
 
89
        # directories does not result in both being loaded.
 
90
        # get a file name we can use which is also a valid attribute
 
91
        # for accessing in activeattributes. - we cannot give import parameters.
 
92
        tempattribute = "0"
 
93
        self.failIf(tempattribute in self.activeattributes)
 
94
        # set a place for the plugins to record their loading, and at the same
 
95
        # time validate that the location the plugins should record to is
 
96
        # valid and correct.
 
97
        bzrlib.tests.test_plugins.TestOneNamedPluginOnly.activeattributes \
 
98
            [tempattribute] = []
 
99
        self.failUnless(tempattribute in self.activeattributes)
 
100
        # create two plugin directories
 
101
        os.mkdir('first')
 
102
        os.mkdir('second')
 
103
        # write a plugin that will record when its loaded in the 
 
104
        # tempattribute list.
 
105
        template = ("from bzrlib.tests.test_plugins import TestOneNamedPluginOnly\n"
 
106
                    "TestOneNamedPluginOnly.activeattributes[%r].append('%s')\n")
 
107
        print >> file(os.path.join('first', 'plugin.py'), 'w'), template % (tempattribute, 'first')
 
108
        print >> file(os.path.join('second', 'plugin.py'), 'w'), template % (tempattribute, 'second')
 
109
        try:
 
110
            bzrlib.plugin.load_from_dirs(['first', 'second'])
 
111
            self.assertEqual(['first'], self.activeattributes[tempattribute])
 
112
        finally:
 
113
            # remove the plugin 'plugin'
 
114
            del self.activeattributes[tempattribute]
 
115
            if getattr(bzrlib.plugins, 'plugin', None):
 
116
                del bzrlib.plugins.plugin
 
117
        self.failIf(getattr(bzrlib.plugins, 'plugin', None))
 
118
 
 
119
 
 
120
class TestAllPlugins(TestCaseInTempDir):
 
121
 
 
122
    def test_plugin_appears_in_all_plugins(self):
 
123
        # This test tests a new plugin appears in bzrlib.plugin.all_plugins().
 
124
        # check the plugin is not loaded already
 
125
        self.failIf(getattr(bzrlib.plugins, 'plugin', None))
 
126
        # write a plugin that _cannot_ fail to load.
 
127
        print >> file('plugin.py', 'w'), ""
 
128
        try:
 
129
            bzrlib.plugin.load_from_dirs(['.'])
 
130
            self.failUnless('plugin' in bzrlib.plugin.all_plugins())
 
131
            self.failUnless(getattr(bzrlib.plugins, 'plugin', None))
 
132
            self.assertEqual(bzrlib.plugin.all_plugins()['plugin'],
 
133
                             bzrlib.plugins.plugin)
 
134
        finally:
 
135
            # remove the plugin 'plugin'
 
136
            if getattr(bzrlib.plugins, 'plugin', None):
 
137
                del bzrlib.plugins.plugin
 
138
        self.failIf(getattr(bzrlib.plugins, 'plugin', None))