/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_registry.py

  • Committer: Jelmer Vernooij
  • Date: 2018-05-06 11:48:54 UTC
  • mto: This revision was merged to the branch mainline in revision 6960.
  • Revision ID: jelmer@jelmer.uk-20180506114854-h4qd9ojaqy8wxjsd
Move .mailmap to root.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_registry.py
84
84
        a_registry.register('one', 'one')
85
85
        self.assertRaises(KeyError, a_registry.register, 'one', 'two')
86
86
        self.assertRaises(KeyError, a_registry.register, 'one', 'two',
87
 
                          override_existing=False)
 
87
                                    override_existing=False)
88
88
 
89
89
        a_registry.register('one', 'two', override_existing=True)
90
90
        self.assertEqual('two', a_registry.get('one'))
105
105
 
106
106
        # We should be able to handle a callable to get information
107
107
        help_calls = []
108
 
 
109
108
        def generic_help(reg, key):
110
109
            help_calls.append(key)
111
110
            return 'generic help for %s' % (key,)
147
146
                          ('six', 'this is my help'),
148
147
                          ('three', 'generic help for three'),
149
148
                          ('two', 'help text for two'),
150
 
                          ], sorted((key, a_registry.get_help(key))
 
149
                         ], sorted((key, a_registry.get_help(key))
151
150
                                    for key in a_registry.keys()))
152
151
 
153
152
        # We don't know what order it was called in, but we should get
187
186
                          ('one', 'string info'),
188
187
                          ('three', ['a', 'list']),
189
188
                          ('two', 2),
190
 
                          ], sorted((key, a_registry.get_info(key))
 
189
                         ], sorted((key, a_registry.get_info(key))
191
190
                                    for key in a_registry.keys()))
192
191
 
193
192
    def test_get_prefix(self):
210
209
        a_registry.register_alias('two', 'one')
211
210
        a_registry.register_alias('three', 'one', info='own info')
212
211
        self.assertEqual(a_registry.get('one'), a_registry.get('two'))
213
 
        self.assertEqual(a_registry.get_help('one'),
214
 
                         a_registry.get_help('two'))
215
 
        self.assertEqual(a_registry.get_info('one'),
216
 
                         a_registry.get_info('two'))
 
212
        self.assertEqual(a_registry.get_help('one'), a_registry.get_help('two'))
 
213
        self.assertEqual(a_registry.get_info('one'), a_registry.get_info('two'))
217
214
        self.assertEqual('own info', a_registry.get_info('three'))
218
215
        self.assertEqual({'two': 'one', 'three': 'one'}, a_registry.aliases())
219
 
        self.assertEqual(
220
 
            {'one': ['three', 'two']},
221
 
            {k: sorted(v) for (k, v) in viewitems(a_registry.alias_map())})
 
216
        self.assertEqual({'one': ['three', 'two']},
 
217
                         {k: sorted(v) for (k, v) in viewitems(a_registry.alias_map())})
222
218
 
223
219
    def test_registry_alias_exists(self):
224
220
        a_registry = registry.Registry()
247
243
        _registry = registry.Registry()
248
244
 
249
245
        def register_more():
250
 
            _registry.register('hidden', None)
 
246
           _registry.register('hidden', None)
251
247
 
252
248
        # Avoid closing over self by binding local variable
253
249
        self.registry = _registry
301
297
        """
302
298
        os.mkdir('tmp')
303
299
        plugin_name = 'bzr_plugin_a_%s' % (osutils.rand_chars(4),)
304
 
        with open('tmp/' + plugin_name + '.py', 'wb') as f:
305
 
            f.write(contents)
 
300
        with open('tmp/'+plugin_name+'.py', 'wb') as f: f.write(contents)
306
301
        return plugin_name
307
302
 
308
303
    def create_simple_plugin(self):
309
304
        return self.create_plugin_file(
310
 
            b'object1 = "foo"\n'
311
 
            b'\n\n'
312
 
            b'def function(a,b,c):\n'
313
 
            b'    return a,b,c\n'
314
 
            b'\n\n'
315
 
            b'class MyClass(object):\n'
316
 
            b'    def __init__(self, a):\n'
317
 
            b'      self.a = a\n'
318
 
            b'\n\n'
 
305
            'object1 = "foo"\n'
 
306
            '\n\n'
 
307
            'def function(a,b,c):\n'
 
308
            '    return a,b,c\n'
 
309
            '\n\n'
 
310
            'class MyClass(object):\n'
 
311
            '    def __init__(self, a):\n'
 
312
            '      self.a = a\n'
 
313
            '\n\n'
319
314
        )
320
315
 
321
316
    def test_lazy_import_registry_foo(self):
373
368
    def test_lazy_import_get_module(self):
374
369
        a_registry = registry.Registry()
375
370
        a_registry.register_lazy('obj', "breezy.tests.test_registry",
376
 
                                 'object1')
 
371
            'object1')
377
372
        self.assertEqual("breezy.tests.test_registry",
378
 
                         a_registry._get_module("obj"))
 
373
            a_registry._get_module("obj"))
379
374
 
380
375
    def test_normal_get_module(self):
381
376
        class AThing(object):
383
378
        a_registry = registry.Registry()
384
379
        a_registry.register("obj", AThing())
385
380
        self.assertEqual("breezy.tests.test_registry",
386
 
                         a_registry._get_module("obj"))
 
381
            a_registry._get_module("obj"))