1 files changed, 17 insertions, 1 deletions
diff --git a/spec/models/concerns/deployment_platform_spec.rb b/spec/models/concerns/deployment_platform_spec.rb
index 2378f400540..e2fc8a5d127 100644
--- a/spec/models/concerns/deployment_platform_spec.rb
+++ b/spec/models/concerns/deployment_platform_spec.rb
@@ -5,7 +5,7 @@ require 'rails_helper'
 describe DeploymentPlatform do
   let(:project) { create(:project) }
 
-  describe '#deployment_platform' do
+  shared_examples '#deployment_platform' do
     subject { project.deployment_platform }
 
     context 'with no Kubernetes configuration on CI/CD, no Kubernetes Service' do
@@ -84,4 +84,20 @@ describe DeploymentPlatform do
       end
     end
   end
+
+  context 'legacy implementation' do
+    before do
+      stub_feature_flags(clusters_cte: false)
+    end
+
+    include_examples '#deployment_platform'
+  end
+
+  context 'CTE implementation' do
+    before do
+      stub_feature_flags(clusters_cte: true)
+    end
+
+    include_examples '#deployment_platform'
+  end
 end