Fresh start

1 files changed, 241 insertions, 0 deletions

diff --git a/jni/ruby/test/rubygems/test_gem_source.rb b/jni/ruby/test/rubygems/test_gem_source.rb
new file mode 100644
index 0000000..7d23eee
--- /dev/null
+++ b/jni/ruby/test/rubygems/test_gem_source.rb
@@ -0,0 +1,241 @@
+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_api_uri
+    assert_equal @source.api_uri, @source.uri
+  end
+
+  def test_api_uri_resolved_from_remote_fetcher
+    uri = URI.parse "http://gem.example/foo"
+    @fetcher.api_endpoints[uri] = URI.parse "http://api.blah"
+
+    src = Gem::Source.new uri
+    assert_equal URI.parse("http://api.blah"), src.api_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
+
+    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}"
+
+    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}"
+
+    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_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
+
+end
+