Merge branch 'ce-11199-add-a-rake-task-to-seed-insights-data-for-a-given-project' into 'master'

Provide a new gitlab:seed:issues task

See merge request gitlab-org/gitlab-ce!27454

2 files changed, 92 insertions, 0 deletions

diff --git a/lib/quality/seeders/issues.rb b/lib/quality/seeders/issues.rb
new file mode 100644
index 00000000000..4c8cb6e97cc
--- /dev/null
+++ b/lib/quality/seeders/issues.rb
@@ -0,0 +1,58 @@
+# frozen_string_literal: true
+
+# rubocop:disable CodeReuse/ActiveRecord
+module Quality
+  module Seeders
+    class Issues
+      DEFAULT_BACKFILL_WEEKS = 52
+      DEFAULT_AVERAGE_ISSUES_PER_WEEK = 10
+
+      attr_reader :project, :user
+
+      def initialize(project:)
+        @project = project
+      end
+
+      def seed(backfill_weeks: DEFAULT_BACKFILL_WEEKS, average_issues_per_week: DEFAULT_AVERAGE_ISSUES_PER_WEEK)
+        created_at = backfill_weeks.to_i.weeks.ago
+        team = project.team.users
+        created_issues_count = 0
+
+        loop do
+          rand(average_issues_per_week * 2).times do
+            params = {
+              title: FFaker::Lorem.sentence(6),
+              description: FFaker::Lorem.sentence,
+              created_at: created_at + rand(6).days,
+              state: %w[opened closed].sample,
+              milestone: project.milestones.sample,
+              assignee_ids: Array(team.pluck(:id).sample(3)),
+              labels: labels.join(',')
+            }
+            issue = ::Issues::CreateService.new(project, team.sample, params).execute
+
+            if issue.persisted?
+              created_issues_count += 1
+              print '.' # rubocop:disable Rails/Output
+            end
+          end
+
+          created_at += 1.week
+
+          break if created_at > Time.now
+        end
+
+        created_issues_count
+      end
+
+      private
+
+      def labels
+        @labels_pool ||= project.labels.limit(rand(3)).pluck(:title).tap do |labels_array|
+          labels_array.concat(project.group.labels.limit(rand(3)).pluck(:title)) if project.group
+        end
+      end
+    end
+  end
+end
+# rubocop:enable CodeReuse/ActiveRecord
diff --git a/lib/tasks/gitlab/seed.rake b/lib/tasks/gitlab/seed.rake
new file mode 100644
index 00000000000..155ba979b36
--- /dev/null
+++ b/lib/tasks/gitlab/seed.rake
@@ -0,0 +1,34 @@
+namespace :gitlab do
+  namespace :seed do
+    desc "GitLab | Seed | Seeds issues"
+    task :issues, [:project_full_path] => :environment do |t, args|
+      projects =
+        if args.project_full_path
+          project = Project.find_by_full_path(args.project_full_path)
+
+          unless project
+            error_message = "Project '#{args.project_full_path}' does not exist!"
+            potential_projects = Project.search(args.project_full_path)
+
+            if potential_projects.present?
+              error_message += " Did you mean '#{potential_projects.first.full_path}'?"
+            end
+
+            puts error_message.color(:red)
+            exit 1
+          end
+
+          [project]
+        else
+          Project.find_each
+        end
+
+      projects.each do |project|
+        puts "\nSeeding issues for the '#{project.full_path}' project"
+        seeder = Quality::Seeders::Issues.new(project: project)
+        issues_created = seeder.seed(backfill_weeks: 5, average_issues_per_week: 2)
+        puts "\n#{issues_created} issues created!"
+      end
+    end
+  end
+end