def test_assign_members_to_projects_back(self, mock_input, mock_menu, mock_project_list, mock_print): mock_project_list.return_value = {'project_name': 'project_id'} mock_menu().draw_menu.return_value = 'BACK' gpm = GPM() gpm.assign_members_to_projects() mock_print.assert_called_once() mock_input.assert_called_once() mock_menu().draw_menu.assert_called_once() mock_project_list.assert_called_once()
def test_assign_members_to_projects_member_exceeded( self, mock_db, mock_input, mock_menu, mock_member_list, mock_project_list, mock_print): mock_member_list.return_value = { 'name': 'member_id', 'name2': 'member_id2' } mock_project_list.return_value = {'project_name': 'project_id'} mock_db().get_project_members.return_value = [['member_id']] mock_db().get_project_members_required.return_value = [1] mock_menu().draw_menu.return_value = 'project_name' gpm = GPM() gpm.assign_members_to_projects() mock_member_list.assert_called_once() self.assertEqual(mock_menu().draw_menu.call_count, 1) self.assertEqual(mock_input.call_count, 1) self.assertEqual(mock_print.call_count, 3) mock_db().get_project_members_required.assert_called_once() mock_db().get_project_members.assert_called_once()
def test_assign_members_to_projects(self, mock_db, mock_input, mock_menu, mock_member_list, mock_project_list, mock_member_request, mock_print): mock_member_list.return_value = { 'name': 'member_id', 'name2': 'member_id2' } mock_project_list.return_value = {'project_name': 'project_id'} mock_db().get_project_members.return_value = [['member_id']] mock_db().get_project_members_required.return_value = [2] mock_menu().draw_menu.side_effect = ['project_name', 'name2'] gpm = GPM() gpm.assign_members_to_projects() mock_member_list.assert_called_once() mock_member_request.assert_called_once() self.assertEqual(mock_menu().draw_menu.call_count, 2) self.assertEqual(mock_input.call_count, 2) self.assertEqual(mock_print.call_count, 2) mock_db().get_project_members_required.assert_called_once() mock_db().get_project_members.assert_called_once()