free_mutant/lib/mutant/meta/example/dsl.rb

module Mutant
  module Meta
    class Example

      # Example DSL
      class DSL
        include AST::Sexp

        # Run DSL on block
        #
        # @return [Example]
        #
        # @api private
        def self.run(file, block)
          instance = new(file)
          instance.instance_eval(&block)
          instance.example
        end

        # Initialize DSL context
        #
        # @return [undefined]
        #
        # @api private
        def initialize(file)
          @file = file
          @source = nil
          @expected = []
        end

        # Example captured by DSL
        #
        # @return [Example]
        #
        # @raise [RuntimeError]
        #   in case example cannot be build
        #
        # @api private
        def example
          fail 'source not defined' unless @source
          Example.new(@file, @source, @expected)
        end

      private

        # Set original source
        #
        # @param [String,Parser::AST::Node] input
        #
        # @return [self]
        #
        # @api private
        def source(input)
          fail 'source already defined' if @source
          @source = node(input)

          self
        end

        # Add expected mutation
        #
        # @param [String,Parser::AST::Node] input
        #
        # @return [self]
        #
        # @api private
        def mutation(input)
          node = node(input)
          if @expected.include?(node)
            fail "Node for input: #{input.inspect} is already expected"
          end
          @expected << node

          self
        end

        # Add singleton mutations
        #
        # @return [undefined]
        #
        # @api private
        def singleton_mutations
          mutation('nil')
          mutation('self')
        end

        # Helper method to coerce input to node
        #
        # @param [String,Parser::AST::Node] input
        #
        # @return [Parser::AST::Node]
        #
        # @raise [RuntimeError]
        #   in case input cannot be coerced
        #
        # @api private
        def node(input)
          case input
          when String
            Unparser::Preprocessor.run(Parser::CurrentRuby.parse(input))
          when Parser::AST::Node
            input
          else
            fail "Cannot coerce to node: #{source.inspect}"
          end
        end

      end # DSL
    end # Example
  end # Meta
end # Mutant