bdb 0.1.0

data/test/env_test.rb

@@ -0,0 +1,101 @@
+require 'test_helper'
+
+class EnvTest < Test::Unit::TestCase
+    
+  def test_cachesize
+    env = Bdb::Env.new(0)
+    env.cachesize = 1024*1024*500
+    assert_equal 1024*1024*500, env.cachesize
+
+    env.cachesize = 1024*1024*1024*3
+    assert_equal 1024*1024*1024*3, env.cachesize
+
+    env.cachesize = 1024*1024*1024*3+1
+    assert_equal 1024*1024*1024*3+1, env.cachesize
+  end
+  
+  def test_flags
+    env = Bdb::Env.new(0)
+
+    env.flags_on = Bdb::DB_AUTO_COMMIT | Bdb::DB_DSYNC_DB
+    assert_equal Bdb::DB_AUTO_COMMIT | Bdb::DB_DSYNC_DB, env.flags    
+
+    env.flags_off = Bdb::DB_AUTO_COMMIT | Bdb::DB_DSYNC_DB
+    assert_equal 0, env.flags    
+  end
+    
+  def test_list_dbs
+    env = Bdb::Env.new(0)
+    assert env.list_dbs.empty?
+    
+    db = env.db
+    assert_equal db, env.list_dbs.first
+    db.close(0)
+  end
+    
+  def test_set_and_get_timeout
+    env = Bdb::Env.new(0)
+    env.set_timeout(10, Bdb::DB_SET_LOCK_TIMEOUT)
+    assert_equal 10, env.get_timeout(Bdb::DB_SET_LOCK_TIMEOUT)
+  end
+  
+  def test_set_and_get_tx_max
+    env = Bdb::Env.new(0)
+    env.set_tx_max(100)
+    assert_equal 100, env.get_tx_max
+  end
+  
+  def test_set_and_get_lk_detect
+    env = Bdb::Env.new(0)
+    env.set_lk_detect(Bdb::DB_LOCK_MAXWRITE)
+    assert_equal Bdb::DB_LOCK_MAXWRITE, env.get_lk_detect
+  end
+    
+  def test_set_and_get_lk_max_locks
+    env = Bdb::Env.new(0)
+    env.set_lk_max_locks(10_000)
+    assert_equal 10_000, env.get_lk_max_locks
+  end
+  
+  def test_set_and_get_lk_max_objects
+    env = Bdb::Env.new(0)
+    env.set_lk_max_objects(10_000)
+    assert_equal 10_000, env.get_lk_max_objects
+  end
+  
+  def test_set_and_get_shm_key
+    env = Bdb::Env.new(0)
+    env.set_shm_key(2506400)
+    assert_equal 2506400, env.get_shm_key
+  end
+    
+  def test_set_and_get_data_dir
+    env = Bdb::Env.new(0)
+    env.set_data_dir('/tmp')
+    assert_equal ['/tmp'], env.get_data_dirs
+  end
+  
+  def test_set_lg_dir
+    env = Bdb::Env.new(0)
+    env.set_lg_dir('/tmp')
+    assert_equal '/tmp', env.get_lg_dir
+  end
+  
+  def test_set_tmp_dir
+    env = Bdb::Env.new(0)
+    env.set_tmp_dir('/tmp')
+    assert_equal '/tmp', env.get_tmp_dir    
+  end
+
+  def test_get_home
+    env = Bdb::Env.new(0)
+    tmp_dir = File.expand_path(File.join(File.dirname(__FILE__), 'tmp'))
+    mkdir_p tmp_dir
+    env.open(tmp_dir, Bdb::DB_CREATE, 0);
+    assert_equal tmp_dir, env.get_home
+    env.close
+  ensure
+    rm_rf tmp_dir
+  end
+  
+end

data/test/simple_test.rb

