1 files changed, 117 insertions, 0 deletions
diff --git a/test/rubygems/test_gem_stream_ui.rb b/test/rubygems/test_gem_stream_ui.rb
new file mode 100644
index 0000000000..a8564ba94f
--- /dev/null
+++ b/test/rubygems/test_gem_stream_ui.rb
@@ -0,0 +1,117 @@
+require 'test/unit'
+require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
+require 'rubygems/user_interaction'
+
+class TestGemStreamUI < RubyGemTestCase
+
+  module IsTty
+    attr_accessor :tty
+
+    def tty?
+      @tty = true unless defined? @tty
+      return @tty
+    end
+
+    alias_method :isatty, :tty?
+  end
+
+  def setup
+    super
+
+    @cfg = Gem.configuration
+
+    @in = StringIO.new
+    @out = StringIO.new
+    @err = StringIO.new
+
+    @in.extend IsTty
+
+    @sui = Gem::StreamUI.new @in, @out, @err
+  end
+
+  def test_ask
+    timeout(1) do
+      expected_answer = "Arthur, King of the Britons"
+      @in.string = "#{expected_answer}\n"
+      actual_answer = @sui.ask("What is your name?")
+      assert_equal expected_answer, actual_answer
+    end
+  end
+
+  def test_ask_no_tty
+    @in.tty = false
+
+    timeout(0.1) do
+      answer = @sui.ask("what is your favorite color?")
+      assert_equal nil, answer
+    end
+  end
+
+  def test_ask_yes_no_no_tty_with_default
+    @in.tty = false
+
+    timeout(0.1) do
+      answer = @sui.ask_yes_no("do coconuts migrate?", false)
+      assert_equal false, answer
+
+      answer = @sui.ask_yes_no("do coconuts migrate?", true)
+      assert_equal true, answer
+    end
+  end
+
+  def test_ask_yes_no_no_tty_without_default
+    @in.tty = false
+
+    timeout(0.1) do
+      assert_raises(Gem::OperationNotSupportedError) do
+        @sui.ask_yes_no("do coconuts migrate?")
+      end
+    end
+  end
+
+  def test_choose_from_list
+    @in.puts "1"
+    @in.rewind
+
+    result = @sui.choose_from_list 'which one?', %w[foo bar]
+
+    assert_equal ['foo', 0], result
+    assert_equal "which one?\n 1. foo\n 2. bar\n> ", @out.string
+  end
+
+  def test_choose_from_list_EOF
+    result = @sui.choose_from_list 'which one?', %w[foo bar]
+
+    assert_equal [nil, nil], result
+    assert_equal "which one?\n 1. foo\n 2. bar\n> ", @out.string
+  end
+
+  def test_proress_reporter_silent_nil
+    @cfg.verbose = nil
+    reporter = @sui.progress_reporter 10, 'hi'
+    assert_kind_of Gem::StreamUI::SilentProgressReporter, reporter
+  end
+
+  def test_proress_reporter_silent_false
+    @cfg.verbose = false
+    reporter = @sui.progress_reporter 10, 'hi'
+    assert_kind_of Gem::StreamUI::SilentProgressReporter, reporter
+    assert_equal "", @out.string
+  end
+
+  def test_proress_reporter_simple
+    @cfg.verbose = true
+    reporter = @sui.progress_reporter 10, 'hi'
+    assert_kind_of Gem::StreamUI::SimpleProgressReporter, reporter
+    assert_equal "hi\n", @out.string
+  end
+
+  def test_proress_reporter_verbose
+    @cfg.verbose = 0
+    reporter = @sui.progress_reporter 10, 'hi'
+    assert_kind_of Gem::StreamUI::VerboseProgressReporter, reporter
+    assert_equal "hi\n", @out.string
+  end
+
+end
+