1 files changed, 20 insertions, 0 deletions
diff --git a/spec/factories/analytics/cycle_analytics/stages.rb b/spec/factories/analytics/cycle_analytics/stages.rb
new file mode 100644
index 00000000000..4f6f38f6f33
--- /dev/null
+++ b/spec/factories/analytics/cycle_analytics/stages.rb
@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :cycle_analytics_stage, class: 'Analytics::CycleAnalytics::Stage' do
+    transient do
+      project { nil }
+    end
+
+    sequence(:name) { |n| "Stage ##{n}" }
+    start_event_identifier { Gitlab::Analytics::CycleAnalytics::StageEvents::MergeRequestCreated.identifier }
+    end_event_identifier { Gitlab::Analytics::CycleAnalytics::StageEvents::MergeRequestMerged.identifier }
+
+    namespace { association(:group) }
+    value_stream { association(:cycle_analytics_value_stream, namespace: namespace) }
+
+    after(:build) do |stage, evaluator|
+      stage.namespace = evaluator.project.reload.project_namespace if evaluator.project
+    end
+  end
+end