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()
