free_mutant/lib/mutant/mutator.rb
2015-11-15 23:13:43 +00:00

118 lines
2.2 KiB
Ruby

module Mutant
# Generator for mutations
class Mutator
include Adamantium::Flat, AbstractType
# Run mutator on input
#
# @param [Object] input
# the input to mutate
#
# @param [Mutator] parent
#
# @return [self]
def self.each(input, parent = nil, &block)
return to_enum(__method__, input, parent) unless block_given?
REGISTRY.lookup(input).new(input, parent, block)
self
end
# Register node class handler
#
# @return [undefined]
def self.handle(*types)
types.each do |type|
REGISTRY.register(type, self)
end
end
private_class_method :handle
# Mutation input
#
# @return [Object]
attr_reader :input
# Parent context of input
#
# @return [Object]
attr_reader :parent
private
# Initialize object
#
# @param [Object] input
# @param [Object] parent
# @param [#call(node)] block
#
# @return [undefined]
def initialize(input, parent, block)
@input, @parent, @block = input, parent, block
@seen = Set.new
guard(input)
dispatch
end
# Test if generated object is not guarded from emitting
#
# @param [Object] object
#
# @return [Boolean]
def new?(object)
!@seen.include?(object)
end
# Add object to guarded values
#
# @param [Object] object
#
# @return [undefined]
def guard(object)
@seen << object
end
# Dispatch node generations
#
# @return [undefined]
abstract_method :dispatch
# Emit generated mutation if object is not equivalent to input
#
# @param [Object] object
#
# @return [undefined]
def emit(object)
return unless new?(object)
guard(object)
emit!(object)
end
# Call block with node
#
# @param [Parser::AST::Node] node
#
# @return [self]
def emit!(node)
@block.call(node)
self
end
# Run input with mutator
#
# @return [undefined]
def run(mutator)
mutator.new(input, self, method(:emit))
end
# Shortcut to create a new unfrozen duplicate of input
#
# @return [Object]
def dup_input
input.dup
end
end # Mutator
end # Mutant