# frozen_string_literal: true require 'rubygems/test_case' require 'rubygems/commands/pristine_command' class TestGemCommandsPristineCommand < Gem::TestCase def setup super common_installer_setup @cmd = Gem::Commands::PristineCommand.new end def test_execute a = util_spec 'a' do |s| s.executables = %w[foo] s.files = %w[bin/foo lib/a.rb] end write_file File.join(@tempdir, 'lib', 'a.rb') do |fp| fp.puts "puts __FILE__" end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a foo_path = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo' a_rb_path = File.join @gemhome, 'gems', a.full_name, 'lib', 'a.rb' write_file foo_path do |io| io.puts 'I changed it!' end write_file a_rb_path 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 assert_equal "puts __FILE__\n", File.read(a_rb_path), a_rb_path out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect end def test_execute_user_install FileUtils.chmod 0555, @gemhome a = util_spec "a" do |s| s.executables = %w[foo] s.files = %w[bin/foo lib/a.rb] end write_file File.join(@tempdir, "lib", "a.rb") do |fp| fp.puts "puts __FILE__" end write_file File.join(@tempdir, "bin", "foo") do |fp| fp.puts "#!/usr/bin/ruby" end install_gem_user(a) Gem::Specification.dirs = [Gem.dir, Gem.user_dir] foo_path = File.join(Gem.user_dir, "gems", a.full_name, "bin", "foo") a_rb_path = File.join(Gem.user_dir, "gems", a.full_name, "lib", "a.rb") write_file foo_path do |io| io.puts("I changed it!") end write_file a_rb_path 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 assert_equal "puts __FILE__\n", File.read(a_rb_path), a_rb_path out = @ui.output.split("\n") assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect ensure FileUtils.chmod(0755, @gemhome) end def test_execute_all a = util_spec 'a' do |s| s.executables = %w[foo] end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a gem_bin = File.join @gemhome, 'gems', a.full_name, 'bin', 'foo' gem_stub = File.join @gemhome, 'bin', 'foo' FileUtils.rm gem_bin FileUtils.rm gem_stub @cmd.handle_options %w[--all] use_ui @ui do @cmd.execute end assert File.exist?(gem_bin) assert File.exist?(gem_stub) out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect end def test_execute_env_shebang a = util_spec 'a' do |s| s.executables = %w[foo] s.files = %w[bin/foo] end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a gem_exec = File.join @gemhome, 'bin', 'foo' FileUtils.rm gem_exec @cmd.handle_options %w[--all --env-shebang] use_ui @ui do @cmd.execute end assert_path_exist gem_exec ruby_exec = sprintf Gem.default_exec_format, 'ruby' bin_env = win_platform? ? "" : %w[/usr/bin/env /bin/env].find {|f| File.executable?(f) } + " " assert_match %r{\A#!\s*#{bin_env}#{ruby_exec}}, File.read(gem_exec) end def test_execute_extensions_explicit a = util_spec 'a' do |s| s.extensions << 'ext/a/extconf.rb' end ext_path = File.join @tempdir, 'ext', 'a', 'extconf.rb' write_file ext_path do |io| io.write <<-'RUBY' File.open "Makefile", "w" do |f| f.puts "clean:\n\techo cleaned\n" f.puts "all:\n\techo built\n" f.puts "install:\n\techo installed\n" end RUBY end b = util_spec 'b' install_gem a install_gem b @cmd.options[:extensions] = true @cmd.options[:extensions_set] = true @cmd.options[:args] = [] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal 'Restoring gems to pristine condition...', out.shift assert_equal 'Building native extensions. This could take a while...', out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect end def test_execute_no_extension a = util_spec 'a' do |s| s.extensions << 'ext/a/extconf.rb' end ext_path = File.join @tempdir, 'ext', 'a', 'extconf.rb' write_file ext_path do |io| io.write "# extconf.rb\nrequire 'mkmf'; create_makefile 'a'" end install_gem a @cmd.options[:args] = %w[a] @cmd.options[:extensions] = false use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal 'Restoring gems to pristine condition...', out.shift assert_equal "Skipped #{a.full_name}, it needs to compile an extension", out.shift assert_empty out, out.inspect end def test_execute_with_extension_with_build_args a = util_spec 'a' do |s| s.extensions << 'ext/a/extconf.rb' end ext_path = File.join @tempdir, 'ext', 'a', 'extconf.rb' write_file ext_path do |io| io.write <<-'RUBY' File.open "Makefile", "w" do |f| f.puts "clean:\n\techo cleaned\n" f.puts "all:\n\techo built\n" f.puts "install:\n\techo installed\n" end RUBY end build_args = %w[--with-awesome=true --sweet] install_gem a, :build_args => build_args @cmd.options[:args] = %w[a] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal 'Restoring gems to pristine condition...', out.shift assert_equal "Building native extensions with: '--with-awesome=true --sweet'", out.shift assert_equal "This could take a while...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect end def test_execute_many a = util_spec 'a' b = util_spec 'b' install_gem a install_gem b @cmd.options[:args] = %w[a b] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_equal "Restored #{b.full_name}", out.shift assert_empty out, out.inspect end def test_skip a = util_spec 'a' b = util_spec 'b' install_gem a install_gem b @cmd.options[:args] = %w[a b] @cmd.options[:skip] = 'a' use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Skipped #{a.full_name}, it was given through options", out.shift assert_equal "Restored #{b.full_name}", out.shift assert_empty out, out.inspect end def test_skip_many_gems a = util_spec 'a' b = util_spec 'b' c = util_spec 'c' install_gem a install_gem b install_gem c @cmd.options[:args] = %w[a b c] @cmd.options[:skip] = ['a', 'c'] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Skipped #{a.full_name}, it was given through options", out.shift assert_equal "Restored #{b.full_name}", out.shift assert_equal "Skipped #{c.full_name}, it was given through options", out.shift assert_empty out, out.inspect end def test_execute_many_multi_repo a = util_spec 'a' install_gem a Gem.clear_paths gemhome2 = File.join @tempdir, 'gemhome2' Gem.use_paths gemhome2, [gemhome2, @gemhome] b = util_spec 'b' install_gem b @cmd.options[:args] = %w[a b] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" assert_equal "Restoring gems to pristine condition...", out.shift assert_equal "Restored #{a.full_name}", out.shift assert_equal "Restored #{b.full_name}", out.shift assert_empty out, out.inspect assert_path_exist File.join(@gemhome, "gems", 'a-2') assert_path_not_exist File.join(gemhome2, "gems", 'a-2') assert_path_exist File.join(gemhome2, "gems", 'b-2') assert_path_not_exist File.join(@gemhome, "gems", 'b-2') end def test_execute_missing_cache_gem specs = spec_fetcher do |fetcher| fetcher.gem 'a', 1 fetcher.gem 'a', 2 fetcher.gem 'a', 3 fetcher.gem 'a', '3.a' end FileUtils.rm specs['a-2'].cache_file @cmd.options[:args] = %w[a] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" [ "Restoring gems to pristine condition...", "Restored a-1", "Cached gem for a-2 not found, attempting to fetch...", "Restored a-2", "Restored a-3.a", "Restored a-3", ].each do |line| assert_equal line, out.shift end assert_empty out, out.inspect end def test_execute_missing_cache_gem_when_multi_repo specs = spec_fetcher do |fetcher| fetcher.gem 'a', 1 fetcher.gem 'b', 1 end FileUtils.rm_rf File.join(@gemhome, 'gems', 'a-1') FileUtils.rm_rf File.join(@gemhome, 'gems', 'b-1') install_gem specs["a-1"] FileUtils.rm File.join(@gemhome, 'cache', 'a-1.gem') Gem.clear_paths gemhome2 = File.join(@tempdir, 'gemhome2') Gem.use_paths gemhome2, [gemhome2, @gemhome] install_gem specs["b-1"] FileUtils.rm File.join(gemhome2, 'cache', 'b-1.gem') @cmd.options[:args] = %w[a b] use_ui @ui do @cmd.execute end out = @ui.output.split "\n" [ "Restoring gems to pristine condition...", "Cached gem for a-1 not found, attempting to fetch...", "Restored a-1", "Cached gem for b-1 not found, attempting to fetch...", "Restored b-1", ].each do |line| assert_equal line, out.shift end assert_empty out, out.inspect assert_empty @ui.error assert_path_exist File.join(@gemhome, "cache", 'a-1.gem') assert_path_not_exist File.join(gemhome2, "cache", 'a-2.gem') assert_path_exist File.join(@gemhome, "gems", 'a-1') assert_path_not_exist File.join(gemhome2, "gems", 'a-1') assert_path_exist File.join(gemhome2, "cache", 'b-1.gem') assert_path_not_exist File.join(@gemhome, "cache", 'b-2.gem') assert_path_exist File.join(gemhome2, "gems", 'b-1') assert_path_not_exist File.join(@gemhome, "gems", 'b-1') end def test_execute_no_gem @cmd.options[:args] = %w[] e = assert_raise Gem::CommandLineError do use_ui @ui do @cmd.execute end end assert_match %r{at least one gem name}, e.message end def test_execute_only_executables a = util_spec 'a' do |s| s.executables = %w[foo] s.files = %w[bin/foo lib/a.rb] end write_file File.join(@tempdir, 'lib', 'a.rb') do |fp| fp.puts "puts __FILE__" end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a gem_lib = File.join @gemhome, 'gems', a.full_name, 'lib', 'a.rb' gem_exec = File.join @gemhome, 'bin', 'foo' FileUtils.rm gem_exec FileUtils.rm gem_lib @cmd.handle_options %w[--all --only-executables] use_ui @ui do @cmd.execute end assert File.exist? gem_exec refute File.exist? gem_lib end def test_execute_only_plugins a = util_spec 'a' do |s| s.executables = %w[foo] s.files = %w[bin/foo lib/a.rb lib/rubygems_plugin.rb] end write_file File.join(@tempdir, 'lib', 'a.rb') do |fp| fp.puts "puts __FILE__" end write_file File.join(@tempdir, 'lib', 'rubygems_plugin.rb') do |fp| fp.puts "puts __FILE__" end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a gem_lib = File.join @gemhome, 'gems', a.full_name, 'lib', 'a.rb' gem_plugin = File.join @gemhome, 'plugins', 'a_plugin.rb' gem_exec = File.join @gemhome, 'bin', 'foo' FileUtils.rm gem_exec FileUtils.rm gem_plugin FileUtils.rm gem_lib @cmd.handle_options %w[--all --only-plugins] use_ui @ui do @cmd.execute end refute File.exist? gem_exec assert File.exist? gem_plugin refute File.exist? gem_lib end def test_execute_bindir a = util_spec 'a' do |s| s.name = "test_gem" s.executables = %w[foo] s.files = %w[bin/foo] end write_file File.join(@tempdir, 'bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end write_file File.join(@tempdir, 'test_bin', 'foo') do |fp| fp.puts "#!/usr/bin/ruby" end install_gem a gem_exec = File.join @gemhome, 'bin', 'foo' gem_bindir = File.join @tempdir, 'test_bin', 'foo' FileUtils.rm gem_exec FileUtils.rm gem_bindir @cmd.handle_options ["--all", "--only-executables", "--bindir", "#{gem_bindir}"] use_ui @ui do @cmd.execute end refute File.exist? gem_exec assert File.exist? gem_bindir end def test_execute_unknown_gem_at_remote_source install_specs util_spec 'a' @cmd.options[:args] = %w[a] use_ui @ui do @cmd.execute end assert_equal([ "Restoring gems to pristine condition...", "Cached gem for a-2 not found, attempting to fetch...", "Skipped a-2, it was not found from cache and remote sources", ], @ui.output.split("\n")) assert_empty @ui.error end def test_execute_default_gem default_gem_spec = new_default_spec("default", "2.0.0.0", nil, "default/gem.rb") install_default_gems(default_gem_spec) @cmd.options[:args] = %w[default] use_ui @ui do @cmd.execute end assert_equal( [ "Restoring gems to pristine condition...", "Skipped default-2.0.0.0, it is a default gem", ], @ui.output.split("\n") ) assert_empty(@ui.error) end def test_execute_multi_platform a = util_spec 'a' do |s| s.extensions << 'ext/a/extconf.rb' end b = util_spec 'b' do |s| s.extensions << 'ext/a/extconf.rb' s.platform = Gem::Platform.new("java") end ext_path = File.join @tempdir, 'ext', 'a', 'extconf.rb' write_file ext_path do |io| io.write <<-'RUBY' File.open "Makefile", "w" do |f| f.puts "clean:\n\techo cleaned\n" f.puts "all:\n\techo built\n" f.puts "install:\n\techo installed\n" end RUBY end install_gem a install_gem b @cmd.options[:extensions] = true @cmd.options[:extensions_set] = true @cmd.options[:args] = [] util_set_arch "x86_64-darwin" do use_ui @ui do @cmd.execute end end out = @ui.output.split "\n" assert_equal 'Restoring gems to pristine condition...', out.shift assert_equal 'Building native extensions. This could take a while...', out.shift assert_equal "Restored #{a.full_name}", out.shift assert_empty out, out.inspect end def test_handle_options @cmd.handle_options %w[] refute @cmd.options[:all] assert @cmd.options[:extensions] refute @cmd.options[:extensions_set] assert_equal Gem::Requirement.default, @cmd.options[:version] end def test_handle_options_extensions @cmd.handle_options %w[--extensions] assert @cmd.options[:extensions] assert @cmd.options[:extensions_set] end end