1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/rubygems/test_gem_source_index.rb
drbrain 5d613c83cc Import RubyGems r1601. [ruby-core:15381].
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@15423 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
2008-02-10 08:00:19 +00:00

429 lines
12 KiB
Ruby

#--
# 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, '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", "= 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