diff options
Diffstat (limited to 'jni/ruby/test/dbm/test_dbm.rb')
-rw-r--r-- | jni/ruby/test/dbm/test_dbm.rb | 631 |
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 |