def test_fire_exception(self, mock_logger): test_exception = "test exception" mock_write = MagicMock(side_effect=ValueError(test_exception)) with patch("railgun.engines.storage_engine.DummyEngine.write", mock_write): r = Railgun(self.config["test"]) self.assertRaises(RailgunException, r.fire, "test") try: r.fire("test") except RailgunException: self.assertTrue(mock_logger.critical.called)
def test_fire(self): test_data = "test data" with patch("railgun.engines.storage_engine.DummyEngine.write") as mock_write: r = Railgun(self.config["test"]) r.fire(test_data) mock_write.assert_called_once_with(test_data)