2008-05-18 06:22:34 +02:00
|
|
|
require 'abstract_unit'
|
2007-12-21 08:48:59 +01:00
|
|
|
|
|
|
|
class AtomicWriteTest < Test::Unit::TestCase
|
|
|
|
def test_atomic_write_without_errors
|
2008-10-27 07:47:01 +01:00
|
|
|
contents = "Atomic Text"
|
2007-12-21 08:48:59 +01:00
|
|
|
File.atomic_write(file_name, Dir.pwd) do |file|
|
|
|
|
file.write(contents)
|
|
|
|
assert !File.exist?(file_name)
|
|
|
|
end
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
assert_equal contents, File.read(file_name)
|
|
|
|
ensure
|
|
|
|
File.unlink(file_name) rescue nil
|
|
|
|
end
|
2008-10-27 07:47:01 +01:00
|
|
|
|
2007-12-21 08:48:59 +01:00
|
|
|
def test_atomic_write_doesnt_write_when_block_raises
|
|
|
|
File.atomic_write(file_name) do |file|
|
|
|
|
file.write("testing")
|
|
|
|
raise "something bad"
|
|
|
|
end
|
|
|
|
rescue
|
|
|
|
assert !File.exist?(file_name)
|
|
|
|
end
|
2008-10-27 07:47:01 +01:00
|
|
|
|
|
|
|
def test_atomic_write_preserves_file_permissions
|
|
|
|
contents = "Atomic Text"
|
|
|
|
File.open(file_name, "w", 0755) do |file|
|
|
|
|
file.write(contents)
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
end
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
assert_equal 0100755, file_mode
|
|
|
|
assert_equal contents, File.read(file_name)
|
|
|
|
|
|
|
|
File.atomic_write(file_name, Dir.pwd) do |file|
|
|
|
|
file.write(contents)
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
end
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
assert_equal 0100755, file_mode
|
|
|
|
assert_equal contents, File.read(file_name)
|
|
|
|
ensure
|
|
|
|
File.unlink(file_name) rescue nil
|
|
|
|
end
|
|
|
|
|
|
|
|
def test_atomic_write_preserves_default_file_permissions
|
|
|
|
contents = "Atomic Text"
|
|
|
|
File.atomic_write(file_name, Dir.pwd) do |file|
|
|
|
|
file.write(contents)
|
|
|
|
assert !File.exist?(file_name)
|
|
|
|
end
|
|
|
|
assert File.exist?(file_name)
|
|
|
|
assert_equal 0100666 ^ File.umask, file_mode
|
|
|
|
assert_equal contents, File.read(file_name)
|
|
|
|
ensure
|
|
|
|
File.unlink(file_name) rescue nil
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|
2008-10-27 07:47:01 +01:00
|
|
|
|
|
|
|
private
|
|
|
|
def file_name
|
|
|
|
"atomic.file"
|
|
|
|
end
|
|
|
|
|
|
|
|
def file_mode
|
|
|
|
File.stat(file_name).mode
|
|
|
|
end
|
2007-12-21 08:48:59 +01:00
|
|
|
end
|