1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
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
17
from bzrlib.builtins import cmd_commit, cmd_log, cmd_status
18
from bzrlib.commands import Command, parse_args
24
from bzrlib.tests import TestCase
26
def parse(options, args):
27
parser = option.get_optparser(dict((o.name, o) for o in options))
28
return parser.parse_args(args)
30
class OptionTests(TestCase):
31
"""Command-line option tests"""
33
def test_parse_args(self):
35
eq = self.assertEquals
36
eq(parse_args(cmd_commit(), ['--help']),
38
eq(parse_args(cmd_commit(), ['--message=biter']),
39
([], {'message': 'biter'}))
40
## eq(parse_args(cmd_log(), '-r 500'.split()),
41
## ([], {'revision': RevisionSpec_int(500)}))
43
def test_no_more_opts(self):
44
"""Terminated options"""
45
self.assertEquals(parse_args(cmd_commit(), ['--', '-file-with-dashes']),
46
(['-file-with-dashes'], {}))
48
def test_option_help(self):
49
"""Options have help strings."""
50
out, err = self.run_bzr_captured(['commit', '--help'])
51
self.assertContainsRe(out, r'--file(.|\n)*file containing commit'
53
self.assertContainsRe(out, r'-h.*--help')
55
def test_option_help_global(self):
56
"""Global options have help strings."""
57
out, err = self.run_bzr_captured(['help', 'status'])
58
self.assertContainsRe(out, r'--show-ids.*show internal object')
60
def test_option_arg_help(self):
61
"""Help message shows option arguments."""
62
out, err = self.run_bzr_captured(['help', 'commit'])
63
self.assertEquals(err, '')
64
self.assertContainsRe(out, r'--file[ =]MSGFILE')
66
def test_unknown_short_opt(self):
67
out, err = self.run_bzr_captured(['help', '-r'], retcode=3)
68
self.assertContainsRe(err, r'no such option')
70
def test_get_short_name(self):
71
file_opt = option.Option.OPTIONS['file']
72
self.assertEquals(file_opt.short_name(), 'F')
73
force_opt = option.Option.OPTIONS['force']
74
self.assertEquals(force_opt.short_name(), None)
76
def test_set_short_name(self):
77
o = option.Option('wiggle')
79
self.assertEqual(o.short_name(), 'w')
81
def test_old_short_names(self):
82
# test the deprecated method for getting and setting short option
85
"access to SHORT_OPTIONS was deprecated in version 0.14."
86
" Set the short option name when constructing the Option.",
87
DeprecationWarning, 2)
89
def capture_warning(message, category, stacklevel=None):
90
_warnings.append((message, category, stacklevel))
91
old_warning_method = symbol_versioning.warn
93
# an example of the kind of thing plugins might want to do through
94
# the old interface - make a new option and then give it a short
96
symbol_versioning.set_warning_method(capture_warning)
97
example_opt = option.Option('example', help='example option')
98
option.Option.SHORT_OPTIONS['w'] = example_opt
99
self.assertEqual(example_opt.short_name(), 'w')
100
self.assertEqual([expected_warning], _warnings)
101
# now check that it can actually be parsed with the registered
103
opts, args = parse([example_opt], ['-w', 'foo'])
104
self.assertEqual(opts.example, True)
105
self.assertEqual(args, ['foo'])
107
symbol_versioning.set_warning_method(old_warning_method)
109
def test_allow_dash(self):
110
"""Test that we can pass a plain '-' as an argument."""
111
self.assertEqual((['-'], {}), parse_args(cmd_commit(), ['-']))
113
def test_conversion(self):
114
options = [option.Option('hello')]
115
opts, args = parse(options, ['--no-hello', '--hello'])
116
self.assertEqual(True, opts.hello)
117
opts, args = parse(options, [])
118
self.assertEqual(option.OptionParser.DEFAULT_VALUE, opts.hello)
119
opts, args = parse(options, ['--hello', '--no-hello'])
120
self.assertEqual(False, opts.hello)
121
options = [option.Option('number', type=int)]
122
opts, args = parse(options, ['--number', '6'])
123
self.assertEqual(6, opts.number)
124
self.assertRaises(errors.BzrCommandError, parse, options, ['--number'])
125
self.assertRaises(errors.BzrCommandError, parse, options,
128
def test_iter_switches(self):
129
opt = option.Option('hello', help='fg')
130
self.assertEqual(list(opt.iter_switches()),
131
[('hello', None, None, 'fg')])
132
opt = option.Option('hello', help='fg', type=int)
133
self.assertEqual(list(opt.iter_switches()),
134
[('hello', None, 'ARG', 'fg')])
135
opt = option.Option('hello', help='fg', type=int, argname='gar')
136
self.assertEqual(list(opt.iter_switches()),
137
[('hello', None, 'GAR', 'fg')])
139
# >>> parse_args('log -r 500'.split())
140
# (['log'], {'revision': [<RevisionSpec_int 500>]})
141
# >>> parse_args('log -r500..600'.split())
142
# (['log'], {'revision': [<RevisionSpec_int 500>, <RevisionSpec_int 600>]})
143
# >>> parse_args('log -vr500..600'.split())
144
# (['log'], {'verbose': True, 'revision': [<RevisionSpec_int 500>, <RevisionSpec_int 600>]})
145
# >>> parse_args('log -rrevno:500..600'.split()) #the r takes an argument
146
# (['log'], {'revision': [<RevisionSpec_revno revno:500>, <RevisionSpec_int 600>]})