#!/usr/bin/env ruby

$LOAD_PATH.unshift File.expand_path('..', __FILE__)
require 'helper'

class TestByteArray < Test::Unit::TestCase
  def test_order
    [
      [(1..100).to_a, :fixnum],
      [('aa'..'zz').to_a, :string],
    ].each do |pair|
      arr, type = pair
      assert_equal arr,
        arr.reverse.map { |e| HBase::ByteArray.new(e) }.sort.map { |ba|
          HBase::Util.from_bytes type, ba.java
        }
    end
  end

  def test_stopkey_bytes_for_prefix
    assert_equal HBase::ByteArray.new("hellp"),
      HBase::ByteArray.new( HBase::ByteArray.new("hello").stopkey_bytes_for_prefix )
    assert_equal HBase::ByteArray.new("BLUF"),
      HBase::ByteArray.new( HBase::ByteArray.new("BLUE").stopkey_bytes_for_prefix )
    assert_nil HBase::ByteArray.new([127, 127, 127].to_java(Java::byte)).stopkey_bytes_for_prefix
    assert_equal HBase::ByteArray.new([126, 127].to_java(Java::byte)),
      HBase::ByteArray.new(
        HBase::ByteArray.new([126, 126, 127, 127, 127, 127].to_java(Java::byte)).stopkey_bytes_for_prefix
      )
  end

  def test_as_hash_key
    hash = {
      HBase::ByteArray.new("Hello") => 1,
      HBase::ByteArray.new("World") => 2
    }
    assert_equal 1, hash[ HBase::ByteArray.new("Hello") ]
    assert_equal 2, hash[ HBase::ByteArray.new("World".to_java_bytes) ]
  end

  def test_concat
    concat = HBase::ByteArray(100) + HBase::ByteArray(200)
    assert_instance_of HBase::ByteArray, concat
    assert_equal 16, concat.to_java_bytes.to_a.length

    assert_equal 100, HBase::Util.from_bytes( :fixnum, concat.to_java_bytes.to_a[0, 8].to_java(Java::byte) )
    assert_equal 200, HBase::Util.from_bytes( :fixnum, concat.java.to_a[8, 8].to_java(Java::byte) )

    assert HBase::ByteArray(100, 200).eql?(concat)
    assert concat.eql?(HBase::ByteArray(100) + 200)
    assert concat.eql?(HBase::ByteArray(100) << 200)

    ba = HBase::ByteArray(100)
    ba += HBase::ByteArray(200)
    assert_equal concat, ba
  end

  def test_default_constructor
    assert_equal 0, HBase::ByteArray().java.to_a.length
    assert_equal 0, HBase::ByteArray.new.java.to_a.length
  end

  def test_slice
    ba = HBase::ByteArray(100, 200, "Hello", 3.14)

    assert_equal 100, ba[0, 8].decode(:fixnum)
    assert_equal 200, ba[8...16].decode(:fixnum)
    assert_equal "Hello", ba[16, 5].decode(:string)
    assert_equal 3.14, ba[21..-1].decode(:float)
    assert_equal "H", ba[16].chr
  end

  def test_length_shift
    ba = HBase::ByteArray(100, 200, "Hello", false, 3.14)

    assert_equal 30, ba.length
    assert_equal 100, ba.shift(:fixnum)
    assert_equal 22, ba.length
    assert_equal 200, ba.shift(:fixnum)
    assert_equal 14, ba.length
    assert_raise(ArgumentError) { ba.shift(:string) }
    assert_equal "Hello", ba.shift(:string, 5)
    assert_equal 9, ba.length
    assert_equal false, ba.shift(:boolean)
    assert_equal 8, ba.length
    assert_equal 3.14, ba.shift(:float)
    assert_equal 0, ba.length

    assert_raise(ArgumentError) { ba.shift(:fixnum) }
  end

  def test_short_int
    assert_equal 1,  HBase::ByteArray(:byte => 12).length
    assert_equal 12, HBase::ByteArray(:byte => 12).decode(:byte)
    assert_raise(RangeError) { HBase::ByteArray(:byte => 128) }

    assert_equal 2,     HBase::ByteArray(:short => 12345).length
    assert_equal 12345, HBase::ByteArray(:short => 12345).decode(:short)
    assert_raise(RangeError) { HBase::ByteArray( :short => 1 << 16 ) }

    assert_equal 4,     HBase::ByteArray(:int => 12345).length
    assert_equal 12345, HBase::ByteArray(:int => 12345).decode(:int)
    assert_raise(RangeError) { HBase::ByteArray.new( :int => 1 << 32 ) }

    ba = HBase::ByteArray( {:int => 10000}, 20000, {:short => 30000}, "Hello" )
    assert_equal 10000, ba[0, 4].decode(:int)
    assert_equal 20000, ba[4, 8].decode(:long)
    assert_equal 30000, ba[12, 2].decode(:short)
    assert_equal "Hell", ba[14, 4].decode(:string)
  end

  def test_each
    ba = HBase::ByteArray("Hello world")
    Hash[ba.to_a.zip("Hello world".each_char.to_a)].each do |ord, chr|
      assert_equal ord.chr, chr
    end
  end

  def test_unshift
    bd = BigDecimal.new("123456789012345678901234567890")
    ba = HBase::ByteArray(bd)
    oid = ba.object_id
    assert_equal oid, ba.unshift(:short => 1000).object_id

    ba.unshift("a", "b", "c")
    assert_equal 'a', ba[0].chr
    assert_equal 'b', ba[1].chr
    assert_equal 'c', ba[2].chr
    assert_equal 1000, ba[3, 2].decode(:short)
    assert_equal bd, ba[5..-1].decode(:bigdecimal)
  end

  def test_append
    ba = HBase::ByteArray()
    oid = ba.object_id
    assert_equal oid, (ba << 100).object_id

    ba << 200 << {:int => 300} << "Hello" << 3.14
    assert_equal 100, ba.shift(:fixnum)
    assert_equal 200, ba.shift(:long)
    assert_equal 300, ba.shift(:int)
    assert_equal "Hello", ba.shift(:string, 5)
    assert_equal 3.14, ba.shift(:float)
  end
end