def test_update_project_back(self, mock_project_list, mock_menu, mock_input): mock_project_list.return_value = {'name1': 'id1'} mock_menu().draw_menu.side_effect = ['BACK'] bdo = BDO() bdo.update_project() mock_input.assert_called_once() mock_menu().draw_menu.assert_called_once() mock_project_list.assert_called_once()
def test_update_project_invalid_area(self, mock_print, mock_menu, mock_input, mock_project_list, mock_validation): mock_input.return_value = '' mock_project_list.return_value = {'name': 'id'} mock_menu().draw_menu.side_effect = ['name', 'AREA_OF_PROJECT'] mock_validation.is_int.return_value = False bdo = BDO() bdo.update_project() self.assertEqual(mock_input.call_count, 4) self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_menu().draw_menu.call_count, 2) mock_project_list.assert_called_once() mock_validation.is_int.assert_called_once()
def test_update_project_type(self, mock_print, mock_menu, mock_input, mock_db, mock_project_list): mock_input.return_value = '' mock_project_list.return_value = {'name': 'id'} mock_menu().draw_menu.side_effect = ['name', 'PROJECT_TYPE', 'road_project'] bdo = BDO() bdo.update_project() self.assertEqual(mock_input.call_count, 3) self.assertEqual(mock_print.call_count, 4) self.assertEqual(mock_menu().draw_menu.call_count, 3) mock_db().update_project.assert_called_once() mock_db().commit_data.assert_called_once() mock_project_list.assert_called_once()
def test_update_project_invalid_end_date(self, mock_print, mock_menu, mock_input, mock_project_list, mock_validation, mock_db): mock_input.return_value = '' mock_project_list.return_value = {'name': 'id'} mock_menu().draw_menu.side_effect = ['name', 'ESTIMATED_END_DATE'] mock_validation.end_date.return_value = False bdo = BDO() bdo.update_project() self.assertEqual(mock_input.call_count, 4) self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_menu().draw_menu.call_count, 2) mock_project_list.assert_called_once() mock_validation.end_date.assert_called_once()
def test_update_project_invalid_labour_numbers(self, mock_print, mock_menu, mock_input, mock_project_list , mock_validation): mock_input.return_value = '' mock_project_list.return_value = {'name': 'id'} mock_menu().draw_menu.side_effect = ['name', 'TOTAL_LABOUR_REQUIRED'] mock_validation.is_int.return_value = False bdo = BDO() bdo.update_project() self.assertEqual(mock_input.call_count, 4) self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_menu().draw_menu.call_count, 2) mock_project_list.assert_called_once() mock_validation.is_int.assert_called_once()