1 files changed, 167 insertions, 0 deletions
diff --git a/jni/ruby/test/rubygems/test_gem_commands_cleanup_command.rb b/jni/ruby/test/rubygems/test_gem_commands_cleanup_command.rb
new file mode 100644
index 0000000..fdaac54
--- /dev/null
+++ b/jni/ruby/test/rubygems/test_gem_commands_cleanup_command.rb
@@ -0,0 +1,167 @@
+require 'rubygems/test_case'
+require 'rubygems/commands/cleanup_command'
+
+class TestGemCommandsCleanupCommand < Gem::TestCase
+
+  def setup
+    super
+
+    @cmd = Gem::Commands::CleanupCommand.new
+
+    @a_1 = util_spec 'a', 1
+    @a_2 = util_spec 'a', 2
+
+    install_gem @a_1
+    install_gem @a_2
+  end
+
+  def test_handle_options_d
+    @cmd.handle_options %w[-d]
+    assert @cmd.options[:dryrun]
+  end
+
+  def test_handle_options_dry_run
+    @cmd.handle_options %w[--dryrun]
+    assert @cmd.options[:dryrun]
+  end
+
+  def test_handle_options_n
+    @cmd.handle_options %w[-n]
+    assert @cmd.options[:dryrun]
+  end
+
+  def test_execute
+    @cmd.options[:args] = %w[a]
+
+    @cmd.execute
+
+    refute_path_exists @a_1.gem_dir
+  end
+
+  def test_execute_all_dependencies
+    @b_1 = util_spec 'b', 1 do |s| s.add_dependency 'a', '1' end
+    @b_2 = util_spec 'b', 2 do |s| s.add_dependency 'a', '2' end
+
+    install_gem @b_1
+    install_gem @b_2
+
+    @cmd.options[:args] = []
+
+    @cmd.execute
+
+    refute_path_exists @a_1.gem_dir
+    refute_path_exists @b_1.gem_dir
+  end
+
+  def test_execute_all
+    gemhome2 = File.join @tempdir, 'gemhome2'
+
+    Gem.ensure_gem_subdirectories gemhome2
+
+    Gem.use_paths @gemhome, gemhome2
+
+    @b_1 = util_spec 'b', 1
+    @b_2 = util_spec 'b', 2
+
+    install_gem @b_1
+    install_gem @b_2
+
+    @cmd.options[:args] = []
+
+    @cmd.execute
+
+    assert_equal @gemhome, Gem.dir, 'GEM_HOME'
+    assert_equal [@gemhome, gemhome2], Gem.path.sort, 'GEM_PATH'
+
+    refute_path_exists @a_1.gem_dir
+    refute_path_exists @b_1.gem_dir
+  end
+
+  def test_execute_all_user
+    @a_1_1 = util_spec 'a', '1.1'
+    @a_1_1 = install_gem_user @a_1_1 # pick up user install path
+
+    Gem::Specification.dirs = [Gem.dir, Gem.user_dir]
+
+    assert_path_exists @a_1.gem_dir
+    assert_path_exists @a_1_1.gem_dir
+
+    @cmd.options[:args] = %w[a]
+
+    @cmd.execute
+
+    refute_path_exists @a_1.gem_dir
+    refute_path_exists @a_1_1.gem_dir
+  end
+
+  def test_execute_all_user_no_sudo
+    FileUtils.chmod 0555, @gemhome
+
+    @a_1_1 = util_spec 'a', '1.1'
+    @a_1_1 = install_gem_user @a_1_1 # pick up user install path
+
+    Gem::Specification.dirs = [Gem.dir, Gem.user_dir]
+
+    assert_path_exists @a_1.gem_dir
+    assert_path_exists @a_1_1.gem_dir
+
+    @cmd.options[:args] = %w[a]
+
+    @cmd.execute
+
+    assert_path_exists @a_1.gem_dir
+    refute_path_exists @a_1_1.gem_dir
+  ensure
+    FileUtils.chmod 0755, @gemhome
+  end unless win_platform?
+
+  def test_execute_dry_run
+    @cmd.options[:args] = %w[a]
+    @cmd.options[:dryrun] = true
+
+    @cmd.execute
+
+    assert_path_exists @a_1.gem_dir
+  end
+
+  def test_execute_keeps_older_versions_with_deps
+    @b_1 = util_spec 'b', 1
+    @b_2 = util_spec 'b', 2
+
+    @c = util_spec 'c', 1 do |s|
+      s.add_dependency 'b', '1'
+    end
+
+    install_gem @c
+    install_gem @b_1
+    install_gem @b_2
+
+    @cmd.options[:args] = []
+
+    @cmd.execute
+
+    assert_path_exists @b_1.gem_dir
+  end
+
+  def test_execute_ignore_default_gem_verbose
+    Gem.configuration.verbose = :really
+
+    @b_1 = util_spec 'b', 1
+    @b_default = new_default_spec "b", "2"
+    @b_2 = util_spec 'b', 3
+
+    install_gem @b_1
+    install_default_specs @b_default
+    install_gem @b_2
+
+    @cmd.options[:args] = []
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    assert_match %r%^Skipped default gems: b-2%, @ui.output
+    assert_empty @ui.error
+  end
+end
+