Merge pull request #17 from github/support-class-based-types

Support class-based schemas and types

.travis.yml

@@ -1,6 +1,6 @@
 language: ruby
 rvm:
-  - 2.2.2
+  - 2.3.8
   - ruby-head
 branches:
   only:

graphql-relay-walker.gemspec

@@ -9,9 +9,9 @@ Gem::Specification.new do |s|
   s.files += Dir.glob('lib/**/*.rb')
   s.homepage = 'https://github.com/github/graphql-relay-walker'
 
-  s.add_dependency 'graphql', '>= 1.5.6'
+  s.add_dependency 'graphql', '~> 1.8'
 
-  s.add_development_dependency 'graphql-client', '~> 0.2'
+  s.add_development_dependency 'graphql-client', '~> 0.15'
   s.add_development_dependency 'rake',           '~> 11.3'
   s.add_development_dependency 'rspec',          '~> 3.5'
 end

lib/graphql/relay/walker/query_builder.rb

@@ -70,7 +70,10 @@ module GraphQL::Relay::Walker
     def inline_fragment_ast(type, with_children: true)
       selections = []
       if with_children
-        type.all_fields.each do |field|
+        # Class-based types return all fields in `.fields`
+        all_fields = type.respond_to?(:all_fields) ? type.all_fields : type.fields.values
+        all_fields = all_fields.sort_by(&:graphql_name)
+        all_fields.each do |field|
           field_type = field.type.unwrap
           if node_field?(field) && include?(field_type)
             selections << node_field_ast(field)
@@ -88,7 +91,7 @@ module GraphQL::Relay::Walker
         nil
       else
         GraphQL::Language::Nodes::InlineFragment.new(
-          type: make_type_name_node(type.name),
+          type: make_type_name_node(type.graphql_name),
           selections: selections,
         )
       end
@@ -114,12 +117,12 @@ module GraphQL::Relay::Walker
       if !required_args_are_present
         nil
       else
-        f_alias = field.name == 'id' ? nil : random_alias
+        f_alias = field.graphql_name == 'id' ? nil : random_alias
         f_args = arguments.map do |name, value|
           GraphQL::Language::Nodes::Argument.new(name: name, value: value)
         end
 
-        GraphQL::Language::Nodes::Field.new(name: field.name, alias: f_alias, arguments: f_args)
+        GraphQL::Language::Nodes::Field.new(name: field.graphql_name, alias: f_alias, arguments: f_args)
       end
     end
 
@@ -213,13 +216,14 @@ module GraphQL::Relay::Walker
     # `node` field that is a relay node. Returns false otherwise.
     def connection_field?(field)
       type = field.type.unwrap
-
+      if type.kind.fields?
         if edges_field = type.get_field('edges')
           edges = edges_field.type.unwrap
           if node_field = edges.get_field('node')
             return node_field?(node_field)
           end
         end
+      end
 
       false
     end
@@ -268,7 +272,7 @@ module GraphQL::Relay::Walker
     end
 
     def valid_input?(type, input)
-      type.valid_isolated_input?(input)
+      type.valid_input?(input, GraphQL::Query::NullContext)
     end
 
     def make_type_name_node(type_name)

spec/client_ext_spec.rb

@@ -3,12 +3,13 @@ require 'graphql/relay/walker'
 require 'graphql/relay/walker/client_ext'
 
 describe GraphQL::Relay::Walker::ClientExt do
-  let(:schema_path)    { 'spec/fixtures/swapi_schema.json' }
-  let(:query_path)     { 'spec/fixtures/swapi_query.graphql' }
-  let(:schema)         { GraphQL::Client.load_schema(schema_path) }
-  let(:client)         { GraphQL::Client.new(schema: schema) }
-
+  [
+    GraphQL::Client.load_schema('spec/fixtures/swapi_schema.json'),
+    GraphQL::Schema.from_definition("spec/fixtures/swapi_schema.graphql")
+  ].each do |schema|
+    describe "with #{schema.class} schema" do
       describe '#walk' do
