def test_create_project_exception(self, mock_gpm_list, mock_db, mock_print): mock_gpm_list.side_effect = Exception bdo = BDO() bdo.create_project() self.assertEqual(mock_print.call_count, 2) mock_gpm_list.assert_called_once() mock_db().rollback_data.assert_called_once()
def test_create_project_invalid_cost_number(self, mock_print, mock_menu, mock_input, mock_validation, mock_gpm_list): mock_input.return_value = '' mock_menu().draw_menu.return_value = '' mock_validation.is_int.side_effect = [True, False] mock_menu().draw_menu.side_effect = ['name', 'project_type'] mock_gpm_list.return_value = {'name': 'id'} bdo = BDO() bdo.create_project() self.assertEqual(mock_validation.is_int.call_count, 2) self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_menu().draw_menu.call_count, 2) self.assertEqual(mock_input.call_count, 6)
def test_create_project_invalid_start_date(self, mock_print, mock_menu, mock_input, mock_validation, mock_gpm_list): mock_input.return_value = '' mock_menu().draw_menu.return_value = '' mock_validation.is_int.return_value = True mock_validation.start_date.return_value = False mock_menu().draw_menu.side_effect = ['name', 'project_type'] mock_gpm_list.return_value = {'name': 'id'} bdo = BDO() bdo.create_project() self.assertEqual(mock_validation.is_int.call_count, 3) self.assertEqual(mock_print.call_count, 3) self.assertEqual(mock_menu().draw_menu.call_count, 2) self.assertEqual(mock_input.call_count, 8) mock_validation.start_date.assert_called_once()