← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_options.py

  • Committer: Aaron Bentley
  • Date: 2012-07-13 19:32:33 UTC
  • mto: This revision was merged to the branch mainline in revision 6540.
  • Revision ID: aaron@aaronbentley.com-20120713193233-l6y0l0twwhd3wmka
Switch to much simpler implementation of restore_uncommitted.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_options.py
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2005-2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
import re
18
18
 
19
19
from bzrlib import (
20
 
    builtins,
21
20
    bzrdir,
22
21
    commands,
 
22
    controldir,
23
23
    errors,
24
24
    option,
25
 
    repository,
26
 
    symbol_versioning,
 
25
    registry,
27
26
    )
28
 
from bzrlib.builtins import cmd_commit, cmd_log, cmd_status
29
 
from bzrlib.commands import Command, parse_args
 
27
from bzrlib.builtins import cmd_commit
 
28
from bzrlib.commands import parse_args
30
29
from bzrlib.tests import TestCase
31
30
from bzrlib.repofmt import knitrepo
32
31
 
41
40
 
42
41
    def test_parse_args(self):
43
42
        """Option parser"""
44
 
        eq = self.assertEquals
45
 
        eq(parse_args(cmd_commit(), ['--help']),
46
 
           ([], {'fixes': [], 'help': True}))
47
 
        eq(parse_args(cmd_commit(), ['--message=biter']),
48
 
           ([], {'fixes': [], 'message': 'biter'}))
 
43
        # XXX: Using cmd_commit makes these tests overly sensitive to changes
 
44
        # to cmd_commit, when they are meant to be about option parsing in
 
45
        # general.
 
46
        self.assertEqual(
 
47
           ([], {'author': [], 'exclude': [], 'fixes': [], 'help': True}),
 
48
           parse_args(cmd_commit(), ['--help']))
 
49
        self.assertEqual(
 
50
           ([], {'author': [], 'exclude': [], 'fixes': [], 'message': 'biter'}),
 
51
           parse_args(cmd_commit(), ['--message=biter']))
49
52
 
50
53
    def test_no_more_opts(self):
51
54
        """Terminated options"""
52
 
        self.assertEquals(parse_args(cmd_commit(), ['--', '-file-with-dashes']),
53
 
                          (['-file-with-dashes'], {'fixes': []}))
 
55
        self.assertEqual(
 
56
            (['-file-with-dashes'], {'author': [], 'exclude': [], 'fixes': []}),
 
57
            parse_args(cmd_commit(), ['--', '-file-with-dashes']))
54
58
 
55
59
    def test_option_help(self):
56
60
        """Options have help strings."""
64
68
        out, err = self.run_bzr('help status')
65
69
        self.assertContainsRe(out, r'--show-ids.*Show internal object.')
66
70
 
 
71
    def test_option_help_global_hidden(self):
 
72
        """Hidden global options have no help strings."""
 
73
        out, err = self.run_bzr('help log')
 
74
        self.assertNotContainsRe(out, r'--message')
 
75
 
67
76
    def test_option_arg_help(self):
68
77
        """Help message shows option arguments."""
69
78
        out, err = self.run_bzr('help commit')
70
 
        self.assertEquals(err, '')
 
79
        self.assertEqual(err, '')
71
80
        self.assertContainsRe(out, r'--file[ =]MSGFILE')
72
81
 
73
82
    def test_unknown_short_opt(self):
81
90
 
82
91
    def test_allow_dash(self):
83
92
        """Test that we can pass a plain '-' as an argument."""
84
 
        self.assertEqual(
85
 
            (['-'], {'fixes': []}), parse_args(cmd_commit(), ['-']))
 
93
        self.assertEqual((['-']), parse_args(cmd_commit(), ['-'])[0])
86
94
 
87
95
    def parse(self, options, args):
88
96
        parser = option.get_optparser(dict((o.name, o) for o in options))
93
101
        opts, args = self.parse(options, ['--no-hello', '--hello'])
