import unittest import json import os import re class TestLoadConfig(unittest.TestCase): def test_main_conf(self): # Load base configurations surok_conf = '/etc/surok/conf/surok.json' # Read config file f = open(surok_conf, 'r') conf = json.loads(f.read()) f.close() self.assertIn('confd', conf) self.assertTrue(os.path.isdir(conf['confd'])) self.assertIn('domain', conf) self.assertIn('wait_time', conf) self.assertIn('lock_dir', conf) self.assertTrue(os.path.isdir(conf['lock_dir'])) class TestLogger(unittest.TestCase): def test_info(self): from surok.logger import make_message m = make_message self.assertIn('INFO', m({'level': 'INFO', 'raw': 'log message'})) def test_warning(self): from surok.logger import make_message m = make_message self.assertIn('WARNING', m({'level': 'WARNING', 'raw': 'log message'})) def test_error(self): from surok.logger import make_message m = make_message self.assertIn('ERROR', m({'level': 'ERROR', 'raw': 'log message'})) def test_info(self): from surok.logger import make_message m = make_message self.assertIn('DEBUG', m({'level': 'DEBUG', 'raw': 'log message'})) if __name__ == '__main__': unittest.main()