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)
