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
  12     @new_repo = "http://beta-gems.example.com"
  13   end
  14
  15   def test_execute
  16     util_setup_spec_fetcher
  17     @cmd.handle_options []
  18
  19     use_ui @ui do
  20       @cmd.execute
  21     end
  22
  23     expected = <<-EOF
  24 *** CURRENT SOURCES ***
  25
  26 #{@gem_repo}
  27     EOF
  28
  29     assert_equal expected, @ui.output
  30     assert_equal '', @ui.error
  31   end
  32
  33   def test_execute_add
  34     util_setup_fake_fetcher
  35
  36     si = Gem::SourceIndex.new
  37     si.add_spec @a1
  38
  39     specs = si.map do |_, spec|
  40       [spec.name, spec.version, spec.original_platform]
  41     end
  42
  43     specs_dump_gz = StringIO.new
  44     Zlib::GzipWriter.wrap specs_dump_gz do |io|
  45       Marshal.dump specs, io
  46     end
  47
  48     @fetcher.data["#{@new_repo}/specs.#{@marshal_version}.gz"] =
  49       specs_dump_gz.string
  50
  51     @cmd.handle_options %W[--add #{@new_repo}]
  52
  53     util_setup_spec_fetcher
  54
  55     use_ui @ui do
  56       @cmd.execute
  57     end
  58
  59     assert_equal [@gem_repo, @new_repo], Gem.sources
  60
  61     expected = <<-EOF
  62 #{@new_repo} added to sources
  63     EOF
  64
  65     assert_equal expected, @ui.output
  66     assert_equal '', @ui.error
  67   end
  68
  69   def test_execute_add_nonexistent_source
  70     util_setup_fake_fetcher
  71
  72     uri = "http://beta-gems.example.com/specs.#{@marshal_version}.gz"
  73     @fetcher.data[uri] = proc do
  74       raise Gem::RemoteFetcher::FetchError.new('it died', uri)
  75     end
  76
  77     Gem::RemoteFetcher.fetcher = @fetcher
  78
  79     @cmd.handle_options %w[--add http://beta-gems.example.com]
  80
  81     util_setup_spec_fetcher
  82
  83     use_ui @ui do
  84       @cmd.execute
  85     end
  86
  87     expected = <<-EOF
  88 Error fetching http://beta-gems.example.com:
  89 \tit died (#{uri})
  90     EOF
  91
  92     assert_equal expected, @ui.output
  93     assert_equal '', @ui.error
  94   end
  95
  96   def test_execute_add_bad_uri
  97     @cmd.handle_options %w[--add beta-gems.example.com]
  98
  99     util_setup_spec_fetcher
 100
 101     use_ui @ui do
 102       @cmd.execute
 103     end
 104
 105     assert_equal [@gem_repo], Gem.sources
 106
 107     expected = <<-EOF
 108 beta-gems.example.com is not a URI
 109     EOF
 110
 111     assert_equal expected, @ui.output
 112     assert_equal '', @ui.error
 113   end
 114
 115   def test_execute_add_legacy
 116     util_setup_fake_fetcher
 117     util_setup_source_info_cache
 118
 119     si = Gem::SourceIndex.new
 120     si.add_spec @a1
 121
 122     @fetcher.data["#{@new_repo}/yaml"] = ''
 123
 124     @cmd.handle_options %W[--add #{@new_repo}]
 125
 126     use_ui @ui do
 127       @cmd.execute
 128     end
 129
 130     assert_equal [@gem_repo], Gem.sources
 131
 132     expected = <<-EOF
 133 WARNING:  RubyGems 1.2+ index not found for:
 134 \t#{@new_repo}
 135
 136 Will cause RubyGems to revert to legacy indexes, degrading performance.
 137     EOF
 138
 139     assert_equal "#{@new_repo} added to sources\n", @ui.output
 140     assert_equal expected, @ui.error
 141   end
 142
 143   def test_execute_clear_all
 144     @cmd.handle_options %w[--clear-all]
 145
 146     util_setup_source_info_cache
 147
 148     cache = Gem::SourceInfoCache.cache
 149     cache.update
 150     cache.write_cache
 151
 152     assert File.exist?(cache.system_cache_file),
 153            'system cache file'
 154     assert File.exist?(cache.latest_system_cache_file),
 155            'latest system cache file'
 156
 157     util_setup_spec_fetcher
 158
 159     fetcher = Gem::SpecFetcher.fetcher
 160
 161     # HACK figure out how to force directory creation via fetcher
 162     #assert File.directory?(fetcher.dir), 'cache dir exists'
 163
 164     use_ui @ui do
 165       @cmd.execute
 166     end
 167
 168     expected = <<-EOF
 169 *** Removed specs cache ***
 170 *** Removed user source cache ***
 171 *** Removed latest user source cache ***
 172 *** Removed system source cache ***
 173 *** Removed latest system source cache ***
 174     EOF
 175
 176     assert_equal expected, @ui.output
 177     assert_equal '', @ui.error
 178
 179     assert !File.exist?(cache.system_cache_file),
 180            'system cache file'
 181     assert !File.exist?(cache.latest_system_cache_file),
 182            'latest system cache file'
 183
 184     assert !File.exist?(fetcher.dir), 'cache dir removed'
 185   end
 186
 187   def test_execute_remove
 188     @cmd.handle_options %W[--remove #{@gem_repo}]
 189
 190     util_setup_spec_fetcher
 191
 192     use_ui @ui do
 193       @cmd.execute
 194     end
 195
 196     expected = "#{@gem_repo} removed from sources\n"
 197
 198     assert_equal expected, @ui.output
 199     assert_equal '', @ui.error
 200   end
 201
 202   def test_execute_remove_no_network
 203     @cmd.handle_options %W[--remove #{@gem_repo}]
 204
 205     util_setup_fake_fetcher
 206
 207     @fetcher.data["#{@gem_repo}Marshal.#{Gem.marshal_version}"] = proc do
 208       raise Gem::RemoteFetcher::FetchError
 209     end
 210
 211     use_ui @ui do
 212       @cmd.execute
 213     end
 214
 215     expected = "#{@gem_repo} removed from sources\n"
 216
 217     assert_equal expected, @ui.output
 218     assert_equal '', @ui.error
 219   end
 220
 221   def test_execute_update
 222     @cmd.handle_options %w[--update]
 223
 224     util_setup_fake_fetcher
 225     source_index = util_setup_spec_fetcher @a1
 226
 227     specs = source_index.map do |name, spec|
 228       [spec.name, spec.version, spec.original_platform]
 229     end
 230
 231     @fetcher.data["#{@gem_repo}specs.#{Gem.marshal_version}.gz"] =
 232       util_gzip Marshal.dump(specs)
 233
 234     latest_specs = source_index.latest_specs.map do |spec|
 235       [spec.name, spec.version, spec.original_platform]
 236     end
 237
 238     @fetcher.data["#{@gem_repo}latest_specs.#{Gem.marshal_version}.gz"] =
 239       util_gzip Marshal.dump(latest_specs)
 240
 241     use_ui @ui do
 242       @cmd.execute
 243     end
 244
 245     assert_equal "source cache successfully updated\n", @ui.output
 246     assert_equal '', @ui.error
 247   end
 248
 249   def test_execute_update_legacy
 250     @cmd.handle_options %w[--update]
 251
 252     util_setup_fake_fetcher
 253     util_setup_source_info_cache
 254     Gem::SourceInfoCache.reset
 255
 256     si = Gem::SourceIndex.new
 257     si.add_spec @a1
 258     @fetcher.data["#{@gem_repo}yaml"] = YAML.dump si
 259     @fetcher.data["#{@gem_repo}Marshal.#{@marshal_version}"] = si.dump
 260
 261     use_ui @ui do
 262       @cmd.execute
 263     end
 264
 265     expected = <<-EOF
 266 Bulk updating Gem source index for: #{@gem_repo}
 267 source cache successfully updated
 268     EOF
 269
 270     assert_equal expected, @ui.output
 271     assert_equal '', @ui.error
 272   end
 273
 274 end
 275