get rid of light url builder and fix wrong spec

lib/gitlab/light_url_builder.rb

-module Gitlab
-  # Similar to UrlBuilder, but using IDs to avoid querying the DB for objects
-  # Useful for using in conjunction with Arel queries.
-  class LightUrlBuilder
-    include Gitlab::Routing.url_helpers
-    include GitlabRoutingHelper
-    include ActionView::RecordIdentifier
-
-    def self.build(*args)
-      new(*args).url
-    end
-
-    def initialize(entity:, project: nil, id:, opts: {})
-      @entity = entity
-      @project = project
-      @id = id
-      @opts = opts
-    end
-
-    def url
-      url_method = "#{@entity}_url"
-      raise NotImplementedError.new("No Light URL builder defined for #{@entity}") unless respond_to?(url_method)
-
-      public_send(url_method)
-    end
-
-    def issue_url
-      namespace_project_issue_url({
-                                    namespace_id: @project.namespace,
-                                    project_id: @project,
-                                    id: @id
-                                  }.merge!(@opts))
-    end
-
-    def user_avatar_url
-      User.find(@id).avatar_url
-    end
-
-    def commit_url
-      namespace_project_commit_url({
-                                     namespace_id: @project.namespace,
-                                     project_id: @project,
-                                     id: @id
-                                   }.merge!(@opts))
-    end
-
-    def merge_request_url
-      namespace_project_merge_request_url({
-                                            namespace_id: @project.namespace,
-                                            project_id: @project,
-                                            id: @id
-                                          }.merge!(@opts))
-    end
-
-    def branch_url
-      namespace_project_commit_url(@project.namespace, @project, @id)
-    end
-
-    def user_url
-      Gitlab::Routing.url_helpers.user_url(@id)
-    end
-
-    def build_url
-      namespace_project_build_url(@project.namespace, @project, @id)
-    end
-  end
-end

spec/lib/gitlab/url_builder_spec.rb

 require 'spec_helper'
 
-describe Gitlab::LightUrlBuilder, lib: true do
+describe Gitlab::UrlBuilder, lib: true do
+  describe '.build' do
     context 'when passing a Commit' do
       it 'returns a proper URL' do
         commit = build_stubbed(:commit)
 
-      url = described_class.build(entity: :commit, project: commit.project, id: commit.id)
+        url = described_class.build(commit)
 
         expect(url).to eq "#{Settings.gitlab['url']}/#{commit.project.path_with_namespace}/commit/#{commit.id}"
       end
@@ -15,7 +16,7 @@ describe Gitlab::LightUrlBuilder, lib: true do
       it 'returns a proper URL' do
         issue = build_stubbed(:issue, iid: 42)
 
-      url = described_class.build(entity: :issue, project: issue.project, id: issue.iid)
+        url = described_class.build(issue)
 
         expect(url).to eq "#{Settings.gitlab['url']}/#{issue.project.path_with_namespace}/issues/#{issue.iid}"
       end
@@ -25,50 +26,94 @@ describe Gitlab::LightUrlBuilder, lib: true do
       it 'returns a proper URL' do
         merge_request = build_stubbed(:merge_request, iid: 42)
 
-      url = described_class.build(entity: :merge_request, project: merge_request.project, id: merge_request.iid)
+        url = described_class.build(merge_request)
 
         expect(url).to eq "#{Settings.gitlab['url']}/#{merge_request.project.path_with_namespace}/merge_requests/#{merge_request.iid}"
       end
     end
 
-  context 'when passing a build' do
+    context 'when passing a Note' do
+      context 'on a Commit' do
         it 'returns a proper URL' do
-      build = build_stubbed(:ci_build, project: build_stubbed(:empty_project))
+          note = build_stubbed(:note_on_commit)
 
-      url = described_class.build(entity: :build, project: build.project, id: build.id)
+          url = described_class.build(note)
 
-      expect(url).to eq "#{Settings.gitlab['url']}/#{build.project.path_with_namespace}/builds/#{build.id}"
+          expect(url).to eq "#{Settings.gitlab['url']}/#{note.project.path_with_namespace}/commit/#{note.commit_id}#note_#{note.id}"
         end
       end
 
-  context 'when passing a branch' do
+      context 'on a Commit Diff' do
         it 'returns a proper URL' do
-      branch = 'branch_name'
-      project = build_stubbed(:empty_project)
+          note = build_stubbed(:diff_note_on_commit)
 
-      url = described_class.build(entity: :branch, project: project, id: branch)
+          url = described_class.build(note)
 
-      expect(url).to eq "#{Settings.gitlab['url']}/#{project.path_with_namespace}/commits/#{branch}"
+          expect(url).to eq "#{Settings.gitlab['url']}/#{note.project.path_with_namespace}/commit/#{note.commit_id}#note_#{note.id}"
         end
       end
 
-  context 'on a User' do
+      context 'on an Issue' do
         it 'returns a proper URL' do
-      user = build_stubbed(:user)
+          issue = create(:issue, iid: 42)
+          note = build_stubbed(:note_on_issue, noteable: issue)
 
-      url = described_class.build(entity: :user, id: user.username)
+          url = described_class.build(note)
 
-      expect(url).to eq "#{Settings.gitlab['url']}/#{user.username}"
+          expect(url).to eq "#{Settings.gitlab['url']}/#{issue.project.path_with_namespace}/issues/#{issue.iid}#note_#{note.id}"
         end
       end
 
-  context 'on a user avatar' do
+      context 'on a MergeRequest' do
         it 'returns a proper URL' do
-      user = create(:user)
+          merge_request = create(:merge_request, iid: 42)
+          note = build_stubbed(:note_on_merge_request, noteable: merge_request)
 
-      url = described_class.build(entity: :user_avatar, id: user.id)
+          url = described_class.build(note)
 
-      expect(url).to eq user.avatar_url
+          expect(url).to eq "#{Settings.gitlab['url']}/#{merge_request.project.path_with_namespace}/merge_requests/#{merge_request.iid}#note_#{note.id}"
+        end
+      end
+
+      context 'on a MergeRequest Diff' do
+        it 'returns a proper URL' do
+          merge_request = create(:merge_request, iid: 42)
+          note = build_stubbed(:diff_note_on_merge_request, noteable: merge_request)
+
+          url = described_class.build(note)
+
+          expect(url).to eq "#{Settings.gitlab['url']}/#{merge_request.project.path_with_namespace}/merge_requests/#{merge_request.iid}#note_#{note.id}"
+        end
+      end
+
+      context 'on a ProjectSnippet' do
+        it 'returns a proper URL' do
+          project_snippet = create(:project_snippet)
+          note = build_stubbed(:note_on_project_snippet, noteable: project_snippet)
+
+          url = described_class.build(note)
+
+          expect(url).to eq "#{Settings.gitlab['url']}/#{project_snippet.project.path_with_namespace}/snippets/#{note.noteable_id}#note_#{note.id}"
+        end
+      end
+
+      context 'on another object' do
+        it 'returns a proper URL' do
+          project = build_stubbed(:project)
+
+          expect { described_class.build(project) }.
+            to raise_error(NotImplementedError, 'No URL builder defined for Project')
+        end
+      end
+    end
+
+    context 'when passing a WikiPage' do
+      it 'returns a proper URL' do
+        wiki_page = build(:wiki_page)
+        url = described_class.build(wiki_page)
+
+        expect(url).to eq "#{Gitlab.config.gitlab.url}#{wiki_page.wiki.wiki_base_path}/#{wiki_page.slug}"
+      end
     end
   end
 end