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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#!/usr/bin/env python
# sugar-lint: disable
from ConfigParser import ConfigParser
from __init__ import tests
from sugar_network.toolkit import Option
class OptionsTest(tests.Test):
def test_LoadDirs(self):
self.touch(
('d/1', ['[section]', 'p1=1', 'p2=2']),
('d/2', ['[section]', 'p2=22', 'p3=33']),
)
p1 = Option(name='p1')
p2 = Option(name='p2')
p3 = Option(name='p3')
Option.seek('section', [p1, p2, p3])
Option.load(['d'])
self.assertEqual('1', p1.value)
self.assertEqual('22', p2.value)
self.assertEqual('33', p3.value)
def test_SaveChangedOptions(self):
option = Option(name='option')
Option.seek('section', [option])
self.touch(('config', [
'[section]',
'option = 1',
]))
Option.load(['./config'])
self.assertEqual('1', option.value)
option.value = '2'
Option.save()
option.value = ''
Option.load(['./config'])
self.assertEqual('2', option.value)
def test_PreserveNonSeekConfigOnSave(self):
option = Option(name='option')
Option.seek('section', [option])
self.touch(('config', [
'[foo]',
'o1 = 1 # foo',
'[section]',
'option = 1',
'[bar]',
'o2 = 2 # bar',
]))
Option.load(['./config'])
self.assertEqual('1', option.value)
option.value = '2'
Option.save()
self.assertEqual('\n'.join([
'[foo]',
'o1 = 1 # foo',
'',
'[section]',
'option = 2',
'',
'[bar]',
'o2 = 2 # bar',
'\n',
]),
file('config').read())
def test_SaveOnlyChangedOptions(self):
option1 = Option(name='option1', default='1')
option2 = Option(name='option2', default='2')
option3 = Option(name='option3', default='3')
Option.seek('section', [option1, option2, option3])
Option.load(['./config'])
self.assertEqual('1', option1.value)
self.assertEqual('2', option2.value)
self.assertEqual('3', option3.value)
Option.save()
parser = ConfigParser()
parser.read('config')
assert not parser.has_option('section', 'option1')
assert not parser.has_option('section', 'option2')
assert not parser.has_option('section', 'option3')
option1.value = '1_'
Option.save()
parser = ConfigParser()
parser.read('config')
self.assertEqual('1_', parser.get('section', 'option1'))
assert not parser.has_option('section', 'option2')
assert not parser.has_option('section', 'option3')
option2.value = '2_'
Option.save()
parser = ConfigParser()
parser.read('config')
self.assertEqual('1_', parser.get('section', 'option1'))
self.assertEqual('2_', parser.get('section', 'option2'))
assert not parser.has_option('section', 'option3')
if __name__ == '__main__':
tests.main()
|