ruby/test/rubygems/test_gem_resolver_api_set.rb

# frozen_string_literal: true

require_relative "helper"

class TestGemResolverAPISet < Gem::TestCase
  def setup
    super

    @dep_uri = Gem::URI "#{@gem_repo}info/"
  end

  def test_initialize
    set = Gem::Resolver::APISet.new

    assert_equal Gem::URI("https://index.rubygems.org/info/"),            set.dep_uri
    assert_equal Gem::URI("https://index.rubygems.org/"),                 set.uri
    assert_equal Gem::Source.new(Gem::URI("https://index.rubygems.org")), set.source
  end

  def test_initialize_deeper_uri
    set = Gem::Resolver::APISet.new "https://rubygemsserver.com/mygems/info"

    assert_equal Gem::URI("https://rubygemsserver.com/mygems/info"),       set.dep_uri
    assert_equal Gem::URI("https://rubygemsserver.com/"),                  set.uri
    assert_equal Gem::Source.new(Gem::URI("https://rubygemsserver.com/")), set.source
  end

  def test_initialize_uri
    set = Gem::Resolver::APISet.new @dep_uri

    assert_equal Gem::URI("#{@gem_repo}info/"), set.dep_uri
    assert_equal Gem::URI(@gem_repo.to_s), set.uri
  end

  def test_find_all
    spec_fetcher

    data = [
      { name: "a",
        number: "1",
        platform: "ruby",
        dependencies: [] },
    ]

    @fetcher.data["#{@dep_uri}a"] = "---\n1  "

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    expected = [
      Gem::Resolver::APISpecification.new(set, data.first),
    ]

    assert_equal expected, set.find_all(a_dep)
  end

  def test_find_all_prereleases
    spec_fetcher

    data = [
      { name: "a",
        number: "1",
        platform: "ruby",
        dependencies: [] },
      { name: "a",
        number: "2.a",
        platform: "ruby",
        dependencies: [] },
    ]

    @fetcher.data["#{@dep_uri}a"] = "---\n1\n2.a"

    set = Gem::Resolver::APISet.new @dep_uri
    set.prerelease = true

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    expected = [
      Gem::Resolver::APISpecification.new(set, data.first),
      Gem::Resolver::APISpecification.new(set, data.last),
    ]

    assert_equal expected, set.find_all(a_dep)
  end

  def test_find_all_cache
    spec_fetcher

    data = [
      { name: "a",
        number: "1",
        platform: "ruby",
        dependencies: [] },
    ]

    @fetcher.data["#{@dep_uri}a"] = "---\n1  "

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    set.prefetch [a_dep]

    expected = [
      Gem::Resolver::APISpecification.new(set, data.first),
    ]

    assert_equal expected, set.find_all(a_dep)

    @fetcher.data.delete "#{@dep_uri}a"
  end

  def test_find_all_local
    set = Gem::Resolver::APISet.new @dep_uri
    set.remote = false

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    assert_empty set.find_all(a_dep)
  end

  def test_find_all_missing
    spec_fetcher

    @fetcher.data["#{@dep_uri}a"] = "---"

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    assert_empty set.find_all(a_dep)

    @fetcher.data.delete "#{@dep_uri}a"

    assert_empty set.find_all(a_dep)
  end

  def test_find_all_not_found
    spec_fetcher

    @fetcher.data["#{@dep_uri}/a"] =
      proc do
        raise Gem::RemoteFetcher::FetchError
      end

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil

    assert_empty set.find_all(a_dep)

    @fetcher.data.delete "#{@dep_uri}a"

    assert_empty set.find_all(a_dep)
  end

  def test_prefetch
    spec_fetcher

    @fetcher.data["#{@dep_uri}a"] = "---\n1  \n"
    @fetcher.data["#{@dep_uri}b"] = "---"

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil
    b_dep = Gem::Resolver::DependencyRequest.new dep("b"), nil

    set.prefetch [a_dep, b_dep]

    assert_equal %w[a-1], set.find_all(a_dep).map(&:full_name)
    assert_empty          set.find_all(b_dep)
  end

  def test_prefetch_cache
    spec_fetcher

    @fetcher.data["#{@dep_uri}a"] = "---\n1  \n"

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil
    b_dep = Gem::Resolver::DependencyRequest.new dep("b"), nil

    set.prefetch [a_dep]

    @fetcher.data.delete "#{@dep_uri}a"
    @fetcher.data["#{@dep_uri}?b"] = "---"

    set.prefetch [a_dep, b_dep]
  end

  def test_prefetch_cache_missing
    spec_fetcher

    @fetcher.data["#{@dep_uri}a"] = "---\n1  \n"
    @fetcher.data["#{@dep_uri}b"] = "---"

    set = Gem::Resolver::APISet.new @dep_uri

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil
    b_dep = Gem::Resolver::DependencyRequest.new dep("b"), nil

    set.prefetch [a_dep, b_dep]

    @fetcher.data.delete "#{@dep_uri}a"
    @fetcher.data.delete "#{@dep_uri}b"

    set.prefetch [a_dep, b_dep]
  end

  def test_prefetch_local
    spec_fetcher

    @fetcher.data["#{@dep_uri}a"] = "---\n1  \n"
    @fetcher.data["#{@dep_uri}b"] = "---"

    set = Gem::Resolver::APISet.new @dep_uri
    set.remote = false

    a_dep = Gem::Resolver::DependencyRequest.new dep("a"), nil
    b_dep = Gem::Resolver::DependencyRequest.new dep("b"), nil

    set.prefetch [a_dep, b_dep]

    assert_empty set.instance_variable_get :@data
  end
end