1 files changed, 128 insertions, 0 deletions
diff --git a/jni/ruby/test/rubygems/test_gem_available_set.rb b/jni/ruby/test/rubygems/test_gem_available_set.rb
new file mode 100644
index 0000000..2b51544
--- /dev/null
+++ b/jni/ruby/test/rubygems/test_gem_available_set.rb
@@ -0,0 +1,128 @@
+require 'rubygems/test_case'
+require 'rubygems/available_set'
+require 'rubygems/security'
+
+class TestGemAvailableSet < Gem::TestCase
+
+  def setup
+    super
+
+    @source = Gem::Source.new(@gem_repo)
+  end
+
+  def test_add_and_empty
+    a1, _ = util_gem 'a', '1'
+
+    set = Gem::AvailableSet.new
+    assert set.empty?
+
+    set.add a1, @source
+
+    refute set.empty?
+
+    assert_equal [a1], set.all_specs
+  end
+
+  def test_find_all
+    a1,  a1_gem  = util_gem 'a', 1
+    a1a, a1a_gem = util_gem 'a', '1.a'
+
+    a1_source  = Gem::Source::SpecificFile.new a1_gem
+    a1a_source = Gem::Source::SpecificFile.new a1a_gem
+
+    set = Gem::AvailableSet.new
+    set.add a1,  a1_source
+    set.add a1a, a1a_source
+
+    dep = Gem::Resolver::DependencyRequest.new dep('a'), nil
+
+    assert_equal %w[a-1], set.find_all(dep).map { |spec| spec.full_name }
+
+    dep = Gem::Resolver::DependencyRequest.new dep('a', '>= 0.a'), nil
+
+    assert_equal %w[a-1 a-1.a],
+                 set.find_all(dep).map { |spec| spec.full_name }.sort
+  end
+
+  def test_match_platform
+    a1, _ = util_gem 'a', '1' do |g|
+      g.platform = "something-weird-yep"
+    end
+
+    a1c, _ = util_gem 'a', '2' do |g|
+      g.platform = Gem::Platform.local
+    end
+
+    a2, _ = util_gem 'a', '2'
+
+    set = Gem::AvailableSet.new
+    set.add a1, @source
+    set.add a1c, @source
+    set.add a2, @source
+
+    set.match_platform!
+
+    assert_equal [a1c, a2], set.all_specs
+  end
+
+  def test_best
+    a1, _ = util_gem 'a', '1'
+    a2, _  = util_gem 'a', '2'
+
+    set = Gem::AvailableSet.new
+    set.add a1, @source
+    set.add a2, @source
+
+    set.pick_best!
+
+    assert_equal [a2], set.all_specs
+  end
+
+  def test_remove_installed_bang
+    a1, _ = util_gem 'a', '1'
+
+    a1.activate
+
+    set = Gem::AvailableSet.new
+    set.add a1, @source
+
+    dep = Gem::Dependency.new "a", ">= 0"
+
+    set.remove_installed! dep
+
+    assert set.empty?
+  end
+
+  def test_sorted_normal_versions
+    a1, _ = util_gem 'a', '1'
+    a2, _  = util_gem 'a', '2'
+
+    set = Gem::AvailableSet.new
+    set.add a1, @source
+    set.add a2, @source
+
+    g = set.sorted
+
+    assert_equal a2, g[0].spec
+    assert_equal a1, g[1].spec
+  end
+
+  def test_sorted_respect_pre
+    a1a, _ = util_gem 'a', '1.a'
+    a1,  _ = util_gem 'a', '1'
+    a2a, _ = util_gem 'a', '2.a'
+    a2,  _ = util_gem 'a', '2'
+    a3a, _ = util_gem 'a', '3.a'
+
+    set = Gem::AvailableSet.new
+    set.add a1, @source
+    set.add a1a, @source
+    set.add a3a, @source
+    set.add a2a, @source
+    set.add a2, @source
+
+    g = set.sorted.map { |t| t.spec }
+
+    assert_equal [a3a, a2, a2a, a1, a1a], g
+  end
+end