printer: use wrap util function in more places (#2934)

graphql · Feb 21, 2021 · d0eae6f · d0eae6f
1 parent b6b2789
commit d0eae6f
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/src/language/printer.js b/src/language/printer.js
@@ -180,12 +180,7 @@ const printDocASTReducer: any = {
   UnionTypeDefinition: ({ description, name, directives, types }) =>
     wrap('', description, '\n') +
     join(
-      [
-        'union',
-        name,
-        join(directives, ' '),
-        types && types.length !== 0 ? '= ' + join(types, ' | ') : '',
-      ],
+      ['union', name, join(directives, ' '), wrap('= ', join(types, ' | '))],
       ' ',
     ),
 
@@ -253,7 +248,7 @@ const printDocASTReducer: any = {
         'extend union',
         name,
         join(directives, ' '),
-        types && types.length !== 0 ? '= ' + join(types, ' | ') : '',
+        wrap('= ', join(types, ' | ')),
       ],
       ' ',
     ),