Add specs for cluster counters

1 files changed, 21 insertions, 3 deletions

diff --git a/spec/controllers/projects/clusters_controller_spec.rb b/spec/controllers/projects/clusters_controller_spec.rb
index 3be3054f59f..3fe90cec651 100644
--- a/spec/controllers/projects/clusters_controller_spec.rb
+++ b/spec/controllers/projects/clusters_controller_spec.rb
@@ -14,15 +14,26 @@ describe Projects::ClustersController do
       end
 
       context 'when project has one or more clusters' do
-        let(:cluster) { create(:cluster, :project, :provided_by_gcp) }
-        let(:project) { cluster.project }
+        let(:project) { create(:project) }
+        let(:clusters) { create_list(:cluster, 2, :provided_by_gcp, projects: [project]) }
+
+        before do
+          clusters.last.enabled = false
+        end
 
         it 'lists available clusters' do
           go
 
           expect(response).to have_gitlab_http_status(:ok)
           expect(response).to render_template(:index)
-          expect(assigns(:clusters)).to eq([cluster])
+          expect(assigns(:clusters)).to eq(clusters)
+        end
+
+        it 'assigns counters to correct values' do
+          go
+
+          expect(assigns(:active_count)).to eq(project.clusters.enabled.count)
+          expect(assigns(:inactive_count)).to eq(project.clusters.disabled.count)
         end
       end
 
@@ -36,6 +47,13 @@ describe Projects::ClustersController do
           expect(response).to render_template(:index)
           expect(assigns(:clusters)).to eq([])
         end
+
+        it 'assigns counters to zero' do
+          go
+
+          expect(assigns(:active_count)).to eq(0)
+          expect(assigns(:inactive_count)).to eq(0)
+        end
       end
     end