← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_options.py

(jameinel) Allow 'bzr serve' to interpret SIGHUP as a graceful shutdown.
 (bug #795025) (John A Meinel)

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
19
19
from bzrlib import (
20
20
    bzrdir,
21
21
    commands,
 
22
    controldir,
22
23
    errors,
23
24
    option,
 
25
    registry,
24
26
    )
25
27
from bzrlib.builtins import cmd_commit
26
 
from bzrlib.commands import Command, parse_args
 
28
from bzrlib.commands import parse_args
27
29
from bzrlib.tests import TestCase
28
30
from bzrlib.repofmt import knitrepo
29
31
 
107
109
        self.assertFalse(option.Option('foo', hidden=False).is_hidden('foo'))
108
110
 
109
111
    def test_registry_conversion(self):
110
 
        registry = bzrdir.BzrDirFormatRegistry()
111
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
112
 
        registry.register_metadir('two', 'RepositoryFormatKnit1', 'two help')
113
 
        registry.register_metadir('hidden', 'RepositoryFormatKnit1',
 
112
        registry = controldir.ControlDirFormatRegistry()
 
113
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
114
        bzrdir.register_metadir(registry, 'two', 'RepositoryFormatKnit1', 'two help')
 
115
        bzrdir.register_metadir(registry, 'hidden', 'RepositoryFormatKnit1',
114
116
            'two help', hidden=True)
115
117
        registry.set_default('one')
116
118
        options = [option.RegistryOption('format', '', registry, str)]
177
179
        self.assertEqual('test option', my_option.help)
178
180
 
179
181
    def test_help(self):
180
 
        registry = bzrdir.BzrDirFormatRegistry()
181
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
182
 
        registry.register_metadir('two',
 
182
        registry = controldir.ControlDirFormatRegistry()
 
183
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
184
        bzrdir.register_metadir(registry, 'two',
183
185
            'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
184
186
            'two help',
185
187
            )
186
 
        registry.register_metadir('hidden', 'RepositoryFormat7', 'hidden help',
 
188
        bzrdir.register_metadir(registry, 'hidden', 'RepositoryFormat7', 'hidden help',
187
189
            hidden=True)
188
190
        registry.set_default('one')
189
191
        options = [option.RegistryOption('format', 'format help', registry,
205
207
        opt = option.Option('hello', help='fg', type=int, argname='gar')
206
208
        self.assertEqual(list(opt.iter_switches()),
207
209
                         [('hello', None, 'GAR', 'fg')])
208
 
        registry = bzrdir.BzrDirFormatRegistry()
209
 
        registry.register_metadir('one', 'RepositoryFormat7', 'one help')
210
 
        registry.register_metadir('two',
 
210
        registry = controldir.ControlDirFormatRegistry()
 
211
        bzrdir.register_metadir(registry, 'one', 'RepositoryFormat7', 'one help')
 
212
        bzrdir.register_metadir(registry, 'two',
211
213
                'bzrlib.repofmt.knitrepo.RepositoryFormatKnit1',
212
214
                'two help',
213
215
                )
386
388
                    + '\n'.join(msgs))
387
389
 
388
390
    def test_is_hidden(self):
389
 
        registry = bzrdir.BzrDirFormatRegistry()
390
 
        registry.register_metadir('hidden', 'HiddenFormat',
 
391
        registry = controldir.ControlDirFormatRegistry()
 
392
        bzrdir.register_metadir(registry, 'hidden', 'HiddenFormat',
391
393
            'hidden help text', hidden=True)
392
 
        registry.register_metadir('visible', 'VisibleFormat',
 
394
        bzrdir.register_metadir(registry, 'visible', 'VisibleFormat',
393
395
            'visible help text', hidden=False)
394
396
        format = option.RegistryOption('format', '', registry, str)
395
397
        self.assertTrue(format.is_hidden('hidden'))
396
398
        self.assertFalse(format.is_hidden('visible'))
397
399
 
 
400
    def test_short_name(self):
 
401
        registry = controldir.ControlDirFormatRegistry()
 
402
        opt = option.RegistryOption('format', help='', registry=registry)
 
403
        self.assertEquals(None, opt.short_name())
 
404
        opt = option.RegistryOption('format', short_name='F', help='',
 
405
            registry=registry)
 
406
        self.assertEquals('F', opt.short_name())
 
407
 
398
408
    def test_option_custom_help(self):
399
409
        the_opt = option.Option.OPTIONS['help']
400
410
        orig_help = the_opt.help[:]
403
413
        self.assertEqual('suggest lottery numbers', my_opt.help)
404
414
        self.assertEqual(orig_help, the_opt.help)
405
415
 
 
416
    def test_short_value_switches(self):
 
417
        reg = registry.Registry()
 
418
        reg.register('short', 'ShortChoice')
 
419
        reg.register('long', 'LongChoice')
 
420
        ropt = option.RegistryOption('choice', '', reg, value_switches=True,
 
421
            short_value_switches={'short': 's'})
 
422
        opts, args = parse([ropt], ['--short'])
 
423
        self.assertEqual('ShortChoice', opts.choice)
 
424
        opts, args = parse([ropt], ['-s'])
 
425
        self.assertEqual('ShortChoice', opts.choice)
 
426
 
406
427
 
407
428
class TestVerboseQuietLinkage(TestCase):
408
429