graphql_devise 0.18.1 → 1.0.1

data/lib/graphql_devise/resource_loader.rb

@@ -6,95 +6,127 @@ module GraphqlDevise
       @resource           = resource
       @options            = options
       @routing            = routing
-      @default_operations = GraphqlDevise::DefaultOperations::MUTATIONS.merge(GraphqlDevise::DefaultOperations::QUERIES)
+      @default_operations = DefaultOperations::MUTATIONS.merge(DefaultOperations::QUERIES)
     end
 
     def call(query, mutation)
       # clean_options responds to all keys defined in GraphqlDevise::MountMethod::SUPPORTED_OPTIONS
-      clean_options = GraphqlDevise::MountMethod::OptionSanitizer.new(@options).call!
+      clean_options = MountMethod::OptionSanitizer.new(@options).call!
 
-      model = if @resource.is_a?(String)
-        ActiveSupport::Deprecation.warn(<<-DEPRECATION.strip_heredoc, caller)
-          Providing a String as the model you want to mount is deprecated and will be removed in a future version of
-          this gem. Please use the actual model constant instead.
-
-          EXAMPLE
-
-          GraphqlDevise::ResourceLoader.new(User) # instead of GraphqlDevise::ResourceLoader.new('User')
-
-          mount_graphql_devise_for User # instead of mount_graphql_devise_for 'User'
-        DEPRECATION
-        @resource.constantize
-      else
-        @resource
+      unless @resource.is_a?(Class)
+        raise(
+          InvalidMountOptionsError,
+          'A class must be provided when mounting a model. String values are no longer supported.'
+        )
       end
 
       # Necesary when mounting a resource via route file as Devise forces the reloading of routes
-      return clean_options if GraphqlDevise.resource_mounted?(model) && @routing
+      return clean_options if ::GraphqlDevise.resource_mounted?(@resource) && @routing
 
       validate_options!(clean_options)
 
       authenticatable_type = clean_options.authenticatable_type.presence ||
                              "Types::#{@resource}Type".safe_constantize ||
-                             GraphqlDevise::Types::AuthenticatableType
+                             Types::AuthenticatableType
 
-      prepared_mutations = prepare_mutations(model, clean_options, authenticatable_type)
+      prepared_mutations = prepare_mutations(@resource, clean_options, authenticatable_type)
 
       if prepared_mutations.any? && mutation.blank?
-        raise GraphqlDevise::Error, 'You need to provide a mutation type unless all mutations are skipped'
+        raise ::GraphqlDevise::Error, 'You need to provide a mutation type unless all mutations are skipped'
       end
 
-      prepared_mutations.each do |action, prepared_mutation|
-        mutation.field(action, mutation: prepared_mutation, authenticate: false)
+      begin
+        prepared_mutations.each do |action, prepared_mutation|
+          mutation.field(action, mutation: prepared_mutation, authenticate: false)
+        end
+      rescue ArgumentError
+        raise unless Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
+
+        raise_undefined_field_class_error('MutationType')
       end
 
-      prepared_resolvers = prepare_resolvers(model, clean_options, authenticatable_type)
+      prepared_resolvers = prepare_resolvers(@resource, clean_options, authenticatable_type)
 
       if prepared_resolvers.any? && query.blank?
-        raise GraphqlDevise::Error, 'You need to provide a query type unless all queries are skipped'
+        raise ::GraphqlDevise::Error, 'You need to provide a query type unless all queries are skipped'
       end
 
-      prepared_resolvers.each do |action, resolver|
-        query.field(action, resolver: resolver, authenticate: false)
+      begin
+        prepared_resolvers.each do |action, resolver|
+          query.field(action, resolver: resolver, authenticate: false)
+        end
+      rescue ArgumentError
+        raise unless Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
+
+        raise_undefined_field_class_error('QueryType')
       end
 
-      GraphqlDevise.add_mapping(GraphqlDevise.to_mapping_name(@resource).to_sym, @resource)
-      GraphqlDevise.mount_resource(model) if @routing
+      ::GraphqlDevise.add_mapping(::GraphqlDevise.to_mapping_name(@resource).to_sym, @resource)
+      ::GraphqlDevise.mount_resource(@resource) if @routing
 
       clean_options
     end
 
     private
 
