Class: Evilution::Mutator::Operator::BlockPassRemoval

Inherits:
Base
  • Object
show all
Defined in:
lib/evilution/mutator/operator/block_pass_removal.rb

Instance Attribute Summary

Attributes inherited from Base

#mutations

Instance Method Summary collapse

Methods inherited from Base

#call, clear_parse_cache!, #initialize, operator_name, parsed_tree_for

Constructor Details

This class inherits a constructor from Evilution::Mutator::Base

Instance Method Details

#visit_call_node(node) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 
# File 'lib/evilution/mutator/operator/block_pass_removal.rb', line 6

def visit_call_node(node)
  if node.block.is_a?(Prism::BlockArgumentNode)
    block_node = node.block
    call_start = node.location.start_offset
    node_end = call_start + node.location.length
    block_end = block_node.location.start_offset + block_node.location.length

    prefix = @file_source.byteslice(call_start...block_node.location.start_offset).rstrip
    suffix = @file_source.byteslice(block_end...node_end)

    # Clean up: remove trailing comma from prefix, remove empty parens
    prefix = prefix.sub(/,\s*\z/, "")
    replacement = "#{prefix}#{suffix}".sub(/\(\s*\)/, "")

    add_mutation(
      offset: call_start,
      length: node.location.length,
      replacement: replacement,
      node: node
    )
  end

  super
end