require 'helper'

class TestBtree < Test::Unit::TestCase
  class BTCompare < BDB1::Btree
    def bdb1_bt_compare(a, b)
      a <=> b
    end
  end

  class AZ < BDB1::Btree
    def bdb1_store_key(a)
      "xx_" + a
    end
    def bdb1_fetch_key(a)
      a.sub(/^xx_/, '')
    end
    def bdb1_store_value(a)
      "yy_" + a
    end
    def bdb1_fetch_value(a)
      a.sub(/^yy_/, '')
    end
  end

  def test_00_error
    assert_raises(BDB1::Fatal, "invalid name") do
      BDB1::Btree.new(tmpdir("."), "a")
    end
    assert_raises(BDB1::Fatal, "invalid Env") do
      BDB1::Btree.open(tmpdir("aa"), "env" => 1)
    end
    assert_raises(TypeError) { BDB1::Btree.new(tmpdir("aa"), "set_cachesize" => "a") }
    assert_raises(TypeError) { BDB1::Btree.new(tmpdir("aa"), "set_pagesize" => "a") }
    assert_raises(TypeError) { BDB1::Btree.new(tmpdir("aa"), "set_bt_minkey" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_bt_compare" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_bt_prefix" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_fetch_key" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_store_key" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_fetch_value" => "a") }
    assert_raises(BDB1::Fatal) { BDB1::Btree.new(tmpdir("aa"), "set_store_value" => "a") }
    assert_raises(TypeError) { BDB1::Btree.new(tmpdir("aa"), "set_lorder" => "a") }
  end

  def test_01_file
    sub_file_init
    sub_file_get_set
    sub_file_delete
    sub_file_cursor
    sub_file_reopen
    sub_file_dup
    sub_file_close
  end

  def sub_file_init
    assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree.new(tmpdir("aa"), "a"), "<open>")
  end

  def sub_file_get_set
    assert_equal(true, @bdb.empty?, "<empty>")
    assert_equal("alpha", @bdb["alpha"] = "alpha", "<set value>")
    assert_equal(false, @bdb.empty?, "<empty>")
    assert_equal("alpha", @bdb["alpha"], "<retrieve value>")
    assert_equal("alpha", @bdb.fetch("alpha"), "<fetch value>")
    assert_equal("beta", @bdb.fetch("xxx", "beta"), "<fetch nil>")
    assert_equal("xxx", @bdb.fetch("xxx") {|x| x}, "<fetch nil>")
    assert_raises(IndexError) { @bdb.fetch("xxx") }
    assert_raises(ArgumentError) { @bdb.fetch("xxx", "beta") {} }
    assert_equal(nil, @bdb["gamma"] = nil, "<set nil>")
    assert_equal(nil, @bdb["gamma"], "<retrieve nil>")
    assert(@bdb.key?("alpha") == "alpha", "<has key>")
    assert_equal(false, @bdb.key?("unknown"), "<has unknown key>")
    assert(@bdb.value?(nil), "<has nil>")
    assert(@bdb.value?("alpha"), "<has value>")
    assert_equal(false, @bdb.value?("unknown"), "<has unknown value>")
    assert_equal(false, @bdb.put("alpha", "gamma", BDB1::NOOVERWRITE), "<nooverwrite>")
    assert_equal("alpha", @bdb["alpha"], "<must not be changed>")
    assert(@bdb.both?("alpha", "alpha"), "<has both>")
    assert(! @bdb.both?("alpha", "beta"), "<don't has both>")
    assert(! @bdb.both?("unknown", "alpha"), "<don't has both>")
    assert_equal([1, 2, 3], @bdb["array"] = [1, 2, 3], "<array>")
    assert_equal([1, 2, 3].to_s, @bdb["array"], "<retrieve array>")
    assert_equal({"a" => "b"}, @bdb["hash"] = {"a" => "b"}, "<hash>")
    assert_equal({"a" => "b"}.to_s, @bdb["hash"], "<retrieve hash>")
    assert(@bdb.sync, "<sync>")
  end

  def sub_file_delete
    size = @bdb.size
    i = 0
    @bdb.each do |key, value|
      assert_equal(@bdb, @bdb.delete(key), "<delete value>")
      i += 1
    end
    assert(size == i, "<delete count>")
    assert_equal(0, @bdb.size, "<empty>")
    $hash = {}
    (33 .. 126).each do |i|
      key = i.to_s * 5
      @bdb[key] = i.to_s * 7
      $hash[key] = i.to_s * 7
      assert_equal(@bdb[key], $hash[key], "<set #{key}>")
    end
    assert_equal(@bdb.size, $hash.size, "<size after load>")
    if @bdb.respond_to?(:select)
      assert_raises(ArgumentError) { @bdb.select("xxx") {}}
      assert_equal([], @bdb.select { false }, "<select none>")
      assert_equal($hash.values.sort, @bdb.select { true }.sort, "<select all>")
    end
    arr0 = []
    @bdb.each_key {|key| arr0 << key }
    assert_equal($hash.keys.sort, arr0.sort, "<each key>")
    arr1 = []
    @bdb.reverse_each_key {|key| arr1 << key }
    assert_equal($hash.keys.sort, arr1.sort, "<reverse each key>")
    assert_equal(arr0, arr1.reverse, "<reverse>")
    assert_equal($hash.invert, @bdb.invert, "<invert>")
    @bdb.each do |key, value|
      if rand < 0.5
        assert_equal(@bdb, @bdb.delete(key), "<delete value>")
        $hash.delete(key)
      end
    end
    assert_equal(@bdb.size, $hash.size, "<size after load>")
    @bdb.each do |key, value|
      assert_equal($hash[key], value, "<after delete>")
    end
    @bdb.each do |key, value|
      assert($hash.key?(key), "<key after delete>")
      assert_equal(@bdb, @bdb.delete(key), "<delete value>")
    end
  end

  def sub_file_cursor
    array = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
    array.each do |x|
      assert_equal(x, @bdb[x] = x, "<set value>")
    end
    assert(array.size == @bdb.size, "<set count>")
    arr = []
    @bdb.each_value do |x|
      arr << x
    end
    assert_equal(array, arr.sort, "<order>")
    arr = []
    @bdb.reverse_each_value do |x|
      arr << x
    end
    assert_equal(array, arr.sort, "<reverse order>")
    arr = @bdb.reject {|k, v| k == "e" || v == "i" }
    has = array.reject {|k, v| k == "e" || k == "i" }
    assert_equal(has, arr.keys.sort, "<reject>")
    @bdb.reject! {|k, v| k == "e" || v == "i" }
    array.reject! {|k, v| k == "e" || k == "i" }
    assert_equal(array, @bdb.keys.sort, "<keys after reject>")
    assert_equal(array, @bdb.values.sort, "<values after reject>")
  end

  def sub_file_reopen
    assert_equal(nil, @bdb.close, "<close>")
    assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree.open(tmpdir("aa"), "w",
	"set_flags" => BDB1::DUP),
      "<reopen with DB_DUP>")
    assert_equal(0, @bdb.size, "<must be 0 after reopen>")
  end

  def sub_file_dup
    assert_equal("a", @bdb["0"] = "a", "<set dup>")
    assert_equal("b", @bdb["0"] = "b", "<set dup>")
    assert_equal("c", @bdb["0"] = "c", "<set dup>")
    assert_equal("d", @bdb["0"] = "d", "<set dup>")
    assert_equal("aa", @bdb["1"] = "aa", "<set dup>")
    assert_equal("bb", @bdb["1"] = "bb", "<set dup>")
    assert_equal("cc", @bdb["1"] = "cc", "<set dup>")
    assert_equal("aaa", @bdb["2"] = "aaa", "<set dup>")
    assert_equal("bbb", @bdb["2"] = "bbb", "<set dup>")
    assert_equal("aaaa", @bdb["3"] = "aaaa", "<set dup>")
    rep = [["a", "b", "c", "d"], ['aa', 'bb', 'cc'], ['aaa', 'bbb'], ['aaaa']]
    for i in [0, 1, 2, 3]
      k0, v0 = [], []
      @bdb.duplicates(i.to_s).each {|k, v| k0 << k; v0 << v}
      assert_equal(k0, [i.to_s] * (4 - i), "<dup key #{i}>")
      assert_equal(v0.sort, rep[i], "<dup val #{i}>")
      k0, v0 = [], []
      @bdb.each_dup(i.to_s) {|k, v| k0 << k; v0 << v}
      assert_equal(k0, [i.to_s] * (4 - i), "<dup key #{i}>")
      assert_equal(v0.sort, rep[i], "<dup val #{i}>")
      v0 = []
      @bdb.each_dup_value(i.to_s) {|v|  v0 << v}
      assert_equal(v0.sort, rep[i], "<dup val #{i}>")
    end
  end

  def sub_file_close
    assert_equal(nil, @bdb.close, "<close>")
  end

  def test_02_memory
    sub_memory_open
    sub_memory_get_set
    sub_memory_close
  end

  def sub_memory_open
    assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree.open, "<open in memory>")
    assert_equal(0, @bdb.size, "<must be 0 after open>")
  end

  def sub_memory_get_set
    (33 .. 126).each do |i|
      key = i.to_s * 5
      val = i.to_s * 7
      assert_equal(val, @bdb[key] = val, "<set in memory>")
    end
    assert_equal(94, @bdb.size, "<length in memory>")
  end

  def sub_memory_close
    assert_equal(nil, @bdb.close, "<close>")
  end

  def test_03_btree
    sub_btree_delete_1
  end

  def intern_btree_delete
    @hash = {}
    File.foreach(examplesdir("wordtest")) do |line|
      line.chomp!
      @hash[line] = line.reverse
      @bdb[line] = line.reverse
    end
    @bdb.each do |k, v|
      assert_equal(@hash[k], v, "<value>")
    end
    @bdb.delete_if {|k, v| k[0] == ?a}
    @hash.delete_if {|k, v| k[0] == ?a}
    assert_equal(@bdb.size, @hash.size, "<size after delete_if>")
    @bdb.each do |k, v|
      assert_equal(@hash[k], v, "<value>")
    end
  end

  def sub_btree_delete_1
    assert_kind_of(BTCompare,
      @bdb = BTCompare.open(tmpdir("aa"), "w",
        "set_pagesize" => 1024,
        "set_cachesize" => 32 * 1024),
      "<open>")
    intern_btree_delete
  end

  def test_04_btree
    sub_btree_delete_2
    sub_index
    sub_convert
    sub_sh
    sub_sh_call
    sub_create
  end

  def sub_btree_delete_2
    assert_kind_of(BDB1::Btree,
      @bdb = BDB1::Btree.open(tmpdir("aa"), "w",
        "set_bt_compare" => proc {|a, b| a <=> b},
        "set_pagesize" => 1024,
        "set_cachesize" => 32 * 1024),
      "<open>")
    intern_btree_delete
  end

  def sub_index
    lines = @hash.keys
    array = []
    10.times do
      h = lines[rand(lines.size - 1)]
      array.push h
      assert_equal(@hash.index(h.reverse), @bdb.index(h.reverse), "<index>")
    end
    assert_equal(@hash.values_at(array), @bdb.values_at(array), "<values_at>")
  end

  def sub_convert
    h = @bdb.to_hash
    h.each do |k, v|
      assert_equal(v, @hash[k], "<to_hash>")
    end
    @hash.each do |k, v|
      assert_equal(v, h[k], "<to_hash>")
    end
    h1 = @hash.to_a
    h2 = @bdb.to_a
    assert_equal(h1.size, h2.size, "<equal>")
  end

  def intern_sh
    val = 'a' .. 'zz'
    val.each do |l|
      assert_equal(l, @bdb[l] = l, "<store>")
    end
    @bdb.each do |k, v|
      assert_equal(k, v, "<fetch>")
    end
    assert_equal(nil, @bdb.close, "<close>")
    assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree.open(tmpdir("aa")), "<sh>")
    val.each do |l|
      assert_equal("yy_#{l}", @bdb["xx_#{l}"], "<fetch value>")
    end
    @bdb.each do |k, v|
      assert_equal("xx_", k[0, 3], "<fetch key>")
      assert_equal("yy_", v[0, 3], "<fetch key>")
    end
    assert_equal(nil, @bdb.close, "<close>")
    clean_tmpdir
  end

  def sub_sh
    assert_equal(nil, @bdb.close, "<close>")
    assert_kind_of(BDB1::Btree, @bdb = AZ.open(tmpdir("aa"), "w"), "<sh>")
    intern_sh
  end

  def sub_sh_call
    @bdb= BDB1::Btree.new(tmpdir("aa"), "w",
      "set_store_key" => proc {|a| "xx_" + a },
      "set_fetch_key" => proc {|a| a.sub(/^xx_/, '') },
      "set_store_value" => proc {|a| "yy_" + a },
      "set_fetch_value" => proc {|a| a.sub(/^yy_/, '') })
    assert_kind_of(BDB1::Btree, @bdb)
    intern_sh
  end


  def intern_create(opt = true)
    if opt
      assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree[@hash], "<create>")
    else
      aa = @hash.to_a.flatten
      assert_kind_of(BDB1::Btree, @bdb = BDB1::Btree[*aa], "<create>")
    end
    assert_equal(@bdb.size, @hash.size, "<size after load>")
    if @bdb.respond_to?(:select)
      assert_raises(ArgumentError) { @bdb.select("xxx") {}}
      assert_equal([], @bdb.select { false }, "<select none>")
      assert_equal(@hash.values.sort, @bdb.select { true }.sort, "<select all>")
    end
    arr0 = []
    @bdb.each_key {|key| arr0 << key }
    assert_equal(@hash.keys.sort, arr0.sort, "<each key>")
    arr0 = []
    @bdb.each_value {|value| arr0 << value }
    assert_equal(@hash.values.sort, arr0.sort, "<each value>")
    @bdb.close
    clean_tmpdir
  end

  def sub_create
    intern_create
    #intern_create(false) # Change 1.9
  end
end