-
Notifications
You must be signed in to change notification settings - Fork 169
/
test_global_config.py
37 lines (28 loc) · 1 KB
/
test_global_config.py
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
import unittest
import tempfile
import shutil
import os
import tuned.consts as consts
import tuned.utils.global_config as global_config
class GlobalConfigTestCase(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.test_dir = tempfile.mkdtemp()
with open(cls.test_dir + '/test_config','w') as f:
f.write('test_option = hello\ntest_bool = 1\ntest_size = 12MB\n'\
+ 'false_bool=0\n')
cls._global_config = global_config.GlobalConfig(\
cls.test_dir + '/test_config')
def test_get(self):
self.assertEqual(self._global_config.get('test_option'), 'hello')
def test_get_bool(self):
self.assertTrue(self._global_config.get_bool('test_bool'))
self.assertFalse(self._global_config.get_bool('false_bool'))
def test_get_size(self):
self.assertEqual(self._global_config.get_size('test_size'),\
12*1024*1024)
self._global_config.set('test_size','bad_value')
self.assertIsNone(self._global_config.get_size('test_size'))
@classmethod
def tearDownClass(cls):
shutil.rmtree(cls.test_dir)