Use ci_commits.ref (Pipeline#ref) to find builds

1 files changed, 3 insertions, 2 deletions

diff --git a/spec/models/build_spec.rb b/spec/models/build_spec.rb
index a57f0b6886c..53064138a50 100644
--- a/spec/models/build_spec.rb
+++ b/spec/models/build_spec.rb
@@ -5,7 +5,8 @@ describe Ci::Build, models: true do
 
   let(:pipeline) do
     create(:ci_pipeline, project: project,
-                         sha: project.commit.id)
+                         sha: project.commit.id,
+                         ref: 'fix')
   end
 
   let(:build) { create(:ci_build, pipeline: pipeline) }
@@ -698,7 +699,7 @@ describe Ci::Build, models: true do
     end
 
     it 'returns builds from ref' do
-      builds = project.latest_success_builds_for('HEAD')
+      builds = project.latest_success_builds_for('fix')
 
       expect(builds).to contain_exactly(build)
     end