From 2537b0c1e0619f03799092ed59c51692baa2cf24 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9my=20Coutable?= <remy@rymai.me>
Date: Wed, 17 Apr 2019 12:23:06 +0200
Subject: Provide a new gitlab:seed:issues task
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Rémy Coutable <remy@rymai.me>
---
 lib/quality/seeders/issues.rb | 58 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 lib/quality/seeders/issues.rb

(limited to 'lib/quality')

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
-- 
cgit v1.2.3