|
3
|
1 |
#!/usr/bin/env python3
|
|
|
2 |
|
|
|
3 |
from pycolib.configparser import ConfigParser, ConfigParserError
|
|
|
4 |
|
|
|
5 |
import unittest
|
|
|
6 |
import os.path
|
|
|
7 |
|
|
|
8 |
|
|
|
9 |
class TestConfigParser(unittest.TestCase):
|
|
|
10 |
|
|
|
11 |
def setUp(self):
|
|
|
12 |
self.config = ConfigParser()
|
|
4
|
13 |
self.config.add_option('arg_str')
|
|
|
14 |
self.config.add_option('arg_int', int, 1)
|
|
3
|
15 |
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
|
16 |
self.data_path = os.path.join(script_path, 'data')
|
|
|
17 |
|
|
4
|
18 |
def test_add_option(self):
|
|
3
|
19 |
name = 'argname'
|
|
4
|
20 |
self.config.add_option(name, int, 1)
|
|
|
21 |
self.assertEqual(self.config.options[name]['type'], int)
|
|
|
22 |
self.assertEqual(self.config.options[name]['default'], 1)
|
|
|
23 |
self.assertEqual(self.config.values[name], 1)
|
|
3
|
24 |
|
|
|
25 |
def test_load_badname(self):
|
|
|
26 |
self.assertRaises(IOError, self.config.load, 'bad/file_name')
|
|
|
27 |
|
|
|
28 |
def test_load_badargs(self):
|
|
|
29 |
fname = os.path.join(self.data_path, 'test_badargs.conf')
|
|
|
30 |
self.assertRaises(ConfigParserError, self.config.load, fname)
|
|
|
31 |
|
|
|
32 |
def test_load_ok(self):
|
|
|
33 |
fname = os.path.join(self.data_path, 'test_ok.conf')
|
|
|
34 |
self.config.load(fname)
|
|
|
35 |
self.assertEqual(self.config.arg_str, '1')
|
|
|
36 |
self.assertEqual(self.config.arg_int, 2)
|
|
|
37 |
|
|
|
38 |
|
|
|
39 |
if __name__ == '__main__':
|
|
|
40 |
unittest.main()
|