Skip to content

Commit

Permalink
Add assertions for nesting level of prompt
Browse files Browse the repository at this point in the history
  • Loading branch information
@aycabta
aycabta committed Aug 17, 2020
1 parent f1a775a commit 46230bc
Showing 1 changed file with 91 additions and 81 deletions.
172 changes: 91 additions & 81 deletions test/irb/test_ruby_lex.rb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -45,202 +45,212 @@ def assert_nesting_level(lines, expected)

def test_auto_indent
input_with_correct_indents = [
Row.new(%q(def each_top_level_statement), nil, 2),
Row.new(%q( initialize_input), nil, 2),
Row.new(%q( catch(:TERM_INPUT) do), nil, 4),
Row.new(%q( loop do), nil, 6),
Row.new(%q( begin), nil, 8),
Row.new(%q( prompt), nil, 8),
Row.new(%q( unless l = lex), nil, 10),
Row.new(%q( throw :TERM_INPUT if @line == ''), nil, 10),
Row.new(%q( else), 8, 10),
Row.new(%q( @line_no += l.count("\n")), nil, 10),
Row.new(%q( next if l == "\n"), nil, 10),
Row.new(%q( @line.concat l), nil, 10),
Row.new(%q( if @code_block_open or @ltype or @continue or @indent > 0), nil, 12),
Row.new(%q( next), nil, 12),
Row.new(%q( end), 10, 10),
Row.new(%q( end), 8, 8),
Row.new(%q( if @line != "\n"), nil, 10),
Row.new(%q( @line.force_encoding(@io.encoding)), nil, 10),
Row.new(%q( yield @line, @exp_line_no), nil, 10),
Row.new(%q( end), 8, 8),
Row.new(%q( break if @io.eof?), nil, 8),
Row.new(%q( @line = ''), nil, 8),
Row.new(%q( @exp_line_no = @line_no), nil, 8),
Row.new(%q( ), nil, 8),
Row.new(%q( @indent = 0), nil, 8),
Row.new(%q( rescue TerminateLineInput), 6, 8),
Row.new(%q( initialize_input), nil, 8),
Row.new(%q( prompt), nil, 8),
Row.new(%q( end), 6, 6),
Row.new(%q( end), 4, 4),
Row.new(%q( end), 2, 2),
Row.new(%q(end), 0, 0),
Row.new(%q(def each_top_level_statement), nil, 2, 1),
Row.new(%q( initialize_input), nil, 2, 1),
Row.new(%q( catch(:TERM_INPUT) do), nil, 4, 2),
Row.new(%q( loop do), nil, 6, 3),
Row.new(%q( begin), nil, 8, 4),
Row.new(%q( prompt), nil, 8, 4),
Row.new(%q( unless l = lex), nil, 10, 5),
Row.new(%q( throw :TERM_INPUT if @line == ''), nil, 10, 5),
Row.new(%q( else), 8, 10, 5),
Row.new(%q( @line_no += l.count("\n")), nil, 10, 5),
Row.new(%q( next if l == "\n"), nil, 10, 5),
Row.new(%q( @line.concat l), nil, 10, 5),
Row.new(%q( if @code_block_open or @ltype or @continue or @indent > 0), nil, 12, 6),
Row.new(%q( next), nil, 12, 6),
Row.new(%q( end), 10, 10, 5),
Row.new(%q( end), 8, 8, 4),
Row.new(%q( if @line != "\n"), nil, 10, 5),
Row.new(%q( @line.force_encoding(@io.encoding)), nil, 10, 5),
Row.new(%q( yield @line, @exp_line_no), nil, 10, 5),
Row.new(%q( end), 8, 8, 4),
Row.new(%q( break if @io.eof?), nil, 8, 4),
Row.new(%q( @line = ''), nil, 8, 4),
Row.new(%q( @exp_line_no = @line_no), nil, 8, 4),
Row.new(%q( ), nil, 8, 4),
Row.new(%q( @indent = 0), nil, 8, 4),
Row.new(%q( rescue TerminateLineInput), 6, 8, 4),
Row.new(%q( initialize_input), nil, 8, 4),
Row.new(%q( prompt), nil, 8, 4),
Row.new(%q( end), 6, 6, 3),
Row.new(%q( end), 4, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q(end), 0, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_braces_on_their_own_line
input_with_correct_indents = [
Row.new(%q(if true), nil, 2),
Row.new(%q( [), nil, 4),
Row.new(%q( ]), 2, 2),
Row.new(%q(end), 0, 0),
Row.new(%q(if true), nil, 2, 1),
Row.new(%q( [), nil, 4, 2),
Row.new(%q( ]), 2, 2, 1),
Row.new(%q(end), 0, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_multiple_braces_in_a_line
input_with_correct_indents = [
Row.new(%q([[[), nil, 6),
Row.new(%q( ]), 4, 4),
Row.new(%q( ]), 2, 2),
Row.new(%q(]), 0, 0),
Row.new(%q([<<FOO]), nil, 0),
Row.new(%q(hello), nil, 0),
Row.new(%q(FOO), nil, 0),
Row.new(%q([[[), nil, 6, 3),
Row.new(%q( ]), 4, 4, 2),
Row.new(%q( ]), 2, 2, 1),
Row.new(%q(]), 0, 0, 0),
Row.new(%q([<<FOO]), nil, 0, -1),
Row.new(%q(hello), nil, 0, -1),
Row.new(%q(FOO), nil, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_a_closed_brace_and_not_closed_brace_in_a_line
input_with_correct_indents = [
Row.new(%q(p() {), nil, 2),
Row.new(%q(}), 0, 0),
Row.new(%q(p() {), nil, 2, 1),
Row.new(%q(}), 0, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_incomplete_coding_magic_comment
input_with_correct_indents = [
Row.new(%q(#coding:u), nil, 0),
Row.new(%q(#coding:u), nil, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_incomplete_encoding_magic_comment
input_with_correct_indents = [
Row.new(%q(#encoding:u), nil, 0),
Row.new(%q(#encoding:u), nil, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_incomplete_emacs_coding_magic_comment
input_with_correct_indents = [
Row.new(%q(# -*- coding: u), nil, 0),
Row.new(%q(# -*- coding: u), nil, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_incomplete_vim_coding_magic_comment
input_with_correct_indents = [
Row.new(%q(# vim:set fileencoding=u), nil, 0),
Row.new(%q(# vim:set fileencoding=u), nil, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_mixed_rescue
input_with_correct_indents = [
Row.new(%q(def m), nil, 2),
Row.new(%q( begin), nil, 4),
Row.new(%q( begin), nil, 6),
Row.new(%q( x = a rescue 4), nil, 6),
Row.new(%q( y = [(a rescue 5)]), nil, 6),
Row.new(%q( [x, y]), nil, 6),
Row.new(%q( rescue => e), 4, 6),
Row.new(%q( raise e rescue 8), nil, 6),
Row.new(%q( end), 4, 4),
Row.new(%q( rescue), 2, 4),
Row.new(%q( raise rescue 11), nil, 4),
Row.new(%q( end), 2, 2),
Row.new(%q(rescue => e), 0, 2),
Row.new(%q( raise e rescue 14), nil, 2),
Row.new(%q(end), 0, 0),
Row.new(%q(def m), nil, 2, 1),
Row.new(%q( begin), nil, 4, 2),
Row.new(%q( begin), nil, 6, 3),
Row.new(%q( x = a rescue 4), nil, 6, 3),
Row.new(%q( y = [(a rescue 5)]), nil, 6, 3),
Row.new(%q( [x, y]), nil, 6, 3),
Row.new(%q( rescue => e), 4, 6, 3),
Row.new(%q( raise e rescue 8), nil, 6, 3),
Row.new(%q( end), 4, 4, 2),
Row.new(%q( rescue), 2, 4, 2),
Row.new(%q( raise rescue 11), nil, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q(rescue => e), 0, 2, 1),
Row.new(%q( raise e rescue 14), nil, 2, 1),
Row.new(%q(end), 0, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

def test_oneliner_method_definition
input_with_correct_indents = [
Row.new(%q(class A), nil, 2),
Row.new(%q( def foo0), nil, 4),
Row.new(%q( 3), nil, 4),
Row.new(%q( end), 2, 2),
Row.new(%q( def foo1()), nil, 4),
Row.new(%q( 3), nil, 4),
Row.new(%q( end), 2, 2),
Row.new(%q( def foo2(a, b)), nil, 4),
Row.new(%q( a + b), nil, 4),
Row.new(%q( end), 2, 2),
Row.new(%q( def foo3 a, b), nil, 4),
Row.new(%q( a + b), nil, 4),
Row.new(%q( end), 2, 2),
Row.new(%q( def bar0() = 3), nil, 2),
Row.new(%q( def bar1(a) = a), nil, 2),
Row.new(%q( def bar2(a, b) = a + b), nil, 2),
Row.new(%q(end), 0, 0),
Row.new(%q(class A), nil, 2, 1),
Row.new(%q( def foo0), nil, 4, 2),
Row.new(%q( 3), nil, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q( def foo1()), nil, 4, 2),
Row.new(%q( 3), nil, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q( def foo2(a, b)), nil, 4, 2),
Row.new(%q( a + b), nil, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q( def foo3 a, b), nil, 4, 2),
Row.new(%q( a + b), nil, 4, 2),
Row.new(%q( end), 2, 2, 1),
Row.new(%q( def bar0() = 3), nil, 2, 1),
Row.new(%q( def bar1(a) = a), nil, 2, 1),
Row.new(%q( def bar2(a, b) = a + b), nil, 2, 1),
Row.new(%q(end), 0, 0, 0),
]

lines = []
input_with_correct_indents.each do |row|
lines << row.content
assert_indenting(lines, row.current_line_spaces, false)
assert_indenting(lines, row.new_line_spaces, true)
assert_nesting_level(lines, row.nesting_level)
end
end

Expand Down

0 comments on commit 46230bc

Please sign in to comment.