test/rubygems/test_gem_commands_sources_command.rb

   1 require 'test/unit'
   2 require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
   3 require 'rubygems/commands/sources_command'
   4
   5 class TestGemCommandsSourcesCommand < RubyGemTestCase
   6
   7   def setup
   8     super
   9
  10     @cmd = Gem::Commands::SourcesCommand.new
  11   end
  12
  13   def test_execute
  14     util_setup_source_info_cache
  15     @cmd.handle_options []
  16
  17     use_ui @ui do
  18       @cmd.execute
  19     end
  20
  21     expected = <<-EOF
  22 *** CURRENT SOURCES ***
  23
  24 #{@gem_repo}
  25     EOF
  26
  27     assert_equal expected, @ui.output
  28     assert_equal '', @ui.error
  29   end
  30
  31   def test_execute_add
  32     util_setup_fake_fetcher
  33
  34     si = Gem::SourceIndex.new
  35     si.add_spec @a1
  36
  37     @fetcher.data["http://beta-gems.example.com/Marshal.#{@marshal_version}"] =
  38       si.dump
  39
  40     @cmd.handle_options %w[--add http://beta-gems.example.com]
  41
  42     util_setup_source_info_cache
  43
  44     use_ui @ui do
  45       @cmd.execute
  46     end
  47
  48     expected = <<-EOF
  49 Bulk updating Gem source index for: http://beta-gems.example.com/
  50 http://beta-gems.example.com added to sources
  51     EOF
  52
  53     assert_equal expected, @ui.output
  54     assert_equal '', @ui.error
  55
  56     Gem::SourceInfoCache.cache.flush
  57     assert_equal %W[http://beta-gems.example.com #{@gem_repo}],
  58                  Gem::SourceInfoCache.cache_data.keys.sort
  59   end
  60
  61   def test_execute_add_nonexistent_source
  62     util_setup_fake_fetcher
  63
  64     @fetcher.data["http://beta-gems.example.com/Marshal.#{@marshal_version}"] =
  65       proc do
  66         raise Gem::RemoteFetcher::FetchError, 'it died'
  67       end
  68
  69     Gem::RemoteFetcher.instance_variable_set :@fetcher, @fetcher
  70
  71     @cmd.handle_options %w[--add http://beta-gems.example.com]
  72
  73     util_setup_source_info_cache
  74
  75     use_ui @ui do
  76       @cmd.execute
  77     end
  78
  79     expected = <<-EOF
  80 Error fetching http://beta-gems.example.com:
  81 \tit died
  82     EOF
  83
  84     assert_equal expected, @ui.output
  85     assert_equal '', @ui.error
  86   end
  87
  88   def test_execute_add_bad_uri
  89     @cmd.handle_options %w[--add beta-gems.example.com]
  90
  91     util_setup_source_info_cache
  92
  93     use_ui @ui do
  94       @cmd.execute
  95     end
  96
  97     expected = <<-EOF
  98 beta-gems.example.com is not a URI
  99     EOF
 100
 101     assert_equal expected, @ui.output
 102     assert_equal '', @ui.error
 103   end
 104
 105   def test_execute_clear_all
 106     @cmd.handle_options %w[--clear-all]
 107
 108     util_setup_source_info_cache
 109
 110     cache = Gem::SourceInfoCache.cache
 111     cache.update
 112     cache.write_cache
 113
 114     assert File.exist?(cache.system_cache_file),
 115            'system cache file'
 116     assert File.exist?(cache.latest_system_cache_file),
 117            'latest system cache file'
 118
 119     use_ui @ui do
 120       @cmd.execute
 121     end
 122
 123     expected = <<-EOF
 124 *** Removed user source cache ***
 125 *** Removed latest user source cache ***
 126 *** Removed system source cache ***
 127 *** Removed latest system source cache ***
 128     EOF
 129
 130     assert_equal expected, @ui.output
 131     assert_equal '', @ui.error
 132
 133     assert !File.exist?(cache.system_cache_file),
 134            'system cache file'
 135     assert !File.exist?(cache.latest_system_cache_file),
 136            'latest system cache file'
 137
 138   end
 139
 140   def test_execute_remove
 141     @cmd.handle_options %W[--remove #{@gem_repo}]
 142
 143     util_setup_source_info_cache
 144
 145     use_ui @ui do
 146       @cmd.execute
 147     end
 148
 149     expected = "#{@gem_repo} removed from sources\n"
 150
 151     assert_equal expected, @ui.output
 152     assert_equal '', @ui.error
 153
 154     Gem::SourceInfoCache.cache.flush
 155     assert_equal [], Gem::SourceInfoCache.cache_data.keys
 156   end
 157
 158   def test_execute_remove_no_network
 159     @cmd.handle_options %W[--remove #{@gem_repo}]
 160
 161     util_setup_fake_fetcher
 162
 163     @fetcher.data["#{@gem_repo}/Marshal.#{Gem.marshal_version}"] = proc do
 164       raise Gem::RemoteFetcher::FetchError
 165     end
 166
 167     use_ui @ui do
 168       @cmd.execute
 169     end
 170
 171     expected = "#{@gem_repo} removed from sources\n"
 172
 173     assert_equal expected, @ui.output
 174     assert_equal '', @ui.error
 175
 176     Gem::SourceInfoCache.cache.flush
 177     assert_equal [], Gem::SourceInfoCache.cache_data.keys
 178   end
 179
 180   def test_execute_update
 181     @cmd.handle_options %w[--update]
 182
 183     util_setup_source_info_cache
 184     Gem::SourceInfoCache.reset
 185
 186     util_setup_fake_fetcher
 187     si = Gem::SourceIndex.new
 188     si.add_spec @a1
 189     @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = si.dump
 190
 191     use_ui @ui do
 192       @cmd.execute
 193     end
 194
 195     expected = <<-EOF
 196 Bulk updating Gem source index for: #{@gem_repo}/
 197 source cache successfully updated
 198     EOF
 199
 200     assert_equal expected, @ui.output
 201     assert_equal '', @ui.error
 202   end
 203
 204 end
 205