+        let(:client) { GraphQL::Client.new(schema: schema) }
         it 'allows passing additional variables through to GraphQLClient#query' do
           expected = { variables: { 'foo' => 'bar', 'id' => '12345' }, context: {} }
           expect(client).to receive(:query).with(anything, expected).and_return({})
@@ -22,4 +23,6 @@ describe GraphQL::Relay::Walker::ClientExt do
           client.walk(from_id: '12345', context: { viewer: viewer })
         end
       end
+    end
+  end
 end

spec/fixtures/swapi_query.graphql

@@ -2,6 +2,20 @@ query($id: ID!) {
   node(id: $id) {
     id
     ... on Film {
+      characterConnection(first: 5) {
+        edges {
+          node {
+            id
+          }
+        }
+      }
+      planetConnection(first: 5) {
+        edges {
+          node {
+            id
+          }
+        }
+      }
       speciesConnection(first: 5) {
         edges {
           node {
@@ -23,25 +37,8 @@ query($id: ID!) {
           }
         }
       }
-      characterConnection(first: 5) {
-        edges {
-          node {
-            id
-          }
-        }
-      }
-      planetConnection(first: 5) {
-        edges {
-          node {
-            id
-          }
-        }
-      }
     }
     ... on Person {
-      homeworld {
-        id
-      }
       filmConnection(first: 5) {
         edges {
           node {
@@ -49,6 +46,9 @@ query($id: ID!) {
           }
         }
       }
+      homeworld {
+        id
+      }
       species {
         id
       }
@@ -68,14 +68,14 @@ query($id: ID!) {
       }
     }
     ... on Planet {
-      residentConnection(first: 5) {
+      filmConnection(first: 5) {
         edges {
           node {
             id
           }
         }
       }
-      filmConnection(first: 5) {
+      residentConnection(first: 5) {
         edges {
           node {
             id
@@ -84,6 +84,13 @@ query($id: ID!) {
       }
     }
     ... on Species {
+      filmConnection(first: 5) {
+        edges {
+          node {
+            id
+          }
+        }
+      }
       homeworld {
         id
       }
@@ -94,23 +101,16 @@ query($id: ID!) {
           }
         }
       }
-      filmConnection(first: 5) {
-        edges {
-          node {
-            id
-          }
-        }
-      }
     }
     ... on Starship {
-      pilotConnection(first: 5) {
+      filmConnection(first: 5) {
         edges {
           node {
             id
           }
         }
       }
-      filmConnection(first: 5) {
+      pilotConnection(first: 5) {
         edges {
           node {
             id
@@ -119,14 +119,14 @@ query($id: ID!) {
       }
     }
     ... on Vehicle {
-      pilotConnection(first: 5) {
+      filmConnection(first: 5) {
         edges {
           node {
             id
           }
         }
       }
-      filmConnection(first: 5) {
+      pilotConnection(first: 5) {
         edges {
           node {
             id

spec/query_builder_spec.rb

@@ -2,14 +2,16 @@ require 'graphql/relay/walker'
 require 'graphql/client'
 
 describe GraphQL::Relay::Walker::QueryBuilder do
-  let(:schema_path)    { 'spec/fixtures/swapi_schema.json' }
+  [
+    GraphQL::Client.load_schema('spec/fixtures/swapi_schema.json'),
+    GraphQL::Schema.from_definition("spec/fixtures/swapi_schema.graphql")
+  ].each do |schema|
+    describe "With #{schema.class} schema" do
       let(:query_path)     { 'spec/fixtures/swapi_query.graphql' }
-  let(:schema)         { GraphQL::Client.load_schema(schema_path) }
       let(:client)         { GraphQL::Client.new(schema: schema) }
       let(:query_builder)  { described_class.new(schema) }
       let(:ast)            { query_builder.ast }
       let(:query_string)   { query_builder.query_string }
-
       describe 'ast' do
         subject { ast }
 
@@ -53,4 +55,6 @@ describe GraphQL::Relay::Walker::QueryBuilder do
 
         children + children.map { |child| nodes(child) }.flatten
       end
+    end
+  end
 end