Merge pull request rmosolgo#4652 from rmosolgo/enum-printer-fix

Fix enum variable default values in sanitized_query_string

Author: rmosolgo

lib/graphql/language/sanitized_printer.rb

@@ -203,7 +203,12 @@ def value_to_ast(value, type)
             [value].map { |v| value_to_ast(v, type.of_type) }
           end
         when "ENUM"
-          GraphQL::Language::Nodes::Enum.new(name: value)
+          if value.is_a?(GraphQL::Language::Nodes::Enum)
+            # if it was a default value, it's already wrapped
+            value
+          else
+            GraphQL::Language::Nodes::Enum.new(name: value)
+          end
         else
           value
         end

spec/graphql/language/sanitized_printer_spec.rb

@@ -344,5 +344,37 @@ def sanitize_string(query_string, inline_variables: true, **options)
     query = GraphQL::Query.new(SanitizeTest::CustomSanitizedPrinterSchema, query_str)
     assert_equal expected_query_string, query.sanitized_query_string
   end
-end
 
+  it "properly prints enum variable default values" do
+    class EnumSchema < GraphQL::Schema
+      class Grouping < GraphQL::Schema::Enum
+        value "DAY"
+      end
+
+      class Query < GraphQL::Schema::Object
+        field :things, [String] do
+          argument :group, Grouping
+        end
+
+        def things(group:)
+          [group]
+        end
+      end
+
+      query(Query)
+    end
+    query_string = <<~EOS
+      query(
+        $group: Grouping = DAY
+      ) {
+        things(group: $group)
+      }
+    EOS
+
+    query = ::GraphQL::Query.new(EnumSchema, query_string)
+
+    assert_equal ["DAY"], query.result["data"]["things"]
+    expected_query_string = "query {\n  things(group: DAY)\n}"
+    assert_equal expected_query_string, query.sanitized_query_string
+  end
+end