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.rb
2019-12-13 20:19:33 +09:00

252 lines
7.2 KiB
Ruby

# frozen_string_literal: true
require 'rubygems/test_case'
require 'rubygems/source'
require 'rubygems/indexer'
class TestGemSource < Gem::TestCase
def tuple(*args)
Gem::NameTuple.new(*args)
end
def setup
super
@specs = spec_fetcher do |fetcher|
fetcher.spec 'a', '1.a'
fetcher.gem 'a', 1
fetcher.spec 'a', 2
fetcher.spec 'b', 2
end
@source = Gem::Source.new(@gem_repo)
end
def test_initialize_invalid_uri
assert_raises URI::InvalidURIError do
Gem::Source.new 'git@example:a.git'
end
end
def test_initialize_git
repository = 'git@example:a.git'
source = Gem::Source::Git.new 'a', repository, 'master', false
assert_equal repository, source.uri
end
def test_cache_dir_escapes_windows_paths
uri = URI.parse("file:///C:/WINDOWS/Temp/gem_repo")
root = Gem.spec_cache_dir
cache_dir = @source.cache_dir(uri).gsub(root, '')
assert cache_dir !~ /:/, "#{cache_dir} should not contain a :"
end
def test_dependency_resolver_set_bundler_api
response = Net::HTTPResponse.new '1.1', 200, 'OK'
response.uri = URI('http://example') if response.respond_to? :uri
@fetcher.data["#{@gem_repo}api/v1/dependencies"] = response
set = @source.dependency_resolver_set
assert_kind_of Gem::Resolver::APISet, set
end
def test_dependency_resolver_set_file_uri
skip 'install builder gem' unless defined? Builder::XChar
Gem::Indexer.new(@tempdir).generate_index
source = Gem::Source.new "file://#{@tempdir}/"
set = source.dependency_resolver_set
assert_kind_of Gem::Resolver::IndexSet, set
end
def test_dependency_resolver_set_marshal_api
set = @source.dependency_resolver_set
assert_kind_of Gem::Resolver::IndexSet, set
end
def test_fetch_spec
a1 = @specs['a-1']
spec_uri = "#{@gem_repo}#{Gem::MARSHAL_SPEC_DIR}#{a1.spec_name}"
spec = @source.fetch_spec tuple('a', Gem::Version.new(1), 'ruby')
assert_equal a1.full_name, spec.full_name
cache_dir = @source.cache_dir URI.parse(spec_uri)
cache_file = File.join cache_dir, a1.spec_name
assert File.exist?(cache_file)
end
def test_fetch_spec_cached
a1 = @specs['a-1']
spec_uri = "#{@gem_repo}/#{Gem::MARSHAL_SPEC_DIR}#{a1.spec_name}"
@fetcher.data["#{spec_uri}.rz"] = nil
cache_dir = @source.cache_dir URI.parse(spec_uri)
FileUtils.mkdir_p cache_dir
cache_file = File.join cache_dir, a1.spec_name
File.open cache_file, 'wb' do |io|
Marshal.dump a1, io
end
spec = @source.fetch_spec tuple('a', Gem::Version.new(1), 'ruby')
assert_equal a1.full_name, spec.full_name
end
def test_fetch_spec_platform
specs = spec_fetcher do |fetcher|
fetcher.legacy_platform
end
spec = @source.fetch_spec tuple('pl', Gem::Version.new(1), 'i386-linux')
assert_equal specs['pl-1-x86-linux'].full_name, spec.full_name
end
def test_fetch_spec_platform_ruby
spec = @source.fetch_spec tuple('a', Gem::Version.new(1), nil)
assert_equal @specs['a-1'].full_name, spec.full_name
spec = @source.fetch_spec tuple('a', Gem::Version.new(1), '')
assert_equal @specs['a-1'].full_name, spec.full_name
end
def test_load_specs
released = @source.load_specs(:released).map { |spec| spec.full_name }
assert_equal %W[a-2 a-1 b-2], released
cache_dir = File.join Gem.spec_cache_dir, 'gems.example.com%80'
assert File.exist?(cache_dir), "#{cache_dir} does not exist"
cache_file = File.join cache_dir, "specs.#{Gem.marshal_version}"
assert File.exist?(cache_file)
end
def test_load_specs_cached
latest_specs = @source.load_specs :latest
# Make sure the cached version is actually different:
latest_specs << Gem::NameTuple.new('cached', Gem::Version.new('1.0.0'), 'ruby')
@fetcher.data["#{@gem_repo}latest_specs.#{Gem.marshal_version}.gz"] = nil
@fetcher.data["#{@gem_repo}latest_specs.#{Gem.marshal_version}"] =
' ' * Marshal.dump(latest_specs).length
cache_dir = File.join Gem.spec_cache_dir, 'gems.example.com%80'
FileUtils.mkdir_p cache_dir
cache_file = File.join cache_dir, "latest_specs.#{Gem.marshal_version}"
File.open cache_file, 'wb' do |io|
Marshal.dump latest_specs, io
end
cached_specs = @source.load_specs :latest
assert_equal latest_specs, cached_specs
end
def test_load_specs_cached_empty
latest_specs = @source.load_specs :latest
# Make sure the cached version is actually different:
latest_specs << Gem::NameTuple.new('fixed', Gem::Version.new('1.0.0'), 'ruby')
# Setup valid data on the 'remote'
@fetcher.data["#{@gem_repo}latest_specs.#{Gem.marshal_version}.gz"] =
util_gzip(Marshal.dump(latest_specs))
cache_dir = File.join Gem.spec_cache_dir, 'gems.example.com%80'
FileUtils.mkdir_p cache_dir
cache_file = File.join cache_dir, "latest_specs.#{Gem.marshal_version}"
File.open cache_file, 'wb' do |io|
# Setup invalid data in the cache:
io.write Marshal.dump(latest_specs)[0, 10]
end
fixed_specs = @source.load_specs :latest
assert_equal latest_specs, fixed_specs
end
def test_load_specs_from_unavailable_uri
src = Gem::Source.new("http://not-there.nothing")
assert_raises Gem::RemoteFetcher::FetchError do
src.load_specs :latest
end
end
def test_spaceship
remote = @source
specific = Gem::Source::SpecificFile.new @specs['a-1'].cache_file
installed = Gem::Source::Installed.new
local = Gem::Source::Local.new
assert_equal(0, remote. <=>(remote), 'remote <=> remote')
assert_equal(-1, remote. <=>(specific), 'remote <=> specific')
assert_equal(1, specific. <=>(remote), 'specific <=> remote')
assert_equal(-1, remote. <=>(local), 'remote <=> local')
assert_equal(1, local. <=>(remote), 'local <=> remote')
assert_equal(-1, remote. <=>(installed), 'remote <=> installed')
assert_equal(1, installed.<=>(remote), 'installed <=> remote')
no_uri = @source.dup
no_uri.instance_variable_set :@uri, nil
assert_equal(-1, remote. <=>(no_uri), 'remote <=> no_uri')
end
def test_spaceship_order_is_preserved_when_uri_differs
sourceA = Gem::Source.new "http://example.com/a"
sourceB = Gem::Source.new "http://example.com/b"
assert_equal(0, sourceA. <=>(sourceA), 'sourceA <=> sourceA')
assert_equal(1, sourceA. <=>(sourceB), 'sourceA <=> sourceB')
assert_equal(1, sourceB. <=>(sourceA), 'sourceB <=> sourceA')
end
def test_update_cache_eh
assert @source.update_cache?
end
def test_update_cache_eh_home_nonexistent
FileUtils.rmdir Gem.user_home
refute @source.update_cache?
end
def test_typo_squatting
rubygems_source = Gem::Source.new("https://rubgems.org")
assert rubygems_source.typo_squatting?("rubygems.org")
assert rubygems_source.typo_squatting?("rubyagems.org")
assert rubygems_source.typo_squatting?("rubyasgems.org")
refute rubygems_source.typo_squatting?("rubysertgems.org")
end
def test_typo_squatting_custom_distance_threshold
rubygems_source = Gem::Source.new("https://rubgems.org")
distance_threshold = 5
assert rubygems_source.typo_squatting?("rubysertgems.org", distance_threshold)
end
end