summaryrefslogtreecommitdiff
path: root/jni/ruby/test/dbm
diff options
context:
space:
mode:
Diffstat (limited to 'jni/ruby/test/dbm')
-rw-r--r--jni/ruby/test/dbm/test_dbm.rb631
1 files changed, 631 insertions, 0 deletions
diff --git a/jni/ruby/test/dbm/test_dbm.rb b/jni/ruby/test/dbm/test_dbm.rb
new file mode 100644
index 0000000..c783b49
--- /dev/null
+++ b/jni/ruby/test/dbm/test_dbm.rb
@@ -0,0 +1,631 @@
+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
+ 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)
+
+ # Fixnum
+ 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'
+ 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(RuntimeError) { d["k"] = "v" }
+ }
+ end
+ end
+end