1 files changed, 10 insertions, 2 deletions
diff --git a/spec/lib/gitlab/analytics/cycle_analytics/base_query_builder_spec.rb b/spec/lib/gitlab/analytics/cycle_analytics/base_query_builder_spec.rb
index 271022e7c55..59b2bacea50 100644
--- a/spec/lib/gitlab/analytics/cycle_analytics/base_query_builder_spec.rb
+++ b/spec/lib/gitlab/analytics/cycle_analytics/base_query_builder_spec.rb
@@ -10,10 +10,10 @@ RSpec.describe Gitlab::Analytics::CycleAnalytics::BaseQueryBuilder do
 
   let(:params) { { current_user: user } }
   let(:records) do
-    stage = build(:cycle_analytics_project_stage, {
+    stage = build(:cycle_analytics_stage, {
       start_event_identifier: :merge_request_created,
       end_event_identifier: :merge_request_merged,
-      project: project
+      namespace: project.reload.project_namespace
     })
     described_class.new(stage: stage, params: params).build.to_a
   end
@@ -25,6 +25,14 @@ RSpec.describe Gitlab::Analytics::CycleAnalytics::BaseQueryBuilder do
     freeze_time
   end
 
+  context 'when an unknown parent class is given' do
+    it 'raises error' do
+      stage = instance_double('Analytics::CycleAnalytics::Stage', parent: Issue.new)
+
+      expect { described_class.new(stage: stage) }.to raise_error(/unknown parent_class: Issue/)
+    end
+  end
+
   describe 'date range parameters' do
     context 'when filters by only the `from` parameter' do
       before do