Commit f9c8822a authored by Steve Azzopardi's avatar Steve Azzopardi

Create `get_build` for project model

Inside of `Projects::ArtifactsController` and
`Projects::BuildArtifactsController` we fetching the build by id using
active record directly which violates `CodeReuse/ActiveRecord` rubocop
rule. Create `get_build` inside of `project` model which does the same
thing.
parent 7ac32ae2
...@@ -85,11 +85,9 @@ class Projects::ArtifactsController < Projects::ApplicationController ...@@ -85,11 +85,9 @@ class Projects::ArtifactsController < Projects::ApplicationController
end end
end end
# rubocop: disable CodeReuse/ActiveRecord
def build_from_id def build_from_id
project.builds.find_by(id: params[:job_id]) if params[:job_id] project.get_build(params[:job_id]) if params[:job_id]
end end
# rubocop: enable CodeReuse/ActiveRecord
def build_from_ref def build_from_ref
return unless @ref_name return unless @ref_name
......
...@@ -44,11 +44,9 @@ class Projects::BuildArtifactsController < Projects::ApplicationController ...@@ -44,11 +44,9 @@ class Projects::BuildArtifactsController < Projects::ApplicationController
@job ||= job_from_id || job_from_ref @job ||= job_from_id || job_from_ref
end end
# rubocop: disable CodeReuse/ActiveRecord
def job_from_id def job_from_id
project.builds.find_by(id: params[:build_id]) if params[:build_id] project.get_build(params[:build_id]) if params[:build_id]
end end
# rubocop: enable CodeReuse/ActiveRecord
def job_from_ref def job_from_ref
return unless @ref_name return unless @ref_name
......
...@@ -652,7 +652,11 @@ class Project < ActiveRecord::Base ...@@ -652,7 +652,11 @@ class Project < ActiveRecord::Base
end end
def latest_successful_build_for!(job_name, ref = default_branch) def latest_successful_build_for!(job_name, ref = default_branch)
latest_successful_build_for(job_name, ref) or raise ActiveRecord::RecordNotFound.new("Couldn't find job #{job_name}", job_name) latest_successful_build_for(job_name, ref) || raise(ActiveRecord::RecordNotFound.new("Couldn't find job #{job_name}"))
end
def get_build(id)
builds.find_by(id: id)
end end
def merge_base_commit(first_commit_id, second_commit_id) def merge_base_commit(first_commit_id, second_commit_id)
......
...@@ -2000,6 +2000,29 @@ describe Project do ...@@ -2000,6 +2000,29 @@ describe Project do
end end
end end
describe '#get_build' do
let(:project) { create(:project, :repository) }
let(:ci_pipeline) { create(:ci_pipeline, project: project) }
context 'when build exists' do
context 'build is associated with project' do
let(:build) { create(:ci_build, :success, pipeline: ci_pipeline) }
it { expect(project.get_build(build.id)).to eq(build) }
end
context 'build is not associated with project' do
let(:build) { create(:ci_build, :success) }
it { expect(project.get_build(build.id)).to be_nil }
end
end
context 'build does not exists' do
it { expect(project.get_build(rand 100)).to be_nil }
end
end
describe '#import_status' do describe '#import_status' do
context 'with import_state' do context 'with import_state' do
it 'returns the right status' do it 'returns the right status' do
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment