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
|
#!/usr/bin/env python
# coding: utf-8
import shutil
import tempfile
import unittest
from tempfile import NamedTemporaryFile
import can
class LoadFileConfigTest(unittest.TestCase):
configuration = {
'default': {'interface': 'virtual', 'channel': '0'},
'one': {'interface': 'virtual', 'channel': '1'},
'two': {'channel': '2'},
'three': {'extra': 'extra value'},
}
def setUp(self):
# Create a temporary directory
self.test_dir = tempfile.mkdtemp()
def tearDown(self):
# Remove the directory after the test
shutil.rmtree(self.test_dir)
def _gen_configration_file(self, sections):
with NamedTemporaryFile(mode='w', dir=self.test_dir,
delete=False) as tmp_config_file:
content = []
for section in sections:
content.append("[{}]".format(section))
for k, v in self.configuration[section].items():
content.append("{} = {}".format(k, v))
tmp_config_file.write('\n'.join(content))
return tmp_config_file.name
def test_config_file_with_default(self):
tmp_config = self._gen_configration_file(['default'])
config = can.util.load_file_config(path=tmp_config)
self.assertEqual(config, self.configuration['default'])
def test_config_file_with_default_and_section(self):
tmp_config = self._gen_configration_file(['default', 'one'])
default = can.util.load_file_config(path=tmp_config)
self.assertEqual(default, self.configuration['default'])
one = can.util.load_file_config(path=tmp_config, section='one')
self.assertEqual(one, self.configuration['one'])
def test_config_file_with_section_only(self):
tmp_config = self._gen_configration_file(['one'])
config = can.util.load_file_config(path=tmp_config, section='one')
self.assertEqual(config, self.configuration['one'])
def test_config_file_with_section_and_key_in_default(self):
expected = self.configuration['default'].copy()
expected.update(self.configuration['two'])
tmp_config = self._gen_configration_file(['default', 'two'])
config = can.util.load_file_config(path=tmp_config, section='two')
self.assertEqual(config, expected)
def test_config_file_with_section_missing_interface(self):
expected = self.configuration['two'].copy()
tmp_config = self._gen_configration_file(['two'])
config = can.util.load_file_config(path=tmp_config, section='two')
self.assertEqual(config, expected)
def test_config_file_extra(self):
expected = self.configuration['default'].copy()
expected.update(self.configuration['three'])
tmp_config = self._gen_configration_file(['default', 'three'])
config = can.util.load_file_config(path=tmp_config, section='three')
self.assertEqual(config, expected)
def test_config_file_with_non_existing_section(self):
expected = {}
tmp_config = self._gen_configration_file([
'default', 'one', 'two', 'three'])
config = can.util.load_file_config(path=tmp_config, section='zero')
self.assertEqual(config, expected)
if __name__ == '__main__':
unittest.main()
|
