60 lines
1.6 KiB
Ruby
60 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../../abstract_unit"
|
|
require "active_support/core_ext/module/remove_method"
|
|
|
|
module RemoveMethodTests
|
|
class A
|
|
def do_something
|
|
1
|
|
end
|
|
|
|
def do_something_protected
|
|
1
|
|
end
|
|
protected :do_something_protected
|
|
|
|
def do_something_private
|
|
1
|
|
end
|
|
private :do_something_private
|
|
|
|
class << self
|
|
def do_something_else
|
|
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_not_respond_to RemoveMethodTests::A.new, :do_something
|
|
end
|
|
|
|
def test_remove_singleton_method_from_an_object
|
|
RemoveMethodTests::A.class_eval {
|
|
remove_possible_singleton_method(:do_something_else)
|
|
}
|
|
assert_not_respond_to RemoveMethodTests::A, :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
|