Fix non-null list fields queries

src/generate.ts

@@ -1,4 +1,5 @@
 import {
+  GraphQLOutputType,
   GraphQLType,
   isEnumType,
   isListType,
@@ -79,18 +80,23 @@ function generateSelector(s: Selector, depth = 0, nonNull = false): string {
     return generateType(s.type)
   }
 
+  const code =
+    generateFields(s, depth) +
+    generateFragments(s) +
+    generateInlineFragments(s, depth - 1)
+
+  if (isNonNullType(s.type)) {
+    return wrapInArray(s.type.ofType, code)
+  }
+
   return (
-    generateArray(s, depth) +
+    wrapInArray(s.type, code) +
     (isNullableType(s.type) && !nonNull ? ' | null' : '')
   )
 }
 
-function generateArray(s: Selector, depth: number): string {
-  const code =
-    generateFields(s, depth) +
-    generateFragments(s) +
-    generateInlineFragments(s, depth - 1)
-  return isListType(s.type) ? 'Array<' + code + '>' : code
+function wrapInArray(t: GraphQLOutputType | undefined, code: string) {
+  return isListType(t) ? `Array<${code}>` : code
 }
 
 function generateFragments(s: Selector): string {

src/index.test.ts

@@ -11,6 +11,8 @@ const schema = buildSchema(`
   
   type Query {
     User(login: String!): User
+    Users: [User!]
+    NonNullUsers: [User!]!
   }
   
   type User {
@@ -151,3 +153,35 @@ test('inline interfaces', () => {
   expect(code).includes('barks: boolean | null\n')
   expect(code).includes('meows: boolean | null\n')
 })
+
+test('query with list', () => {
+  const source = new Source(`
+    query Nullable {
+      Users {
+        name
+      }
+    }
+  `)
+  const code = transpile(schema, source)
+  expect(code).includes(`{
+  Users: Array<{
+    name: string | null
+  }> | null
+}`)
+})
+
+test('query with non-null list', () => {
+  const source = new Source(`
+    query NonNullable {
+      NonNullUsers {
+        name
+      }
+    }
+  `)
+  const code = transpile(schema, source)
+  expect(code).includes(`{
+  NonNullUsers: Array<{
+    name: string | null
+  }>
+}`)
+})