From 724b208a5d03d2a5e81ab406411296e98fd6d528 Mon Sep 17 00:00:00 2001
From: Dmitry Kovanikov <kovanikov@gmail.com>
Date: Fri, 4 Sep 2015 12:41:57 +0300
Subject: [PATCH] [test] Add readLine() tests

---
 .../kotlin/idea/console/KotlinReplTest.kt     | 47 +++++++++++++++----
 1 file changed, 39 insertions(+), 8 deletions(-)

diff --git a/idea/tests/org/jetbrains/kotlin/idea/console/KotlinReplTest.kt b/idea/tests/org/jetbrains/kotlin/idea/console/KotlinReplTest.kt
index b77245c5d7d..f63a17e564a 100644
--- a/idea/tests/org/jetbrains/kotlin/idea/console/KotlinReplTest.kt
+++ b/idea/tests/org/jetbrains/kotlin/idea/console/KotlinReplTest.kt
@@ -42,6 +42,19 @@ public class KotlinReplTest : PlatformTestCase() {
         super.tearDown()
     }
 
+    private fun sendCommand(command: String) {
+        consoleRunner.consoleView.editorDocument.setText(command)
+        consoleRunner.executor.executeCommand()
+    }
+
+    private fun waitForExpectedOutput(expectedOutput: String) {
+        val endsWithPredicate = { x: String -> x.trim().endsWith(expectedOutput) }
+
+        val historyText = checkHistoryUpdate { endsWithPredicate(it) }
+
+        assertTrue(endsWithPredicate(historyText), "'$expectedOutput' should be printed but document text is:\n$historyText")
+    }
+
     private fun checkHistoryUpdate(maxIterations: Int = 20, sleepTime: Long = 500, stopPredicate: (String) -> Boolean): String {
         val consoleView = consoleRunner.consoleView as LanguageConsoleImpl
 
@@ -59,14 +72,8 @@ public class KotlinReplTest : PlatformTestCase() {
     }
 
     private fun testSimpleCommand(command: String, expectedOutput: String) {
-        consoleRunner.consoleView.editorDocument.setText(command)
-        consoleRunner.executor.executeCommand()
-
-        val endsWithTwo = { x: String -> x.trim().endsWith(expectedOutput) }
-
-        val historyText = checkHistoryUpdate { endsWithTwo(it) }
-
-        assertTrue(endsWithTwo(historyText), "'$expectedOutput' should be printed but document text is:\n$historyText")
+        sendCommand(command)
+        waitForExpectedOutput(expectedOutput)
     }
 
     @Test fun testRunPossibility() {
@@ -83,4 +90,28 @@ public class KotlinReplTest : PlatformTestCase() {
     @Test fun testOnePlusOne() = testSimpleCommand("1 + 1", "2")
     @Test fun testPrintlnText() = "Hello, console world!" let { testSimpleCommand("println(\"$it\")", it) }
     @Test fun testDivisionByZeroException() = testSimpleCommand("1 / 0", "java.lang.ArithmeticException: / by zero")
+
+    @Test fun testReadLineSingle() {
+        val readLineText = "ReadMe!"
+
+        sendCommand("val a = readLine()")
+        sendCommand(readLineText)
+        sendCommand("a")
+        waitForExpectedOutput(readLineText)
+    }
+
+    @Test fun testReadLineMultiple() {
+        val readLineTextA = "ReadMe A!"
+        val readLineTextB = "ReadMe B!"
+
+        sendCommand("val a = readLine()\n" +
+                    "val b = readLine()")
+        sendCommand(readLineTextA)
+        sendCommand(readLineTextB)
+
+        sendCommand("a")
+        waitForExpectedOutput(readLineTextA)
+        sendCommand("b")
+        waitForExpectedOutput(readLineTextB)
+    }
 }
\ No newline at end of file