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
|
#!/usr/bin/env python
import shutil
import tempfile
import unittest
from tempfile import NamedTemporaryFile
import can
class LoadFileConfigTest(unittest.TestCase):
configuration = {
"default": {"interface": "virtual", "channel": "0"},
"one": {"interface": "kvaser", "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(f"[{section}]")
for k, v in self.configuration[section].items():
content.append(f"{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"])
config = can.util.load_file_config(path=tmp_config)
self.assertEqual(config, self.configuration["default"])
config.update(can.util.load_file_config(path=tmp_config, section="one"))
self.assertEqual(config, 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)
config.update(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)
config.update(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)
config.update(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)
config.update(can.util.load_file_config(path=tmp_config, section="three"))
self.assertEqual(config, expected)
def test_config_file_with_non_existing_section(self):
expected = self.configuration["default"].copy()
tmp_config = self._gen_configration_file(["default", "one", "two", "three"])
config = can.util.load_file_config(path=tmp_config)
config.update(can.util.load_file_config(path=tmp_config, section="zero"))
self.assertEqual(config, expected)
if __name__ == "__main__":
unittest.main()
|
