From 5a48f20c33ad9a70cfc0353689a63d044e2412d2 Mon Sep 17 00:00:00 2001
From: nobu <nobu@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Mon, 30 Jan 2017 03:21:15 +0000
Subject: test for prompt

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@57457 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/irb/test_ruby-lex.rb | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

(limited to 'test/irb')

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 = []
-- 
cgit v1.2.3