surok/build/tests.py

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()