81 lines
2.4 KiB
Python
81 lines
2.4 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('wait_time', conf)
|
|
self.assertIn('lock_dir', conf)
|
|
self.assertTrue(os.path.isdir(conf['lock_dir']))
|
|
|
|
|
|
class TestLogger(unittest.TestCase):
|
|
|
|
def test_debug(self):
|
|
from surok.logger import Logger
|
|
m = Logger()
|
|
self.assertIn('DEBUG', m.testing('debug','log message'))
|
|
|
|
def test_info(self):
|
|
from surok.logger import Logger
|
|
m = Logger()
|
|
self.assertIn('INFO', m.testing('info','log message'))
|
|
|
|
def test_warning(self):
|
|
from surok.logger import Logger
|
|
m = Logger()
|
|
self.assertIn('WARNING', m.testing('warning','log message'))
|
|
|
|
def test_error(self):
|
|
from surok.logger import Logger
|
|
m = Logger()
|
|
self.assertIn('ERROR', m.testing('error','log message'))
|
|
|
|
|
|
class TestMemcachedDiscovery(unittest.TestCase):
|
|
|
|
def test_discovery_memcache(self):
|
|
from surok.system import discovery_memcached
|
|
from surok.discovery import Discovery
|
|
# 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()
|
|
d=Discovery(conf)
|
|
self.assertEqual(discovery_memcached(conf), [])
|
|
|
|
|
|
class TestGetGroup(unittest.TestCase):
|
|
|
|
def test_get_group_from_service(self):
|
|
from surok.discovery import DiscoveryTemplate
|
|
d=DiscoveryTemplate({})
|
|
self.assertEqual('xxx.yyy.zzz',d.get_group({'group':'xxx.yyy.zzz'}, {}))
|
|
|
|
def test_get_group_from_env(self):
|
|
from surok.discovery import DiscoveryTemplate
|
|
d=DiscoveryTemplate({})
|
|
self.assertEqual('xxx.yyy.zzz',d.get_group({}, {'env':{'SUROK_DISCOVERY_GROUP':'xxx.yyy.zzz'}}))
|
|
|
|
def test_get_group_from_marathon_id(self):
|
|
from surok.discovery import DiscoveryTemplate
|
|
d=DiscoveryTemplate({})
|
|
self.assertEqual('xxx.yyy.zzz',d.get_group({}, {'env':{'MARATHON_APP_ID':'/zzz/yyy/xxx/www'}}))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|