1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
import os, sys
import Bcfg2.Options
class TestOption(object):
def test__init(self):
o = Bcfg2.Options.Option('foo', False, cmd='-F')
try:
p = Bcfg2.Options.Option('foo', False, cmd='--F')
assert False
except Bcfg2.Options.OptionFailure:
pass
def test_parse(self):
o = Bcfg2.Options.Option('foo', 'test4', cmd='-F', env='TEST2',
odesc='bar', cf=('communication', 'password'))
o.parse([], ['-F', 'test'])
assert o.value == 'test'
o.parse([('-F', 'test2')], [])
assert o.value == 'test2'
os.environ['TEST2'] = 'test3'
o.parse([], [])
assert o.value == 'test3'
del os.environ['TEST2']
o.parse([], [])
print o.value
assert o.value == 'foobat'
o.cf = ('communication', 'pwd')
o.parse([], [])
print o.value
assert o.value == 'test4'
o.cf = False
o.parse([], [])
assert o.value == 'test4'
def test_cook(self):
# check that default value isn't cooked
o1 = Bcfg2.Options.Option('foo', 'test4', cook=Bcfg2.Options.bool_cook)
o1.parse([], [])
assert o1.value == 'test4'
o2 = Bcfg2.Options.Option('foo', False, cmd='-F')
o2.parse([('-F', '')], [])
assert o2.value == True
class TestOptionSet(object):
def test_buildGetopt(self):
opts = [('foo', Bcfg2.Options.Option('foo', 'test1', cmd='-G')),
('bar', Bcfg2.Options.Option('foo', 'test2')),
('baz', Bcfg2.Options.Option('foo', 'test1', cmd='-H', odesc='1'))]
os = Bcfg2.Options.OptionSet(opts)
res = os.buildGetopt()
assert 'H:' in res and 'G' in res and len(res) == 3
def test_buildLongGetopt(self):
opts = [('foo', Bcfg2.Options.Option('foo', 'test1', cmd='-G')),
('bar', Bcfg2.Options.Option('foo', 'test2')),
('baz', Bcfg2.Options.Option('foo', 'test1', cmd='--H',
odesc='1', long_arg=True))]
os = Bcfg2.Options.OptionSet(opts)
res = os.buildLongGetopt()
print res
assert 'H=' in res and len(res) == 1
def test_parse(self):
opts = [('foo', Bcfg2.Options.Option('foo', 'test1', cmd='-G')),
('bar', Bcfg2.Options.Option('foo', 'test2')),
('baz', Bcfg2.Options.Option('foo', 'test1', cmd='-H', odesc='1'))]
os = Bcfg2.Options.OptionSet(opts)
try:
os.parse(['-G', '-H'])
assert False
except SystemExit:
pass
os2 = Bcfg2.Options.OptionSet(opts)
try:
os2.parse(['-h'])
assert False
except SystemExit:
pass
os3 = Bcfg2.Options.OptionSet(opts)
os3.parse(['-G'])
assert os3['foo'] == True
class TestOptionParser(object):
def test__init(self):
opts = [('foo', Bcfg2.Options.Option('foo', 'test1', cmd='-h')),
('bar', Bcfg2.Options.Option('foo', 'test2')),
('baz', Bcfg2.Options.Option('foo', 'test1', cmd='-H', odesc='1'))]
os1 = Bcfg2.Options.OptionParser(opts)
assert Bcfg2.Options.Option.cfpath == '/etc/bcfg2.conf'
sys.argv = ['foo', '-C', '/usr/local/etc/bcfg2.conf']
os2 = Bcfg2.Options.OptionParser(opts)
assert Bcfg2.Options.Option.cfpath == '/usr/local/etc/bcfg2.conf'
sys.argv = []
os3 = Bcfg2.Options.OptionParser(opts)
assert Bcfg2.Options.Option.cfpath == '/etc/bcfg2.conf'
|
