diff options
-rw-r--r-- | test/irb/test_ruby-lex.rb | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/test/irb/test_ruby-lex.rb b/test/irb/test_ruby-lex.rb index 79528676e4..d1355549b7 100644 --- a/test/irb/test_ruby-lex.rb +++ b/test/irb/test_ruby-lex.rb @@ -9,6 +9,10 @@ module TestIRB @scanner = RubyLex.new end + def teardown + RubyLex.debug_level = 0 + end + def test_set_input_proc called = false @scanner.set_input(nil) {called = true; nil} @@ -40,6 +44,49 @@ module TestIRB assert_equal([[src, 1]], top_level_statement(src)) end + def test_prompt + prompts = [] + @scanner.set_prompt {|*a| + a << @scanner.instance_variable_get(:@lex_state) + unless prompts.last == a + prompts << a + end + } + src = "#{<<-"begin;"}#{<<~"end;"}" + begin; + if false or + true + " + " + ' + ' + else + nil + nil + end + end; + assert_equal([[src, 1]], top_level_statement(src)) + expected = [ + [nil, 0, false], + [nil, 1, true], + [nil, 1, false], + ['"', 1, false], + [nil, 1, false], + ["'", 1, false], + [nil, 1, false], + [nil, 1, true], # FIXME: just after `else' should be `false' + [nil, 1, false], + [nil, 1, false], + [nil, 0, false], + ] + srcs = src.lines + assert_equal(expected.size, prompts.size) + expected.each_with_index {|e, i| + assert_equal(i + 1, prompts[i][3]) + assert_equal(e, prompts[i][0..2], "#{i+1}: #{srcs[i]} # #{prompts[i]}") + } + end + def top_level_statement(lines) input = InputLines.new(lines, "r") scanned = [] |