ruby/test/rubygems/test_gem_source.rb

254 строки
7.2 KiB
Ruby

# frozen_string_literal: true
require_relative "helper"
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_raise 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")
@fetcher.data[@gem_repo] = response
set = @source.dependency_resolver_set
assert_kind_of Gem::Resolver::APISet, set
end
def test_dependency_resolver_set_file_uri
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_raise 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_false_positive
rubygems_source = Gem::Source.new("https://rubygems.org")
refute rubygems_source.typo_squatting?("rubygems.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