Add a minimal GraphQL API

1 files changed, 29 insertions, 0 deletions

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