Add organization membership deactivate and reactivate API methods (#264)

* Add organization membership deactivate and reactivate API methods.

Author: mattgd

tests/test_user_management.py

@@ -389,6 +389,38 @@ def test_list_organization_memberships_returns_metadata(
         dict_oms = oms.to_dict()
         assert dict_oms["metadata"]["params"]["organization_id"] == "org_12345"
 
+    def test_list_organization_memberships_with_multiple_statuses_returns_metadata(
+        self, mock_organization_memberships, capture_and_mock_request
+    ):
+        _, request_kwargs = capture_and_mock_request(
+            "get", mock_organization_memberships, 200
+        )
+
+        oms = self.user_management.list_organization_memberships(
+            organization_id="org_12345",
+            statuses=["active", "inactive"],
+        )
+
+        assert request_kwargs["params"]["statuses"] == "active,inactive"
+        dict_oms = oms.to_dict()
+        assert dict_oms["metadata"]["params"]["organization_id"] == "org_12345"
+
+    def test_list_organization_memberships_with_a_single_status_returns_metadata(
+        self, mock_organization_memberships, capture_and_mock_request
+    ):
+        _, request_kwargs = capture_and_mock_request(
+            "get", mock_organization_memberships, 200
+        )
+
+        oms = self.user_management.list_organization_memberships(
+            organization_id="org_12345",
+            statuses=["inactive"],
+        )
+
+        assert request_kwargs["params"]["statuses"] == "inactive"
+        dict_oms = oms.to_dict()
+        assert dict_oms["metadata"]["params"]["organization_id"] == "org_12345"
+
     def test_delete_organization_membership(self, capture_and_mock_request):
         url, request_kwargs = capture_and_mock_request("delete", None, 200)
 
@@ -397,6 +429,34 @@ def test_delete_organization_membership(self, capture_and_mock_request):
         assert url[0].endswith("user_management/organization_memberships/om_ABCDE")
         assert user is None
 
+    def test_deactivate_organization_membership(
+        self, mock_organization_membership, capture_and_mock_request
+    ):
+        url, request_kwargs = capture_and_mock_request(
+            "put", mock_organization_membership, 200
+        )
+
+        om = self.user_management.deactivate_organization_membership("om_ABCDE")
+
+        assert url[0].endswith(
+            "user_management/organization_memberships/om_ABCDE/deactivate"
+        )
+        assert om["id"] == "om_ABCDE"
+
+    def test_reactivate_organization_membership(
+        self, mock_organization_membership, capture_and_mock_request
+    ):
+        url, request_kwargs = capture_and_mock_request(
+            "put", mock_organization_membership, 200
+        )
+
+        om = self.user_management.reactivate_organization_membership("om_ABCDE")
+
+        assert url[0].endswith(
+            "user_management/organization_memberships/om_ABCDE/reactivate"
+        )
+        assert om["id"] == "om_ABCDE"
+
     def test_authorization_url_throws_value_error_without_redirect_uri(self):
         connection_id = "connection_123"
         login_hint = "[email protected]"

workos/user_management.py

@@ -28,6 +28,12 @@
 USER_DETAIL_PATH = "user_management/users/{0}"
 ORGANIZATION_MEMBERSHIP_PATH = "user_management/organization_memberships"
 ORGANIZATION_MEMBERSHIP_DETAIL_PATH = "user_management/organization_memberships/{0}"
+ORGANIZATION_MEMBERSHIP_DEACTIVATE_PATH = (
+    ORGANIZATION_MEMBERSHIP_DETAIL_PATH + "/deactivate"
+)
+ORGANIZATION_MEMBERSHIP_REACTIVATE_PATH = (
+    ORGANIZATION_MEMBERSHIP_DETAIL_PATH + "/reactivate"
+)
 USER_AUTHORIZATION_PATH = "user_management/authorize"
 USER_AUTHENTICATE_PATH = "user_management/authenticate"
 USER_SEND_PASSWORD_RESET_PATH = "user_management/password_reset/send"
@@ -290,6 +296,7 @@ def list_organization_memberships(
         self,
         user_id=None,
         organization_id=None,
+        statuses=None,
         limit=None,
         before=None,
         after=None,
@@ -300,6 +307,7 @@ def list_organization_memberships(
         Kwargs:
             user_id (str): Filter Organization Memberships by user. (Optional)
             organization_id (str): Filter Organization Memberships by organization. (Optional)
+            statuses (list): Filter Organization Memberships by status. (Optional)
             limit (int): Maximum number of records to return. (Optional)
             before (str): Pagination cursor to receive records before a provided Organization Membership ID. (Optional)
             after (str): Pagination cursor to receive records after a provided Organization Membership ID. (Optional)
@@ -315,9 +323,13 @@ def list_organization_memberships(
             limit = RESPONSE_LIMIT
             default_limit = True
 
+        if statuses is not None:
+            statuses = ",".join(statuses)
+
         params = {
             "user_id": user_id,
             "organization_id": organization_id,
+            "statuses": statuses,
             "limit": limit,
             "before": before,
             "after": after,
@@ -364,6 +376,38 @@ def delete_organization_membership(self, organization_membership_id):
             token=workos.api_key,
         )
 
+    def deactivate_organization_membership(self, organization_membership_id):
+        """Deactivate an organization membership.
+
+        Args:
+            organization_membership_id (str) -  The unique ID of the Organization Membership.
+        Returns:
+            dict: OrganizationMembership response from WorkOS.
+        """
+        response = self.request_helper.request(
+            ORGANIZATION_MEMBERSHIP_DEACTIVATE_PATH.format(organization_membership_id),
+            method=REQUEST_METHOD_PUT,
+            token=workos.api_key,
+        )
+
+        return WorkOSOrganizationMembership.construct_from_response(response).to_dict()
+
+    def reactivate_organization_membership(self, organization_membership_id):
+        """Reactivates an organization membership.
+
+        Args:
+            organization_membership_id (str) -  The unique ID of the Organization Membership.
+        Returns:
+            dict: OrganizationMembership response from WorkOS.
+        """
+        response = self.request_helper.request(
+            ORGANIZATION_MEMBERSHIP_REACTIVATE_PATH.format(organization_membership_id),
+            method=REQUEST_METHOD_PUT,
+            token=workos.api_key,
+        )
+
+        return WorkOSOrganizationMembership.construct_from_response(response).to_dict()
+
     def get_authorization_url(
         self,
         redirect_uri,