def test_configure_logging_file(self, fileconfig_mock):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.return_value = '/path/to/logging.conf'
cfg.configure_logging()
cfgparser.get.assert_called_once_with('daemon', 'logging_config')
fileconfig_mock.assert_called_once_with('/path/to/logging.conf')
def test_from_config_list(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.return_value = 'one, two,\tthree'
ret = {}
cfg._from_config(ret, 'sec', 'one', opt_type='list')
cfgparser.get.assert_called_with('sec', 'one')
self.assertEqual(['one', 'two', 'three'], ret['one'])
def test_configure_logging(self, basicconfig_mock):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.side_effect = NoOptionError(None, None)
cfg.configure_logging()
cfgparser.get.assert_called_once_with('daemon', 'logging_config')
basicconfig_mock.assert_called_once_with(
stream=sys.stdout, format=ANY, level=ANY)
def test_from_config_float(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.getfloat.return_value = 3.14
ret = {}
cfg._from_config(ret, 'sec', 'one', opt_type='float')
cfgparser.getfloat.assert_called_with('sec', 'one')
self.assertEqual(3.14, ret['one'])
def test_from_config_bool(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.getboolean.return_value = True
ret = {}
cfg._from_config(ret, 'sec', 'one', opt_type='bool')
cfgparser.getboolean.assert_called_with('sec', 'one')
self.assertEqual(True, ret['one'])
def test_from_config_missing(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.side_effect = NoOptionError('sec', 'one')
ret = {}
cfg._from_config(ret, 'sec', 'one')
cfgparser.get.assert_called_with('sec', 'one')
self.assertFalse('one' in ret)
def test_from_config_int(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.getint.return_value = 13
ret = {}
cfg._from_config(ret, 'sec', 'one', opt_type='int')
cfgparser.getint.assert_called_with('sec', 'one')
self.assertEqual(13, ret['one'])
def test_from_config_str(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.return_value = 'three'
ret = {}
cfg._from_config(ret, 'sec', 'onetest', dict_key='one')
cfgparser.get.assert_called_with('sec', 'onetest')
self.assertEqual('three', ret['one'])
def test_get_pidfile(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.side_effect = ['/var/run/test',
NoOptionError('daemon', 'pidfile')]
self.assertEqual('/var/run/test', cfg.get_pidfile())
cfgparser.get.assert_called_with('daemon', 'pidfile')
self.assertEqual(None, cfg.get_pidfile())
cfgparser.get.assert_called_with('daemon', 'pidfile')
def test_get_rlimits(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.getint.side_effect = [65535,
NoOptionError('daemon', 'max_fd')]
self.assertEqual([(resource.RLIMIT_NOFILE, (65535, 65535))],
list(cfg.get_rlimits()))
cfgparser.getint.assert_called_with('daemon', 'max_fd')
self.assertEqual([], list(cfg.get_rlimits()))
cfgparser.getint.assert_called_with('daemon', 'max_fd')
def test_get_worker_privileges(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.side_effect = ['test_user',
'test_group',
'1234']
self.assertEqual(('test_user', 'test_group', 1234),
cfg.get_worker_privileges())
self.assertEqual(3, cfgparser.get.call_count)
cfgparser.get.assert_any_call('daemon', 'user')
cfgparser.get.assert_any_call('daemon', 'group')
cfgparser.get.assert_any_call('daemon', 'umask')
def test_get_stdio_redirects(self):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.get.side_effect = ['test_stdout',
'test_stderr',
'test_stdin']
self.assertEqual(('test_stdout', 'test_stderr', 'test_stdin'),
cfg.get_stdio_redirects())
self.assertEqual(3, cfgparser.get.call_count)
cfgparser.get.assert_any_call('daemon', 'stdout')
cfgparser.get.assert_any_call('daemon', 'stderr')
cfgparser.get.assert_any_call('daemon', 'stdin')
def test_configure_amqp(self, set_mock):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.sections.return_value = ['one', 'amqp']
cfgparser.get.side_effect = [
'testhost',
'testuser',
'testpass',
NoOptionError('amqp', 'virtual_host')]
cfgparser.getint.return_value = 5672
cfgparser.getfloat.side_effect = [30.0, 10.0]
cfg.configure_amqp()
set_mock.assert_called_with(
userid='testuser',
password='******',
host='testhost',
port=5672,
heartbeat=30.0,
connect_timeout=10.0)
def test_configure_http(self, set_mock, reset_mock):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.sections.return_value = ['one', 'http:test']
cfgparser.get.side_effect = ['testhost',
'testuser',
'testpass',
NoOptionError(None, None),
NoOptionError(None, None)]
cfgparser.getint.side_effect = [3306,
NoOptionError(None, None)]
cfgparser.getboolean.side_effect = [True]
cfg.configure_http()
reset_mock.assert_called_with()
set_mock.assert_called_with('test',
host='testhost',
port=3306,
user='******',
password='******',
ssl=True)
def test_configure_mysql(self, set_mock, reset_mock):
cfgparser = MagicMock()
cfg = Configuration(cfgparser)
cfgparser.sections.return_value = ['one', 'mysql:test']
cfgparser.get.side_effect = [
'testuser',
'testpass',
'testhost',
'testdb',
'testcharset',
NoOptionError('mysql:test', 'unix_socket')]
cfgparser.getint.side_effect = [3306, 10]
cfg.configure_mysql()
reset_mock.assert_called_with()
set_mock.assert_called_with(
'test',
user='******',
password='******',
host='testhost',
port=3306,
database='testdb',
charset='testcharset',
connect_timeout=10)
