module Mutant
  class Mutator
    class Node
      class Define < self

        handle(Rubinius::AST::Define)
        handle(Rubinius::AST::DefineSingletonScope)

      private

        # Emit mutations
        #
        # @return [undefined]
        #
        # @api private
        #
        def dispatch
          emit_attribute_mutations(:body)
          emit_attribute_mutations(:arguments)
        end
      end

      class DefineSingleton < self

        handle(Rubinius::AST::DefineSingleton)

      private

        # Emit mutations
        #
        # @return [undefined]
        #
        # @api private
        #
        def dispatch
          emit_attribute_mutations(:body)
        end
      end
    end
  end
end