test/rubygems/test_gem_commands_pristine_command.rb

   1 require 'test/unit'
   2 require File.join(File.expand_path(File.dirname(__FILE__)), 'gemutilities')
   3 require 'rubygems/commands/pristine_command'
   4
   5 class TestGemCommandsPristineCommand < RubyGemTestCase
   6
   7   def setup
   8     super
   9     @cmd = Gem::Commands::PristineCommand.new
  10   end
  11
  12   def test_execute
  13     a = quick_gem 'a' do |s| s.executables = %w[foo] end
  14     FileUtils.mkdir_p File.join(@tempdir, 'bin')
  15     File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
  16       fp.puts "#!/usr/bin/ruby"
  17     end
  18
  19     install_gem a
  20
  21     foo_path = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo'
  22
  23     File.open foo_path, 'w' do |io|
  24       io.puts 'I changed it!'
  25     end
  26
  27     @cmd.options[:args] = %w[a]
  28
  29     use_ui @ui do
  30       @cmd.execute
  31     end
  32
  33     assert_equal "#!/usr/bin/ruby\n", File.read(foo_path), foo_path
  34
  35     out = @ui.output.split "\n"
  36
  37     assert_equal "Restoring gem(s) to pristine condition...", out.shift
  38     assert_equal "Restored #{a.full_name}", out.shift
  39     assert out.empty?, out.inspect
  40   end
  41
  42   def test_execute_all
  43     a = quick_gem 'a' do |s| s.executables = %w[foo] end
  44     FileUtils.mkdir_p File.join(@tempdir, 'bin')
  45     File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
  46       fp.puts "#!/usr/bin/ruby"
  47     end
  48
  49     install_gem a
  50
  51     gem_bin = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo'
  52
  53     FileUtils.rm gem_bin
  54
  55     @cmd.handle_options %w[--all]
  56
  57     use_ui @ui do
  58       @cmd.execute
  59     end
  60
  61     assert File.exist?(gem_bin)
  62
  63     out = @ui.output.split "\n"
  64
  65     assert_equal "Restoring gem(s) to pristine condition...", out.shift
  66     assert_equal "Restored #{a.full_name}", out.shift
  67     assert out.empty?, out.inspect
  68   end
  69
  70   def test_execute_missing_cache_gem
  71     a = quick_gem 'a' do |s| s.executables = %w[foo] end
  72     FileUtils.mkdir_p File.join(@tempdir, 'bin')
  73     File.open File.join(@tempdir, 'bin', 'foo'), 'w' do |fp|
  74       fp.puts "#!/usr/bin/ruby"
  75     end
  76
  77     install_gem a
  78
  79     FileUtils.rm File.join(@gemhome, 'cache', "#{a.full_name}.gem")
  80
  81     @cmd.options[:args] = %w[a]
  82
  83     use_ui @ui do
  84       @cmd.execute
  85     end
  86
  87     out = @ui.output.split "\n"
  88
  89     assert_equal "Restoring gem\(s\) to pristine condition...", out.shift
  90     assert out.empty?, out.inspect
  91
  92     assert_equal "ERROR:  Cached gem for #{a.full_name} not found, use `gem install` to restore\n",
  93                  @ui.error
  94   end
  95
  96   def test_execute_no_gem
  97     @cmd.options[:args] = %w[]
  98
  99     e = assert_raise Gem::CommandLineError do
 100       use_ui @ui do
 101         @cmd.execute
 102       end
 103     end
 104
 105     assert_match %r|specify a gem name|, e.message
 106   end
 107
 108 end
 109