test/test_tmpdir.rb

   1 # frozen_string_literal: true
   2 require 'test/unit'
   3 require 'tmpdir'
   4
   5 class TestTmpdir < Test::Unit::TestCase
   6   def test_tmpdir_modifiable
   7     tmpdir = Dir.tmpdir
   8     assert_not_predicate(tmpdir, :frozen?)
   9     tmpdir_org = tmpdir.dup
  10     tmpdir << "foo"
  11     assert_equal(tmpdir_org, Dir.tmpdir)
  12   end
  13
  14   def test_world_writable
  15     omit "no meaning on this platform" if /mswin|mingw/ =~ RUBY_PLATFORM
  16     Dir.mktmpdir do |tmpdir|
  17       # ToDo: fix for parallel test
  18       envs = %w[TMPDIR TMP TEMP]
  19       oldenv = envs.each_with_object({}) {|v, h| h[v] = ENV.delete(v)}
  20       begin
  21         envs.each do |e|
  22           tmpdirx = File.join(tmpdir, e)
  23           ENV[e] = tmpdirx
  24           assert_not_equal(tmpdirx, assert_warn('') {Dir.tmpdir})
  25           File.write(tmpdirx, "")
  26           assert_not_equal(tmpdirx, assert_warn(/\A#{e} is not a directory/) {Dir.tmpdir})
  27           File.unlink(tmpdirx)
  28           ENV[e] = tmpdir
  29           assert_equal(tmpdir, Dir.tmpdir)
  30           File.chmod(0555, tmpdir)
  31           assert_not_equal(tmpdir, assert_warn(/\A#{e} is not writable/) {Dir.tmpdir})
  32           File.chmod(0777, tmpdir)
  33           assert_not_equal(tmpdir, assert_warn(/\A#{e} is world-writable/) {Dir.tmpdir})
  34           newdir = Dir.mktmpdir("d", tmpdir) do |dir|
  35             assert_file.directory? dir
  36             assert_equal(tmpdir, File.dirname(dir))
  37             dir
  38           end
  39           assert_file.not_exist?(newdir)
  40           File.chmod(01777, tmpdir)
  41           assert_equal(tmpdir, Dir.tmpdir)
  42           ENV[e] = nil
  43         end
  44       ensure
  45         ENV.update(oldenv)
  46       end
  47     end
  48   end
  49
  50   def test_tmpdir_not_empty_parent
  51     Dir.mktmpdir do |tmpdir|
  52       envs = %w[TMPDIR TMP TEMP]
  53       oldenv = envs.each_with_object({}) {|v, h| h[v] = ENV.delete(v)}
  54       ENV[envs[0]] = ""
  55       ENV[envs[1]] = tmpdir
  56       assert_equal(tmpdir, Dir.tmpdir)
  57     ensure
  58       ENV.update(oldenv)
  59     end
  60   end
  61
  62   def test_no_homedir
  63     bug7547 = '[ruby-core:50793]'
  64     home, ENV["HOME"] = ENV["HOME"], nil
  65     dir = assert_nothing_raised(bug7547) do
  66       break Dir.mktmpdir("~")
  67     end
  68     assert_match(/\A~/, File.basename(dir), bug7547)
  69   ensure
  70     ENV["HOME"] = home
  71     Dir.rmdir(dir) if dir
  72   end
  73
  74   def test_mktmpdir_nil
  75     Dir.mktmpdir(nil) {|d|
  76       assert_kind_of(String, d)
  77     }
  78   end
  79
  80   def test_mktmpdir_mutate
  81     bug16918 = '[ruby-core:98563]'
  82     assert_nothing_raised(bug16918) do
  83       assert_mktmpdir_traversal do |traversal_path|
  84         Dir.mktmpdir(traversal_path + 'foo') do |actual|
  85           actual << "foo"
  86         end
  87       end
  88     end
  89   end
  90
  91   def test_mktmpdir_traversal
  92     assert_mktmpdir_traversal do |traversal_path|
  93       Dir.mktmpdir(traversal_path + 'foo') do |actual|
  94         actual
  95       end
  96     end
  97   end
  98
  99   def test_mktmpdir_traversal_array
 100     assert_mktmpdir_traversal do |traversal_path|
 101       Dir.mktmpdir([traversal_path, 'foo']) do |actual|
 102         actual
 103       end
 104     end
 105   end
 106
 107   def assert_mktmpdir_traversal
 108     Dir.mktmpdir do |target|
 109       target = target.chomp('/') + '/'
 110       traversal_path = target.sub(/\A\w:/, '') # for DOSISH
 111       traversal_path = Array.new(target.count('/')-2, '..').join('/') + traversal_path
 112       [File::SEPARATOR, File::ALT_SEPARATOR].compact.each do |separator|
 113         actual = yield traversal_path.tr('/', separator)
 114         assert_not_send([File.absolute_path(actual), :start_with?, target])
 115       end
 116     end
 117   end
 118
 119   def test_ractor
 120     assert_ractor(<<~'end;', require: "tmpdir")
 121       r = Ractor.new do
 122         Dir.mktmpdir() do |d|
 123           Ractor.yield d
 124           Ractor.receive
 125         end
 126       end
 127       dir = r.take
 128       assert_file.directory? dir
 129       r.send true
 130       r.take
 131       assert_file.not_exist? dir
 132     end;
 133   end
 134 end