def test_delete_gpm_exception(self, mock_gpm_list, mock_db, mock_print):
mock_gpm_list.side_effect = Exception
bdo = BDO()
bdo.delete_gpm()
self.assertEqual(mock_print.call_count, 2)
mock_gpm_list.assert_called_once()
mock_db().rollback_data.assert_called_once()
def test_delete_gpm_menu1_back(self, mock_gpm_list, mock_menu, mock_print, mock_input):
mock_gpm_list.return_value = {'name1': 'id1'}
mock_menu().draw_menu.side_effect = ['BACK']
bdo = BDO()
bdo.delete_gpm()
mock_input.assert_called_once()
mock_print.assert_called_once()
mock_menu().draw_menu.assert_called_once()
mock_gpm_list.assert_called_once()
def test_delete_gpm_menu2_back(self, mock_gpm_list, mock_menu, mock_print, mock_os, mock_input):
mock_gpm_list.return_value = {'name1': 'id1', 'name2': 'id2'}
mock_menu().draw_menu.side_effect = ['name1', 'BACK']
bdo = BDO()
bdo.delete_gpm()
self.assertEqual(mock_input.call_count, 2)
self.assertEqual(mock_print.call_count, 2)
self.assertEqual(mock_menu().draw_menu.call_count, 2)
mock_os.system.assert_called_once_with('clear')
mock_gpm_list.assert_called_once()
def test_delete_gpm_no_alternative(self, mock_gpm_list, mock_menu, mock_print, mock_os, mock_input):
mock_gpm_list.return_value = {'name1': 'id1'}
mock_menu().draw_menu.side_effect = ['name1']
bdo = BDO()
bdo.delete_gpm()
self.assertEqual(mock_print.call_count, 3)
mock_os.system.assert_called_once()
mock_input.assert_called_once()
mock_menu().draw_menu.assert_called_once()
mock_gpm_list.assert_called_once()
def test_delete_gpm(self, mock_gpm_list, mock_menu, mock_print, mock_db, mock_os, mock_delete_member, mock_input):
mock_gpm_list.return_value = {'name1': 'id1', 'name2': 'id2'}
mock_menu().draw_menu.side_effect = ['name1', 'name2']
mock_db().get_gpm_projects.return_value = ['dummy']
mock_db().get_project_members.return_value = [['dummy']]
bdo = BDO()
bdo.delete_gpm()
self.assertEqual(mock_menu().draw_menu.call_count, 2)
self.assertEqual(mock_input.call_count, 2)
self.assertEqual(mock_print.call_count, 2)
mock_os.system.assert_called_once()
mock_delete_member.assert_called_once()
mock_db().get_project_members.assert_called_once()
mock_db().get_gpm_projects.assert_called_once()
mock_gpm_list.assert_called_once()
mock_db().update_member_gpm.assert_called_once()
mock_db().update_user.assert_called_once()
mock_db().commit_data.assert_called_once()
