bdb/test/simple_test.rb

94 lines
1.9 KiB
Ruby

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