Sha256: ba6bc93d242fc3df8c10848f43a2d6aa545efb815fb7d72378ce53429c1af237
Contents?: true
Size: 1.31 KB
Versions: 9
Compression:
Stored size: 1.31 KB
Contents
require 'active_support'
class Class
def add_allow_switch(method, options={})
default = options[:default] || false
class_eval do
cattr_accessor "allow_#{method}"
self.send("allow_#{method}=", default)
alias_method "original_#{method}", method
eval %{
def #{method}(*args)
if allow_#{method}
original_#{method}(*args)
else
raise RuntimeError, "You're trying to call `#{method}' on `#{self}', which you probably don't want in a test."
end
end
}, binding, __FILE__, __LINE__
end
end
end
class Module
def add_allow_switch(method, options={})
default = options[:default] || false
mattr_accessor "allow_#{method}"
send("allow_#{method}=", default)
unless respond_to?(:__metaclass___)
def __metaclass__
class << self; self; end
end
end
__metaclass__.class_eval do
alias_method "original_#{method}", method
eval %{
def #{method}(*args)
if allow_#{method}
original_#{method}(*args)
else
raise RuntimeError, "You're trying to call `#{method}' on `#{self}', which you probably don't want in a test."
end
end
}, binding, __FILE__, __LINE__
end
end
end
Version data entries
9 entries across 9 versions & 3 rubygems