Testing an implementation

85071271 · Laura Montemayor · 1d01381a · 85071271 · 85071271
--- a/spec/factories/ci/processable.rb
+++ b/spec/factories/ci/processable.rb
@@ -17,15 +17,21 @@

    transient do
      stage { 'test' }
+      explicit_stage { false }
    end

    after(:build) do |processable, evaluator|
      processable.stage = evaluator.stage
-      processable.ci_stage = build(:ci_stage)
+
+      if processable.ci_stage
+        evaluator.explicit_stage = true
+      else
+        processable.ci_stage = build(:ci_stage)
+      end
    end

    before(:create) do |processable, evaluator|
-      next if processable.ci_stage
+      next if evaluator.explicit_stage

      if ci_stage = processable.pipeline.stages.find_by(name: evaluator.stage)
        processable.ci_stage = ci_stage

--- a/spec/factories/commit_statuses.rb
+++ b/spec/factories/commit_statuses.rb
@@ -59,6 +59,18 @@
      status { 'scheduled' }
    end

+    # rubocop:disable RSpec/FactoryBot/StrategyInCallback
+    before(:create) do |commit_status, evaluator|
+      commit_status.ci_stage = if ci_stage = commit_status.pipeline.stages.find_by(name: evaluator.stage)
+                                 ci_stage
+                               else
+                                 create(:ci_stage, pipeline: commit_status.pipeline,
+                                   project: commit_status.project || evaluator.project,
+                                   name: evaluator.stage, position: evaluator.stage_idx, status: 'created')
+                               end
+    end
+    # rubocop:enable RSpec/FactoryBot/StrategyInCallback
+
    after(:build) do |build, evaluator|
      build.project ||= build.pipeline.project
    end