blob: e9975b346e8e31da5b7f895e68637b85437b25e9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
# frozen_string_literal: true
require_relative "test_helper"
return unless defined?(RubyVM::InstructionSequence)
module Prism
class NewlineTest < TestCase
base = File.expand_path("../", __FILE__)
filepaths = Dir["*.rb", base: base] - %w[encoding_test.rb errors_test.rb parser_test.rb static_literals_test.rb unescape_test.rb]
filepaths.each do |relative|
define_method("test_newline_flags_#{relative}") do
assert_newlines(base, relative)
end
end
private
def assert_newlines(base, relative)
filepath = File.join(base, relative)
source = File.read(filepath, binmode: true, external_encoding: Encoding::UTF_8)
expected = rubyvm_lines(source)
result = Prism.parse_file(filepath)
assert_empty result.errors
actual = prism_lines(result)
source.each_line.with_index(1) do |line, line_number|
# Lines like `while (foo = bar)` result in two line flags in the
# bytecode but only one newline flag in the AST. We need to remove the
# extra line flag from the bytecode to make the test pass.
if line.match?(/while \(/)
index = expected.index(line_number)
expected.delete_at(index) if index
end
# Lines like `foo =` where the value is on the next line result in
# another line flag in the bytecode but only one newline flag in the
# AST.
if line.match?(/^\s+\w+ =$/)
if source.lines[line_number].match?(/^\s+case/)
actual[actual.index(line_number)] += 1
else
actual.delete_at(actual.index(line_number))
end
end
if line.match?(/^\s+\w+ = \[$/)
if !expected.include?(line_number) && !expected.include?(line_number + 2)
actual[actual.index(line_number)] += 1
end
end
end
assert_equal expected, actual
end
def ignore_warnings
previous_verbosity = $VERBOSE
$VERBOSE = nil
yield
ensure
$VERBOSE = previous_verbosity
end
def rubyvm_lines(source)
queue = [ignore_warnings { RubyVM::InstructionSequence.compile(source) }]
lines = []
while iseq = queue.shift
lines.concat(iseq.trace_points.filter_map { |line, event| line if event == :line })
iseq.each_child { |insn| queue << insn unless insn.label.start_with?("ensure in ") }
end
lines.sort
end
def prism_lines(result)
result.mark_newlines!
queue = [result.value]
newlines = []
while node = queue.shift
queue.concat(node.compact_child_nodes)
newlines << result.source.line(node.location.start_offset) if node&.newline?
end
newlines.sort
end
end
end
|
