зеркало из https://github.com/github/ruby.git
117 строки
2.6 KiB
Ruby
117 строки
2.6 KiB
Ruby
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
|
|
|