class T::Private::ClassUtils::ReplacedMethod

def restore
  # The check below would also catch this, but this makes the failure mode much clearer
  if @restored
    raise "Method '#{@name}' on '#{@mod}' was already restored"
  end
  if @mod.instance_method(@name) != @new_method
    raise "Trying to restore #{@mod}##{@name} but the method has changed since the call to replace_method"
  end
  @restored = true
  if @overwritten
    # The original method was overwritten. Overwrite again to restore it.
    T::Configuration.without_ruby_warnings do
      @mod.send(:define_method, @old_method.name, @old_method)
    end
  else
    # The original method was in an ancestor. Restore it by removing the overriding method.
    @mod.send(:remove_method, @old_method.name)
  end
  # Restore the visibility. Note that we need to do this even when we call remove_method
  # above, because the module may have set custom visibility for a method it inherited.
  @mod.send(@visibility, @old_method.name)
  nil
end