94
102
        self.assertEqual(True, opts.hello)
95
103
        opts, args = self.parse(options, [])
96
 
        self.assertEqual(option.OptionParser.DEFAULT_VALUE, opts.hello)
 
104
        self.assertFalse(hasattr(opts, 'hello'))
97
105
        opts, args = self.parse(options, ['--hello', '--no-hello'])
98
106
        self.assertEqual(False, opts.hello)
99
107
        options = [option.Option('number', type=int)]
104
112
        self.assertRaises(errors.BzrCommandError, self.parse, options,
105
113
                          ['--no-number'])
106
114
 
 
115
    def test_is_hidden(self):
 
116
        self.assertTrue(option.Option('foo', hidden=True).is_hidden('foo'))
 
117
        self.assertFalse(option.Option('foo', hidden=False).is_hidden('foo'))
 
118
 
107
119
    def test_registry_conversion(self):
108
 
        registry = bzrdir.BzrDirFormatRegistry()
109
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
110
 
        registry.register_metadir('two', 'RepositoryFormatKnit1', 'two help')
111
 
        registry.register_metadir('hidden', 'RepositoryFormatKnit1',
 
120
        registry = controldir.ControlDirFormatRegistry()
 
121
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
122
        bzrdir.register_metadir(registry, 'two', 'RepositoryFormatKnit1', 'two help')
 
123
        bzrdir.register_metadir(registry, 'hidden', 'RepositoryFormatKnit1',
112
124
            'two help', hidden=True)
113
125
        registry.set_default('one')
114
126
        options = [option.RegistryOption('format', '', registry, str)]
148
160
 
149
161
    def test_registry_converter(self):
150
162
        options = [option.RegistryOption('format', '',
151
 
                   bzrdir.format_registry, bzrdir.format_registry.make_bzrdir)]
 
163
                   controldir.format_registry, controldir.format_registry.make_bzrdir)]
152
164
        opts, args = self.parse(options, ['--format', 'knit'])
153
165
        self.assertIsInstance(opts.format.repository_format,
154
166
                              knitrepo.RepositoryFormatKnit1)
155
167
 
 
168
    def test_lazy_registry(self):
 
169
        options = [option.RegistryOption('format', '',
 
170
                   lazy_registry=('bzrlib.controldir','format_registry'),
 
171
                   converter=str)]
 
172
        opts, args = self.parse(options, ['--format', 'knit'])
 
173
        self.assertEqual({'format': 'knit'}, opts)
 
174
        self.assertRaises(
 
175
            errors.BadOptionValue, self.parse, options, ['--format', 'BAD'])
 
176
 
156
177
    def test_from_kwargs(self):
157
178
        my_option = option.RegistryOption.from_kwargs('my-option',
158
179
            help='test option', short='be short', be_long='go long')
166
187
        self.assertEqual('test option', my_option.help)
167
188
 
168
189
    def test_help(self):
169
 
        registry = bzrdir.BzrDirFormatRegistry()
