1  ############################ Copyrights and license ############################
 2  #                                                                              #
 3  # Copyright 2021 James Simpson <james@snowterminal.com>                        #
 4  #                                                                              #
 5  # This file is part of PyGithub.                                               #
 6  # http://pygithub.readthedocs.io/                                              #
 7  #                                                                              #
 8  # PyGithub is free software: you can redistribute it and/or modify it under    #
 9  # the terms of the GNU Lesser General Public License as published by the Free  #
10  # Software Foundation, either version 3 of the License, or (at your option)    #
11  # any later version.                                                           #
12  #                                                                              #
13  # PyGithub is distributed in the hope that it will be useful, but WITHOUT ANY  #
14  # WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS    #
15  # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more #
16  # details.                                                                     #
17  #                                                                              #
18  # You should have received a copy of the GNU Lesser General Public License     #
19  # along with PyGithub. If not, see <http://www.gnu.org/licenses/>.             #
20  #                                                                              #
21  ################################################################################
22  
23  from . import Framework
24  
25  
26  class Organization2072(Framework.TestCase):
27      def setUp(self):
28          super().setUp()
29          self.org = self.g.get_organization("TestOrganization2072")
30  
31      def testCancelInvitation(self):
32          self.assertFalse(any([i for i in self.org.invitations() if i.email == "foo@bar.org"]))
33          self.org.invite_user(email="foo@bar.org")
34          self.assertTrue(any([i for i in self.org.invitations() if i.email == "foo@bar.org"]))
35          invitation = [i for i in self.org.invitations() if i.email == "foo@bar.org"][0]
36          self.assertTrue(self.org.cancel_invitation(invitation))