def test_handle_event(self, mock_decrypt): @patch("Bcfg2.Server.Plugins.Cfg.CfgGenerator.handle_event") @patch("Bcfg2.Options.setup.lax_decryption", False) def inner(mock_handle_event): def reset(): mock_decrypt.reset_mock() mock_handle_event.reset_mock() def get_event_data(obj, event): obj.data = "encrypted" mock_handle_event.side_effect = get_event_data mock_decrypt.side_effect = lambda d, **kw: "plaintext" event = Mock() ceg = self.get_obj() ceg.handle_event(event) mock_handle_event.assert_called_with(ceg, event) mock_decrypt.assert_called_with("encrypted") self.assertEqual(ceg.data, "plaintext") reset() mock_decrypt.side_effect = EVPError self.assertRaises(PluginExecutionError, ceg.handle_event, event) inner() # to perform the tests from the parent test object, we # make bruteforce_decrypt just return whatever data was # given to it mock_decrypt.side_effect = lambda d, **kw: d TestCfgGenerator.test_handle_event(self)
def test_handle_event(self, mock_decrypt): @patch("Bcfg2.Server.Plugins.Cfg.CfgGenerator.handle_event") @patch("Bcfg2.Options.setup.lax_decryption", False) def inner(mock_handle_event): def reset(): mock_decrypt.reset_mock() mock_handle_event.reset_mock() def get_event_data(obj, event): obj.data = "encrypted" mock_handle_event.side_effect = get_event_data mock_decrypt.side_effect = lambda d, **kw: "plaintext" event = Mock() ceg = self.get_obj() ceg.handle_event(event) mock_handle_event.assert_called_with(ceg, event) mock_decrypt.assert_called_with("encrypted") self.assertEqual(ceg.data, "plaintext") reset() mock_decrypt.side_effect = EVPError self.assertRaises(PluginExecutionError, ceg.handle_event, event) inner() # to perform the tests from the parent test object, we # make bruteforce_decrypt just return whatever data was # given to it mock_decrypt.side_effect = lambda d, **kw: d TestCfgGenerator.test_handle_event(self)
def test_handle_event(self): @patch("Bcfg2.Server.Plugins.Cfg.CfgGenerator.handle_event") def inner(mock_handle_event): cgg = self.get_obj() cgg.loader = Mock() cgg.data = "template data" event = Mock() cgg.handle_event(event) cgg.loader.load.assert_called_with(cgg.name, cls=NewTextTemplate, encoding=cgg.encoding) cgg.loader.reset_mock() cgg.loader.load.side_effect = OSError self.assertRaises(PluginExecutionError, cgg.handle_event, event) cgg.loader.load.assert_called_with(cgg.name, cls=NewTextTemplate, encoding=cgg.encoding) inner() loader_cls = self.test_obj.__loader_cls__ self.test_obj.__loader_cls__ = Mock TestCfgGenerator.test_handle_event(self) self.test_obj.__loader_cls__ = loader_cls
def test_handle_event(self): @patch("Bcfg2.Server.Plugins.Cfg.CfgGenerator.handle_event") def inner(mock_handle_event): cgg = self.get_obj() cgg.loader = Mock() cgg.data = "template data" event = Mock() cgg.handle_event(event) cgg.loader.load.assert_called_with(cgg.name, cls=NewTextTemplate, encoding=cgg.encoding) cgg.loader.reset_mock() cgg.loader.load.side_effect = OSError self.assertRaises(PluginExecutionError, cgg.handle_event, event) cgg.loader.load.assert_called_with(cgg.name, cls=NewTextTemplate, encoding=cgg.encoding) inner() loader_cls = self.test_obj.__loader_cls__ self.test_obj.__loader_cls__ = Mock TestCfgGenerator.test_handle_event(self) self.test_obj.__loader_cls__ = loader_cls
def test_get_data(self): ceg = self.get_obj() ceg.data = None entry = lxml.etree.Element("Path", name="/test.txt") metadata = Mock() self.assertRaises(PluginExecutionError, ceg.get_data, entry, metadata) TestCfgGenerator.test_get_data(self)
def test_get_data(self): ceg = self.get_obj() ceg.data = None entry = lxml.etree.Element("Path", name="/test.txt") metadata = Mock() self.assertRaises(PluginExecutionError, ceg.get_data, entry, metadata) TestCfgGenerator.test_get_data(self)
def test__init(self): TestCfgGenerator.test__init(self) cgg = self.get_obj() self.assertIsInstance(cgg.loader, cgg.__loader_cls__)
def setUp(self): TestCfgGenerator.setUp(self) set_setup_default("repository", datastore)
def test__init(self): TestCfgGenerator.test__init(self) cgg = self.get_obj() self.assertIsInstance(cgg.loader, cgg.__loader_cls__)
def setUp(self): TestCfgGenerator.setUp(self) set_setup_default("repository", datastore)
def setUp(self): TestCfgGenerator.setUp(self)
def setUp(self): TestCfgGenerator.setUp(self)
def setUp(self): TestCfgGenerator.setUp(self) TestStructFile.setUp(self)
def setUp(self): TestCfgGenerator.setUp(self) TestStructFile.setUp(self)