1 files changed, 61 insertions, 0 deletions
diff --git a/spec/graphql/resolvers/merge_requests_resolver_spec.rb b/spec/graphql/resolvers/merge_requests_resolver_spec.rb
new file mode 100644
index 00000000000..ab3c426b2cd
--- /dev/null
+++ b/spec/graphql/resolvers/merge_requests_resolver_spec.rb
@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe Resolvers::MergeRequestsResolver do
+  include GraphqlHelpers
+
+  set(:project) { create(:project, :repository) }
+  set(:merge_request_1) { create(:merge_request, :simple, source_project: project, target_project: project) }
+  set(:merge_request_2) { create(:merge_request, :rebased, source_project: project, target_project: project) }
+
+  set(:other_project) { create(:project, :repository) }
+  set(:other_merge_request) { create(:merge_request, source_project: other_project, target_project: other_project) }
+
+  let(:iid_1) { merge_request_1.iid }
+  let(:iid_2) { merge_request_2.iid }
+
+  let(:other_iid) { other_merge_request.iid }
+
+  describe '#resolve' do
+    it 'batch-resolves by target project full path and individual IID' do
+      result = batch(max_queries: 2) do
+        resolve_mr(project, iid: iid_1) + resolve_mr(project, iid: iid_2)
+      end
+
+      expect(result).to contain_exactly(merge_request_1, merge_request_2)
+    end
+
+    it 'batch-resolves by target project full path and IIDS' do
+      result = batch(max_queries: 2) do
+        resolve_mr(project, iids: [iid_1, iid_2])
+      end
+
+      expect(result).to contain_exactly(merge_request_1, merge_request_2)
+    end
+
+    it 'can batch-resolve merge requests from different projects' do
+      result = batch(max_queries: 3) do
+        resolve_mr(project, iid: iid_1) +
+          resolve_mr(project, iid: iid_2) +
+          resolve_mr(other_project, iid: other_iid)
+      end
+
+      expect(result).to contain_exactly(merge_request_1, merge_request_2, other_merge_request)
+    end
+
+    it 'resolves an unknown iid to be empty' do
+      result = batch { resolve_mr(project, iid: -1) }
+
+      expect(result).to be_empty
+    end
+
+    it 'resolves empty iids to be empty' do
+      result = batch { resolve_mr(project, iids: []) }
+
+      expect(result).to be_empty
+    end
+  end
+
+  def resolve_mr(project, args)
+    resolve(described_class, obj: project, args: args)
+  end
+end