Add a test for `ci_environments_status` Gitaly N+1

spec/controllers/projects/merge_requests_controller_spec.rb

@@ -870,22 +870,19 @@ describe Projects::MergeRequestsController do
         end
       end
 
-      # we're trying to reduce the overall number of queries for this method.
-      # set a hard limit for now. https://gitlab.com/gitlab-org/gitlab-ce/issues/52287
-      it 'keeps queries in check' do
-        control_count = ActiveRecord::QueryRecorder.new { get_ci_environments_status }.count
-
-        expect(control_count).to be <= 137
-      end
+      context 'when multiple environments with deployments are present' do
+        let(:another_environment) { create(:environment, project: forked) }
 
-      it 'has no N+1 issues for environments', :request_store, retry: 0 do
+        it 'has no N+1 SQL issues for environments', :request_store, retry: 0 do
           # First run to insert test data from lets, which does take up some 30 queries
           get_ci_environments_status
 
           control_count = ActiveRecord::QueryRecorder.new(skip_cached: false) { get_ci_environments_status }.count
 
-        environment2 = create(:environment, project: forked)
-        create(:deployment, :succeed, environment: environment2, sha: sha, ref: 'master', deployable: build)
+          create(:deployment, :succeed, environment: another_environment,
+                                        sha: sha,
+                                        ref: 'master',
+                                        deployable: build)
 
           # TODO address the last 11 queries
           # See https://gitlab.com/gitlab-org/gitlab-ce/issues/63952 (5 queries)
@@ -894,6 +891,27 @@ describe Projects::MergeRequestsController do
           expect { get_ci_environments_status }.not_to exceed_all_query_limit(control_count + leeway)
         end
 
+        it 'has no N+1 Gitaly requests for deployments', :request_store do
+          expect(merge_request).to be_present
+
+          create(:deployment, :succeed, environment: another_environment,
+                                        sha: sha,
+                                        ref: 'master',
+                                        deployable: build)
+
+          expect { get_ci_environments_status }
+            .not_to change { Gitlab::GitalyClient.get_request_count }
+        end
+      end
+
+      # we're trying to reduce the overall number of queries for this method.
+      # set a hard limit for now. https://gitlab.com/gitlab-org/gitlab-ce/issues/52287
+      it 'keeps queries in check' do
+        control_count = ActiveRecord::QueryRecorder.new { get_ci_environments_status }.count
+
+        expect(control_count).to be <= 137
+      end
+
       def get_ci_environments_status(extra_params = {})
         params = {
           namespace_id: merge_request.project.namespace.to_param,