gdbm 0.9

data/test/test_gdbm-1.9.1.rb

@@ -0,0 +1,723 @@
+require 'test/unit'
+require 'tmpdir'
+
+begin
+  require 'gdbm'
+rescue LoadError
+end
+
+#This is needed because JRuby does not have it implemented yet,
+#but it's in MRI 1.9 and in the tests below (remove_entry_secure)
+class File::Stat
+	def world_writable?
+		(self.mode & 0002) != 0
+	end
+end
+
+if defined? GDBM
+  require 'tmpdir'
+  require 'fileutils'
+
+  class TestGDBM < Test::Unit::TestCase
+    def TestGDBM.uname_s
+      require 'rbconfig'
+      case RbConfig::CONFIG['target_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
+        RbConfig::CONFIG['target_os']
+      end
+    end
+    SYSTEM = uname_s
+
+    def setup
+      @tmpdir = Dir.mktmpdir("tmptest_gdbm")
+      @prefix = "tmptest_gdbm_#{$$}"
+      @path = "#{@tmpdir}/#{@prefix}_"
+      assert_instance_of(GDBM, @gdbm = GDBM.new(@path))
+
+      # prepare to make readonly GDBM file
+      GDBM.open("#{@tmpdir}/#{@prefix}_rdonly", 0400) {|gdbm|
+        gdbm['foo'] = 'FOO'
+      }
+      assert_instance_of(GDBM, @gdbm_rdonly = GDBM.new("#{@tmpdir}/#{@prefix}_rdonly", nil))
+    end
+    def teardown
+      assert_nil(@gdbm.close)
+      assert_nil(@gdbm_rdonly.close)
+      ObjectSpace.each_object(GDBM) do |obj|
+        obj.close unless obj.closed?
+      end
+      FileUtils.remove_entry_secure @tmpdir
+    end
+
+    def check_size(expect, gdbm=@gdbm)
+      assert_equal(expect, gdbm.size)
+      n = 0
+      gdbm.each { n+=1 }
+      assert_equal(expect, n)
+      if expect == 0
+        assert_equal(true, gdbm.empty?)
+      else
+        assert_equal(false, gdbm.empty?)
+      end
+    end
+
+    def have_fork?
+      begin
+        fork{}
+        true
+      rescue NotImplementedError
+        false
+      end
+    end
+
+    def test_s_new_has_no_block
+      # GDBM.new ignore the block
+      foo = true
+      assert_instance_of(GDBM, gdbm = GDBM.new("#{@tmpdir}/#{@prefix}") { foo = false })
+      assert_equal(foo, true)
+      assert_nil(gdbm.close)
+    end
+    def test_s_open_create_new
+      return if /^CYGWIN_9/ =~ SYSTEM
+
+      save_mask = File.umask(0)
+      begin
+        assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}"))
+        gdbm.close
+        assert_equal(File.stat("#{@tmpdir}/#{@prefix}").mode & 0777, 0666)
+        assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}2", 0644))
+        gdbm.close
+        assert_equal(File.stat("#{@tmpdir}/#{@prefix}2").mode & 0777, 0644)
+      ensure
+        File.umask save_mask
+      end
+    end
+    def test_s_open_no_create
+      assert_nil(gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", nil),
+                 "this test is failed on libgdbm 1.8.0")
+    ensure
+      gdbm.close if gdbm
+    end
+    def test_s_open_3rd_arg
+      assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0644,
+                                                GDBM::FAST))
+      gdbm.close
+
+      # gdbm 1.8.0 specific
+      if defined? GDBM::SYNC
+        assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0644,
+                                                  GDBM::SYNC))
+        gdbm.close
+      end
+      # gdbm 1.8.0 specific
+      if defined? GDBM::NOLOCK
+        assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0644,
+                                                  GDBM::NOLOCK))
+        gdbm.close
+      end
+    end
+    def test_s_open_with_block
+      assert_equal(GDBM.open("#{@tmpdir}/#{@prefix}") { :foo }, :foo)
+    end
+    def test_s_open_lock
+      return unless have_fork?	# snip this test
+      pid = fork() {
+        assert_instance_of(GDBM, gdbm  = GDBM.open("#{@tmpdir}/#{@prefix}", 0644))
+        sleep 2
+      }
+      begin
+        sleep 1
+        assert_raise(Errno::EWOULDBLOCK) {
+          begin
+            assert_instance_of(GDBM, gdbm2 = GDBM.open("#{@tmpdir}/#{@prefix}", 0644))
+          rescue Errno::EAGAIN, Errno::EACCES
+            raise Errno::EWOULDBLOCK
+          end
+        }
+      ensure
+        Process.wait pid
+      end
+    end
+
+=begin
+    # Is it guaranteed on many OS?
+    def test_s_open_lock_one_process
+      # locking on one process
+      assert_instance_of(GDBM, gdbm  = GDBM.open("#{@tmpdir}/#{@prefix}", 0644))
+      assert_raise(Errno::EWOULDBLOCK) {
+        begin
+          GDBM.open("#{@tmpdir}/#{@prefix}", 0644)
+        rescue Errno::EAGAIN
+          raise Errno::EWOULDBLOCK
+        end
+      }
+    end
+=end
+
+    def test_s_open_nolock
+      # gdbm 1.8.0 specific
+      if not defined? GDBM::NOLOCK
+        return
+      end
+      return unless have_fork?	# snip this test
+
+      pid = fork() {
+        assert_instance_of(GDBM, gdbm  = GDBM.open("#{@tmpdir}/#{@prefix}", 0644,
+                                                  GDBM::NOLOCK))
+        sleep 2
+      }
+      sleep 1
+      begin
+        gdbm2 = nil
+        assert_nothing_raised(Errno::EWOULDBLOCK, Errno::EAGAIN, Errno::EACCES) {
+          assert_instance_of(GDBM, gdbm2 = GDBM.open("#{@tmpdir}/#{@prefix}", 0644))
+        }
+      ensure
+        Process.wait pid
+        gdbm2.close if gdbm2
+      end
+
+      p Dir.glob("#{@tmpdir}/#{@prefix}*") if $DEBUG
+
+      pid = fork() {
+        assert_instance_of(GDBM, gdbm  = GDBM.open("#{@tmpdir}/#{@prefix}", 0644))
+        sleep 2
+      }
+      begin
+        sleep 1
+        gdbm2 = nil
+        assert_nothing_raised(Errno::EWOULDBLOCK, Errno::EAGAIN, Errno::EACCES) {
+          # this test is failed on Cygwin98 (???)
+          assert_instance_of(GDBM, gdbm2 = GDBM.open("#{@tmpdir}/#{@prefix}", 0644,
+                                                     GDBM::NOLOCK))
+        }
+      ensure
+        Process.wait pid
+        gdbm2.close if gdbm2
+      end
+    end
+
+    def test_s_open_error
+      assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0))
+      assert_raise(Errno::EACCES) {
+        GDBM.open("#{@tmpdir}/#{@prefix}", 0)
+      }
+      gdbm.close
+    end
+
+    def test_close
+      assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}"))
+      assert_nil(gdbm.close)
+
+      # closed GDBM file
+      assert_raise(RuntimeError) { gdbm.close }
+    end
+
+    def test_aref
+      assert_equal('bar', @gdbm['foo'] = 'bar')
+      assert_equal('bar', @gdbm['foo'])
+
+      assert_nil(@gdbm['bar'])
+    end
+
+    def test_fetch
+      assert_equal('bar', @gdbm['foo']='bar')
+      assert_equal('bar', @gdbm.fetch('foo'))
+
+      # key not found
+      assert_raise(IndexError) {
+        @gdbm.fetch('bar')
+      }
+
+      # test for `ifnone' arg
+      assert_equal('baz', @gdbm.fetch('bar', 'baz'))
+
+      # test for `ifnone' block
+      assert_equal('foobar', @gdbm.fetch('bar') {|key| 'foo' + key })
+    end
+
+    def test_aset
+      num = 0
+      2.times {|i|
+        assert_equal('foo', @gdbm['foo'] = 'foo')
+        assert_equal('foo', @gdbm['foo'])
+        assert_equal('bar', @gdbm['foo'] = 'bar')
+        assert_equal('bar', @gdbm['foo'])
+
+        num += 1 if i == 0
+        assert_equal(num, @gdbm.size)
+
+        # assign nil
+        assert_equal('', @gdbm['bar'] = '')
+        assert_equal('', @gdbm['bar'])
+
+        num += 1 if i == 0
+        assert_equal(num, @gdbm.size)
+
+        # empty string
+        assert_equal('', @gdbm[''] = '')
+        assert_equal('', @gdbm[''])
+
+        num += 1 if i == 0
+        assert_equal(num, @gdbm.size)
+
+        # Fixnum
+        assert_equal('200', @gdbm['100'] = '200')
+        assert_equal('200', @gdbm['100'])
+
+        num += 1 if i == 0
+        assert_equal(num, @gdbm.size)
+
+        # Big key and value
+        assert_equal('y' * 100, @gdbm['x' * 100] = 'y' * 100)
+        assert_equal('y' * 100, @gdbm['x' * 100])
+
+        num += 1 if i == 0
+        assert_equal(num, @gdbm.size)
+      }
+    end
+
+    def test_key
+      assert_equal('bar', @gdbm['foo'] = 'bar')
+      assert_equal('foo', @gdbm.key('bar'))
+      assert_nil(@gdbm['bar'])
+    end
+
+    def test_values_at
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+      assert_equal(values.reverse, @gdbm.values_at(*keys.reverse))
+    end
+
+    def test_select_with_block
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+      ret = @gdbm.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, @gdbm.size)
+      num.times {|i|
+        i = i.to_s
+        @gdbm[i] = i
+      }
+      assert_equal(num, @gdbm.size)
+
+      @gdbm.shift
+
+      assert_equal(num - 1, @gdbm.size)
+    end
+
+    def test_empty?
+      assert_equal(true, @gdbm.empty?)
+      @gdbm['foo'] = 'FOO'
+      assert_equal(false, @gdbm.empty?)
+    end
+
+    def test_each_pair
+      n = 0
+      @gdbm.each_pair { n += 1 }
+      assert_equal(0, n)
+
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      n = 0
+      ret = @gdbm.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(@gdbm, ret)
+    end
+
+    def test_each_value
+      n = 0
+      @gdbm.each_value { n += 1 }
+      assert_equal(0, n)
+
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      n = 0
+      ret = @gdbm.each_value {|val|
+        assert_not_nil(key = @gdbm.key(val))
+        assert_not_nil(i = keys.index(key))
+        assert_equal(val, values[i])
+
+        n += 1
+      }
+      assert_equal(keys.size, n)
+      assert_equal(@gdbm, ret)
+    end
+
+    def test_each_key
+      n = 0
+      @gdbm.each_key { n += 1 }
+      assert_equal(0, n)
+
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      n = 0
+      ret = @gdbm.each_key {|key|
+        assert_not_nil(i = keys.index(key))
+        assert_equal(@gdbm[key], values[i])
+
+        n += 1
+      }
+      assert_equal(keys.size, n)
+      assert_equal(@gdbm, ret)
+    end
+
+    def test_keys
+      assert_equal([], @gdbm.keys)
+
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      assert_equal(keys.sort, @gdbm.keys.sort)
+      assert_equal(values.sort, @gdbm.values.sort)
+    end
+
+    def test_values
+      test_keys
+    end
+
+    def test_shift
+      assert_nil(@gdbm.shift)
+      assert_equal(0, @gdbm.size)
+
+      keys = %w(foo bar baz)
+      values = %w(FOO BAR BAZ)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      ret_keys = []
+      ret_values = []
+      while ret = @gdbm.shift
+        ret_keys.push ret[0]
+        ret_values.push ret[1]
+
+        assert_equal(keys.size - ret_keys.size, @gdbm.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(@gdbm.delete(key))
+      assert_equal(0, @gdbm.size)
+
+      @gdbm[keys[0]], @gdbm[keys[1]], @gdbm[keys[2]] = values
+
+      assert_equal('BAR', @gdbm.delete(key))
+      assert_nil(@gdbm[key])
+      assert_equal(2, @gdbm.size)
+
+      assert_nil(@gdbm.delete(key))
+
+      if /^CYGWIN_9/ !~ SYSTEM
+        assert_raise(GDBMError) {
+          @gdbm_rdonly.delete("foo")
+        }
+
+        assert_nil(@gdbm_rdonly.delete("bar"))
+      end
+    end
+    def test_delete_with_block
+      key = 'no called block'
+      @gdbm[key] = 'foo'
+      assert_equal('foo', @gdbm.delete(key) {|k| k.replace 'called block'})
+      assert_equal('no called block', key)
+      assert_equal(0, @gdbm.size)
+
+      key = 'no called block'
+      assert_equal(:blockval,
+                    @gdbm.delete(key) {|k| k.replace 'called block'; :blockval})
+      assert_equal('called block', key)
+      assert_equal(0, @gdbm.size)
+    end
+
+    def test_delete_if
+      v = "0"
+      100.times {@gdbm[v] = v; v = v.next}
+
+      ret = @gdbm.delete_if {|key, val| key.to_i < 50}
+      assert_equal(@gdbm, ret)
+      check_size(50, @gdbm)
+
+      ret = @gdbm.delete_if {|key, val| key.to_i >= 50}
+      assert_equal(@gdbm, ret)
+      check_size(0, @gdbm)
+
+      # break
+      v = "0"
+      100.times {@gdbm[v] = v; v = v.next}
+      check_size(100, @gdbm)
+      n = 0;
+      @gdbm.delete_if {|key, val|
+        break if n > 50
+        n+=1
+        true
+      }
+      assert_equal(51, n)
+      check_size(49, @gdbm)
+
+      @gdbm.clear
+
+      # raise
+      v = "0"
+      100.times {@gdbm[v] = v; v = v.next}
+      check_size(100, @gdbm)
+      n = 0;
+      begin
+        @gdbm.delete_if {|key, val|
+          raise "runtime error" if n > 50
+          n+=1
+          true
+        }
+      rescue
+      end
+      assert_equal(51, n)
+      check_size(49, @gdbm)
+    end
+
+    def test_reject
+      v = "0"
+      100.times {@gdbm[v] = v; v = v.next}
+
+      hash = @gdbm.reject {|key, val| key.to_i < 50}
+      assert_instance_of(Hash, hash)
+      assert_equal(100, @gdbm.size)
+
+      assert_equal(50, hash.size)
+      hash.each_pair {|key,val|
+        assert_equal(false, key.to_i < 50)
+        assert_equal(key, val)
+      }
+
+      hash = @gdbm.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; @gdbm[v] = v}
+
+      assert_equal(@gdbm, @gdbm.clear)
+
+      # validate GDBM#size
+      i = 0
+      @gdbm.each { i += 1 }
+      assert_equal(@gdbm.size, i)
+      assert_equal(0, i)
+    end
+
+    def test_invert
+      v = "0"
+      100.times {@gdbm[v] = v; v = v.next}
+
+      hash = @gdbm.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}
+
+      @gdbm["101"] = "101"
+      @gdbm.update hash
+      assert_equal(101, @gdbm.size)
+      @gdbm.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}
+
+      @gdbm["101"] = "101"
+      @gdbm.replace hash
+      assert_equal(100, @gdbm.size)
+      @gdbm.each_pair {|key, val|
+        assert_equal(key.to_i, val.to_i)
+      }
+    end
+
+    def test_reorganize
+      size1 = File.size(@path)
+      i = "1"
+      1000.times {i = i.next; @gdbm[i] = i}
+      @gdbm.clear
+      @gdbm.sync
+
+      size2 = File.size(@path)
+      @gdbm.reorganize
+      size3 = File.size(@path)
+
+      # p [size1, size2, size3]
+      assert_equal(true, size1 < size2)
+      # this test is failed on Cygwin98. `GDBM version 1.8.0, as of May 19, 1999'
+      assert_equal(true, size3 < size2)
+      assert_equal(size1, size3)
+    end
+
+    def test_sync
+      assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0666, GDBM::FAST))
+      assert_equal(gdbm.sync, gdbm)
+      gdbm.close
+      assert_instance_of(GDBM, gdbm = GDBM.open("#{@tmpdir}/#{@prefix}", 0666))
+      assert_equal(gdbm.sync, gdbm)
+      gdbm.close
+    end
+
+    def test_cachesize=
+        assert_equal(@gdbm.cachesize = 1024, 1024)
+    end
+
+    def test_fastmode=
+        assert_equal(@gdbm.fastmode = true, true)
+    end
+
+    def test_syncmode=
+        assert_equal(@gdbm.syncmode = true, true)
+    end
+
+    def test_haskey?
+      assert_equal('bar', @gdbm['foo']='bar')
+      assert_equal(true,  @gdbm.has_key?('foo'))
+      assert_equal(false, @gdbm.has_key?('bar'))
+    end
+
+    def test_has_value?
+      assert_equal('bar', @gdbm['foo']='bar')
+      assert_equal(true,  @gdbm.has_value?('bar'))
+      assert_equal(false, @gdbm.has_value?('foo'))
+    end
+
+    def test_to_a
+      v = "0"
+      100.times {v = v.next; @gdbm[v] = v}
+
+      ary = @gdbm.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; @gdbm[v] = v}
+
+      hash = @gdbm.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 TestGDBM2 < Test::Unit::TestCase
+    def setup
+      @tmproot = Dir.mktmpdir('ruby-gdbm')
+    end
+
+    def teardown
+      FileUtils.remove_entry_secure @tmproot if File.directory?(@tmproot)
+    end
+
+    def test_reader_open_notexist
+      assert_raise(Errno::ENOENT) {
+        GDBM.open("#{@tmproot}/a", 0666, GDBM::READER)
+      }
+    end
+
+    def test_writer_open_notexist
+      assert_raise(Errno::ENOENT) {
+        GDBM.open("#{@tmproot}/a", 0666, GDBM::WRITER)
+      }
+    end
+
+    def test_wrcreat_open_notexist
+      v = GDBM.open("#{@tmproot}/a", 0666, GDBM::WRCREAT)
+      assert_instance_of(GDBM, v)
+      v.close
+    end
+
+    def test_newdb_open_notexist
+      v = GDBM.open("#{@tmproot}/a", 0666, GDBM::NEWDB)
+      assert_instance_of(GDBM, v)
+      v.close
+    end
+
+    def test_reader_open
+      GDBM.open("#{@tmproot}/a.dbm") {} # create a db.
+      v = GDBM.open("#{@tmproot}/a.dbm", nil, GDBM::READER) {|d|
+        assert_raise(GDBMError) { d["k"] = "v" }
+        true
+      }
+      assert(v)
+    end
+
+    def test_newdb_open
+      GDBM.open("#{@tmproot}/a.dbm") {|dbm|
+        dbm["k"] = "v"
+      } 
+      v = GDBM.open("#{@tmproot}/a.dbm", nil, GDBM::NEWDB) {|d|
+        assert_equal(0, d.length)
+        assert_nil(d["k"])
+        true
+      }
+      assert(v)
+    end
+
+    def test_freeze
+      GDBM.open("#{@tmproot}/a.dbm") {|d|
+        d.freeze
+        assert_raise(RuntimeError) { d["k"] = "v" }
+      }
+    end
+  end
+end