1 files changed, 140 insertions, 0 deletions
diff --git a/test/did_you_mean/spell_checking/test_method_name_check.rb b/test/did_you_mean/spell_checking/test_method_name_check.rb
new file mode 100644
index 0000000000..f3a6b1c7c7
--- /dev/null
+++ b/test/did_you_mean/spell_checking/test_method_name_check.rb
@@ -0,0 +1,140 @@
+require_relative '../helper'
+
+class MethodNameCheckTest < Test::Unit::TestCase
+  include DidYouMean::TestHelper
+
+  class User
+    def friends; end
+    def first_name; end
+    def descendants; end
+    def call_incorrect_private_method
+      raiae NoMethodError
+    end
+
+    def raise_no_method_error
+      self.firstname
+    rescue NoMethodError => e
+      raise e, e.message, e.backtrace
+    end
+
+    protected
+    def the_protected_method; end
+
+    private
+    def friend; end
+    def the_private_method; end
+
+    class << self
+      def load; end
+    end
+  end
+
+  module UserModule
+    def from_module; end
+  end
+
+  def setup
+    @user = User.new.extend(UserModule)
+  end
+
+  def test_corrections_include_instance_method
+    error = assert_raise(NoMethodError){ @user.flrst_name }
+
+    assert_correction :first_name, error.corrections
+    assert_match "Did you mean?  first_name",  error.to_s
+  end
+
+  def test_corrections_include_private_method
+    error = assert_raise(NoMethodError){ @user.friend }
+
+    assert_correction :friends, error.corrections
+    assert_match "Did you mean?  friends", error.to_s
+  end
+
+  def test_corrections_include_method_from_module
+    error = assert_raise(NoMethodError){ @user.fr0m_module }
+
+    assert_correction :from_module, error.corrections
+    assert_match "Did you mean?  from_module", error.to_s
+  end
+
+  def test_corrections_include_class_method
+    error = assert_raise(NoMethodError){ User.l0ad }
+
+    assert_correction :load, error.corrections
+    assert_match "Did you mean?  load", error.to_s
+  end
+
+  def test_private_methods_should_not_be_suggested
+    error = assert_raise(NoMethodError){ User.new.the_protected_method }
+    refute_includes error.corrections, :the_protected_method
+
+    error = assert_raise(NoMethodError){ User.new.the_private_method }
+    refute_includes error.corrections, :the_private_method
+  end
+
+  def test_corrections_when_private_method_is_called_with_args
+    error = assert_raise(NoMethodError){ @user.call_incorrect_private_method }
+
+    assert_correction :raise, error.corrections
+    assert_match "Did you mean?  raise", error.to_s
+  end
+
+  def test_exclude_methods_on_nil
+    error = assert_raise(NoMethodError){ nil.map }
+    assert_empty error.corrections
+  end
+
+  def test_does_not_exclude_custom_methods_on_nil
+    def nil.empty?
+    end
+
+    error = assert_raise(NoMethodError){ nil.empty }
+    assert_correction :empty?, error.corrections
+  ensure
+    NilClass.class_eval { undef empty? }
+  end
+
+  def test_does_not_append_suggestions_twice
+    error = assert_raise NoMethodError do
+      begin
+        @user.firstname
+      rescue NoMethodError => e
+        raise e, e.message, e.backtrace
+      end
+    end
+
+    assert_equal 1, error.to_s.scan(/Did you mean/).count
+  end
+
+  def test_does_not_append_suggestions_three_times
+    error = assert_raise NoMethodError do
+      begin
+        @user.raise_no_method_error
+      rescue NoMethodError => e
+        raise e, e.message, e.backtrace
+      end
+    end
+
+    assert_equal 1, error.to_s.scan(/Did you mean/).count
+  end
+
+  def test_suggests_corrections_on_nested_error
+    error = assert_raise NoMethodError do
+      begin
+        @user.firstname
+      rescue NoMethodError
+        @user.firstname
+      end
+    end
+
+    assert_equal 1, error.to_s.scan(/Did you mean/).count
+  end
+
+  def test_suggests_yield
+    error = assert_raise(NoMethodError) { yeild(1) }
+
+    assert_correction :yield, error.corrections
+    assert_match "Did you mean?  yield", error.to_s
+  end
+end