Add activate_membership method

jacquerie · sigmavirus24 · commit 3b15ba019c6a · 2018-07-28T16:46:58.000-05:00
Adds a method to activate a pending organization membership, as described in https://git.io/fNEo9.
diff --git a/src/github3/github.py b/src/github3/github.py
@@ -70,6 +70,28 @@ def _repr(self):
             return '<GitHub [{!r}]>'.format(self.session.auth)
         return '<Anonymous GitHub at 0x{0:x}>'.format(id(self))
 
+    @requires_auth
+    def activate_membership(self, organization):
+        """Activate the membership to an organization.
+
+        :param organization:
+            the organization or organization login for which to activate the
+            membership
+        :type organization:
+            str
+        :type organization:
+            :class:`~github3.orgs.Organization`
+        :returns:
+            the activated membership
+        :rtype:
+            :class:`~github3.orgs.Membership`
+        """
+        organization_name = getattr(organization, 'login', organization)
+        url = self._build_url('user', 'memberships', 'orgs', organization_name)
+        data = {'state': 'active'}
+        _json = self._json(self._patch(url, data=json.dumps(data)), 200)
+        return self._instance_or_null(orgs.Membership, _json)
+
     @requires_auth
     def add_email_addresses(self, addresses=[]):
         """Add the addresses to the authenticated user's account.
diff --git a/tests/unit/test_github.py b/tests/unit/test_github.py
@@ -19,6 +19,13 @@ class TestGitHub(helper.UnitHelper):
     described_class = GitHub
     example_data = None
 
+    def test_activate_membership(self):
+        self.instance.activate_membership('github3py')
+        self.session.patch.assert_called_once_with(
+            url_for('user/memberships/orgs/github3py'),
+            data='{"state": "active"}'
+        )
+
     def test_add_email_addresses(self):
         """Verify request to add email addresses for a user."""
         self.instance.add_email_addresses(['example1@example.com',
@@ -1285,6 +1292,10 @@ class TestGitHubRequiresAuthentication(
     described_class = GitHub
     example_data = None
 
+    def test_activate_membership(self):
+        """Verify that activating a membership requires authentication."""
+        self.assert_requires_auth(self.instance.activate_membership)
+
     def test_add_email_addresses(self):
         """Verify a user must be authenticated to add email addresses."""
         self.assert_requires_auth(self.instance.add_email_addresses, [])
