activesupport/test/core_ext/module/remove_method_test.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

# frozen_string_literal: true
require "abstract_unit"
require "active_support/core_ext/module/remove_method"

module RemoveMethodTests
  class A
    def do_something
      return 1
    end

    def do_something_protected
      return 1
    end
    protected :do_something_protected

    def do_something_private
      return 1
    end
    private :do_something_private

    class << self
      def do_something_else
        return 2
      end
    end
  end
end

class RemoveMethodTest < ActiveSupport::TestCase
  def test_remove_method_from_an_object
    RemoveMethodTests::A.class_eval {
      remove_possible_method(:do_something)
    }
    assert !RemoveMethodTests::A.new.respond_to?(:do_something)
  end

  def test_remove_singleton_method_from_an_object
    RemoveMethodTests::A.class_eval {
      remove_possible_singleton_method(:do_something_else)
    }
    assert !RemoveMethodTests::A.respond_to?(:do_something_else)
  end

  def test_redefine_method_in_an_object
    RemoveMethodTests::A.class_eval {
      redefine_method(:do_something) { return 100 }
      redefine_method(:do_something_protected) { return 100 }
      redefine_method(:do_something_private) { return 100 }
    }
    assert_equal 100, RemoveMethodTests::A.new.do_something
    assert_equal 100, RemoveMethodTests::A.new.send(:do_something_protected)
    assert_equal 100, RemoveMethodTests::A.new.send(:do_something_private)

    assert RemoveMethodTests::A.public_method_defined? :do_something
    assert RemoveMethodTests::A.protected_method_defined? :do_something_protected
    assert RemoveMethodTests::A.private_method_defined? :do_something_private
  end
end