1 files changed, 73 insertions, 0 deletions
diff --git a/test/readline/test_readline.rb b/test/readline/test_readline.rb
new file mode 100644
index 000000000..e7093f7bd
--- /dev/null
+++ b/test/readline/test_readline.rb
@@ -0,0 +1,73 @@
+begin
+  require "readline"
+rescue LoadError
+end
+
+if defined?(Readline)
+
+require "test/unit"
+require "tempfile"
+
+class TestReadline < Test::Unit::TestCase
+  def test_readline
+    stdin = Tempfile.new("test_readline_stdin")
+    stdout = Tempfile.new("test_readline_stdout")
+    begin
+      stdin.write("hello\n")
+      stdin.close
+      stdout.close
+      line = replace_stdio(stdin.path, stdout.path) { Readline.readline("> ") }
+      assert_equal("hello", line)
+      assert_equal(true, line.tainted?)
+      stdout.open
+      assert_equal("> ", stdout.read(2))
+      assert_raises(SecurityError) do
+        Thread.start {
+          $SAFE = 1
+          replace_stdio(stdin.path, stdout.path) do
+            Readline.readline("> ".taint)
+          end
+        }.join
+      end
+      assert_raises(SecurityError) do
+        Thread.start {
+          $SAFE = 4
+          replace_stdio(stdin.path, stdout.path) { Readline.readline("> ") }
+        }.join
+      end
+    ensure
+      stdin.close(true)
+      stdout.close(true)
+    end
+  end
+
+  def test_completion_append_character
+    Readline.completion_append_character = "x"
+    assert_equal("x", Readline.completion_append_character)
+    Readline.completion_append_character = "xyz"
+    assert_equal("x", Readline.completion_append_character)
+    Readline.completion_append_character = nil
+    assert_equal(nil, Readline.completion_append_character)
+    Readline.completion_append_character = ""
+    assert_equal(nil, Readline.completion_append_character)
+  end
+
+  private
+
+  def replace_stdio(stdin_path, stdout_path)
+    orig_stdin = STDIN.dup
+    orig_stdout = STDOUT.dup
+    STDIN.reopen(stdin_path, "r")
+    STDOUT.reopen(stdout_path, "w")
+    begin
+      yield
+    ensure
+      STDIN.reopen(orig_stdin)
+      STDOUT.reopen(orig_stdout)
+      orig_stdin.close
+      orig_stdout.close
+    end
+  end
+end
+
+end