require 'runit/testcase'
require 'runit/cui/testrunner'
if $".grep(/\bsdbm.so\b/).empty?
begin
require './sdbm'
rescue LoadError
require 'sdbm'
end
end
def uname_s
require 'rbconfig'
case Config::CONFIG['host_os']
when 'cygwin'
require 'Win32API'
uname = Win32API.new('cygwin1', 'uname', 'P', 'I')
utsname = ' ' * 100
raise 'cannot get system name' if uname.call(utsname) == -1
utsname.unpack('A20' * 5)[0]
else
Config::CONFIG['host_os']
end
end
SYSTEM = uname_s
class TestSDBM < RUNIT::TestCase
def setup
@path = "tmptest_sdbm_"
assert_instance_of(SDBM, @sdbm = SDBM.new(@path))
end
def teardown
assert_nil(@sdbm.close)
GC.start
File.delete *Dir.glob("tmptest_sdbm*").to_a
p Dir.glob("tmptest_sdbm*") if $DEBUG
end
def check_size(expect, sdbm=@sdbm)
assert_equals(expect, sdbm.size)
n = 0
sdbm.each { n+=1 }
assert_equals(expect, n)
if expect == 0
assert_equals(true, sdbm.empty?)
else
assert_equals(false, sdbm.empty?)
end
end
def test_version
assert(! SDBM.const_defined?(:VERSION))
end
def test_s_new_has_no_block
# SDBM.new ignore the block
foo = true
assert_instance_of(SDBM, sdbm = SDBM.new("tmptest_sdbm") { foo = false })
assert_equals(foo, true)
assert_nil(sdbm.close)
end
def test_s_open_no_create
assert_nil(sdbm = SDBM.open("tmptest_sdbm", nil))
ensure
sdbm.close if sdbm
end
def test_s_open_with_block
assert_equals(SDBM.open("tmptest_sdbm") { :foo }, :foo)
end
=begin
# Is it guaranteed on many OS?
def test_s_open_lock_one_process
# locking on one process
assert_instance_of(SDBM, sdbm = SDBM.open("tmptest_sdbm", 0644))
assert_exception(Errno::EWOULDBLOCK) {
begin
SDBM.open("tmptest_sdbm", 0644)
rescue Errno::EAGAIN
raise Errno::EWOULDBLOCK
end
}
end
=end
def test_s_open_nolock
# sdbm 1.8.0 specific
if not defined? SDBM::NOLOCK
return
end
fork() {
assert_instance_of(SDBM, sdbm = SDBM.open("tmptest_sdbm", 0644,
SDBM::NOLOCK))
sleep 2
}
sleep 1
begin
sdbm2 = nil
assert_no_exception(Errno::EWOULDBLOCK, Errno::EAGAIN, Errno::EACCES) {
assert_instance_of(SDBM, sdbm2 = SDBM.open("tmptest_sdbm", 0644))
}
ensure
Process.wait
sdbm2.close if sdbm2
end
p Dir.glob("tmptest_sdbm*") if $DEBUG
fork() {
assert_instance_of(SDBM, sdbm = SDBM.open("tmptest_sdbm", 0644))
sleep 2
}
begin
sleep 1
sdbm2 = nil
assert_no_exception(Errno::EWOULDBLOCK, Errno::EAGAIN, Errno::EACCES) {
# this test is failed on Cygwin98 (???)
assert_instance_of(SDBM, sdbm2 = SDBM.open("tmptest_sdbm", 0644,
SDBM::NOLOCK))
}
ensure
Process.wait
sdbm2.close if sdbm2
end
end
def test_s_open_error
assert_instance_of(SDBM, sdbm = SDBM.open("tmptest_sdbm", 0))
assert_exception(Errno::EACCES) {
SDBM.open("tmptest_sdbm", 0)
}
sdbm.close
end
def test_close
assert_instance_of(SDBM, sdbm = SDBM.open("tmptest_sdbm"))
assert_nil(sdbm.close)
# closed SDBM file
assert_exception(SDBMError) { sdbm.close }
end
def test_aref
assert_equals('bar', @sdbm['foo'] = 'bar')
assert_equals('bar', @sdbm['foo'])
assert_nil(@sdbm['bar'])
end
def test_fetch
assert_equals('bar', @sdbm['foo']='bar')
assert_equals('bar', @sdbm.fetch('foo'))
# key not found
assert_exception(IndexError) {
@sdbm.fetch('bar')
}
# test for `ifnone' arg
assert_equals('baz', @sdbm.fetch('bar', 'baz'))
# test for `ifnone' block
assert_equals('foobar', @sdbm.fetch('bar') {|key| 'foo' + key })
end
def test_aset
num = 0
2.times {|i|
assert_equals('foo', @sdbm['foo'] = 'foo')
assert_equals('foo', @sdbm['foo'])
assert_equals('bar', @sdbm['foo'] = 'bar')
assert_equals('bar', @sdbm['foo'])
num += 1 if i == 0
assert_equals(num, @sdbm.size)
# assign nil
assert_equals('', @sdbm['bar'] = '')
assert_equals('', @sdbm['bar'])
num += 1 if i == 0
assert_equals(num, @sdbm.size)
# empty string
assert_equals('', @sdbm[''] = '')
assert_equals('', @sdbm[''])
num += 1 if i == 0
assert_equals(num, @sdbm.size)
# Fixnum
assert_equals('200', @sdbm['100'] = '200')
assert_equals('200', @sdbm['100'])
num += 1 if i == 0
assert_equals(num, @sdbm.size)
# Big key and value
assert_equals('y' * 100, @sdbm['x' * 100] = 'y' * 100)
assert_equals('y' * 100, @sdbm['x' * 100])
num += 1 if i == 0
assert_equals(num, @sdbm.size)
}
end
def test_index
assert_equals('bar', @sdbm['foo'] = 'bar')
assert_equals('foo', @sdbm.index('bar'))
assert_nil(@sdbm['bar'])
end
def test_indexes
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
assert_equals(values.reverse, @sdbm.indexes(*keys.reverse))
end
def test_values_at
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
assert_equals(values.reverse, @sdbm.values_at(*keys.reverse))
end
def test_select_with_block
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
ret = @sdbm.select {|k,v|
assert_equals(k.upcase, v)
k != "bar"
}
assert_equals([['baz', 'BAZ'], ['foo', 'FOO']],
ret.sort)
end
def test_length
num = 10
assert_equals(0, @sdbm.size)
num.times {|i|
i = i.to_s
@sdbm[i] = i
}
assert_equals(num, @sdbm.size)
@sdbm.shift
assert_equals(num - 1, @sdbm.size)
end
def test_empty?
assert_equals(true, @sdbm.empty?)
@sdbm['foo'] = 'FOO'
assert_equals(false, @sdbm.empty?)
end
def test_each_pair
n = 0
@sdbm.each_pair { n += 1 }
assert_equals(0, n)
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
n = 0
ret = @sdbm.each_pair {|key, val|
assert_not_nil(i = keys.index(key))
assert_equals(val, values[i])
n += 1
}
assert_equals(keys.size, n)
assert_equals(@sdbm, ret)
end
def test_each_value
n = 0
@sdbm.each_value { n += 1 }
assert_equals(0, n)
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
n = 0
ret = @sdbm.each_value {|val|
assert_not_nil(key = @sdbm.index(val))
assert_not_nil(i = keys.index(key))
assert_equals(val, values[i])
n += 1
}
assert_equals(keys.size, n)
assert_equals(@sdbm, ret)
end
def test_each_key
n = 0
@sdbm.each_key { n += 1 }
assert_equals(0, n)
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
n = 0
ret = @sdbm.each_key {|key|
assert_not_nil(i = keys.index(key))
assert_equals(@sdbm[key], values[i])
n += 1
}
assert_equals(keys.size, n)
assert_equals(@sdbm, ret)
end
def test_keys
assert_equals([], @sdbm.keys)
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
assert_equals(keys.sort, @sdbm.keys.sort)
assert_equals(values.sort, @sdbm.values.sort)
end
def test_values
test_keys
end
def test_shift
assert_nil(@sdbm.shift)
assert_equals(0, @sdbm.size)
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
ret_keys = []
ret_values = []
while ret = @sdbm.shift
ret_keys.push ret[0]
ret_values.push ret[1]
assert_equals(keys.size - ret_keys.size, @sdbm.size)
end
assert_equals(keys.sort, ret_keys.sort)
assert_equals(values.sort, ret_values.sort)
end
def test_delete
keys = %w(foo bar baz)
values = %w(FOO BAR BAZ)
key = keys[1]
assert_nil(@sdbm.delete(key))
assert_equals(0, @sdbm.size)
@sdbm[keys[0]], @sdbm[keys[1]], @sdbm[keys[2]] = values
assert_equals('BAR', @sdbm.delete(key))
assert_nil(@sdbm[key])
assert_equals(2, @sdbm.size)
assert_nil(@sdbm.delete(key))
end
def test_delete_with_block
key = 'no called block'
@sdbm[key] = 'foo'
assert_equals('foo', @sdbm.delete(key) {|k| k.replace 'called block'})
assert_equals('no called block', key)
assert_equals(0, @sdbm.size)
key = 'no called block'
assert_equals(:blockval,
@sdbm.delete(key) {|k| k.replace 'called block'; :blockval})
assert_equals('called block', key)
assert_equals(0, @sdbm.size)
end
def test_delete_if
v = "0"
100.times {@sdbm[v] = v; v = v.next}
ret = @sdbm.delete_if {|key, val| key.to_i < 50}
assert_equals(@sdbm, ret)
check_size(50, @sdbm)
ret = @sdbm.delete_if {|key, val| key.to_i >= 50}
assert_equals(@sdbm, ret)
check_size(0, @sdbm)
# break
v = "0"
100.times {@sdbm[v] = v; v = v.next}
check_size(100, @sdbm)
n = 0;
@sdbm.delete_if {|key, val|
break if n > 50
n+=1
true
}
assert_equals(51, n)
check_size(49, @sdbm)
@sdbm.clear
# raise
v = "0"
100.times {@sdbm[v] = v; v = v.next}
check_size(100, @sdbm)
n = 0;
begin
@sdbm.delete_if {|key, val|
raise "runtime error" if n > 50
n+=1
true
}
rescue
end
assert_equals(51, n)
check_size(49, @sdbm)
end
def test_reject
v = "0"
100.times {@sdbm[v] = v; v = v.next}
hash = @sdbm.reject {|key, val| key.to_i < 50}
assert_instance_of(Hash, hash)
assert_equals(100, @sdbm.size)
assert_equals(50, hash.size)
hash.each_pair {|key,val|
assert_equals(false, key.to_i < 50)
assert_equals(key, val)
}
hash = @sdbm.reject {|key, val| key.to_i < 100}
assert_instance_of(Hash, hash)
assert_equals(true, hash.empty?)
end
def test_clear
v = "1"
100.times {v = v.next; @sdbm[v] = v}
assert_equals(@sdbm, @sdbm.clear)
# validate SDBM#size
i = 0
@sdbm.each { i += 1 }
assert_equals(@sdbm.size, i)
assert_equals(0, i)
end
def test_invert
v = "0"
100.times {@sdbm[v] = v; v = v.next}
hash = @sdbm.invert
assert_instance_of(Hash, hash)
assert_equals(100, hash.size)
hash.each_pair {|key, val|
assert_equals(key.to_i, val.to_i)
}
end
def test_update
hash = {}
v = "0"
100.times {v = v.next; hash[v] = v}
@sdbm["101"] = "101"
@sdbm.update hash
assert_equals(101, @sdbm.size)
@sdbm.each_pair {|key, val|
assert_equals(key.to_i, val.to_i)
}
end
def test_replace
hash = {}
v = "0"
100.times {v = v.next; hash[v] = v}
@sdbm["101"] = "101"
@sdbm.replace hash
assert_equals(100, @sdbm.size)
@sdbm.each_pair {|key, val|
assert_equals(key.to_i, val.to_i)
}
end
def test_haskey?
assert_equals('bar', @sdbm['foo']='bar')
assert_equals(true, @sdbm.has_key?('foo'))
assert_equals(false, @sdbm.has_key?('bar'))
end
def test_has_value?
assert_equals('bar', @sdbm['foo']='bar')
assert_equals(true, @sdbm.has_value?('bar'))
assert_equals(false, @sdbm.has_value?('foo'))
end
def test_to_a
v = "0"
100.times {v = v.next; @sdbm[v] = v}
ary = @sdbm.to_a
assert_instance_of(Array, ary)
assert_equals(100, ary.size)
ary.each {|key,val|
assert_equals(key.to_i, val.to_i)
}
end
def test_to_hash
v = "0"
100.times {v = v.next; @sdbm[v] = v}
hash = @sdbm.to_hash
assert_instance_of(Hash, hash)
assert_equals(100, hash.size)
hash.each {|key,val|
assert_equals(key.to_i, val.to_i)
}
end
end
if $0 == __FILE__
if ARGV.size == 0
suite = RUNIT::TestSuite.new
suite.add_test(TestSDBM.suite)
else
suite = RUNIT::TestSuite.new
ARGV.each do |testmethod|
suite.add_test(TestSDBM.new(testmethod))
end
end
RUNIT::CUI::TestRunner.run(suite)
end