# frozen_string_literal: true require_relative "helper" 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") Gem::Specification.reset @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