def test_delete_scenario_calls_data_handler(self, mock__get_scenarios, mock__data_handler): mapper = ScenarioMapper() mapper.delete_scenario(1, 1) mock__data_handler.assert_called_with( 'DELETE FROM scenario WHERE id = %s', (1, ), False)
def test_scenarios_calls_parser(self, mock_data_handler): mapper = ScenarioMapper() model = ScenarioResponseModel() model.naam = 'Naam' scenarios = mapper.get_scenarios(1) mock_data_handler.assert_called_with(1)
def test_add_scenario_calls_data_handler(self, mock__get_scenarios, mock_stored_procedure): model = ScenarioResponseModel() mapper = ScenarioMapper() mapper.add_scenario(model, 1) mock_stored_procedure.assert_called_with( 'maak_scenario_en_stop_in_project', ('scenario', 'toelichting', 1))
def test_update_scenario_calls_data_handler(self, mock__get_scenarios, mock__data_handler): model = ScenarioResponseModel() mapper = ScenarioMapper() mapper.update_scenario(model, 1) mock__data_handler.assert_called_with("UPDATE scenario " \ "SET naam = %s, toelichting = %s " \ "WHERE id = %s", ('scenario', 'toelichting', 1), False)
def test_update_jaar_in_gebeurtenis_in_scenario_call_get_scenarios( self, mock__get_scenarios, mock__data_handler): # Arrange mapper = ScenarioMapper() # Act mapper.update_jaar_in_gebeurtenis_in_scenario(1, 1, 1, 1, 1) # Assert mock__get_scenarios.assert_called()
def test_get_scenario_by_id_calls_get_scenarios_with_details( self, mock_get_scenarios_with_details): # Arrange mapper = ScenarioMapper() # Act mapper.get_scenario_by_id(1, 1) # Assert mock_get_scenarios_with_details.assert_called()
def test_remove_jaar_from_gebeurtenis_call_get_scenarios( self, mock__get_scenarios, mock__data_handler): # Arrange mapper = ScenarioMapper() # Act mapper.remove_jaar_from_gebeurtenis(1, 1, 1, 1) # Assert mock__get_scenarios.assert_called()
def test_add_gebeurtenis_to_scenario_call_get_scenarios( self, mock__get_scenarios, mock__data_handler): # Arrange mapper = ScenarioMapper() # Act mapper.add_gebeurtenis_to_scenario(1, 1, 1) # Assert mock__get_scenarios.assert_called()
def test_data_handler_throws_exception(self): with self.assertRaises(Exception) as excep: mapper = ScenarioMapper() mapper._data_handler("WRONGQUERY", "", False)
def test_data_handler_returns_None(self, mock_db_connection): mapper = ScenarioMapper() result = mapper._data_handler("", "", False) assert result == None
def test_data_handler_calls_connection(self, mock_db_connection): mapper = ScenarioMapper() mapper._data_handler("", "", True) mock_db_connection.assert_called_with()
def test_db_connection(self, mock_connect): mapper = ScenarioMapper() mapper._db_connection() mock_connect.assert_called()
def __init__(self): self.repository = ScenarioMapper() self.scenarios = [] self.JAREN = 100
class ScenarioController(object): def __init__(self): self.repository = ScenarioMapper() self.scenarios = [] self.JAREN = 100 """"create_scenario geeft het meegegeven scenario mee aan de repository zodat het scenario kan worden toegevoegd""" def create_scenario(self, model: ScenarioResponseModel, project_id: int): self.repository.add_scenario(model, project_id) """get_all_scenario's roept get_scenarios() aan in de repository en geeft alle opgehaalde scenarios terug""" def get_all_scenarios(self, project_id: int): return self.repository.get_scenarios(project_id) """delete_scenario geeft het meegegeven id mee aan de repository zodat die scenario met het meegegeven id kan verwijderen """ def delete_scenario(self, id: int, project_id: int): return self.repository.delete_scenario(id, project_id) """update_scenario geeft het meegegeven scenario mee aan de repository zodat het meegegeven scenario bijgewerkt kan worden """ def update_scenario(self, model: ScenarioResponseModel, project_id: int): return self.repository.update_scenario(model, project_id) """get_scenario_from_id selecteert het gewenste scenario en geeft deze terug.""" def add_gebeurtenis_to_scenario(self, scenario_id, gebeurtenis_id, project_id): return self.repository.add_gebeurtenis_to_scenario( scenario_id, gebeurtenis_id, project_id) """verwijder een gebeurtenis van een scenario""" def remove_gebeurtenis_from_scenario(self, scenario_id, gebeurtenis_id, project_id: int): return self.repository.remove_gebeurtenis_from_scenario( scenario_id, gebeurtenis_id, project_id) """voeg een jaar toe aan een gebeurtenis binnen een scenario""" def add_jaar_to_gebeurtenis_in_scenario(self, scenario_id, gebeurtenis_id, jaar, waarde, project_id): return self.repository.add_jaar_to_gebeurtenis_in_scenario( scenario_id, gebeurtenis_id, jaar, waarde, project_id) """pas de waarde van een jaar aan""" def update_jaar_in_gebeurtenis_in_scenario(self, scenario_id, gebeurtenis_id, jaar, waarde, project_id): return self.repository.update_jaar_in_gebeurtenis_in_scenario( scenario_id, gebeurtenis_id, jaar, waarde, project_id) """verwijder een jaar uit een gebeurtenis""" def remove_jaar_from_gebeurtenis_in_scenario(self, scenario_id, gebeurtenis_id, jaar, project_id): return self.repository.remove_jaar_from_gebeurtenis( scenario_id, gebeurtenis_id, jaar, project_id) def get_scenario_from_id(self, project_id: int, id: int, waccc: float = 5, waccw: float = 1, jaren: int = 100, cumulatief_berekenen: bool = True, cashflow_berekenen: bool = True, waardering_berekenen: bool = True): calc = LCVCalculator(waccc, waccw, cumulatief_berekenen, cashflow_berekenen, waardering_berekenen) list = self.repository.get_scenarios(project_id) for scenario in list: if self._is_id_gelijk(scenario.id, id): scenario.result = calc.bereken_lcv_voor_scenario( scenario, jaren) return scenario return None """get_multiple_scenarios_from_ids selecteert de gewenste scenario's en geeft deze terug.""" def get_multiple_scenarios_from_ids(self, project_id: int, ids: List[int], waccc: float = 5, waccw: float = 1, jaren: int = 100, cumulatief_berekenen: bool = True, cashflow_berekenen: bool = True, waardering_berekenen: bool = True): calc = LCVCalculator(waccc, waccw, cumulatief_berekenen, cashflow_berekenen, waardering_berekenen) list = self.repository.get_scenarios(project_id) geselecteerde_scenarios: List[ScenarioResponseModel] = [] for id in ids: for scenario in list: if self._is_id_gelijk(scenario.id, id): scenario.result = calc.bereken_lcv_voor_scenario( scenario, jaren) geselecteerde_scenarios.append(scenario) return geselecteerde_scenarios def _is_id_gelijk(self, scenario_id, opgegeven_id): return scenario_id == opgegeven_id
def import_to_db_scenario(self, json_model_array, project_id): mapper = ScenarioMapper() for i in range(len(json_model_array)): mapper.add_scenario(json_model_array[i], project_id)