зеркало из https://github.com/github/ruby.git
635 строки
15 KiB
Ruby
635 строки
15 KiB
Ruby
# frozen_string_literal: true
|
|
require 'test/unit'
|
|
require 'tmpdir'
|
|
|
|
begin
|
|
require 'dbm'
|
|
rescue LoadError
|
|
end
|
|
|
|
if defined? DBM
|
|
require 'tmpdir'
|
|
require 'fileutils'
|
|
|
|
class TestDBM_RDONLY < Test::Unit::TestCase
|
|
def TestDBM_RDONLY.uname_s
|
|
require 'rbconfig'
|
|
case RbConfig::CONFIG['target_os']
|
|
when 'cygwin'
|
|
require 'etc'
|
|
Etc.uname[:sysname]
|
|
else
|
|
RbConfig::CONFIG['target_os']
|
|
end
|
|
end
|
|
SYSTEM = uname_s
|
|
|
|
def setup
|
|
@tmpdir = Dir.mktmpdir("tmptest_dbm")
|
|
@prefix = "tmptest_dbm_#{$$}"
|
|
@path = "#{@tmpdir}/#{@prefix}_"
|
|
|
|
# prepare to make readonly DBM file
|
|
DBM.open("#{@tmpdir}/#{@prefix}_rdonly") {|dbm|
|
|
dbm['foo'] = 'FOO'
|
|
}
|
|
|
|
File.chmod(0400, *Dir.glob("#{@tmpdir}/#{@prefix}_rdonly.*"))
|
|
|
|
assert_instance_of(DBM, @dbm_rdonly = DBM.new("#{@tmpdir}/#{@prefix}_rdonly", nil))
|
|
end
|
|
def teardown
|
|
assert_nil(@dbm_rdonly.close)
|
|
ObjectSpace.each_object(DBM) do |obj|
|
|
obj.close unless obj.closed?
|
|
end
|
|
FileUtils.remove_entry_secure @tmpdir
|
|
end
|
|
|
|
def test_delete_rdonly
|
|
skip("skipped because root can read anything") if Process.uid == 0
|
|
|
|
if /^CYGWIN_9/ !~ SYSTEM
|
|
assert_raise(DBMError) {
|
|
@dbm_rdonly.delete("foo")
|
|
}
|
|
|
|
assert_nil(@dbm_rdonly.delete("bar"))
|
|
end
|
|
end
|
|
|
|
def test_fetch_not_found
|
|
notfound = nil
|
|
result = Object.new
|
|
assert_same(result, @dbm_rdonly.fetch("bar") {|k| notfound = k; result})
|
|
assert_equal("bar", notfound)
|
|
assert_predicate(notfound, :tainted?)
|
|
end
|
|
end
|
|
|
|
class TestDBM < Test::Unit::TestCase
|
|
def setup
|
|
@tmpdir = Dir.mktmpdir("tmptest_dbm")
|
|
@prefix = "tmptest_dbm_#{$$}"
|
|
@path = "#{@tmpdir}/#{@prefix}_"
|
|
assert_instance_of(DBM, @dbm = DBM.new(@path))
|
|
end
|
|
def teardown
|
|
assert_nil(@dbm.close) unless @dbm.closed?
|
|
ObjectSpace.each_object(DBM) do |obj|
|
|
obj.close unless obj.closed?
|
|
end
|
|
FileUtils.remove_entry_secure @tmpdir
|
|
end
|
|
|
|
def check_size(expect, dbm=@dbm)
|
|
assert_equal(expect, dbm.size)
|
|
n = 0
|
|
dbm.each { n+=1 }
|
|
assert_equal(expect, n)
|
|
if expect == 0
|
|
assert_equal(true, dbm.empty?)
|
|
else
|
|
assert_equal(false, dbm.empty?)
|
|
end
|
|
end
|
|
|
|
def test_dbmfile_suffix
|
|
@dbm.close
|
|
prefix = File.basename(@path)
|
|
suffixes = Dir.entries(@tmpdir).grep(/\A#{Regexp.escape prefix}/) { $' }.sort
|
|
pagname = "#{@path}.pag"
|
|
dirname = "#{@path}.dir"
|
|
dbname = "#{@path}.db"
|
|
case DBM::VERSION
|
|
when /\bNDBM\b/
|
|
assert_equal(%w[.dir .pag], suffixes)
|
|
assert(File.zero?(pagname))
|
|
assert(File.zero?(dirname))
|
|
when /\bGDBM\b/
|
|
assert_equal(%w[.dir .pag], suffixes)
|
|
assert(!File.zero?(pagname))
|
|
assert(!File.zero?(dirname))
|
|
pag = File.binread(pagname, 16)
|
|
pag_magics = [
|
|
0x13579ace, # GDBM_OMAGIC
|
|
0x13579acd, # GDBM_MAGIC32
|
|
0x13579acf, # GDBM_MAGIC64
|
|
]
|
|
assert_operator(pag_magics, :include?,
|
|
pag.unpack("i")[0]) # native endian, native int.
|
|
if !File.identical?(pagname, dirname)
|
|
dir = File.binread(dirname, 16)
|
|
assert_equal("GDBM", dir[0, 4])
|
|
end
|
|
when /\bBerkeley DB\b/
|
|
assert_equal(%w[.db], suffixes)
|
|
assert(!File.zero?(dbname))
|
|
db = File.binread(dbname, 16)
|
|
assert(db[0,4].unpack("N") == [0x00061561] || # Berkeley DB 1
|
|
db[12,4].unpack("L") == [0x00061561]) # Berkeley DBM 2 or later.
|
|
when /\bQDBM\b/
|
|
assert_equal(%w[.dir .pag], suffixes)
|
|
assert(!File.zero?(pagname))
|
|
assert(!File.zero?(dirname))
|
|
dir = File.binread(dirname, 16)
|
|
assert_equal("[depot]\0\v", dir[0, 9])
|
|
pag = File.binread(pagname, 16)
|
|
if [1].pack("s") == "\x00\x01" # big endian
|
|
assert_equal("[DEPOT]\n\f", pag[0, 9])
|
|
else # little endian
|
|
assert_equal("[depot]\n\f", pag[0, 9])
|
|
end
|
|
end
|
|
if suffixes == %w[.db]
|
|
assert_match(/\bBerkeley DB\b/, DBM::VERSION)
|
|
end
|
|
end
|
|
|
|
def test_s_new_has_no_block
|
|
# DBM.new ignore the block
|
|
foo = true
|
|
assert_instance_of(DBM, dbm = DBM.new("#{@tmpdir}/#{@prefix}") { foo = false })
|
|
assert_equal(foo, true)
|
|
assert_nil(dbm.close)
|
|
end
|
|
|
|
def test_s_open_no_create
|
|
skip "dbm_open() is broken on libgdbm 1.8.0 or prior (#{DBM::VERSION})" if /GDBM version 1\.(?:[0-7]\b|8\.0)/ =~ DBM::VERSION
|
|
assert_nil(dbm = DBM.open("#{@tmpdir}/#{@prefix}", nil))
|
|
ensure
|
|
dbm.close if dbm
|
|
end
|
|
|
|
def test_s_open_with_block
|
|
assert_equal(DBM.open("#{@tmpdir}/#{@prefix}") { :foo }, :foo)
|
|
end
|
|
|
|
def test_close
|
|
assert_instance_of(DBM, dbm = DBM.open("#{@tmpdir}/#{@prefix}"))
|
|
assert_nil(dbm.close)
|
|
|
|
# closed DBM file
|
|
assert_raise(DBMError) { dbm.close }
|
|
end
|
|
|
|
def test_aref
|
|
assert_equal('bar', @dbm['foo'] = 'bar')
|
|
assert_equal('bar', @dbm['foo'])
|
|
|
|
assert_nil(@dbm['bar'])
|
|
end
|
|
|
|
def test_fetch
|
|
assert_equal('bar', @dbm['foo']='bar')
|
|
assert_equal('bar', @dbm.fetch('foo'))
|
|
|
|
# key not found
|
|
assert_raise(IndexError) {
|
|
@dbm.fetch('bar')
|
|
}
|
|
|
|
# test for `ifnone' arg
|
|
assert_equal('baz', @dbm.fetch('bar', 'baz'))
|
|
|
|
# test for `ifnone' block
|
|
assert_equal('foobar', @dbm.fetch('bar') {|key| 'foo' + key })
|
|
end
|
|
|
|
def test_aset
|
|
num = 0
|
|
2.times {|i|
|
|
assert_equal('foo', @dbm['foo'] = 'foo')
|
|
assert_equal('foo', @dbm['foo'])
|
|
assert_equal('bar', @dbm['foo'] = 'bar')
|
|
assert_equal('bar', @dbm['foo'])
|
|
|
|
num += 1 if i == 0
|
|
assert_equal(num, @dbm.size)
|
|
|
|
# assign nil
|
|
assert_equal('', @dbm['bar'] = '')
|
|
assert_equal('', @dbm['bar'])
|
|
|
|
num += 1 if i == 0
|
|
assert_equal(num, @dbm.size)
|
|
|
|
# empty string
|
|
assert_equal('', @dbm[''] = '')
|
|
assert_equal('', @dbm[''])
|
|
|
|
num += 1 if i == 0
|
|
assert_equal(num, @dbm.size)
|
|
|
|
# Integer
|
|
assert_equal('200', @dbm['100'] = '200')
|
|
assert_equal('200', @dbm['100'])
|
|
|
|
num += 1 if i == 0
|
|
assert_equal(num, @dbm.size)
|
|
|
|
# Big key and value
|
|
assert_equal('y' * 100, @dbm['x' * 100] = 'y' * 100)
|
|
assert_equal('y' * 100, @dbm['x' * 100])
|
|
|
|
num += 1 if i == 0
|
|
assert_equal(num, @dbm.size)
|
|
}
|
|
end
|
|
|
|
def test_key
|
|
assert_equal('bar', @dbm['foo'] = 'bar')
|
|
assert_equal('foo', @dbm.key('bar'))
|
|
assert_nil(@dbm['bar'])
|
|
end
|
|
|
|
def test_values_at
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
assert_equal(values.reverse, @dbm.values_at(*keys.reverse))
|
|
end
|
|
|
|
def test_select_with_block
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
ret = @dbm.select {|k,v|
|
|
assert_equal(k.upcase, v)
|
|
k != "bar"
|
|
}
|
|
assert_equal([['baz', 'BAZ'], ['foo', 'FOO']],
|
|
ret.sort)
|
|
end
|
|
|
|
def test_length
|
|
num = 10
|
|
assert_equal(0, @dbm.size)
|
|
num.times {|i|
|
|
i = i.to_s
|
|
@dbm[i] = i
|
|
}
|
|
assert_equal(num, @dbm.size)
|
|
|
|
@dbm.shift
|
|
|
|
assert_equal(num - 1, @dbm.size)
|
|
end
|
|
|
|
def test_empty?
|
|
assert_equal(true, @dbm.empty?)
|
|
@dbm['foo'] = 'FOO'
|
|
assert_equal(false, @dbm.empty?)
|
|
end
|
|
|
|
def test_each_pair
|
|
n = 0
|
|
@dbm.each_pair { n += 1 }
|
|
assert_equal(0, n)
|
|
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
n = 0
|
|
ret = @dbm.each_pair {|key, val|
|
|
assert_not_nil(i = keys.index(key))
|
|
assert_equal(val, values[i])
|
|
|
|
n += 1
|
|
}
|
|
assert_equal(keys.size, n)
|
|
assert_equal(@dbm, ret)
|
|
end
|
|
|
|
def test_each_value
|
|
n = 0
|
|
@dbm.each_value { n += 1 }
|
|
assert_equal(0, n)
|
|
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
n = 0
|
|
ret = @dbm.each_value {|val|
|
|
assert_not_nil(key = @dbm.key(val))
|
|
assert_not_nil(i = keys.index(key))
|
|
assert_equal(val, values[i])
|
|
|
|
n += 1
|
|
}
|
|
assert_equal(keys.size, n)
|
|
assert_equal(@dbm, ret)
|
|
end
|
|
|
|
def test_each_key
|
|
n = 0
|
|
@dbm.each_key { n += 1 }
|
|
assert_equal(0, n)
|
|
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
n = 0
|
|
ret = @dbm.each_key {|key|
|
|
assert_not_nil(i = keys.index(key))
|
|
assert_equal(@dbm[key], values[i])
|
|
|
|
n += 1
|
|
}
|
|
assert_equal(keys.size, n)
|
|
assert_equal(@dbm, ret)
|
|
end
|
|
|
|
def test_keys
|
|
assert_equal([], @dbm.keys)
|
|
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
assert_equal(keys.sort, @dbm.keys.sort)
|
|
assert_equal(values.sort, @dbm.values.sort)
|
|
end
|
|
|
|
def test_values
|
|
test_keys
|
|
end
|
|
|
|
def test_shift
|
|
assert_nil(@dbm.shift)
|
|
assert_equal(0, @dbm.size)
|
|
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
ret_keys = []
|
|
ret_values = []
|
|
while ret = @dbm.shift
|
|
ret_keys.push ret[0]
|
|
ret_values.push ret[1]
|
|
|
|
assert_equal(keys.size - ret_keys.size, @dbm.size)
|
|
end
|
|
|
|
assert_equal(keys.sort, ret_keys.sort)
|
|
assert_equal(values.sort, ret_values.sort)
|
|
end
|
|
|
|
def test_delete
|
|
keys = %w(foo bar baz)
|
|
values = %w(FOO BAR BAZ)
|
|
key = keys[1]
|
|
|
|
assert_nil(@dbm.delete(key))
|
|
assert_equal(0, @dbm.size)
|
|
|
|
@dbm[keys[0]], @dbm[keys[1]], @dbm[keys[2]] = values
|
|
|
|
assert_equal('BAR', @dbm.delete(key))
|
|
assert_nil(@dbm[key])
|
|
assert_equal(2, @dbm.size)
|
|
|
|
assert_nil(@dbm.delete(key))
|
|
end
|
|
|
|
def test_delete_with_block
|
|
key = 'no called block'
|
|
@dbm[key] = 'foo'
|
|
assert_equal('foo', @dbm.delete(key) {|k| k.replace 'called block'; :blockval})
|
|
assert_equal(0, @dbm.size)
|
|
|
|
key = 'no called block'.dup
|
|
assert_equal(:blockval, @dbm.delete(key) {|k| k.replace 'called block'; :blockval})
|
|
assert_equal(0, @dbm.size)
|
|
end
|
|
|
|
def test_delete_if
|
|
v = "0"
|
|
100.times {@dbm[v] = v; v = v.next}
|
|
|
|
ret = @dbm.delete_if {|key, val| key.to_i < 50}
|
|
assert_equal(@dbm, ret)
|
|
check_size(50, @dbm)
|
|
|
|
ret = @dbm.delete_if {|key, val| key.to_i >= 50}
|
|
assert_equal(@dbm, ret)
|
|
check_size(0, @dbm)
|
|
|
|
# break
|
|
v = "0"
|
|
100.times {@dbm[v] = v; v = v.next}
|
|
check_size(100, @dbm)
|
|
n = 0;
|
|
@dbm.delete_if {|key, val|
|
|
break if n > 50
|
|
n+=1
|
|
true
|
|
}
|
|
assert_equal(51, n)
|
|
check_size(49, @dbm)
|
|
|
|
@dbm.clear
|
|
|
|
# raise
|
|
v = "0"
|
|
100.times {@dbm[v] = v; v = v.next}
|
|
check_size(100, @dbm)
|
|
n = 0;
|
|
begin
|
|
@dbm.delete_if {|key, val|
|
|
raise "runtime error" if n > 50
|
|
n+=1
|
|
true
|
|
}
|
|
rescue RuntimeError
|
|
end
|
|
assert_equal(51, n)
|
|
check_size(49, @dbm)
|
|
end
|
|
|
|
def test_reject
|
|
v = "0"
|
|
100.times {@dbm[v] = v; v = v.next}
|
|
|
|
hash = @dbm.reject {|key, val| key.to_i < 50}
|
|
assert_instance_of(Hash, hash)
|
|
assert_equal(100, @dbm.size)
|
|
|
|
assert_equal(50, hash.size)
|
|
hash.each_pair {|key,val|
|
|
assert_equal(false, key.to_i < 50)
|
|
assert_equal(key, val)
|
|
}
|
|
|
|
hash = @dbm.reject {|key, val| key.to_i < 100}
|
|
assert_instance_of(Hash, hash)
|
|
assert_equal(true, hash.empty?)
|
|
end
|
|
|
|
def test_clear
|
|
v = "1"
|
|
100.times {v = v.next; @dbm[v] = v}
|
|
|
|
assert_equal(@dbm, @dbm.clear)
|
|
|
|
# validate DBM#size
|
|
i = 0
|
|
@dbm.each { i += 1 }
|
|
assert_equal(@dbm.size, i)
|
|
assert_equal(0, i)
|
|
end
|
|
|
|
def test_invert
|
|
v = "0"
|
|
100.times {@dbm[v] = v; v = v.next}
|
|
|
|
hash = @dbm.invert
|
|
assert_instance_of(Hash, hash)
|
|
assert_equal(100, hash.size)
|
|
hash.each_pair {|key, val|
|
|
assert_equal(key.to_i, val.to_i)
|
|
}
|
|
end
|
|
|
|
def test_update
|
|
hash = {}
|
|
v = "0"
|
|
100.times {v = v.next; hash[v] = v}
|
|
|
|
@dbm["101"] = "101"
|
|
@dbm.update hash
|
|
assert_equal(101, @dbm.size)
|
|
@dbm.each_pair {|key, val|
|
|
assert_equal(key.to_i, val.to_i)
|
|
}
|
|
end
|
|
|
|
def test_replace
|
|
hash = {}
|
|
v = "0"
|
|
100.times {v = v.next; hash[v] = v}
|
|
|
|
@dbm["101"] = "101"
|
|
@dbm.replace hash
|
|
assert_equal(100, @dbm.size)
|
|
@dbm.each_pair {|key, val|
|
|
assert_equal(key.to_i, val.to_i)
|
|
}
|
|
end
|
|
|
|
def test_haskey?
|
|
assert_equal('bar', @dbm['foo']='bar')
|
|
assert_equal(true, @dbm.has_key?('foo'))
|
|
assert_equal(false, @dbm.has_key?('bar'))
|
|
end
|
|
|
|
def test_has_value?
|
|
assert_equal('bar', @dbm['foo']='bar')
|
|
assert_equal(true, @dbm.has_value?('bar'))
|
|
assert_equal(false, @dbm.has_value?('foo'))
|
|
end
|
|
|
|
def test_to_a
|
|
v = "0"
|
|
100.times {v = v.next; @dbm[v] = v}
|
|
|
|
ary = @dbm.to_a
|
|
assert_instance_of(Array, ary)
|
|
assert_equal(100, ary.size)
|
|
ary.each {|key,val|
|
|
assert_equal(key.to_i, val.to_i)
|
|
}
|
|
end
|
|
|
|
def test_to_hash
|
|
v = "0"
|
|
100.times {v = v.next; @dbm[v] = v}
|
|
|
|
hash = @dbm.to_hash
|
|
assert_instance_of(Hash, hash)
|
|
assert_equal(100, hash.size)
|
|
hash.each {|key,val|
|
|
assert_equal(key.to_i, val.to_i)
|
|
}
|
|
end
|
|
end
|
|
|
|
class TestDBM2 < Test::Unit::TestCase
|
|
def setup
|
|
@tmproot = Dir.mktmpdir('ruby-dbm')
|
|
end
|
|
|
|
def teardown
|
|
FileUtils.remove_entry_secure @tmproot if File.directory?(@tmproot)
|
|
end
|
|
|
|
def test_version
|
|
assert_instance_of(String, DBM::VERSION)
|
|
end
|
|
|
|
def test_reader_open_notexist
|
|
assert_raise(Errno::ENOENT) {
|
|
DBM.open("#{@tmproot}/a", 0666, DBM::READER)
|
|
}
|
|
end
|
|
|
|
def test_writer_open_notexist
|
|
skip "dbm_open() is broken on libgdbm 1.8.0 or prior (#{DBM::VERSION})" if /GDBM version 1\.(?:[0-7]\b|8\.0)/ =~ DBM::VERSION
|
|
assert_raise(Errno::ENOENT) {
|
|
DBM.open("#{@tmproot}/a", 0666, DBM::WRITER)
|
|
}
|
|
end
|
|
|
|
def test_wrcreat_open_notexist
|
|
v = DBM.open("#{@tmproot}/a", 0666, DBM::WRCREAT)
|
|
assert_instance_of(DBM, v)
|
|
v.close
|
|
end
|
|
|
|
def test_newdb_open_notexist
|
|
v = DBM.open("#{@tmproot}/a", 0666, DBM::NEWDB)
|
|
assert_instance_of(DBM, v)
|
|
v.close
|
|
end
|
|
|
|
def test_reader_open
|
|
DBM.open("#{@tmproot}/a") {} # create a db.
|
|
v = DBM.open("#{@tmproot}/a", nil, DBM::READER) {|d|
|
|
# Errno::EPERM is raised on Solaris which use ndbm.
|
|
# DBMError is raised on Debian which use gdbm.
|
|
assert_raise(Errno::EPERM, DBMError) { d["k"] = "v" }
|
|
true
|
|
}
|
|
assert(v)
|
|
end
|
|
|
|
def test_newdb_open
|
|
DBM.open("#{@tmproot}/a") {|dbm|
|
|
dbm["k"] = "v"
|
|
}
|
|
v = DBM.open("#{@tmproot}/a", nil, DBM::NEWDB) {|d|
|
|
assert_equal(0, d.length)
|
|
assert_nil(d["k"])
|
|
true
|
|
}
|
|
assert(v)
|
|
end
|
|
|
|
def test_freeze
|
|
DBM.open("#{@tmproot}/a") {|d|
|
|
d.freeze
|
|
assert_raise(FrozenError) { d["k"] = "v" }
|
|
}
|
|
end
|
|
end
|
|
end
|