← 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] bzr.dev

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