148 lines
4.5 KiB
Ruby
148 lines
4.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module CycleAnalyticsHelpers
|
|
include GitHelpers
|
|
|
|
def create_commit_referencing_issue(issue, branch_name: generate(:branch))
|
|
project.repository.add_branch(user, branch_name, 'master')
|
|
create_commit("Commit for ##{issue.iid}", issue.project, user, branch_name)
|
|
end
|
|
|
|
def create_commit(message, project, user, branch_name, count: 1, commit_time: nil, skip_push_handler: false)
|
|
repository = project.repository
|
|
oldrev = repository.commit(branch_name)&.sha || Gitlab::Git::BLANK_SHA
|
|
|
|
if Timecop.frozen?
|
|
mock_gitaly_multi_action_dates(repository, commit_time)
|
|
end
|
|
|
|
commit_shas = Array.new(count) do |index|
|
|
commit_sha = repository.create_file(user, generate(:branch), "content", message: message, branch_name: branch_name)
|
|
repository.commit(commit_sha)
|
|
|
|
commit_sha
|
|
end
|
|
|
|
return if skip_push_handler
|
|
|
|
Git::BranchPushService.new(
|
|
project,
|
|
user,
|
|
change: {
|
|
oldrev: oldrev,
|
|
newrev: commit_shas.last,
|
|
ref: 'refs/heads/master'
|
|
}
|
|
).execute
|
|
end
|
|
|
|
def create_cycle(user, project, issue, mr, milestone, pipeline)
|
|
issue.update!(milestone: milestone)
|
|
pipeline.run
|
|
|
|
ci_build = create(:ci_build, pipeline: pipeline, status: :success, author: user)
|
|
|
|
merge_merge_requests_closing_issue(user, project, issue)
|
|
ProcessCommitWorker.new.perform(project.id, user.id, mr.commits.last.to_hash)
|
|
|
|
ci_build
|
|
end
|
|
|
|
def create_merge_request_closing_issue(user, project, issue, message: nil, source_branch: nil, commit_message: 'commit message')
|
|
if !source_branch || project.repository.commit(source_branch).blank?
|
|
source_branch = generate(:branch)
|
|
project.repository.add_branch(user, source_branch, 'master')
|
|
end
|
|
|
|
# Cycle analytic specs often test with frozen times, which causes metrics to be
|
|
# pinned to the current time. For example, in the plan stage, we assume that an issue
|
|
# milestone has been created before any code has been written. We add a second
|
|
# to ensure that the plan time is positive.
|
|
create_commit(commit_message, project, user, source_branch, commit_time: Time.now + 1.second, skip_push_handler: true)
|
|
|
|
opts = {
|
|
title: 'Awesome merge_request',
|
|
description: message || "Fixes #{issue.to_reference}",
|
|
source_branch: source_branch,
|
|
target_branch: 'master'
|
|
}
|
|
|
|
mr = MergeRequests::CreateService.new(project, user, opts).execute
|
|
NewMergeRequestWorker.new.perform(mr, user)
|
|
mr
|
|
end
|
|
|
|
def merge_merge_requests_closing_issue(user, project, issue)
|
|
merge_requests = Issues::ReferencedMergeRequestsService
|
|
.new(project, user)
|
|
.closed_by_merge_requests(issue)
|
|
|
|
merge_requests.each { |merge_request| MergeRequests::MergeService.new(project, user, sha: merge_request.diff_head_sha).execute(merge_request) }
|
|
end
|
|
|
|
def deploy_master(user, project, environment: 'production')
|
|
dummy_job =
|
|
case environment
|
|
when 'production'
|
|
dummy_production_job(user, project)
|
|
when 'staging'
|
|
dummy_staging_job(user, project)
|
|
else
|
|
raise ArgumentError
|
|
end
|
|
|
|
dummy_job.success! # State machine automatically update associated deployment/environment record
|
|
end
|
|
|
|
def dummy_production_job(user, project)
|
|
new_dummy_job(user, project, 'production')
|
|
end
|
|
|
|
def dummy_staging_job(user, project)
|
|
new_dummy_job(user, project, 'staging')
|
|
end
|
|
|
|
def dummy_pipeline(project)
|
|
create(:ci_pipeline,
|
|
sha: project.repository.commit('master').sha,
|
|
ref: 'master',
|
|
source: :push,
|
|
project: project,
|
|
protected: false)
|
|
end
|
|
|
|
def new_dummy_job(user, project, environment)
|
|
create(:ci_build,
|
|
:with_deployment,
|
|
project: project,
|
|
user: user,
|
|
environment: environment,
|
|
ref: 'master',
|
|
tag: false,
|
|
name: 'dummy',
|
|
stage: 'dummy',
|
|
pipeline: dummy_pipeline(project),
|
|
protected: false)
|
|
end
|
|
|
|
def mock_gitaly_multi_action_dates(repository, commit_time)
|
|
allow(repository.raw).to receive(:multi_action).and_wrap_original do |m, *args|
|
|
new_date = commit_time || Time.now
|
|
branch_update = m.call(*args)
|
|
|
|
if branch_update.newrev
|
|
_, opts = args
|
|
|
|
commit = rugged_repo(repository).rev_parse(branch_update.newrev)
|
|
|
|
branch_update.newrev = commit.amend(
|
|
update_ref: "#{Gitlab::Git::BRANCH_REF_PREFIX}#{opts[:branch_name]}",
|
|
author: commit.author.merge(time: new_date),
|
|
committer: commit.committer.merge(time: new_date)
|
|
)
|
|
end
|
|
|
|
branch_update
|
|
end
|
|
end
|
|
end
|