ruby/test/test_tempfile.rb

349 строки
7.5 KiB
Ruby

require 'test/unit'
require 'tempfile'
require 'thread'
class TestTempfile < Test::Unit::TestCase
def initialize(*)
super
@tempfile = nil
end
def tempfile(*args, &block)
t = Tempfile.new(*args, &block)
@tempfile = (t unless block)
end
def teardown
if @tempfile
@tempfile.close!
end
end
def test_basic
t = tempfile("foo")
path = t.path
t.write("hello world")
t.close
assert_equal "hello world", File.read(path)
end
def test_saves_in_dir_tmpdir_by_default
t = tempfile("foo")
assert_equal Dir.tmpdir, File.dirname(t.path)
bug3733 = '[ruby-dev:42089]'
assert_nothing_raised(SecurityError, bug3733) {
proc {$SAFE = 1; File.expand_path(Dir.tmpdir)}.call
}
end
def test_saves_in_given_directory
subdir = File.join(Dir.tmpdir, "tempfile-test-#{rand}")
Dir.mkdir(subdir)
begin
tempfile = Tempfile.new("foo", subdir)
tempfile.close
begin
assert_equal subdir, File.dirname(tempfile.path)
ensure
tempfile.unlink
end
ensure
Dir.rmdir(subdir)
end
end
def test_basename
t = tempfile("foo")
assert_match(/^foo/, File.basename(t.path))
end
def test_default_basename
t = tempfile
assert_file.exist?(t.path)
end
def test_basename_with_suffix
t = tempfile(["foo", ".txt"])
assert_match(/^foo/, File.basename(t.path))
assert_match(/\.txt$/, File.basename(t.path))
end
def test_unlink
t = tempfile("foo")
path = t.path
t.close
assert_file.exist?(path)
t.unlink
assert !File.exist?(path)
assert_nil t.path
end
def test_unlink_silently_fails_on_windows
tempfile = tempfile("foo")
path = tempfile.path
begin
assert_nothing_raised do
tempfile.unlink
end
ensure
tempfile.close
File.unlink(path) if File.exist?(path)
end
end
def test_unlink_before_close_works_on_posix_systems
tempfile = tempfile("foo")
begin
path = tempfile.path
tempfile.unlink
assert !File.exist?(path)
tempfile.write("hello ")
tempfile.write("world\n")
tempfile.rewind
assert_equal "hello world\n", tempfile.read
ensure
tempfile.close
tempfile.unlink
end
end unless /mswin|mingw/ =~ RUBY_PLATFORM
def test_close_and_close_p
t = tempfile("foo")
assert !t.closed?
t.close
assert t.closed?
end
def test_close_with_unlink_now_true_works
t = tempfile("foo")
path = t.path
t.close(true)
assert t.closed?
assert_nil t.path
assert !File.exist?(path)
end
def test_close_with_unlink_now_true_does_not_unlink_if_already_unlinked
t = tempfile("foo")
path = t.path
t.unlink
File.open(path, "w").close
begin
t.close(true)
assert_file.exist?(path)
ensure
File.unlink(path) rescue nil
end
end unless /mswin|mingw/ =~ RUBY_PLATFORM
def test_close_bang_works
t = tempfile("foo")
path = t.path
t.close!
assert t.closed?
assert_nil t.path
assert !File.exist?(path)
end
def test_close_bang_does_not_unlink_if_already_unlinked
t = tempfile("foo")
path = t.path
t.unlink
File.open(path, "w").close
begin
t.close!
assert_file.exist?(path)
ensure
File.unlink(path) rescue nil
end
end unless /mswin|mingw/ =~ RUBY_PLATFORM
def test_finalizer_does_not_unlink_if_already_unlinked
assert_in_out_err('-rtempfile', <<-'EOS') do |(filename,*), (error,*)|
file = Tempfile.new('foo')
path = file.path
puts path
file.close!
File.open(path, "w").close
EOS
assert_file.exist?(filename)
File.unlink(filename)
assert_nil error
end
assert_in_out_err('-rtempfile', <<-'EOS') do |(filename,*), (error,*)|
file = Tempfile.new('foo')
path = file.path
file.unlink
puts path
File.open(path, "w").close
EOS
if !filename.empty?
# POSIX unlink semantics supported, continue with test
assert_file.exist?(filename)
File.unlink(filename)
end
assert_nil error
end
end unless /mswin|mingw/ =~ RUBY_PLATFORM
def test_close_does_not_make_path_nil
t = tempfile("foo")
t.close
assert_not_nil t.path
end
def test_close_flushes_buffer
t = tempfile("foo")
t.write("hello")
t.close
assert_equal 5, File.size(t.path)
end
def test_tempfile_is_unlinked_when_ruby_exits
assert_in_out_err('-rtempfile', <<-'EOS') do |(filename), (error)|
puts Tempfile.new('foo').path
EOS
assert !File.exist?(filename)
assert_nil(error)
end
end
def test_tempfile_finalizer_does_not_run_if_unlinked
bug8768 = '[ruby-core:56521] [Bug #8768]'
args = %w(--disable-gems -rtempfile)
assert_in_out_err(args, <<-'EOS') do |(filename), (error)|
tmp = Tempfile.new('foo')
puts tmp.path
tmp.close
tmp.unlink
$DEBUG = true
EOS
assert_file.not_exist?(filename)
assert_nil(error, "#{bug8768} we used to get a confusing 'removing ...done' here")
end
end
def test_size_flushes_buffer_before_determining_file_size
t = tempfile("foo")
t.write("hello")
assert_equal 0, File.size(t.path)
assert_equal 5, t.size
assert_equal 5, File.size(t.path)
end
def test_size_works_if_file_is_closed
t = tempfile("foo")
t.write("hello")
t.close
assert_equal 5, t.size
end
def test_concurrency
threads = []
tempfiles = []
lock = Mutex.new
cond = ConditionVariable.new
start = false
4.times do
threads << Thread.new do
lock.synchronize do
while !start
cond.wait(lock)
end
end
result = []
30.times do
result << Tempfile.new('foo')
end
Thread.current[:result] = result
end
end
lock.synchronize do
start = true
cond.broadcast
end
threads.each do |thread|
thread.join
tempfiles |= thread[:result]
end
filenames = tempfiles.map { |f| f.path }
begin
assert_equal filenames.size, filenames.uniq.size
ensure
tempfiles.each do |tempfile|
tempfile.close!
end
end
end
module M
end
def test_extend
o = tempfile("foo")
o.extend M
assert(M === o, "[ruby-dev:32932]")
end
def test_tempfile_encoding_nooption
default_external=Encoding.default_external
t = tempfile("TEST")
t.write("\xE6\x9D\xBE\xE6\xB1\x9F")
t.rewind
assert_equal(default_external,t.read.encoding)
end
def test_tempfile_encoding_ascii8bit
t = tempfile("TEST",:encoding=>"ascii-8bit")
t.write("\xE6\x9D\xBE\xE6\xB1\x9F")
t.rewind
assert_equal(Encoding::ASCII_8BIT,t.read.encoding)
end
def test_tempfile_encoding_ascii8bit2
t = tempfile("TEST",Dir::tmpdir,:encoding=>"ascii-8bit")
t.write("\xE6\x9D\xBE\xE6\xB1\x9F")
t.rewind
assert_equal(Encoding::ASCII_8BIT,t.read.encoding)
end
def test_binmode
t = tempfile("TEST", mode: IO::BINARY)
if IO::BINARY.nonzero?
assert(t.binmode?)
t.open
assert(t.binmode?, 'binmode after reopen')
else
assert_equal(0600, t.stat.mode & 0777)
end
end
def test_create_with_block
path = nil
Tempfile.create("tempfile-create") {|f|
path = f.path
assert(File.exist?(path))
}
assert(!File.exist?(path))
end
def test_create_without_block
path = nil
f = Tempfile.create("tempfile-create")
path = f.path
assert(File.exist?(path))
f.close
assert(File.exist?(path))
ensure
f.close if f && !f.closed?
File.unlink path if path
end
end