Enum with custom values aren't working as expected to trigger subscriptions about graphql-ruby HOT 5 CLOSED

# frozen_string_literal: true
require "spec_helper"

module SubscriptionEnum
  class InMemoryBackend < GraphQL::Subscriptions
    attr_reader :write_subscription_events, :execute_all_events

    def initialize(...)
      super
      @write_subscription_events = []
      @execute_all_events = []
    end

    def write_subscription(_query, events)
      @write_subscription_events.concat(events)
    end

    def execute_all(event, _object)
      @execute_all_events.push(event)
    end
  end

  class MyEnumType < GraphQL::Schema::Enum
    value "ONE", value: "one"
    value "TWO", value: "two"
  end

  class MySubscription < GraphQL::Schema::Subscription
    argument :my_enum, MyEnumType, required: true
    field :my_enum, MyEnumType
  end

  class SubscriptionType < GraphQL::Schema::Object
    field :my_subscription, resolver: MySubscription
  end

  class Schema < GraphQL::Schema
    subscription SubscriptionType
    use InMemoryBackend
  end
end

describe GraphQL::Subscriptions do
  describe "enums" do
    let(:schema) { SubscriptionEnum::Schema }
    let(:implementation) { schema.subscriptions }
    let(:write_subscription_events) { implementation.write_subscription_events }
    let(:execute_all_events) { implementation.execute_all_events }

    describe "pushing updates" do
      focus
      it "sends updated data" do
        query_str = <<-GRAPHQL
          subscription ($myEnum: MyEnum!) {
            mySubscription (myEnum: $myEnum) {
              myEnum
            }
          }
        GRAPHQL

        schema.execute(query_str, variables: { "myEnum" => "ONE" })

        schema.subscriptions.trigger(:mySubscription, { "myEnum" => "ONE" }, nil)

        assert_equal(":mySubscription:myEnum:one", write_subscription_events[0].topic)
        assert_equal(":mySubscription:myEnum:one", execute_all_events[0].topic)
      end
    end
  end
end

 # Normalize enum values to its custom value when triggering subscriptions.
 #
 # In a nutshell, Event#topic normalizes "ENUM_VALUE" to "enum_value" when
 # writing a subscription while when triggering it doesn't.
 #
 # This monkey-patch will allow us to call either way:
 #
 #   RootSchema.subscriptions.trigger(:my_subscription, my_enum: "enum_value")
 #   RootSchema.subscriptions.trigger(:my_subscription, my_enum: "ENUM_VALUE")
 #
 # See https://github.com/rmosolgo/graphql-ruby/issues/4713
 module NormalizeTriggerSubscriptionEnumValues
   def stringify_args(arg_owner, args, context)
     result = super

     if arg_owner.is_a?(Class) && arg_owner < GraphQL::Schema::Enum
       arg_owner.coerce_isolated_input(result) || result
     else
       result
     end
   end
 end

 GraphQL::Subscriptions::Event.singleton_class.prepend(NormalizeTriggerSubscriptionEnumValues)