mongo-ruby-driver/test/bson/byte_buffer_test.rb

209 lines
4.9 KiB
Ruby

# encoding: binary
require File.expand_path("../test_helper", __FILE__)
class ByteBufferTest < Test::Unit::TestCase
include BSON
def setup
@buf = ByteBuffer.new
end
def test_initial_state
assert_equal 0, @buf.position
assert_equal [], @buf.to_a
assert_equal "", @buf.to_s
assert_equal 0, @buf.length
end
def test_nil_get_returns_one_byte
@buf.put_array([1, 2, 3, 4])
@buf.rewind
assert_equal 1, @buf.get
end
def test_unpack
@buf.put_array([17, 2, 3, 4])
assert_equal [17, 2, 3, 4], @buf.to_a
assert_equal ["11020304"], @buf.unpack("H*")
assert_equal ["11020304"], @buf.to_a("H*")
end
def test_one_get_returns_array_length_one
@buf.put_array([1, 2, 3, 4])
@buf.rewind
assert_equal [1], @buf.get(1)
end
def test_zero_get_returns_empty_array
@buf.put_array([1, 2, 3, 4])
@buf.rewind
assert_equal [], @buf.get(0)
end
def test_length
@buf.put_int 3
assert_equal 4, @buf.length
end
def test_default_order
assert_equal :little_endian, @buf.order
end
def test_long_length
@buf.put_long 1027
assert_equal 8, @buf.length
end
def test_get_long
@buf.put_long 1027
@buf.rewind
assert_equal 1027, @buf.get_long
end
def test_get_double
@buf.put_double 41.2
@buf.rewind
assert_equal 41.2, @buf.get_double
end
if defined?(Encoding)
def test_serialize_cstr_converts_encoding_to_utf8
theta = "hello \xC8".force_encoding("ISO-8859-7")
ByteBuffer.serialize_cstr(@buf, theta)
assert_equal "hello \xCE\x98\0", @buf.to_s
assert_equal Encoding.find('binary'), @buf.to_s.encoding
end
def test_serialize_cstr_validates_data_as_utf8
assert_raises(Encoding::UndefinedConversionError) do
ByteBuffer.serialize_cstr(@buf, "hello \xFF")
end
end
else
def test_serialize_cstr_forces_encoding_to_utf8
# Unicode snowman (\u2603)
ByteBuffer.serialize_cstr(@buf, "hello \342\230\203")
assert_equal "hello \342\230\203\0", @buf.to_s
end
def test_serialize_cstr_validates_data_as_utf8
assert_raises(BSON::InvalidStringEncoding) do
ByteBuffer.serialize_cstr(@buf, "hello \xFF")
end
end
end
def test_put_negative_byte
@buf.put(-1)
@buf.rewind
assert_equal 255, @buf.get
assert_equal "\xFF", @buf.to_s
end
def test_put_with_offset
@buf.put(1)
@buf.put(2, 0)
@buf.put(3, 3)
assert_equal "\x02\x00\x00\x03", @buf.to_s
end
def test_put_array_with_offset
@buf.put(1)
@buf.put_array([2, 3], 0)
@buf.put_array([4, 5], 4)
assert_equal "\x02\x03\x00\x00\x04\x05", @buf.to_s
end
def test_put_int_with_offset
@buf.put(1)
@buf.put_int(2, 0)
@buf.put_int(3, 5)
assert_equal "\x02\x00\x00\x00\x00\x03\x00\x00\x00", @buf.to_s
end
def test_put_long_with_offset
@buf.put(1)
@buf.put_long(2, 0)
@buf.put_long(3, 9)
assert_equal(
"\x02\x00\x00\x00\x00\x00\x00\x00" +
"\x00" +
"\x03\x00\x00\x00\x00\x00\x00\x00",
@buf.to_s)
end
def test_put_binary
@buf.put(1)
@buf.put_binary("\x02\x03", 0)
@buf.put_binary("\x04\x05", 4)
assert_equal "\x02\x03\x00\x00\x04\x05", @buf.to_s
end
def test_rewrite
@buf.put_int(0)
@buf.rewind
@buf.put_int(1027)
assert_equal 4, @buf.length
@buf.rewind
assert_equal 1027, @buf.get_int
assert_equal 4, @buf.position
end
def test_prepend_byte_buffer
@buf.put_int(4)
new_buf = ByteBuffer.new([5, 0, 0, 0])
@buf.prepend!(new_buf)
assert_equal [5, 0, 0, 0, 4, 0, 0, 0], @buf.to_a
end
def test_append_byte_buffer
@buf.put_int(4)
new_buf = ByteBuffer.new([5, 0, 0, 0])
@buf.append!(new_buf)
assert_equal [4, 0, 0, 0, 5, 0, 0, 0], @buf.to_a
end
def test_array_as_initial_input
@buf = ByteBuffer.new([5, 0, 0, 0])
assert_equal 4, @buf.size
assert_equal "\x05\x00\x00\x00", @buf.to_s
assert_equal [5, 0, 0, 0], @buf.to_a
@buf.put_int(32)
@buf.rewind
assert_equal 5, @buf.get_int
assert_equal 32, @buf.get_int
end
def test_binary_string_as_initial_input
str = "abcd"
str.force_encoding('binary') if str.respond_to?(:force_encoding)
@buf = ByteBuffer.new(str)
assert_equal "abcd", @buf.to_s
assert_equal [97, 98, 99, 100], @buf.to_a
@buf.put_int(0)
assert_equal [97, 98, 99, 100, 0, 0, 0, 0], @buf.to_a
end
def test_more
assert !@buf.more?
@buf.put_int(5)
assert !@buf.more?
@buf.rewind
assert @buf.more?
@buf.get_int
assert !@buf.more?
end
def test_equality
@buf = ByteBuffer.new("foo")
assert_equal @buf, @buf
assert_equal ByteBuffer.new(""), ByteBuffer.new("")
assert_equal ByteBuffer.new("123"), ByteBuffer.new("123")
assert_not_equal ByteBuffer.new("123"), ByteBuffer.new("1234")
assert_equal @buf, "foo"
assert_not_equal @buf, 123
assert_not_equal @buf, nil
end
end