72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
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'}))
|
|
|
|
|
|
class TestMemcachedDiscovery(unittest.TestCase):
|
|
|
|
def test_discovery_memcache(self):
|
|
from surok.system import discovery_memcached
|
|
|
|
# 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.assertEqual(discovery_memcached(conf), [])
|
|
|
|
|
|
class TestGetGroup(unittest.TestCase):
|
|
|
|
def test_get_group(self):
|
|
from surok.discovery import get_group
|
|
self.assertFalse(get_group({}, {'env': os.environ}))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|