From 9c6c17cbcdb8bf8185fc1b873dcfd08f723e4df5 Mon Sep 17 00:00:00 2001
From: Nick Thomas <nick@gitlab.com>
Date: Wed, 16 Aug 2017 14:04:41 +0100
Subject: Add a minimal GraphQL API

---
 spec/support/helpers/graphql_helpers.rb | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 spec/support/helpers/graphql_helpers.rb

(limited to 'spec/support/helpers')

diff --git a/spec/support/helpers/graphql_helpers.rb b/spec/support/helpers/graphql_helpers.rb
new file mode 100644
index 00000000000..5bb2cf9dd9e
--- /dev/null
+++ b/spec/support/helpers/graphql_helpers.rb
@@ -0,0 +1,29 @@
+module GraphqlHelpers
+  # Run a loader's named resolver
+  def resolve(kls, name, obj: nil, args: {}, ctx: {})
+    kls[name].call(obj, args, ctx)
+  end
+
+  # Runs a block inside a GraphQL::Batch wrapper
+  def batch(max_queries: nil, &blk)
+    wrapper = proc do
+      GraphQL::Batch.batch do
+        result = yield
+
+        if result.is_a?(Array)
+          Promise.all(result)
+        else
+          result
+        end
+      end
+    end
+
+    if max_queries
+      result = nil
+      expect { result = wrapper.call }.not_to exceed_query_limit(max_queries)
+      result
+    else
+      wrapper.call
+    end
+  end
+end
-- 
cgit v1.2.1