+    def raise_undefined_field_class_error(type_class)
+      raise(
+        GraphqlDevise::Error,
+        <<~MESSAGE
+          To use this gem with graphql >= 2.0 you need to use a custom `field_class`
+          on your #{type_class}. You can use the `field_class` defined in this gem like this:
+
+          module Types
+            class #{type_class} < GraphQL::Schema::Object
+              field_class GraphqlDevise::Types::BaseField
+            end
+          end
+
+          If you already have a `field_class` set on your #{type_class}, you can include our concern
+          to make this work
+
+          module Types
+            class BaseField < GraphQL::Schema::Field
+              include GraphqlDevise::FieldAuthentication
+            end
+          end
+
+          The concern will define an initializer, if your custom field logic is not standard, you can look
+          at the `GraphqlDevise::FieldAuthentication` implementation to add what is required to your own
+          field class
+        MESSAGE
+      )
+    end
+
     def prepare_resolvers(model, clean_options, authenticatable_type)
-      GraphqlDevise::MountMethod::OperationPreparer.new(
+      MountMethod::OperationPreparer.new(
         model:                 model,
         custom:                clean_options.operations,
         additional_operations: clean_options.additional_queries,
-        preparer:              GraphqlDevise::MountMethod::OperationPreparers::ResolverTypeSetter.new(authenticatable_type),
-        selected_operations:   GraphqlDevise::MountMethod::OperationSanitizer.call(
-          default: GraphqlDevise::DefaultOperations::QUERIES, only: clean_options.only, skipped: clean_options.skip
+        preparer:              MountMethod::OperationPreparers::ResolverTypeSetter.new(authenticatable_type),
+        selected_operations:   MountMethod::OperationSanitizer.call(
+          default: DefaultOperations::QUERIES, only: clean_options.only, skipped: clean_options.skip
         )
       ).call
     end
 
     def prepare_mutations(model, clean_options, authenticatable_type)
-      GraphqlDevise::MountMethod::OperationPreparer.new(
+      MountMethod::OperationPreparer.new(
         model:                 model,
         custom:                clean_options.operations,
         additional_operations: clean_options.additional_mutations,
-        preparer:              GraphqlDevise::MountMethod::OperationPreparers::MutationFieldSetter.new(authenticatable_type),
-        selected_operations:   GraphqlDevise::MountMethod::OperationSanitizer.call(
-          default: GraphqlDevise::DefaultOperations::MUTATIONS, only: clean_options.only, skipped: clean_options.skip
+        preparer:              MountMethod::OperationPreparers::MutationFieldSetter.new(authenticatable_type),
+        selected_operations:   MountMethod::OperationSanitizer.call(
+          default: DefaultOperations::MUTATIONS, only: clean_options.only, skipped: clean_options.skip
         )
       ).call
     end
 
     def validate_options!(clean_options)
-      GraphqlDevise::MountMethod::OptionsValidator.new(
+      MountMethod::OptionsValidator.new(
         [
-          GraphqlDevise::MountMethod::OptionValidators::SkipOnlyValidator.new(options: clean_options),
-          GraphqlDevise::MountMethod::OptionValidators::ProvidedOperationsValidator.new(
+          MountMethod::OptionValidators::SkipOnlyValidator.new(options: clean_options),
+          MountMethod::OptionValidators::ProvidedOperationsValidator.new(
             options: clean_options, supported_operations: @default_operations
           )
         ]

data/lib/graphql_devise/route_mounter.rb

@@ -0,0 +1,13 @@
+module GraphqlDevise
+  module RouteMounter
+    def mount_graphql_devise_for(resource, options = {})
+      clean_options = ResourceLoader.new(resource, options, true).call(
+        Types::QueryType,
+        Types::MutationType
+      )
+
+      post clean_options.at, to: 'graphql_devise/graphql#auth'
+      get  clean_options.at, to: 'graphql_devise/graphql#auth'
+    end
+  end
+end

data/lib/graphql_devise/schema_plugin.rb

@@ -4,7 +4,7 @@ module GraphqlDevise
   class SchemaPlugin
     # NOTE: Based on GQL-Ruby docs  https://graphql-ruby.org/schema/introspection.html
     INTROSPECTION_FIELDS = ['__schema', '__type', '__typename']
-    DEFAULT_NOT_AUTHENTICATED = ->(field) { raise GraphqlDevise::AuthenticationError, "#{field} field requires authentication" }
+    DEFAULT_NOT_AUTHENTICATED = ->(field) { raise AuthenticationError, "#{field} field requires authentication" }
 
     def initialize(query: nil, mutation: nil, authenticate_default: true, public_introspection: !Rails.env.production?, resource_loaders: [], unauthenticated_proc: DEFAULT_NOT_AUTHENTICATED)
       @query                = query
@@ -30,22 +30,7 @@ module GraphqlDevise
       auth_required = authenticate_option(field, trace_data)
       context       = context_from_data(trace_data)
 
-      if context.key?(:resource_name)
-        ActiveSupport::Deprecation.warn(<<-DEPRECATION.strip_heredoc, caller)
-          Providing `resource_name` as part of the GQL context, or doing so by using the `graphql_context(resource_name)`
-          method on your controller is deprecated and will be removed in a future version of this gem.
-          Please use `gql_devise_context` in you controller instead.
-
-          EXAMPLE
-          include GraphqlDevise::SetUserByToken
-
-          DummySchema.execute(params[:query], context: gql_devise_context(User))
-          DummySchema.execute(params[:query], context: gql_devise_context(User, Admin))
-        DEPRECATION
-      end
-
       if auth_required && !(public_introspection && introspection_field?(field))
-        context = set_current_resource(context)
         raise_on_missing_resource(context, field, auth_required)
       end
 
@@ -56,25 +41,6 @@ module GraphqlDevise
 
     attr_reader :public_introspection
 
-    def set_current_resource(context)
-      controller     = context[:controller]
-      resource_names = Array(context[:resource_name])
-
-      context[:current_resource] ||= resource_names.find do |resource_name|
-        unless Devise.mappings.key?(resource_name)
-          raise(
-            GraphqlDevise::Error,
-            "Invalid resource_name `#{resource_name}` provided to `graphql_context`. Possible values are: #{Devise.mappings.keys}."
-          )
-        end
-
-        found = controller.set_resource_by_token(resource_name)
-        break found if found
-      end
-
-      context
-    end
-
     def raise_on_missing_resource(context, field, auth_required)
       @unauthenticated_proc.call(field.name) if context[:current_resource].blank?
 
@@ -113,7 +79,15 @@ module GraphqlDevise
       auth_required = if trace_data[:context]
         field.metadata[:authenticate]
       else
-        field.graphql_definition.metadata[:authenticate]
+        if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
+          # authenticate will only be defined if "field_class GraphqlDevise::Types::BaseField" is added to the type
+          # returning nil here will use the default value used when mounting the plugin
+          field.try(:authenticate)
+        elsif Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.13.1')
+          field.graphql_definition(silence_deprecation_warning: true).metadata[:authenticate]
+        else
+          field.graphql_definition.metadata[:authenticate]
+        end
       end
 
       auth_required.nil? ? @authenticate_default : auth_required
@@ -121,7 +95,7 @@ module GraphqlDevise
 
     def load_fields
       @resource_loaders.each do |resource_loader|
-        raise Error, 'Invalid resource loader instance' unless resource_loader.instance_of?(GraphqlDevise::ResourceLoader)
+        raise ::GraphqlDevise::Error, 'Invalid resource loader instance' unless resource_loader.instance_of?(ResourceLoader)
 
         resource_loader.call(@query, @mutation)
       end
@@ -132,6 +106,3 @@ module GraphqlDevise
     end
   end
 end
-
-GraphQL::Field.accepts_definitions(authenticate: GraphQL::Define.assign_metadata_key(:authenticate))
-GraphQL::Schema::Field.accepts_definition(:authenticate)

data/lib/graphql_devise/types/authenticatable_type.rb

@@ -2,7 +2,7 @@
 
 module GraphqlDevise
   module Types
-    class AuthenticatableType < GraphQL::Schema::Object
+    class AuthenticatableType < BaseType
       field :email, String, null: false
     end
   end

data/lib/graphql_devise/types/base_field.rb

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+module GraphqlDevise
+  module Types
+    class BaseField < GraphQL::Schema::Field
+      include FieldAuthentication
+    end
+  end
+end

data/lib/graphql_devise/types/base_type.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module GraphqlDevise
+  module Types
+    class BaseType < GraphQL::Schema::Object
+    end
+  end
+end

data/lib/graphql_devise/types/credential_type.rb

@@ -2,7 +2,7 @@
 
 module GraphqlDevise
   module Types
-    class CredentialType < GraphQL::Schema::Object
+    class CredentialType < BaseType
       field :access_token, String, null: false
       field :uid,          String, null: false
       field :token_type,   String, null: false

data/lib/graphql_devise/types/mutation_type.rb

@@ -3,6 +3,7 @@
 module GraphqlDevise
   module Types
     class MutationType < GraphQL::Schema::Object
+      field_class GraphqlDevise::Types::BaseField if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
     end
   end
 end

data/lib/graphql_devise/types/query_type.rb

@@ -3,6 +3,7 @@
 module GraphqlDevise
   module Types
     class QueryType < GraphQL::Schema::Object
+      field_class GraphqlDevise::Types::BaseField if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
     end
   end
 end

data/lib/graphql_devise/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GraphqlDevise
-  VERSION = '0.18.1'.freeze
+  VERSION = '1.0.1'
 end

data/lib/graphql_devise.rb

@@ -1,13 +1,31 @@
 # frozen_string_literal: true
 
 require 'rails'
+require 'rails/generators'
 require 'graphql'
 require 'devise_token_auth'
+require 'zeitwerk'
+
+if Gem::Version.new(GraphQL::VERSION) < Gem::Version.new('2.0')
+  GraphQL::Field.accepts_definitions(authenticate: GraphQL::Define.assign_metadata_key(:authenticate))
+  GraphQL::Schema::Field.accepts_definition(:authenticate)
+end
+
+loader = Zeitwerk::Loader.for_gem
+
+loader.collapse("#{__dir__}/graphql_devise/concerns")
+loader.collapse("#{__dir__}/graphql_devise/errors")
+loader.ignore("#{__dir__}/generators")
+
+loader.inflector.inflect('error_codes' => 'ERROR_CODES')
+loader.inflector.inflect('supported_options' => 'SUPPORTED_OPTIONS')
+
+loader.setup
 
 module GraphqlDevise
   class Error < StandardError; end
 
-  class InvalidMountOptionsError < GraphqlDevise::Error; end
+  class InvalidMountOptionsError < ::GraphqlDevise::Error; end
 
   @schema_loaded     = false
   @mounted_resources = []
@@ -52,32 +70,6 @@ module GraphqlDevise
   end
 end
 
+ActionDispatch::Routing::Mapper.include(GraphqlDevise::RouteMounter)
+
 require 'graphql_devise/engine'
-require 'graphql_devise/version'
-require 'graphql_devise/errors/error_codes'
-require 'graphql_devise/errors/execution_error'
-require 'graphql_devise/errors/user_error'
-require 'graphql_devise/errors/authentication_error'
-require 'graphql_devise/errors/detailed_user_error'
-
-# TODO: Remove when merged into master
-require 'graphql_devise/concerns/authenticatable'
-require 'graphql_devise/concerns/set_user_by_token.rb'
-
-require 'graphql_devise/concerns/controller_methods'
-require 'graphql_devise/schema'
-require 'graphql_devise/types/authenticatable_type'
-require 'graphql_devise/types/credential_type'
-require 'graphql_devise/types/mutation_type'
-require 'graphql_devise/types/query_type'
-require 'graphql_devise/default_operations/mutations'
-require 'graphql_devise/default_operations/resolvers'
-require 'graphql_devise/resolvers/dummy'
-
-require 'graphql_devise/mount_method/option_sanitizer'
-require 'graphql_devise/mount_method/options_validator'
-require 'graphql_devise/mount_method/operation_preparer'
-require 'graphql_devise/mount_method/operation_sanitizer'
-
-require 'graphql_devise/resource_loader'
-require 'graphql_devise/schema_plugin'

data/spec/dummy/app/controllers/api/v1/graphql_controller.rb

@@ -4,6 +4,7 @@ module Api
   module V1
     class GraphqlController < ApplicationController
       include GraphqlDevise::SetUserByToken
+      include ActionController::Cookies
 
       def graphql
         result = DummySchema.execute(params[:query], **execute_params(params))
@@ -15,28 +16,13 @@ module Api
         render json: InterpreterSchema.execute(params[:query], **execute_params(params))
       end
 
-      def failing_resource_name
-        render json: DummySchema.execute(params[:query], context: graphql_context([:user, :fail]))
-      end
-
-      def controller_auth
-        result = DummySchema.execute(
-          params[:query],
-          operation_name: params[:operationName],
-          variables:      ensure_hash(params[:variables]),
-          context:        gql_devise_context(SchemaUser, User)
-        )
-
-        render json: result unless performed?
-      end
-
       private
 
       def execute_params(item)
         {
           operation_name: item[:operationName],
           variables:      ensure_hash(item[:variables]),
-          context:        graphql_context([:user, :schema_user])
+          context:        gql_devise_context(SchemaUser, User)
         }
       end
 

data/spec/dummy/app/graphql/dummy_schema.rb

@@ -10,11 +10,7 @@ class DummySchema < GraphQL::Schema
         User,
         only: [
           :login,
-          :confirm_account,
-          :send_password_reset,
-          :resend_confirmation,
-          :resend_confirmation_with_token,
-          :check_password_token
+          :resend_confirmation_with_token
         ]
       ),
       GraphqlDevise::ResourceLoader.new(Guest, only: [:logout]),

data/spec/dummy/app/graphql/interpreter_schema.rb

@@ -1,8 +1,12 @@
 # frozen_string_literal: true
 
 class InterpreterSchema < GraphQL::Schema
-  use GraphQL::Execution::Interpreter if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.9.0')
-  use GraphQL::Analysis::AST          if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.10.0')
+  if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.9.0') && Gem::Version.new(GraphQL::VERSION) < Gem::Version.new('2.0')
+    use GraphQL::Execution::Interpreter
+  end
+  if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('1.10.0') && Gem::Version.new(GraphQL::VERSION) < Gem::Version.new('2.0')
+    use GraphQL::Analysis::AST
+  end
 
   use GraphqlDevise::SchemaPlugin.new(query: Types::QueryType, authenticate_default: false)
 

data/spec/dummy/app/graphql/mutations/base_mutation.rb

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+module Mutations
+  class BaseMutation < GraphQL::Schema::Mutation
+  end
+end

data/spec/dummy/app/graphql/mutations/update_user.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Mutations
-  class UpdateUser < GraphQL::Schema::Mutation
+  class UpdateUser < BaseMutation
     field :user, Types::UserType, null: false
 
     argument :email, String, required: false
@@ -10,10 +10,8 @@ module Mutations
     def resolve(**attrs)
       user = context[:current_resource]
 
-      schema_url = context[:controller].full_url_without_params
-
       user.update_with_email(
-        attrs.merge(schema_url: schema_url, confirmation_success_url: 'https://google.com')
+        attrs.merge(confirmation_url: 'https://google.com')
       )
 
       { user: user }

data/spec/dummy/app/graphql/types/admin_type.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Types
-  class AdminType < GraphQL::Schema::Object
+  class AdminType < BaseObject
     field :id,    Int,    null: false
     field :email, String, null: false
   end

data/spec/dummy/app/graphql/types/custom_admin_type.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Types
-  class CustomAdminType < GraphQL::Schema::Object
+  class CustomAdminType < BaseObject
     field :email,        String, null: false
     field :custom_field, String, null: false
 

data/spec/dummy/app/graphql/types/mutation_type.rb

@@ -1,7 +1,9 @@
 # frozen_string_literal: true
 
 module Types
-  class MutationType < Types::BaseObject
+  class MutationType < BaseObject
+    field_class GraphqlDevise::Types::BaseField if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
+
     field :dummy_mutation, String, null: false, authenticate: true
     field :update_user, mutation: Mutations::UpdateUser
 

data/spec/dummy/app/graphql/types/query_type.rb

@@ -1,7 +1,9 @@
 # frozen_string_literal: true
 
 module Types
-  class QueryType < Types::BaseObject
+  class QueryType < BaseObject
+    field_class GraphqlDevise::Types::BaseField if Gem::Version.new(GraphQL::VERSION) >= Gem::Version.new('2.0')
+
     field :user, resolver: Resolvers::UserShow
     field :public_field, String, null: false, authenticate: false
     field :private_field, String, null: false, authenticate: true

data/spec/dummy/app/graphql/types/user_type.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Types
-  class UserType < GraphQL::Schema::Object
+  class UserType < BaseObject
     field :id,            Int,    null: false
     field :email,         String, null: false
     field :name,          String, null: false

data/spec/dummy/config/environments/test.rb

@@ -43,5 +43,5 @@ Rails.application.configure do
   config.active_support.deprecation = :stderr
 
   # Raises error for missing translations
-  config.action_view.raise_on_missing_translations = true
+  config.action_view.raise_on_missing_translations = true if Rails::VERSION::MAJOR < 7
 end

data/spec/dummy/config/routes.rb

@@ -1,9 +1,8 @@
 # frozen_string_literal: true
 
 Rails.application.routes.draw do
-  mount_graphql_devise_for 'User', at: '/api/v1/graphql_auth', operations: {
+  mount_graphql_devise_for User, at: '/api/v1/graphql_auth', operations: {
     login:    Mutations::Login,
-    sign_up:  Mutations::SignUp,
     register: Mutations::Register
   }, additional_mutations: {
     register_confirmed_user: Mutations::RegisterConfirmedUser
@@ -14,22 +13,21 @@ Rails.application.routes.draw do
   mount_graphql_devise_for(
     Admin,
     authenticatable_type: Types::CustomAdminType,
-    skip:                 [:sign_up, :register, :check_password_token],
+    skip:                 [:register],
     operations:           {
-      confirm_account:            Resolvers::ConfirmAdminAccount,
       update_password_with_token: Mutations::ResetAdminPasswordWithToken
     },
     at:                   '/api/v1/admin/graphql_auth'
   )
 
   mount_graphql_devise_for(
-    'Guest',
-    only: [:login, :logout, :sign_up, :register],
+    Guest,
+    only: [:login, :logout, :register],
     at:   '/api/v1/guest/graphql_auth'
   )
 
   mount_graphql_devise_for(
-    'Users::Customer',
+    Users::Customer,
     only: [:login],
     at:   '/api/v1/user_customer/graphql_auth'
   )
@@ -37,6 +35,4 @@ Rails.application.routes.draw do
   get '/api/v1/graphql', to: 'api/v1/graphql#graphql'
   post '/api/v1/graphql', to: 'api/v1/graphql#graphql'
   post '/api/v1/interpreter', to: 'api/v1/graphql#interpreter'
-  post '/api/v1/failing', to: 'api/v1/graphql#failing_resource_name'
-  post '/api/v1/controller_auth', to: 'api/v1/graphql#controller_auth'
 end

data/spec/generators/graphql_devise/install_generator_spec.rb

@@ -41,7 +41,7 @@ RSpec.describe GraphqlDevise::InstallGenerator, type: :generator do
     let(:args) { [] }
 
     it 'creates and updated required files' do
-      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for 'User', at: 'graphql_auth'/
+      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for User, at: 'graphql_auth'/
       expect(routes_content).not_to match(dta_route)
 
       assert_file 'config/initializers/devise.rb'
@@ -60,7 +60,7 @@ RSpec.describe GraphqlDevise::InstallGenerator, type: :generator do
     let(:args) { %w[Admin api] }
 
     it 'creates and updated required files' do
-      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for 'Admin', at: 'api'/
+      assert_file 'config/routes.rb', /^\s{2}mount_graphql_devise_for Admin, at: 'api'/
       expect(routes_content).not_to match(dta_route)
 
       assert_file 'config/initializers/devise.rb'