diff options
Diffstat (limited to 'test/rubygems/test_gem_source_index.rb')
| -rw-r--r-- | test/rubygems/test_gem_source_index.rb | 429 |
1 files changed, 429 insertions, 0 deletions
diff --git a/test/rubygems/test_gem_source_index.rb b/test/rubygems/test_gem_source_index.rb new file mode 100644 index 0000000000..befbbe6f67 --- /dev/null +++ b/test/rubygems/test_gem_source_index.rb @@ -0,0 +1,429 @@ +#-- +# Copyright 2006 by Chad Fowler, Rich Kilmer, Jim Weirich and others. +# All rights reserved. +# See LICENSE.txt for permissions. +#++ + +require 'test/unit' +require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities') +require 'rubygems/source_index' +require 'rubygems/config_file' + +class Gem::SourceIndex + public :fetcher, :fetch_bulk_index, :fetch_quick_index, + :find_missing, :gems, :remove_extra, + :update_with_missing, :unzip +end + +class TestGemSourceIndex < RubyGemTestCase + + def setup + super + + util_setup_fake_fetcher + end + + def test_create_from_directory + # TODO + end + + def test_fetcher + assert_equal @fetcher, @source_index.fetcher + end + + def test_fetch_bulk_index_compressed + util_setup_bulk_fetch true + + use_ui @ui do + fetched_index = @source_index.fetch_bulk_index @uri + assert_equal [@gem1.full_name, @gem4.full_name, @gem2.full_name].sort, + fetched_index.gems.map { |n,s| n }.sort + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_fetch_bulk_index_error + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}.Z"] = proc { raise SocketError } + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = proc { raise SocketError } + @fetcher.data["#{@gem_repo}/yaml.Z"] = proc { raise SocketError } + @fetcher.data["#{@gem_repo}/yaml"] = proc { raise SocketError } + + e = assert_raise Gem::RemoteSourceException do + use_ui @ui do + @source_index.fetch_bulk_index @uri + end + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}", paths.shift + assert_equal "#{@gem_repo}/yaml.Z", paths.shift + assert_equal "#{@gem_repo}/yaml", paths.shift + + assert paths.empty?, paths.join(', ') + + assert_equal 'Error fetching remote gem cache: SocketError', + e.message + end + + def test_fetch_bulk_index_fallback + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}.Z"] = + proc { raise SocketError } + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = + proc { raise SocketError } + @fetcher.data["#{@gem_repo}/yaml.Z"] = proc { raise SocketError } + @fetcher.data["#{@gem_repo}/yaml"] = @source_index.to_yaml + + use_ui @ui do + fetched_index = @source_index.fetch_bulk_index @uri + assert_equal [@gem1.full_name, @gem4.full_name, @gem2.full_name].sort, + fetched_index.gems.map { |n,s| n }.sort + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}", paths.shift + assert_equal "#{@gem_repo}/yaml.Z", paths.shift + assert_equal "#{@gem_repo}/yaml", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_fetch_bulk_index_marshal_mismatch + marshal = @source_index.dump + marshal[0] = (Marshal::MAJOR_VERSION - 1).chr + + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = marshal + @fetcher.data["#{@gem_repo}/yaml"] = @source_index.to_yaml + + use_ui @ui do + fetched_index = @source_index.fetch_bulk_index @uri + assert_equal [@gem1.full_name, @gem4.full_name, @gem2.full_name].sort, + fetched_index.gems.map { |n,s| n }.sort + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}", paths.shift + assert_equal "#{@gem_repo}/yaml.Z", paths.shift + assert_equal "#{@gem_repo}/yaml", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_fetch_bulk_index_uncompressed + util_setup_bulk_fetch false + use_ui @ui do + fetched_index = @source_index.fetch_bulk_index @uri + assert_equal [@gem1.full_name, @gem4.full_name, @gem2.full_name].sort, + fetched_index.gems.map { |n,s| n }.sort + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_fetch_quick_index + quick_index = util_zip @gem_names + @fetcher.data["#{@gem_repo}/quick/index.rz"] = quick_index + + quick_index = @source_index.fetch_quick_index @uri + assert_equal [@gem1.full_name, @gem4.full_name, @gem2.full_name].sort, + quick_index.sort + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_fetch_quick_index_error + @fetcher.data["#{@gem_repo}/quick/index.rz"] = + proc { raise Exception } + + e = assert_raise Gem::OperationNotSupportedError do + @source_index.fetch_quick_index @uri + end + + assert_equal 'No quick index found: Exception', e.message + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_find_missing + missing = @source_index.find_missing [@gem3.full_name] + assert_equal [@gem3.full_name], missing + end + + def test_find_missing_none_missing + missing = @source_index.find_missing @gem_names.split + assert_equal [], missing + end + + def test_latest_specs + spec = quick_gem @gem1.name, '0.0.1' + @source_index.add_spec spec + + expected = [ + @gem1.full_name, + @gem2.full_name, + @gem4.full_name, + ].sort + + assert_equal expected, @source_index.latest_specs.map { |s| s.full_name }.sort + end + + def test_outdated + sic = Gem::SourceInfoCache.new + Gem::SourceInfoCache.instance_variable_set :@cache, sic + + assert_equal [], @source_index.outdated + + updated = quick_gem @gem1.name, (@gem1.version.bump) + util_setup_source_info_cache updated + + assert_equal [updated.name], @source_index.outdated + + updated_platform = quick_gem @gem1.name, (updated.version.bump) do |s| + s.platform = Gem::Platform.new 'x86-other_platform1' + end + + util_setup_source_info_cache updated, updated_platform + + assert_equal [updated_platform.name], @source_index.outdated + end + + def test_remove_extra + @source_index.remove_extra [@gem1.full_name] + assert_equal [@gem1.full_name], @source_index.gems.map { |n,s| n } + end + + def test_remove_extra_no_changes + gems = @gem_names.split.sort + @source_index.remove_extra gems + assert_equal gems, @source_index.gems.map { |n,s| n }.sort + end + + def test_search + assert_equal [@gem1, @gem4], @source_index.search("gem_one") + assert_equal [@gem1], @source_index.search("gem_one", "= 0.0.2") + + assert_equal [], @source_index.search("bogusstring") + assert_equal [], @source_index.search("gem_one", "= 3.2.1") + + @a1 = quick_gem 'a', '1' + @a2 = quick_gem 'a', '2' + + source_index = Gem::SourceIndex.new @a1.full_name => @a1, + @a2.full_name => @a2 + + assert_equal [@a1], source_index.search(@a1.name, '= 1') + + r1 = Gem::Requirement.create '= 1' + assert_equal [@a1], source_index.search(@a1.name, r1) + + dep = Gem::Dependency.new @a1.name, r1 + assert_equal [@a1], source_index.search(dep) + end + + def test_search_empty_cache + empty_source_index = Gem::SourceIndex.new({}) + assert_equal [], empty_source_index.search("foo") + end + + def test_search_platform + util_set_arch 'x86-my_platform1' + + a1 = quick_gem 'a', '1' + a1_mine = quick_gem 'a', '1' do |s| + s.platform = Gem::Platform.new 'x86-my_platform1' + end + a1_other = quick_gem 'a', '1' do |s| + s.platform = Gem::Platform.new 'x86-other_platform1' + end + + si = Gem::SourceIndex.new(a1.full_name => a1, a1_mine.full_name => a1_mine, + a1_other.full_name => a1_other) + + dep = Gem::Dependency.new 'a', Gem::Requirement.new('1') + + gems = si.search dep, true + + assert_equal [a1, a1_mine], gems.sort + end + + def test_signature + sig = @source_index.gem_signature('foo-1.2.3') + assert_equal 64, sig.length + assert_match(/^[a-f0-9]{64}$/, sig) + end + + def test_specification + assert_equal @gem1, @source_index.specification(@gem1.full_name) + + assert_nil @source_index.specification("foo-1.2.4") + end + + def test_index_signature + sig = @source_index.index_signature + assert_match(/^[a-f0-9]{64}$/, sig) + end + + def test_unzip + input = "x\234+\316\317MU(I\255(\001\000\021\350\003\232" + assert_equal 'some text', @source_index.unzip(input) + end + + def test_update_bulk + util_setup_bulk_fetch true + + @source_index.gems.replace({}) + assert_equal [], @source_index.gems.keys.sort + + use_ui @ui do + @source_index.update @uri + + assert_equal @gem_names.split, @source_index.gems.keys.sort + end + + paths = @fetcher.paths + + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + assert_equal "#{@gem_repo}/Marshal.#{@marshal_version}.Z", paths.shift + + assert paths.empty?, paths.join(', ') + end + + def test_update_incremental + old_gem_conf = Gem.configuration + Gem.configuration = Gem::ConfigFile.new([]) + + quick_index = util_zip @all_gem_names.join("\n") + @fetcher.data["#{@gem_repo}/quick/index.rz"] = quick_index + + marshal_uri = File.join @gem_repo, "quick", "Marshal.#{@marshal_version}", + "#{@gem3.full_name}.gemspec.rz" + @fetcher.data[marshal_uri] = util_zip Marshal.dump(@gem3) + + use_ui @ui do + @source_index.update @uri + + assert_equal @all_gem_names, @source_index.gems.keys.sort + end + + paths = @fetcher.paths + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + assert_equal marshal_uri, paths.shift + + assert paths.empty?, paths.join(', ') + ensure + Gem.configuration = old_gem_conf + end + + def test_update_incremental_fallback + old_gem_conf = Gem.configuration + Gem.configuration = Gem::ConfigFile.new([]) + + quick_index = util_zip @all_gem_names.join("\n") + @fetcher.data["#{@gem_repo}/quick/index.rz"] = quick_index + + marshal_uri = File.join @gem_repo, "quick", "Marshal.#{@marshal_version}", + "#{@gem3.full_name}.gemspec.rz" + + yaml_uri = "#{@gem_repo}/quick/#{@gem3.full_name}.gemspec.rz" + @fetcher.data[yaml_uri] = util_zip @gem3.to_yaml + + use_ui @ui do + @source_index.update @uri + + assert_equal @all_gem_names, @source_index.gems.keys.sort + end + + paths = @fetcher.paths + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + assert_equal marshal_uri, paths.shift + assert_equal yaml_uri, paths.shift + + assert paths.empty?, paths.join(', ') + ensure + Gem.configuration = old_gem_conf + end + + def test_update_incremental_marshal_mismatch + old_gem_conf = Gem.configuration + Gem.configuration = Gem::ConfigFile.new([]) + + quick_index = util_zip @all_gem_names.join("\n") + @fetcher.data["#{@gem_repo}/quick/index.rz"] = quick_index + + marshal_uri = File.join @gem_repo, "quick", "Marshal.#{@marshal_version}", + "#{@gem3.full_name}.gemspec.rz" + marshal_data = Marshal.dump(@gem3) + marshal_data[0] = (Marshal::MAJOR_VERSION - 1).chr + @fetcher.data[marshal_uri] = util_zip marshal_data + + yaml_uri = "#{@gem_repo}/quick/#{@gem3.full_name}.gemspec.rz" + @fetcher.data[yaml_uri] = util_zip @gem3.to_yaml + + use_ui @ui do + @source_index.update @uri + + assert_equal @all_gem_names, @source_index.gems.keys.sort + end + + paths = @fetcher.paths + assert_equal "#{@gem_repo}/quick/index.rz", paths.shift + assert_equal marshal_uri, paths.shift + assert_equal yaml_uri, paths.shift + + assert paths.empty?, paths.join(', ') + ensure + Gem.configuration = old_gem_conf + end + + def test_update_with_missing + marshal_uri = File.join @gem_repo, "quick", "Marshal.#{@marshal_version}", + "#{@gem3.full_name}.gemspec.rz" + dumped = Marshal.dump(@gem3) + @fetcher.data[marshal_uri] = util_zip(dumped) + + use_ui @ui do + @source_index.update_with_missing @uri, [@gem3.full_name] + end + + spec = @source_index.specification(@gem3.full_name) + # We don't care about the equality of undumped attributes + @gem3.files = spec.files + @gem3.loaded_from = spec.loaded_from + + assert_equal @gem3, spec + end + + def util_setup_bulk_fetch(compressed) + source_index = @source_index.dump + + if compressed then + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}.Z"] = util_zip source_index + else + @fetcher.data["#{@gem_repo}/Marshal.#{@marshal_version}"] = source_index + end + end + +end |