Extract Routing Logic from BillableMembership Entity

app/helpers/gitlab_routing_helper.rb

@@ -166,6 +166,16 @@ module GitlabRoutingHelper
     resend_invite_group_group_member_path(group_member.source, group_member)
   end
 
+  # Members
+  def source_members_url(member)
+    case member.source_type
+    when 'Namespace'
+      group_group_members_url(member.source)
+    when 'Project'
+      project_project_members_url(member.source)
+    end
+  end
+
   # Artifacts
 
   # Rails path generators are slow because they need to do large regex comparisons

ee/lib/ee/api/entities/billable_membership.rb

@@ -4,7 +4,7 @@ module EE
   module API
     module Entities
       class BillableMembership < Grape::Entity
-        include ::Gitlab::Routing
+        include RequestAwareEntity
 
         expose :id
         expose :source_id
@@ -12,12 +12,7 @@ module EE
           member.source.full_name
         end
         expose :source_members_url do |member|
-          case member.source_type
-          when 'Namespace'
-            group_group_members_url(member.source)
-          when 'Project'
-            project_project_members_url(member.source)
-          end
+          source_members_url(member)
         end
         expose :created_at
         expose :expires_at

ee/spec/lib/ee/api/entities/billable_membership_spec.rb

-# frozen_string_literal: true
-
-require 'spec_helper'
-
-RSpec.describe ::EE::API::Entities::BillableMembership do
-  describe '#as_json' do
-    it 'returns source_members_url for a group' do
-      membership = create(:group_member)
-      group_members_url = Gitlab::Routing.url_helpers.group_group_members_url(membership.source)
-
-      expect(described_class.new(membership).as_json[:source_members_url]).to eq(group_members_url)
-    end
-
-    it 'returns source_members_url for a project' do
-      membership = create(:project_member)
-      project_members_url = Gitlab::Routing.url_helpers.project_project_members_url(membership.source)
-
-      expect(described_class.new(membership).as_json[:source_members_url]).to eq(project_members_url)
-    end
-  end
-end

spec/helpers/gitlab_routing_helper_spec.rb

@@ -113,6 +113,24 @@ RSpec.describe GitlabRoutingHelper do
     end
   end
 
+  describe 'members helpers' do
+    describe '#source_members_url' do
+      it 'returns a url to the memberships page for a group membership' do
+        membership = build_stubbed(:group_member)
+        group_members_url = "http://test.host/groups/#{membership.source.full_path}/-/group_members"
+
+        expect(source_members_url(membership)).to eq(group_members_url)
+      end
+
+      it 'returns a url to the memberships page for a project membership' do
+        membership = build_stubbed(:project_member)
+        project_members_url = "http://test.host/#{membership.source.full_path}/-/project_members"
+
+        expect(source_members_url(membership)).to eq(project_members_url)
+      end
+    end
+  end
+
   context 'artifacts' do
     let_it_be(:project) { create(:project) }
     let_it_be(:job) { create(:ci_build, project: project, name: 'test:job', artifacts_expire_at: 1.hour.from_now) }