def test_data_handler_returns_None(self, mock_db_connection):
mapper = ScenarioMapper()
result = mapper._data_handler("", "", False)
assert result == None
def test_data_handler_throws_exception(self):
with self.assertRaises(Exception) as excep:
mapper = ScenarioMapper()
mapper._data_handler("WRONGQUERY", "", False)
def test_data_handler_calls_connection(self, mock_db_connection):
mapper = ScenarioMapper()
mapper._data_handler("", "", True)
mock_db_connection.assert_called_with()