@@ -0,0 +1,93 @@
+require 'test_helper'
+require File.dirname(__FILE__) + '/../lib/bdb/simple'
+
+class SimpleTest < Test::Unit::TestCase
+  def setup
+    @path = File.join(File.dirname(__FILE__), 'tmp')
+    rm_rf @path
+    mkdir @path
+    open
+  end
+  
+  def teardown
+    close
+    rm_rf @path
+  end
+  
+  def open
+    @db  = Bdb::Simple.new(@path)
+    @dbd = Bdb::Simple.new(@path, :name => 'dup', :dup => true)
+  end
+
+  def close
+    @db.close
+    @dbd.close
+  end
+
+  def test_put_and_get
+    @db['key'] = 'data'
+    assert_equal 'data', @db['key']
+
+    @dbd['key'] = 'data1'
+    @dbd['key'] = 'data2'
+    assert_equal ['data1', 'data2'], @dbd['key'].to_a
+  end
+  
+  def test_update
+    @db[:key] = 0
+    close
+
+    pids = []
+    5.times do
+      pids << Process.fork do
+        db = Bdb::Simple.new(@path)
+        10.times do
+          db.update(:key) do |v|
+            sleep(0.1)
+            v + 1
+          end
+        end
+        db.close
+      end
+    end
+    pids.each {|pid| Process.wait(pid)}
+
+    open
+    assert_equal 50, @db[:key]
+  end
+  
+
+  def test_delete
+    @db['key'] = 'data'
+    assert_equal 'data', @db['key']
+
+    @db.delete('key')    
+    assert_nil @db['key']
+  end
+  
+  def test_range
+    (1..10).each {|i| @db[i] = "data#{i}"}
+    
+    assert_equal (3..7).collect {|i| "data#{i}"}, @db[3..7].to_a
+  end
+
+  def test_compare_absolute
+    list = [5, 6, "foo", :bar, "bar", :foo, [1,2,4], true, [1,2,3], false, [1], [2], nil, {}, {:b => 1, :a => 1}, {:b => 2, :a => 1}]
+
+    expected = [nil, false, true, 5, 6, :bar, :foo, "bar", "foo", [1], [1, 2, 3], [1, 2, 4], [2], {}, {:a=>1, :b=>1}, {:a=>1, :b=>2}]
+    assert_equal expected, list.sort {|a,b| Bdb::Simple.compare_absolute(a,b)}
+    100.times do
+      assert_equal expected, list.shuffle.sort {|a,b| Bdb::Simple.compare_absolute(a,b)}
+    end
+  end
+
+  def parallel(n)
+    threads = []
+    n.times do |i|
+      threads << Thread.new do
+        yield(i)
+      end
+    end
+    threads.each { |thread| thread.join }
+  end
+end

data/test/stat_test.rb

@@ -0,0 +1,22 @@
+require 'test_helper'
+
+class DbStat < Test::Unit::TestCase
+  
+  def setup
+    mkdir File.join(File.dirname(__FILE__), 'tmp')
+  end
+  
+  def teardown
+    rm_rf File.join(File.dirname(__FILE__), 'tmp')
+  end
+  
+  def test_stat
+    @db = Bdb::Db.new
+    @db.open(nil, File.join(File.dirname(__FILE__), 'tmp', 'test.db'), nil, Bdb::Db::BTREE, Bdb::DB_CREATE, 0)
+    @db.put(nil, 'key', 'data', 0)
+    stats = @db.stat(nil, 0)
+    assert_equal 1, stats['bt_nkeys']
+    @db.close(0)
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,7 @@
+require 'test/unit'
+require 'fileutils'
+require File.dirname(__FILE__) + '/../ext/bdb'
+
+class Test::Unit::TestCase
+  include FileUtils
+end

data/test/txn_test.rb

