1 files changed, 21 insertions, 2 deletions
diff --git a/spec/controllers/projects/clusters_controller_spec.rb b/spec/controllers/projects/clusters_controller_spec.rb
index 35cbab57037..8ac72df5d20 100644
--- a/spec/controllers/projects/clusters_controller_spec.rb
+++ b/spec/controllers/projects/clusters_controller_spec.rb
@@ -79,8 +79,12 @@ describe Projects::ClustersController do
   end
 
   describe 'GET new' do
-    def go
-      get :new, params: { namespace_id: project.namespace, project_id: project }
+    def go(provider: 'gke')
+      get :new, params: {
+        namespace_id: project.namespace,
+        project_id: project,
+        provider: provider
+      }
     end
 
     describe 'functionality for new cluster' do
@@ -91,6 +95,7 @@ describe Projects::ClustersController do
         end
 
         before do
+          stub_feature_flags(create_eks_clusters: false)
           allow(SecureRandom).to receive(:hex).and_return(key)
         end
 
@@ -100,6 +105,20 @@ describe Projects::ClustersController do
           expect(assigns(:authorize_url)).to include(key)
           expect(session[session_key_for_redirect_uri]).to eq(new_project_cluster_path(project))
         end
+
+        context 'when create_eks_clusters feature flag is enabled' do
+          before do
+            stub_feature_flags(create_eks_clusters: true)
+          end
+
+          context 'when selected provider is gke and no valid gcp token exists' do
+            it 'redirects to gcp authorize_url' do
+              go
+
+              expect(response).to redirect_to(assigns(:authorize_url))
+            end
+          end
+        end
       end
 
       context 'when omniauth has not configured' do