* gems/default_gems: Sync did_you_mean

This commit is contained in:
Yuki Nishijima 2021-12-21 19:03:10 +09:00
Родитель 12fa4f2ace
Коммит 505dfae05d
10 изменённых файлов: 150 добавлений и 13 удалений

Просмотреть файл

@ -86,11 +86,24 @@ require_relative 'did_you_mean/tree_spell_checker'
#
module DidYouMean
# Map of error types and spell checker objects.
SPELL_CHECKERS = Hash.new(NullChecker)
@spell_checkers = Hash.new(NullChecker)
# Returns a sharable hash map of error types and spell checker objects.
def self.spell_checkers
@spell_checkers
end
# Adds +DidYouMean+ functionality to an error using a given spell checker
def self.correct_error(error_class, spell_checker)
SPELL_CHECKERS[error_class.name] = spell_checker
if defined?(Ractor)
new_mapping = { **@spell_checkers, error_class.name => spell_checker }
new_mapping.default = NullChecker
@spell_checkers = Ractor.make_shareable(new_mapping)
else
spell_checkers[error_class.name] = spell_checker
end
error_class.prepend(Correctable) unless error_class < Correctable
end
@ -100,15 +113,23 @@ module DidYouMean
correct_error LoadError, RequirePathChecker if RUBY_VERSION >= '2.8.0'
correct_error NoMatchingPatternKeyError, PatternKeyNameChecker if defined?(::NoMatchingPatternKeyError)
# TODO: Remove on 3.3:
SPELL_CHECKERS = @spell_checkers
deprecate_constant :SPELL_CHECKERS
# Returns the currently set formatter. By default, it is set to +DidYouMean::Formatter+.
def self.formatter
@formatter
if defined?(Reactor)
Ractor.current[:__did_you_mean_formatter__] || Formatter
else
Formatter
end
end
# Updates the primary formatter used to format the suggestions.
def self.formatter=(formatter)
@formatter = formatter
if defined?(Reactor)
Ractor.current[:__did_you_mean_formatter__] = formatter
end
end
@formatter = Formatter.new
end

Просмотреть файл

@ -26,7 +26,7 @@ module DidYouMean
end
def spell_checker
SPELL_CHECKERS[self.class.to_s].new(self)
DidYouMean.spell_checkers[self.class.to_s].new(self)
end
end
end

Просмотреть файл

@ -26,7 +26,7 @@ module DidYouMean
#
# # => nil
#
def message_for(corrections)
def self.message_for(corrections)
corrections.empty? ? "" : "\nDid you mean? #{corrections.join("\n ")}"
end
end

Просмотреть файл

@ -6,6 +6,7 @@ module DidYouMean
NAMES_TO_EXCLUDE = { NilClass => nil.methods }
NAMES_TO_EXCLUDE.default = []
Ractor.make_shareable(NAMES_TO_EXCLUDE) if defined?(Ractor)
# +MethodNameChecker::RB_RESERVED_WORDS+ is the list of reserved words in
# Ruby that take an argument. Unlike
@ -36,6 +37,8 @@ module DidYouMean
yield
)
Ractor.make_shareable(RB_RESERVED_WORDS) if defined?(Ractor)
def initialize(exception)
@method_name = exception.name
@receiver = exception.receiver

Просмотреть файл

@ -8,6 +8,7 @@ module DidYouMean
NAMES_TO_EXCLUDE = { 'foo' => [:fork, :for] }
NAMES_TO_EXCLUDE.default = []
Ractor.make_shareable(NAMES_TO_EXCLUDE) if defined?(Ractor)
# +VariableNameChecker::RB_RESERVED_WORDS+ is the list of all reserved
# words in Ruby. They could be declared like methods are, and a typo would
@ -62,6 +63,8 @@ module DidYouMean
__ENCODING__
)
Ractor.make_shareable(RB_RESERVED_WORDS) if defined?(Ractor)
def initialize(exception)
@name = exception.name.to_s.tr("@", "")
@lvar_names = exception.respond_to?(:local_variables) ? exception.local_variables : []

Просмотреть файл

