ruby-1.9.1-rc1

[splhack/AndroidRuby.git] / lib / ruby-1.9.1-rc1 / test / rubygems / test_gem_commands_pristine_command.rb

diff --git a/lib/ruby-1.9.1-rc1/test/rubygems/test_gem_commands_pristine_command.rb b/lib/ruby-1.9.1-rc1/test/rubygems/test_gem_commands_pristine_command.rb
new file mode 100644 (file)
index 0000000..307cba7
--- /dev/null
+++ b/lib/ruby-1.9.1-rc1/test/rubygems/test_gem_commands_pristine_command.rb
@@ -0,0 +1,108 @@
+require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
+require 'rubygems/commands/pristine_command'
+
+class TestGemCommandsPristineCommand < RubyGemTestCase
+
+  def setup
+    super
+    @cmd = Gem::Commands::PristineCommand.new
+  end
+
+  def test_execute
+    a = quick_gem 'a' do |s| s.executables = %w[foo] end
+    FileUtils.mkdir_p File.join(@tempdir, 'bin')
+    File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
+      fp.puts "#!/usr/bin/ruby"
+    end
+
+    install_gem a
+
+    foo_path = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo'
+
+    File.open foo_path, 'w' do |io|
+      io.puts 'I changed it!'
+    end
+
+    @cmd.options[:args] = %w[a]
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    assert_equal "#!/usr/bin/ruby\n", File.read(foo_path), foo_path
+
+    out = @ui.output.split "\n"
+
+    assert_equal "Restoring gem(s) to pristine condition...", out.shift
+    assert_equal "Restored #{a.full_name}", out.shift
+    assert out.empty?, out.inspect
+  end
+
+  def test_execute_all
+    a = quick_gem 'a' do |s| s.executables = %w[foo] end
+    FileUtils.mkdir_p File.join(@tempdir, 'bin')
+    File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
+      fp.puts "#!/usr/bin/ruby"
+    end
+
+    install_gem a
+
+    gem_bin = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo'
+
+    FileUtils.rm gem_bin
+
+    @cmd.handle_options %w[--all]
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    assert File.exist?(gem_bin)
+
+    out = @ui.output.split "\n"
+
+    assert_equal "Restoring gem(s) to pristine condition...", out.shift
+    assert_equal "Restored #{a.full_name}", out.shift
+    assert out.empty?, out.inspect
+  end
+
+  def test_execute_missing_cache_gem
+    a = quick_gem 'a' do |s| s.executables = %w[foo] end
+    FileUtils.mkdir_p File.join(@tempdir, 'bin')
+    File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
+      fp.puts "#!/usr/bin/ruby"
+    end
+
+    install_gem a
+
+    FileUtils.rm File.join(@gemhome, 'cache', "#{a.full_name}.gem")
+
+    @cmd.options[:args] = %w[a]
+
+    use_ui @ui do
+      @cmd.execute
+    end
+
+    out = @ui.output.split "\n"
+
+    assert_equal "Restoring gem\(s\) to pristine condition...", out.shift
+    assert out.empty?, out.inspect
+
+    assert_equal "ERROR:  Cached gem for #{a.full_name} not found, use `gem install` to restore\n",
+                 @ui.error
+  end
+
+  def test_execute_no_gem
+    @cmd.options[:args] = %w[]
+
+    e = assert_raises Gem::CommandLineError do
+      use_ui @ui do
+        @cmd.execute
+      end
+    end
+
+    assert_match %r|specify a gem name|, e.message
+  end
+
+end
+