152 lines
3 KiB
Ruby
152 lines
3 KiB
Ruby
module Mutant
|
|
class Matcher
|
|
# Abstract base class for matcher that returns method subjects extracted from scope
|
|
class Methods < self
|
|
include AbstractType, Concord.new(:scope)
|
|
|
|
# Enumerate subjects
|
|
#
|
|
# @return [self]
|
|
# if block given
|
|
#
|
|
# @return [Enumerator<Subject>]
|
|
# otherwise
|
|
#
|
|
# @api private
|
|
#
|
|
def each(&block)
|
|
return to_enum unless block_given?
|
|
|
|
methods.each do |method|
|
|
emit_matches(method, &block)
|
|
end
|
|
|
|
self
|
|
end
|
|
|
|
# Return method matcher class
|
|
#
|
|
# @return [Class:Matcher::Method]
|
|
#
|
|
# @api private
|
|
#
|
|
def matcher
|
|
self.class::MATCHER
|
|
end
|
|
|
|
# Return methods
|
|
#
|
|
# @return [Enumerable<Method, UnboundMethod>]
|
|
#
|
|
# @api private
|
|
#
|
|
def methods
|
|
candidate_names.each_with_object([]) do |name, methods|
|
|
method = access(name)
|
|
methods << method if method.owner == candidate_scope
|
|
end
|
|
end
|
|
memoize :methods
|
|
|
|
private
|
|
|
|
# Emit matches for method
|
|
#
|
|
# @param [UnboundMethod, Method] method
|
|
#
|
|
# @return [undefined]
|
|
#
|
|
# @api private
|
|
#
|
|
def emit_matches(method)
|
|
matcher.new(scope, method).each do |subject|
|
|
yield subject
|
|
end
|
|
end
|
|
|
|
# Return candidate names
|
|
#
|
|
# @param [Object] object
|
|
#
|
|
# @return [Enumerable<Symbol>]
|
|
#
|
|
# @api private
|
|
#
|
|
def candidate_names
|
|
object = candidate_scope
|
|
names =
|
|
object.public_instance_methods(false) +
|
|
object.private_instance_methods(false) +
|
|
object.protected_instance_methods(false)
|
|
names.sort
|
|
end
|
|
|
|
# Return candidate scope
|
|
#
|
|
# @return [Class, Module]
|
|
#
|
|
# @api private
|
|
#
|
|
abstract_method :candidate_scope
|
|
|
|
class Singleton < self
|
|
MATCHER = Matcher::Method::Singleton
|
|
|
|
private
|
|
|
|
# Return method for name
|
|
#
|
|
# @param [Symbol] method_name
|
|
#
|
|
# @return [Method]
|
|
#
|
|
# @api private
|
|
#
|
|
def access(method_name)
|
|
scope.method(method_name)
|
|
end
|
|
|
|
# Return candidate scope
|
|
#
|
|
# @return [Class]
|
|
#
|
|
# @api private
|
|
#
|
|
def candidate_scope
|
|
scope.singleton_class
|
|
end
|
|
memoize :candidate_scope, :freezer => :noop
|
|
|
|
end
|
|
|
|
class Instance < self
|
|
MATCHER = Matcher::Method::Instance
|
|
|
|
private
|
|
|
|
# Return method for name
|
|
#
|
|
# @param [Symbol] method_name
|
|
#
|
|
# @return [UnboundMethod]
|
|
#
|
|
# @api private
|
|
#
|
|
def access(method_name)
|
|
scope.instance_method(method_name)
|
|
end
|
|
|
|
# Return candidate scope
|
|
#
|
|
# @return [Class, Module]
|
|
#
|
|
# @api private
|
|
#
|
|
def candidate_scope
|
|
scope
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|
|
end
|