1 files changed, 53 insertions, 0 deletions
diff --git a/spec/lib/api/helpers/members_helpers_spec.rb b/spec/lib/api/helpers/members_helpers_spec.rb
new file mode 100644
index 00000000000..987d5ba9f6c
--- /dev/null
+++ b/spec/lib/api/helpers/members_helpers_spec.rb
@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe API::Helpers::MembersHelpers, feature_category: :subgroups do
+  let(:helper) do
+    Class.new.include(described_class).new
+  end
+
+  describe '#source_members' do
+    subject(:source_members) { helper.source_members(source) }
+
+    shared_examples_for 'returns all direct members' do
+      specify do
+        expect(source_members).to match_array(direct_members)
+      end
+    end
+
+    context 'for a group' do
+      let_it_be(:source) { create(:group) }
+      let_it_be(:direct_members) { create_list(:group_member, 2, group: source) }
+
+      it_behaves_like 'returns all direct members'
+      it_behaves_like 'query with source filters'
+
+      context 'when project_members_index_by_project_namespace feature flag is disabled' do
+        before do
+          stub_feature_flags(project_members_index_by_project_namespace: false)
+        end
+
+        it_behaves_like 'returns all direct members'
+        it_behaves_like 'query with source filters'
+      end
+    end
+
+    context 'for a project' do
+      let_it_be(:source) { create(:project, group: create(:group)) }
+      let_it_be(:direct_members) { create_list(:project_member, 2, project: source) }
+
+      it_behaves_like 'returns all direct members'
+      it_behaves_like 'query without source filters'
+
+      context 'when project_members_index_by_project_namespace feature flag is disabled' do
+        before do
+          stub_feature_flags(project_members_index_by_project_namespace: false)
+        end
+
+        it_behaves_like 'returns all direct members'
+        it_behaves_like 'query with source filters'
+      end
+    end
+  end
+end