@ -9,7 +9,10 @@ module DidYouMean
attr_reader :path
INITIAL_LOAD_PATH = $LOAD_PATH.dup.freeze
ENV_SPECIFIC_EXT = ".#{RbConfig::CONFIG["DLEXT"]}"
Ractor.make_shareable(INITIAL_LOAD_PATH) if defined?(Ractor)
ENV_SPECIFIC_EXT = ".#{RbConfig::CONFIG["DLEXT"]}"
Ractor.make_shareable(ENV_SPECIFIC_EXT) if defined?(Ractor)
private_constant :INITIAL_LOAD_PATH, :ENV_SPECIFIC_EXT

Просмотреть файл

@ -1,3 +1,3 @@
module DidYouMean
VERSION = "1.6.0-alpha"
VERSION = "1.6.0".freeze
end

Просмотреть файл

@ -1,7 +1,7 @@
require_relative '../helper'
class NameErrorExtensionTest < Test::Unit::TestCase
SPELL_CHECKERS = DidYouMean::SPELL_CHECKERS
SPELL_CHECKERS = DidYouMean.spell_checkers
class TestSpellChecker
def initialize(*); end
@ -9,13 +9,14 @@ class NameErrorExtensionTest < Test::Unit::TestCase
end
def setup
@org, SPELL_CHECKERS['NameError'] = SPELL_CHECKERS['NameError'], TestSpellChecker
@original_spell_checker = DidYouMean.spell_checkers['NameError']
DidYouMean.correct_error(NameError, TestSpellChecker)
@error = assert_raise(NameError){ doesnt_exist }
end
def teardown
SPELL_CHECKERS['NameError'] = @org
DidYouMean.correct_error(NameError, @original_spell_checker)
end
def test_message

Просмотреть файл

@ -4,6 +4,10 @@ module DidYouMean
module TestHelper
class << self
attr_reader :root
def ractor_compatible?
defined?(Ractor) && RUBY_VERSION >= "3.1.0"
end
end
if File.file?(File.expand_path('../lib/did_you_mean.rb', __dir__))

Просмотреть файл

@ -0,0 +1,102 @@
return if not DidYouMean::TestHelper.ractor_compatible?
require_relative './helper'
class RactorCompatibilityTest < Test::Unit::TestCase
include DidYouMean::TestHelper
class ::Book; end
class FirstNameError < NameError; end
def test_class_name_suggestion_works_in_ractor
error = Ractor.new {
begin
Boook
rescue NameError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_correction "Book", error.corrections
end
def test_key_name_suggestion_works_in_ractor
error = Ractor.new {
begin
hash = { "foo" => 1, bar: 2 }
hash.fetch(:bax)
rescue KeyError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_correction ":bar", error.corrections
assert_match "Did you mean? :bar", error.to_s
end
def test_method_name_suggestion_works_in_ractor
error = Ractor.new {
begin
self.to__s
rescue NoMethodError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_correction :to_s, error.corrections
assert_match "Did you mean? to_s", error.to_s
end
if defined?(::NoMatchingPatternKeyError)
def test_pattern_key_name_suggestion_works_in_ractor
error = Ractor.new {
begin
eval(<<~RUBY, binding, __FILE__, __LINE__)
hash = {foo: 1, bar: 2, baz: 3}
hash => {fooo:}
fooo = 1 # suppress "unused variable: fooo" warning
RUBY
rescue NoMatchingPatternKeyError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_correction ":foo", error.corrections
assert_match "Did you mean? :foo", error.to_s
end
end
def test_can_raise_other_name_error_in_ractor
error = Ractor.new {
begin
raise FirstNameError, "Other name error"
rescue FirstNameError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_not_match(/Did you mean\?/, error.message)
end
def test_variable_name_suggestion_works_in_ractor
error = Ractor.new {
in_ractor = in_ractor = 1
begin
in_reactor
rescue NameError => e
e.corrections # It is important to call the #corrections method within Ractor.
e
end
}.take
assert_correction :in_ractor, error.corrections
assert_match "Did you mean? in_ractor", error.to_s
end
end