@@ -0,0 +1,74 @@
+require 'test_helper'
+
+class TxnTest < Test::Unit::TestCase
+
+  def setup
+    mkdir File.join(File.dirname(__FILE__), 'tmp')
+    @env = Bdb::Env.new(0)
+    env_flags = Bdb::DB_CREATE |    # Create the environment if it does not already exist.
+                 Bdb::DB_INIT_TXN  | # Initialize transactions
+                 Bdb::DB_INIT_LOCK | # Initialize locking.
+                 Bdb::DB_INIT_LOG  | # Initialize logging
+                 Bdb::DB_INIT_MPOOL  # Initialize the in-memory cache.
+    @env.open(File.join(File.dirname(__FILE__), 'tmp'), env_flags, 0);
+    
+    @db = @env.db
+    @db.open(nil, 'db1.db', nil, Bdb::Db::BTREE, Bdb::DB_CREATE | Bdb::DB_AUTO_COMMIT, 0)    
+  end
+  
+  def teardown
+    @db.close(0)
+    @env.close
+    rm_rf File.join(File.dirname(__FILE__), 'tmp')
+  end
+
+  def test_commit
+    txn = @env.txn_begin(nil, 0)
+    @db.put(txn, 'key', 'value', 0)
+    txn.commit(0)
+    assert_equal 'value', @db.get(nil, 'key', nil, 0)
+  end
+
+  def test_abort
+    txn = @env.txn_begin(nil, 0)
+    @db.put(txn, 'key', 'value', 0)
+    txn.abort
+    assert_nil @db.get(nil, 'key', nil, 0)
+  end
+
+  def test_id
+    txn = @env.txn_begin(nil, 0)
+    @db.put(txn, 'key', 'value', 0)
+    assert txn.tid
+    txn.commit(0)
+    assert_equal 'value', @db.get(nil, 'key', nil, 0)
+  end
+
+  def test_timeout
+    txn = @env.txn_begin(nil, 0)
+    txn.set_timeout(10, Bdb::DB_SET_TXN_TIMEOUT)
+    @db.put(txn, 'key', 'value', 0)
+    txn.commit(0)
+    assert_equal 'value', @db.get(nil, 'key', nil, 0)
+  end
+
+  def test_stat
+    txn = @env.txn_begin(nil, 0)
+    @db.put(txn, 'key', 'value', 0)
+    txn.commit(0)
+    txn_stat = @env.txn_stat(0)
+    assert txn_stat
+    assert txn_stat['st_ncommits'] > 0
+    assert_equal 'value', @db.get(nil, 'key', nil, 0)    
+  end
+
+  def test_stat_active
+    txn = @env.txn_begin(nil, 0)
+    @db.put(txn, 'key', 'value', 0)
+    txn_stat = @env.txn_stat(0)
+    txn.commit(0)
+    assert_equal 1, txn_stat['st_txnarray'].length
+    assert_equal 'value', @db.get(nil, 'key', nil, 0)    
+  end
+
+end

metadata

@@ -0,0 +1,82 @@
+--- !ruby/object:Gem::Specification 
+name: bdb
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Justin Balthrop
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-20 00:00:00 -08:00
+default_executable: 
+dependencies: []
+
+description: Advanced Ruby Berkeley DB library.
+email: code@justinbalthrop.com
+executables: []
+
+extensions: 
+- ext/extconf.rb
+extra_rdoc_files: 
+- LICENSE
+- README.textile
+files: 
+- VERSION
+- ext/bdb.c
+- ext/bdb.h
+- ext/extconf.rb
+- lib/bdb/base.rb
+- lib/bdb/database.rb
+- lib/bdb/environment.rb
+- lib/bdb/partitioned_database.rb
+- lib/bdb/result_set.rb
+- test/benchmark.rb
+- test/cursor_test.rb
+- test/db_test.rb
+- test/env_test.rb
+- test/simple_test.rb
+- test/stat_test.rb
+- test/test_helper.rb
+- test/txn_test.rb
+- LICENSE
+- README.textile
+has_rdoc: true
+homepage: http://github.com/ninjudd/bdb
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- ext
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Ruby Berkeley DB
+test_files: 
+- test/benchmark.rb
+- test/cursor_test.rb
+- test/db_test.rb
+- test/env_test.rb
+- test/simple_test.rb
+- test/stat_test.rb
+- test/test_helper.rb
+- test/txn_test.rb