def test_config_reader_get_different_section(self): """ ConfigReader.grab called with a different section supplied """ config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'some_value' data = config.grab('test', section='other_section', cast=False) self.assertTrue(data, 'some_value')
def test_config_reader_get_cast_to_int(self): """ ConfigReader.grab is able to cast to integers """ config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = '1' data = config.grab('test', cast=True) self.assertTrue(data is 1)
def test_config_reader_get_no_cast(self): """ ConfigReader.grab return data when not casting """ config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'some_value' data = config.grab('test', cast=False) self.assertTrue(data, 'some_value')
def test_config_reader_get_cast_with_lambda(self): """ ConfigReader.grab doesn't cast lambda functions """ config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'lambda: "failure"' data = config.grab('test', cast=True) self.assertTrue(data is 'lambda: "failure"')
def test_config_reader_get_cast_on_string(self): """ ConfigReader.grab is able to return strings when casting """ config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'mystring' data = config.grab('test', cast=True) self.assertTrue(data is 'mystring')
def test_config_reader_get_cast_with_class(self): """ ConfigReader.grab doesn't cast classes """ class SomeClass(object): pass config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'SomeClass' data = config.grab('test', cast=True) self.assertTrue(data, 'SomeClass')
def test_config_reader_get_cast_with_func(self): """ ConfigReader.grab doesn't cast functions """ def some_func(): print('failure') config = alarmer.config.ConfigReader('/some/path', 'test_section') config._config.grab = MagicMock() config._config.grab.return_value = 'some_func' data = config.grab('test', cast=True) self.assertTrue(data is 'some_func')