зеркало из https://github.com/github/ruby.git
254 строки
7.2 KiB
Ruby
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
|