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

* Add organization membership deactivate and reactivate API methods.

Author: mattgd

lib/workos/user_management.rb

@@ -748,7 +748,8 @@ def get_organization_membership(id:)
       #
       # @param [Hash] options
       # @option options [String] user_id The ID of the User.
-      # @option options [String] organization_id Filter Users by the organization they are members of.
+      # @option options [String] organization_id Filter memberships by the organization they are members of.
+      # @option options [Array<String>] statuses Filter memberships by status.
       # @option options [String] limit Maximum number of records to return.
       # @option options [String] order The order in which to paginate records
       # @option options [String] before Pagination cursor to receive records
@@ -816,6 +817,38 @@ def delete_organization_membership(id:)
         response.is_a? Net::HTTPSuccess
       end
 
+      # Deactivate an Organization Membership
+      #
+      # @param [String] id The unique ID of the Organization Membership.
+      #
+      # @return WorkOS::OrganizationMembership
+      def deactivate_organization_membership(id:)
+        response = execute_request(
+          request: put_request(
+            path: "/user_management/organization_memberships/#{id}/deactivate",
+            auth: true,
+          ),
+        )
+
+        WorkOS::OrganizationMembership.new(response.body)
+      end
+
+      # Reactivate an Organization Membership
+      #
+      # @param [String] id The unique ID of the Organization Membership.
+      #
+      # @return WorkOS::OrganizationMembership
+      def reactivate_organization_membership(id:)
+        response = execute_request(
+          request: put_request(
+            path: "/user_management/organization_memberships/#{id}/reactivate",
+            auth: true,
+          ),
+        )
+
+        WorkOS::OrganizationMembership.new(response.body)
+      end
+
       # Gets an Invitation
       #
       # @param [String] id The unique ID of the Invitation.

spec/lib/workos/user_management_spec.rb

@@ -933,6 +933,33 @@
         end
       end
     end
+
+    context 'with statuses option' do
+      it 'returns a list of matching users' do
+        request_args = [
+          '/user_management/organization_memberships?user_id=user_01HXYSZBKQE2N3NHBKZHDP1X5X&'\
+          'statuses=active&statuses=inactive&order=desc&limit=5',
+          'Content-Type' => 'application/json'
+        ]
+
+        expected_request = Net::HTTP::Get.new(*request_args)
+
+        expect(Net::HTTP::Get).to receive(:new).with(*request_args).
+          and_return(expected_request)
+
+        VCR.use_cassette 'user_management/list_organization_memberships/with_statuses_option' do
+          organization_memberships = described_class.list_organization_memberships(
+            user_id: 'user_01HXYSZBKQE2N3NHBKZHDP1X5X',
+            statuses: %w[active inactive],
+            order: 'desc',
+            limit: '5',
+          )
+
+          expect(organization_memberships.data.size).to eq(1)
+          expect(organization_memberships.data[0].user_id).to eq('user_01HXYSZBKQE2N3NHBKZHDP1X5X')
+        end
+      end
+    end
   end
 
   describe '.create_organization_membership' do
@@ -988,6 +1015,56 @@
     end
   end
 
+  describe '.deactivate_organization_membership' do
+    context 'with a valid id' do
+      it 'returns a organization membership' do
+        VCR.use_cassette 'user_management/deactivate_organization_membership' do
+          organization_membership = described_class.deactivate_organization_membership(
+            id: 'om_01HXYT0G3H5QG9YTSHSHFZQE6D',
+          )
+
+          expect(organization_membership.id.instance_of?(String))
+          expect(organization_membership.instance_of?(WorkOS::OrganizationMembership))
+        end
+      end
+    end
+
+    context 'with an invalid id' do
+      it 'returns an error' do
+        expect do
+          described_class.deactivate_organization_membership(
+            id: 'invalid_organization_membership_id',
+          ).to raise_error(WorkOS::APIError)
+        end
+      end
+    end
+  end
+
+  describe '.reactivate_organization_membership' do
+    context 'with a valid id' do
+      it 'returns a organization membership' do
+        VCR.use_cassette 'user_management/reactivate_organization_membership' do
+          organization_membership = described_class.reactivate_organization_membership(
+            id: 'om_01HXYT0G3H5QG9YTSHSHFZQE6D',
+          )
+
+          expect(organization_membership.id.instance_of?(String))
+          expect(organization_membership.instance_of?(WorkOS::OrganizationMembership))
+        end
+      end
+    end
+
+    context 'with an invalid id' do
+      it 'returns an error' do
+        expect do
+          described_class.reactivate_organization_membership(
+            id: 'invalid_organization_membership_id',
+          ).to raise_error(WorkOS::APIError)
+        end
+      end
+    end
+  end
+
   describe '.get_invitation' do
     context 'with a valid id' do
       it 'returns an invitation' do