2017-08-16 13:04:41 +00:00
|
|
|
module GraphqlHelpers
|
2018-05-23 07:55:14 +00:00
|
|
|
# makes an underscored string look like a fieldname
|
|
|
|
# "merge_request" => "mergeRequest"
|
|
|
|
def self.fieldnamerize(underscored_field_name)
|
|
|
|
graphql_field_name = underscored_field_name.to_s.camelize
|
|
|
|
graphql_field_name[0] = graphql_field_name[0].downcase
|
|
|
|
|
|
|
|
graphql_field_name
|
|
|
|
end
|
|
|
|
|
2017-08-16 13:04:41 +00:00
|
|
|
# Run a loader's named resolver
|
2018-05-23 07:55:14 +00:00
|
|
|
def resolve(resolver_class, obj: nil, args: {}, ctx: {})
|
|
|
|
resolver_class.new(object: obj, context: ctx).resolve(args)
|
2017-08-16 13:04:41 +00:00
|
|
|
end
|
|
|
|
|
2018-02-23 15:36:40 +00:00
|
|
|
# Runs a block inside a BatchLoader::Executor wrapper
|
2017-08-16 13:04:41 +00:00
|
|
|
def batch(max_queries: nil, &blk)
|
|
|
|
wrapper = proc do
|
2018-02-23 15:36:40 +00:00
|
|
|
begin
|
|
|
|
BatchLoader::Executor.ensure_current
|
2018-05-21 07:52:24 +00:00
|
|
|
yield
|
2018-02-23 15:36:40 +00:00
|
|
|
ensure
|
|
|
|
BatchLoader::Executor.clear_current
|
2017-08-16 13:04:41 +00:00
|
|
|
end
|
|
|
|
end
|
2018-05-21 07:52:24 +00:00
|
|
|
|
2017-08-16 13:04:41 +00:00
|
|
|
if max_queries
|
|
|
|
result = nil
|
|
|
|
expect { result = wrapper.call }.not_to exceed_query_limit(max_queries)
|
|
|
|
result
|
|
|
|
else
|
|
|
|
wrapper.call
|
|
|
|
end
|
|
|
|
end
|
2018-05-21 07:52:24 +00:00
|
|
|
|
2018-05-23 07:55:14 +00:00
|
|
|
def graphql_query_for(name, attributes = {}, fields = nil)
|
|
|
|
fields ||= all_graphql_fields_for(name.classify)
|
|
|
|
attributes = attributes_to_graphql(attributes)
|
|
|
|
<<~QUERY
|
|
|
|
{
|
|
|
|
#{name}(#{attributes}) {
|
|
|
|
#{fields}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
QUERY
|
|
|
|
end
|
|
|
|
|
|
|
|
def all_graphql_fields_for(class_name)
|
|
|
|
type = GitlabSchema.types[class_name.to_s]
|
2018-05-21 07:52:24 +00:00
|
|
|
return "" unless type
|
|
|
|
|
|
|
|
type.fields.map do |name, field|
|
|
|
|
if scalar?(field)
|
|
|
|
name
|
|
|
|
else
|
|
|
|
"#{name} { #{all_graphql_fields_for(field_type(field))} }"
|
|
|
|
end
|
|
|
|
end.join("\n")
|
|
|
|
end
|
|
|
|
|
2018-05-23 07:55:14 +00:00
|
|
|
def attributes_to_graphql(attributes)
|
|
|
|
attributes.map do |name, value|
|
|
|
|
"#{GraphqlHelpers.fieldnamerize(name.to_s)}: \"#{value}\""
|
|
|
|
end.join(", ")
|
|
|
|
end
|
|
|
|
|
|
|
|
def post_graphql(query, current_user: nil)
|
|
|
|
post api('/', current_user, version: 'graphql'), query: query
|
|
|
|
end
|
|
|
|
|
|
|
|
def graphql_data
|
|
|
|
json_response['data']
|
|
|
|
end
|
|
|
|
|
|
|
|
def graphql_errors
|
|
|
|
json_response['data']
|
2018-05-21 07:52:24 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def scalar?(field)
|
|
|
|
field_type(field).kind.scalar?
|
|
|
|
end
|
|
|
|
|
|
|
|
def field_type(field)
|
|
|
|
if field.type.respond_to?(:of_type)
|
|
|
|
field.type.of_type
|
|
|
|
else
|
|
|
|
field.type
|
|
|
|
end
|
|
|
|
end
|
2017-08-16 13:04:41 +00:00
|
|
|
end
|