1 files changed, 147 insertions, 0 deletions
diff --git a/test/rubygems/test_gem_commands_sources_command.rb b/test/rubygems/test_gem_commands_sources_command.rb
new file mode 100644
index 0000000000..3d1ab801b1
--- /dev/null
+++ b/test/rubygems/test_gem_commands_sources_command.rb
@@ -0,0 +1,147 @@
+require 'test/unit'
+require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
+require 'rubygems/commands/sources_command'
+
+class TestGemCommandsSourcesCommand < RubyGemTestCase
+
+  def setup
+    super
+
+    @cmd = Gem::Commands::SourcesCommand.new
+  end
+
+  def test_execute
+    util_setup_source_info_cache
+    @cmd.handle_options []
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = <<-EOF
+*** CURRENT SOURCES ***
+
+#{@gem_repo}
+    EOF
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+  end
+
+  def test_execute_add
+    util_setup_fake_fetcher
+
+    @si = Gem::SourceIndex.new @gem1.full_name => @gem1.name
+
+    @fetcher.data["http://beta-gems.example.com/Marshal.#{@marshal_version}"] =
+      @si.dump
+
+    @cmd.handle_options %w[--add http://beta-gems.example.com]
+
+    util_setup_source_info_cache
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = <<-EOF
+Bulk updating Gem source index for: http://beta-gems.example.com
+http://beta-gems.example.com added to sources
+    EOF
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+
+    Gem::SourceInfoCache.cache.flush
+    assert_equal %W[http://beta-gems.example.com #{@gem_repo}],
+                 Gem::SourceInfoCache.cache_data.keys.sort
+  end
+
+  def test_execute_add_nonexistent_source
+    util_setup_fake_fetcher
+
+    @si = Gem::SourceIndex.new @gem1.full_name => @gem1.name
+
+    @fetcher.data["http://beta-gems.example.com/Marshal.#{@marshal_version}"] =
+      proc do
+        raise Gem::RemoteFetcher::FetchError, 'it died'
+      end
+
+
+    Gem::RemoteFetcher.instance_variable_set :@fetcher, @fetcher
+
+    @cmd.handle_options %w[--add http://beta-gems.example.com]
+
+    util_setup_source_info_cache
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = <<-EOF
+Error fetching http://beta-gems.example.com:
+\tit died
+    EOF
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+  end
+
+  def test_execute_add_bad_uri
+    @cmd.handle_options %w[--add beta-gems.example.com]
+
+    util_setup_source_info_cache
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = <<-EOF
+beta-gems.example.com is not a URI
+    EOF
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+  end
+
+  def test_execute_remove
+    @cmd.handle_options %W[--remove #{@gem_repo}]
+
+    util_setup_source_info_cache
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = "#{@gem_repo} removed from sources\n"
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+
+    Gem::SourceInfoCache.cache.flush
+    assert_equal [], Gem::SourceInfoCache.cache_data.keys
+  end
+
+  def test_execute_update
+    @cmd.handle_options %w[--update]
+
+    util_setup_source_info_cache
+    util_setup_fake_fetcher
+    @si = Gem::SourceIndex.new @gem1.full_name => @gem1.name
+    @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = @si.dump
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    expected = <<-EOF
+Bulk updating Gem source index for: #{@gem_repo}
+source cache successfully updated
+    EOF
+
+    assert_equal expected, @ui.output
+    assert_equal '', @ui.error
+  end
+
+end
+