ruby/test/test_find.rb

302 строки
7.7 KiB
Ruby

require 'test/unit'
require 'find'
require 'tmpdir'
class TestFind < Test::Unit::TestCase
def test_empty
Dir.mktmpdir {|d|
a = []
Find.find(d) {|f| a << f }
assert_equal([d], a)
}
end
def test_rec
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
File.open("#{d}/b/a", "w"){}
File.open("#{d}/b/b", "w"){}
Dir.mkdir("#{d}/c")
a = []
Find.find(d) {|f| a << f }
assert_equal([d, "#{d}/a", "#{d}/b", "#{d}/b/a", "#{d}/b/b", "#{d}/c"], a)
}
end
def test_relative
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
File.open("#{d}/b/a", "w"){}
File.open("#{d}/b/b", "w"){}
Dir.mkdir("#{d}/c")
a = []
Dir.chdir(d) {
Find.find(".") {|f| a << f }
}
assert_equal([".", "./a", "./b", "./b/a", "./b/b", "./c"], a)
}
end
def test_dont_follow_symlink
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
File.open("#{d}/b/a", "w"){}
File.open("#{d}/b/b", "w"){}
begin
File.symlink("#{d}/b", "#{d}/c")
rescue NotImplementedError
skip "symlink is not supported."
end
a = []
Find.find(d) {|f| a << f }
assert_equal([d, "#{d}/a", "#{d}/b", "#{d}/b/a", "#{d}/b/b", "#{d}/c"], a)
}
end
def test_prune
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
File.open("#{d}/b/a", "w"){}
File.open("#{d}/b/b", "w"){}
Dir.mkdir("#{d}/c")
a = []
Find.find(d) {|f|
a << f
Find.prune if f == "#{d}/b"
}
assert_equal([d, "#{d}/a", "#{d}/b", "#{d}/c"], a)
}
end
def test_countup3
Dir.mktmpdir {|d|
1.upto(3) {|n| File.open("#{d}/#{n}", "w"){} }
a = []
Find.find(d) {|f| a << f }
assert_equal([d, "#{d}/1", "#{d}/2", "#{d}/3"], a)
}
end
def test_countdown3
Dir.mktmpdir {|d|
3.downto(1) {|n| File.open("#{d}/#{n}", "w"){} }
a = []
Find.find(d) {|f| a << f }
assert_equal([d, "#{d}/1", "#{d}/2", "#{d}/3"], a)
}
end
def test_unreadable_dir
skip "no meaning test on Windows" if /mswin|mingw/ =~ RUBY_PLATFORM
Dir.mktmpdir {|d|
Dir.mkdir(dir = "#{d}/dir")
File.open("#{dir}/foo", "w"){}
begin
File.chmod(0300, dir)
a = []
Find.find(d) {|f| a << f }
assert_equal([d, dir], a)
a = []
Find.find(d, ignore_error: true) {|f| a << f }
assert_equal([d, dir], a)
a = []
Find.find(d, ignore_error: true).each {|f| a << f }
assert_equal([d, dir], a)
a = []
assert_raise_with_message(Errno::EACCES, /#{Regexp.quote(dir)}/) do
Find.find(d, ignore_error: false) {|f| a << f }
end
assert_equal([d, dir], a)
a = []
assert_raise_with_message(Errno::EACCES, /#{Regexp.quote(dir)}/) do
Find.find(d, ignore_error: false).each {|f| a << f }
end
assert_equal([d, dir], a)
ensure
File.chmod(0700, dir)
end
}
end
def test_unsearchable_dir
Dir.mktmpdir {|d|
Dir.mkdir(dir = "#{d}/dir")
File.open(file = "#{dir}/foo", "w"){}
begin
File.chmod(0600, dir)
a = []
Find.find(d) {|f| a << f }
assert_equal([d, dir, file], a)
a = []
Find.find(d, ignore_error: true) {|f| a << f }
assert_equal([d, dir, file], a)
a = []
Find.find(d, ignore_error: true).each {|f| a << f }
assert_equal([d, dir, file], a)
skip "no meaning test on Windows" if /mswin|mingw/ =~ RUBY_PLATFORM
a = []
assert_raise_with_message(Errno::EACCES, /#{Regexp.quote(file)}/) do
Find.find(d, ignore_error: false) {|f| a << f }
end
assert_equal([d, dir, file], a)
a = []
assert_raise_with_message(Errno::EACCES, /#{Regexp.quote(file)}/) do
Find.find(d, ignore_error: false).each {|f| a << f }
end
assert_equal([d, dir, file], a)
assert_raise(Errno::EACCES) { File.lstat(file) }
ensure
File.chmod(0700, dir)
end
}
end
def test_dangling_symlink
Dir.mktmpdir {|d|
begin
File.symlink("foo", "#{d}/bar")
rescue NotImplementedError
skip "symlink is not supported."
end
a = []
Find.find(d) {|f| a << f }
assert_equal([d, "#{d}/bar"], a)
assert_raise(Errno::ENOENT) { File.stat("#{d}/bar") }
}
end
def test_dangling_symlink_stat_error
Dir.mktmpdir {|d|
begin
File.symlink("foo", "#{d}/bar")
rescue NotImplementedError
skip "symlink is not supported."
end
assert_raise(Errno::ENOENT) {
Find.find(d) {|f| File.stat(f) }
}
}
end
def test_change_dir_to_file
Dir.mktmpdir {|d|
Dir.mkdir(dir_1 = "#{d}/d1")
File.open(file_a = "#{dir_1}/a", "w"){}
File.open(file_b = "#{dir_1}/b", "w"){}
File.open(file_c = "#{dir_1}/c", "w"){}
Dir.mkdir(dir_d = "#{dir_1}/d")
File.open("#{dir_d}/e", "w"){}
dir_2 = "#{d}/d2"
a = []
Find.find(d) {|f|
a << f
if f == file_b
File.rename(dir_1, dir_2)
File.open(dir_1, "w") {}
end
}
assert_equal([d, dir_1, file_a, file_b, file_c, dir_d], a)
}
end
def test_change_dir_to_symlink_loop
Dir.mktmpdir {|d|
Dir.mkdir(dir_1 = "#{d}/d1")
File.open(file_a = "#{dir_1}/a", "w"){}
File.open(file_b = "#{dir_1}/b", "w"){}
File.open(file_c = "#{dir_1}/c", "w"){}
Dir.mkdir(dir_d = "#{dir_1}/d")
File.open("#{dir_d}/e", "w"){}
dir_2 = "#{d}/d2"
a = []
Find.find(d) {|f|
a << f
if f == file_b
File.rename(dir_1, dir_2)
begin
File.symlink("d1", dir_1)
rescue NotImplementedError
skip "symlink is not supported."
end
end
}
assert_equal([d, dir_1, file_a, file_b, file_c, dir_d], a)
}
end
def test_enumerator
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
File.open("#{d}/b/a", "w"){}
File.open("#{d}/b/b", "w"){}
Dir.mkdir("#{d}/c")
e = Find.find(d)
a = []
e.each {|f| a << f }
assert_equal([d, "#{d}/a", "#{d}/b", "#{d}/b/a", "#{d}/b/b", "#{d}/c"], a)
}
end
def test_encoding_ascii
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
a = []
Find.find(d.encode(Encoding::US_ASCII)) {|f| a << f }
a.each do |i|
assert(Encoding.compatible?(d.encode(Encoding.find('filesystem')), i))
end
}
end
def test_encoding_non_ascii
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
Dir.mkdir("#{d}/b")
euc_jp = Encoding::EUC_JP
win_31j = Encoding::Windows_31J
utf_8 = Encoding::UTF_8
a = []
Find.find(d.encode(euc_jp), d.encode(win_31j), d.encode(utf_8)) {|f| a << [f, f.encoding] }
assert_equal([[d, euc_jp], ["#{d}/a", euc_jp], ["#{d}/b", euc_jp],
[d, win_31j], ["#{d}/a", win_31j], ["#{d}/b", win_31j],
[d, utf_8], ["#{d}/a", utf_8], ["#{d}/b", utf_8]],
a)
if /mswin|mingw/ =~ RUBY_PLATFORM
a = []
Dir.mkdir("#{d}/\u{2660}")
Find.find("#{d}".encode(utf_8)) {|f| a << [f, f.encoding] }
assert_equal([[d, utf_8], ["#{d}/a", utf_8], ["#{d}/b", utf_8], ["#{d}/\u{2660}", utf_8]], a)
end
}
end
class TestInclude < Test::Unit::TestCase
include Find
def test_functional_call
Dir.mktmpdir {|d|
File.open("#{d}/a", "w"){}
a = []
find(d) {|f| a << f }
assert_equal([d, "#{d}/a"], a)
}
end
end
end