Improve helpers in pipeline model specs

1 files changed, 3 insertions, 3 deletions

diff --git a/spec/models/ci/pipeline_spec.rb b/spec/models/ci/pipeline_spec.rb
index edefd258c06..675be77eafd 100644
--- a/spec/models/ci/pipeline_spec.rb
+++ b/spec/models/ci/pipeline_spec.rb
@@ -328,7 +328,7 @@ describe Ci::Pipeline, models: true do
 
       context 'when pipeline becomes blocked' do
         let!(:build) { create_build('build:1') }
-        let!(:action) { create(:ci_build, :manual, pipeline: pipeline) }
+        let!(:action) { create_build('manual:action', :manual) }
 
         before do
           travel_to(current + 1.minute) do
@@ -407,8 +407,8 @@ describe Ci::Pipeline, models: true do
       end
     end
 
-    def create_build(name, queued_at: current, started_from: 0, **opts)
-      create(:ci_build,
+    def create_build(name, *traits, queued_at: current, started_from: 0, **opts)
+      create(:ci_build, *traits,
              name: name,
              pipeline: pipeline,
              queued_at: queued_at,