Welcome to mirror list, hosted at ThFree Co, Russian Federation.

issues.rb « seeders « quality « lib - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 813ff0bf097dcd5487f477045e0793f59f8d4746 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
# frozen_string_literal: true

# rubocop:disable CodeReuse/ActiveRecord
module Quality
  module Seeders
    class Issues
      DEFAULT_BACKFILL_WEEKS = 52
      DEFAULT_AVERAGE_ISSUES_PER_WEEK = 20

      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)
        create_milestones!
        create_team_members!

        created_at = backfill_weeks.to_i.weeks.ago
        team = project.team.users
        created_issues_count = 0

        loop do
          rand(1..average_issues_per_week).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_id: project.milestones.sample&.id,
              assignee_ids: Array(team.pluck(:id).sample(rand(3))),
              due_date: rand(10).days.from_now,
              labels: labels.join(',')
            }.merge(additional_params)

            params[:closed_at] = params[:created_at] + rand(35).days if params[:state] == 'closed'
            create_result = ::Issues::CreateService.new(container: project, current_user: team.sample, params: params, perform_spam_check: false).execute_without_rate_limiting

            if create_result.success?
              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

      # Overriden on Quality::Seeders::Insights::Issues
      def additional_params
        {}
      end

      def create_team_members!
        3.times do |i|
          user = FactoryBot.create(
            :user,
            name: "I User#{i}",
            username: "i-user-#{i}-#{suffix}",
            email: "i-user-#{i}@#{suffix}.com"
          )

          # need owner access to allow changing Issue#created_at
          project.add_owner(user)
        end

        Sidekiq::Worker.skipping_transaction_check do
          AuthorizedProjectUpdate::ProjectRecalculateService.new(project).execute
        end

        # Refind object toreload ProjectTeam association which is memoized at Project model
        @project = Project.find(project.id)
      end

      def create_milestones!
        3.times do |i|
          params = {
            project: project,
            title: "Sprint #{i + suffix}",
            description: FFaker::Lorem.sentence,
            state: [:active, :closed].sample
          }

          FactoryBot.create(:milestone, **params)
        end
      end

      def suffix
        @suffix ||= Time.now.to_i
      end

      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