Create labels_as_hash_ concern, modify params for methodccr/48930_extract_logic_to_concern

3 files changed, 35 insertions, 26 deletions

diff --git a/app/controllers/concerns/labels_as_hash.rb b/app/controllers/concerns/labels_as_hash.rb
new file mode 100644
index 00000000000..1171aa9cf44
--- /dev/null
+++ b/app/controllers/concerns/labels_as_hash.rb
@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+module LabelsAsHash
+  extend ActiveSupport::Concern
+
+  def labels_as_hash(target = nil, params = {})
+    available_labels = LabelsFinder.new(
+      current_user,
+      params
+    ).execute
+
+    label_hashes = available_labels.as_json(only: [:title, :color])
+
+    if target&.respond_to?(:labels)
+      already_set_labels = available_labels & target.labels
+      if already_set_labels.present?
+        titles = already_set_labels.map(&:title)
+        label_hashes.each do |hash|
+          if titles.include?(hash['title'])
+            hash[:set] = true
+          end
+        end
+      end
+    end
+
+    label_hashes
+  end
+end
diff --git a/app/services/projects/autocomplete_service.rb b/app/services/projects/autocomplete_service.rb
index 5286b92ab6b..7b747171d9c 100644
--- a/app/services/projects/autocomplete_service.rb
+++ b/app/services/projects/autocomplete_service.rb
@@ -2,6 +2,7 @@
 
 module Projects
   class AutocompleteService < BaseService
+    include LabelsAsHash
     def issues
       IssuesFinder.new(current_user, project_id: project.id, state: 'opened').execute.select([:iid, :title])
     end
@@ -22,34 +23,14 @@ module Projects
       MergeRequestsFinder.new(current_user, project_id: project.id, state: 'opened').execute.select([:iid, :title])
     end
 
-    def labels_as_hash(target = nil)
-      available_labels = LabelsFinder.new(
-        current_user,
-        project_id: project.id,
-        include_ancestor_groups: true
-      ).execute
-
-      label_hashes = available_labels.as_json(only: [:title, :color])
-
-      if target&.respond_to?(:labels)
-        already_set_labels = available_labels & target.labels
-        if already_set_labels.present?
-          titles = already_set_labels.map(&:title)
-          label_hashes.each do |hash|
-            if titles.include?(hash['title'])
-              hash[:set] = true
-            end
-          end
-        end
-      end
-
-      label_hashes
-    end
-
     def commands(noteable, type)
       return [] unless noteable
 
       QuickActions::InterpretService.new(project, current_user).available_commands(noteable)
     end
+
+    def labels_as_hash(target)
+      super(target, project_id: project.id, include_ancestor_groups: true)
+    end
   end
 end
diff --git a/spec/services/projects/autocomplete_service_spec.rb b/spec/services/projects/autocomplete_service_spec.rb
index e98df375d48..373fe7cb7dd 100644
--- a/spec/services/projects/autocomplete_service_spec.rb
+++ b/spec/services/projects/autocomplete_service_spec.rb
@@ -148,7 +148,7 @@ describe Projects::AutocompleteService do
     let!(:label1) { create(:label, project: project) }
     let!(:label2) { create(:label, project: project) }
     let!(:sub_group_label) { create(:group_label, group: sub_group) }
-    let!(:parent_group_label) { create(:group_label, group: group.parent) }
+    let!(:parent_group_label) { create(:group_label, group: group.parent, group_id: group.id) }
 
     before do
       create(:group_member, group: group, user: user)
@@ -156,7 +156,7 @@ describe Projects::AutocompleteService do
 
     it 'returns labels from project and ancestor groups' do
       service = described_class.new(project, user)
-      results = service.labels_as_hash
+      results = service.labels_as_hash(nil)
       expected_labels = [label1, label2, parent_group_label]
 
       expect_labels_to_equal(results, expected_labels)