170
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
171
 
        registry.register_metadir('two',
 
190
        registry = controldir.ControlDirFormatRegistry()
 
191
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
192
        bzrdir.register_metadir(registry, 'two',
172
193
            'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
173
194
            'two help',
174
195
            )
175
 
        registry.register_metadir('hidden', 'RepositoryFormat7', 'hidden help',
 
196
        bzrdir.register_metadir(registry, 'hidden', 'RepositoryFormat7', 'hidden help',
176
197
            hidden=True)
177
198
        registry.set_default('one')
178
199
        options = [option.RegistryOption('format', 'format help', registry,
194
215
        opt = option.Option('hello', help='fg', type=int, argname='gar')
195
216
        self.assertEqual(list(opt.iter_switches()),
196
217
                         [('hello', None, 'GAR', 'fg')])
197
 
        registry = bzrdir.BzrDirFormatRegistry()
198
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
199
 
        registry.register_metadir('two',
 
218
        registry = controldir.ControlDirFormatRegistry()
 
219
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
220
        bzrdir.register_metadir(registry, 'two',
200
221
                'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
201
222
                'two help',
202
223
                )
214
235
                          ('two', None, None, 'two help'),
215
236
                          ])
216
237
 
 
238
    def test_option_callback_bool(self):
 
239
        "Test booleans get True and False passed correctly to a callback."""
 
240
        cb_calls = []
 
241
        def cb(option, name, value, parser):
 
242
            cb_calls.append((option,name,value,parser))
 
243
        options = [option.Option('hello', custom_callback=cb)]
 
244
        opts, args = self.parse(options, ['--hello', '--no-hello'])
 
245
        self.assertEqual(2, len(cb_calls))
 
246
        opt,name,value,parser = cb_calls[0]
 
247
        self.assertEqual('hello', name)
 
248
        self.assertTrue(value)
 
249
        opt,name,value,parser = cb_calls[1]
 
250
        self.assertEqual('hello', name)
 
251
        self.assertFalse(value)
 
252
 
 
253
    def test_option_callback_str(self):
 
254
        """Test callbacks work for string options both long and short."""
 
255
        cb_calls = []
 
256
        def cb(option, name, value, parser):
 
257
            cb_calls.append((option,name,value,parser))
 
258
        options = [option.Option('hello', type=str, custom_callback=cb,
 
259
            short_name='h')]
 
260
        opts, args = self.parse(options, ['--hello', 'world', '-h', 'mars'])
 
261
        self.assertEqual(2, len(cb_calls))
 
262
        opt,name,value,parser = cb_calls[0]
 
263
        self.assertEqual('hello', name)
 
264
        self.assertEqual('world', value)
 
265
        opt,name,value,parser = cb_calls[1]
 
266
        self.assertEqual('hello', name)
 
267
        self.assertEqual('mars', value)
 
268
 
217
269
 
218
270
class TestListOptions(TestCase):
219
271
    """Tests for ListOption, used to specify lists on the command-line."""
227
279
        opts, args = self.parse(options, ['--hello=world', '--hello=sailor'])
228
280
        self.assertEqual(['world', 'sailor'], opts.hello)
229
281
 
 
282
    def test_list_option_with_dash(self):
 
283
        options = [option.ListOption('with-dash', type=str)]
 
284
        opts, args = self.parse(options, ['--with-dash=world',
 
285
                                          '--with-dash=sailor'])
 
286
        self.assertEqual(['world', 'sailor'], opts.with_dash)
 
287
 
230
288
    def test_list_option_no_arguments(self):
231
289
        options = [option.ListOption('hello', type=str)]
232
290
        opts, args = self.parse(options, [])
250
308
            options, ['--hello=a', '--hello=b', '--hello=-', '--hello=c'])
251
309
        self.assertEqual(['c'], opts.hello)
252
310
 
 
311
    def test_option_callback_list(self):
 
312
        """Test callbacks work for list options."""
 
313
        cb_calls = []
 
314
        def cb(option, name, value, parser):
 
315
            # Note that the value is a reference so copy to keep it
 
316
            cb_calls.append((option,name,value[:],parser))
 
317
        options = [option.ListOption('hello', type=str, custom_callback=cb)]
 
318
        opts, args = self.parse(options, ['--hello=world', '--hello=mars',
 
319
            '--hello=-'])
 
320
        self.assertEqual(3, len(cb_calls))
 
321
        opt,name,value,parser = cb_calls[0]
 
322
        self.assertEqual('hello', name)
 
323
        self.assertEqual(['world'], value)
 
324
        opt,name,value,parser = cb_calls[1]
 
325
        self.assertEqual('hello', name)
 
326
        self.assertEqual(['world', 'mars'], value)
 
327
        opt,name,value,parser = cb_calls[2]
 
328
        self.assertEqual('hello', name)
 
329
        self.assertEqual([], value)
 
330
 
 
331
    def test_list_option_param_name(self):
 
332
        """Test list options can have their param_name set."""
 
333
        options = [option.ListOption('hello', type=str, param_name='greeting')]
 
334
        opts, args = self.parse(
 
335
            options, ['--hello=world', '--hello=sailor'])
 
336
        self.assertEqual(['world', 'sailor'], opts.greeting)
 
337
 
253
338
 
254
339
class TestOptionDefinitions(TestCase):
255
340
    """Tests for options in the Bazaar codebase."""
256
341
 
257
342
    def get_builtin_command_options(self):
258
343
        g = []
259
 
        for cmd_name, cmd_class in sorted(commands.get_all_cmds()):
260
 
            cmd = cmd_class()
 
344
        commands.install_bzr_command_hooks()
 
345
        for cmd_name in sorted(commands.builtin_command_names()):
 
346
            cmd = commands.get_cmd_object(cmd_name)
261
347
            for opt_name, opt in sorted(cmd.options().items()):
262
348
                g.append((cmd_name, opt))
 
349
        self.assert_(g)
263
350
        return g
264
351
 
265
 
    def test_global_options_used(self):
266
 
        # In the distant memory, options could only be declared globally.  Now
267
 
        # we prefer to declare them in the command, unless like -r they really
268
 
        # are used very widely with the exact same meaning.  So this checks
269
 
        # for any that should be garbage collected.
270
 
        g = dict(option.Option.OPTIONS.items())
271
 
        used_globals = {}
272
 
        msgs = []
273
 
        for cmd_name, cmd_class in sorted(commands.get_all_cmds()):
274
 
            for option_or_name in sorted(cmd_class.takes_options):
275
 
                if not isinstance(option_or_name, basestring):
276
 
                    self.assertIsInstance(option_or_name, option.Option)
277
 
                elif not option_or_name in g:
278
 
                    msgs.append("apparent reference to undefined "
279
 
                        "global option %r from %r"
280
 
                        % (option_or_name, cmd_class))
281
 
                else:
282
 
                    used_globals.setdefault(option_or_name, []).append(cmd_name)
283
 
        unused_globals = set(g.keys()) - set(used_globals.keys())
284
 
        # not enforced because there might be plugins that use these globals
285
 
        ## for option_name in sorted(unused_globals):
286
 
        ##    msgs.append("unused global option %r" % option_name)
287
 
        ## for option_name, cmds in sorted(used_globals.items()):
288
 
        ##     if len(cmds) <= 1:
289
 
        ##         msgs.append("global option %r is only used by %r"
290
 
        ##                 % (option_name, cmds))
291
 
        if msgs:
292
 
            self.fail("problems with global option definitions:\n"
293
 
                    + '\n'.join(msgs))
294
 
 
295
352
    def test_option_grammar(self):
296
353
        msgs = []
297
354
        # Option help should be written in sentence form, and have a final
298
 
        # period and be all on a single line, because the display code will
299
 
        # wrap it.
300
 
        option_re = re.compile(r'^[A-Z][^\n]+\.$')
301
 
        for scope, option in self.get_builtin_command_options():
302
 
            if not option.help:
303
 
                msgs.append('%-16s %-16s %s' %
304
 
                       ((scope or 'GLOBAL'), option.name, 'NO HELP'))
305
 
            elif not option_re.match(option.help):
306
 
                msgs.append('%-16s %-16s %s' %
307
 
                        ((scope or 'GLOBAL'), option.name, option.help))
 
355
        # period with an optional bracketed suffix. All the text should be on
 
356
        # one line, because the display code will wrap it.
 
357
        option_re = re.compile(r'^[A-Z][^\n]+\.(?: \([^\n]+\))?$')
 
358
        for scope, opt in self.get_builtin_command_options():
 
359
            for name, _, _, helptxt in opt.iter_switches():
 
360
                if name != opt.name:
 
361
                    name = "/".join([opt.name, name])
 
362
                if not helptxt:
 
363
                    msgs.append('%-16s %-16s %s' %
 
364
                           ((scope or 'GLOBAL'), name, 'NO HELP'))
 
365
                elif not option_re.match(helptxt):
 
366
                    if name.startswith("format/"):
 
367
                        # Don't complain about the odd format registry help
 
368
                        continue
 
369
                    msgs.append('%-16s %-16s %s' %
 
370
                            ((scope or 'GLOBAL'), name, helptxt))
308
371
        if msgs:
309
372
            self.fail("The following options don't match the style guide:\n"
310
373
                    + '\n'.join(msgs))
311
374
 
 
375
 
 
376
class TestOptionMisc(TestCase):
 
377
 
312
378
    def test_is_hidden(self):
313
 
        registry = bzrdir.BzrDirFormatRegistry()
314
 
        registry.register_metadir('hidden', 'HiddenFormat',
 
379
        registry = controldir.ControlDirFormatRegistry()
 
380
        bzrdir.register_metadir(registry, 'hidden', 'HiddenFormat',
315
381
            'hidden help text', hidden=True)
316
 
        registry.register_metadir('visible', 'VisibleFormat',
 
382
        bzrdir.register_metadir(registry, 'visible', 'VisibleFormat',
317
383
            'visible help text', hidden=False)
318
384
        format = option.RegistryOption('format', '', registry, str)
319
385
        self.assertTrue(format.is_hidden('hidden'))
320
386
        self.assertFalse(format.is_hidden('visible'))
 
387
 
 
388
    def test_short_name(self):
 
389
        registry = controldir.ControlDirFormatRegistry()
 
390
        opt = option.RegistryOption('format', help='', registry=registry)
 
391
        self.assertEquals(None, opt.short_name())
 
392
        opt = option.RegistryOption('format', short_name='F', help='',
 
393
            registry=registry)
 
394
        self.assertEquals('F', opt.short_name())
 
395
 
 
396
    def test_option_custom_help(self):
 
397
        the_opt = option.Option.OPTIONS['help']
 
398
        orig_help = the_opt.help[:]
 
399
        my_opt = option.custom_help('help', 'suggest lottery numbers')
 
400
        # Confirm that my_opt has my help and the original is unchanged
 
401
        self.assertEqual('suggest lottery numbers', my_opt.help)
 
402
        self.assertEqual(orig_help, the_opt.help)
 
403
 
 
404
    def test_short_value_switches(self):
 
405
        reg = registry.Registry()
 
406
        reg.register('short', 'ShortChoice')
 
407
        reg.register('long', 'LongChoice')
 
408
        ropt = option.RegistryOption('choice', '', reg, value_switches=True,
 
409
            short_value_switches={'short': 's'})
 
410
        opts, args = parse([ropt], ['--short'])
 
411
        self.assertEqual('ShortChoice', opts.choice)
 
412
        opts, args = parse([ropt], ['-s'])
 
413
        self.assertEqual('ShortChoice', opts.choice)
 
414
 
 
415
 
 
416
class TestVerboseQuietLinkage(TestCase):
 
417
 
 
418
    def check(self, parser, level, args):
 
419
        option._verbosity_level = 0
 
420
        opts, args = parser.parse_args(args)
 
421
        self.assertEqual(level, option._verbosity_level)
 
422
 
 
423
    def test_verbose_quiet_linkage(self):
 
424
        parser = option.get_optparser(option.Option.STD_OPTIONS)
 
425
        self.check(parser, 0, [])
 
426
        self.check(parser, 1, ['-v'])
 
427
        self.check(parser, 2, ['-v', '-v'])
 
428
        self.check(parser, -1, ['-q'])
 
429
        self.check(parser, -2, ['-qq'])
 
430
        self.check(parser, -1, ['-v', '-v', '-q'])
 
431
        self.check(parser, 2, ['-q', '-v', '-v'])
 
432
        self.check(parser, 0, ['--no-verbose'])
 
433
        self.check(parser, 0, ['-v', '-q', '--no-quiet'])