require 'test/test_helper' require 'iconv' class BSONTest < Test::Unit::TestCase include Mongo def setup # We don't pass a DB to the constructor, even though we are about to test # deserialization. This means that when we deserialize, any DBRefs will # have nil @db ivars. That's fine for now. #BSON = BSON.new end def test_string doc = {'doc' => 'hello, world'} bson = bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_valid_utf8_string doc = {'doc' => "aéあ"} bson = bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_invalid_string string = Iconv.conv('iso-8859-1', 'utf-8', 'aé').first doc = {'doc' => string} assert_raise InvalidStringEncoding do BSON.serialize(doc) end end def test_code doc = {'$where' => Code.new('this.a.b < this.b')} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_number doc = {'doc' => 41.99} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_int doc = {'doc' => 42} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) doc = {"doc" => -5600} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) doc = {"doc" => 2147483647} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) doc = {"doc" => -2147483648} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_ordered_hash doc = OrderedHash.new doc["b"] = 1 doc["a"] = 2 doc["c"] = 3 doc["d"] = 4 bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_object doc = {'doc' => {'age' => 42, 'name' => 'Spongebob', 'shoe_size' => 9.5}} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_oid doc = {'doc' => ObjectID.new} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_array doc = {'doc' => [1, 2, 'a', 'b']} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_regex doc = {'doc' => /foobar/i} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) assert_equal doc, doc2 r = doc2['doc'] assert_kind_of RegexpOfHolding, r assert_equal '', r.extra_options_str r.extra_options_str << 'zywcab' assert_equal 'zywcab', r.extra_options_str doc = {'doc' => r} bson_doc = BSON.serialize(doc) doc2 = nil doc2 = BSON.deserialize(bson_doc) assert_equal doc, doc2 r = doc2['doc'] assert_kind_of RegexpOfHolding, r assert_equal 'abcwyz', r.extra_options_str # must be sorted end def test_boolean doc = {'doc' => true} bson = BSON.serialize(doc) assert_equal doc, BSON.deserialize(bson) end def test_date doc = {'date' => Time.now} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) # Mongo only stores up to the millisecond assert_in_delta doc['date'], doc2['date'], 0.001 end def test_date_returns_as_utc doc = {'date' => Time.now} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) assert doc2['date'].utc? end def test_date_before_epoch begin doc = {'date' => Time.utc(1600)} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) # Mongo only stores up to the millisecond assert_in_delta doc['date'], doc2['date'], 0.001 rescue ArgumentError # some versions of Ruby won't let you create pre-epoch Time instances # # TODO figure out how that will work if somebady has saved data # w/ early dates already and is just querying for it. end end def test_dbref oid = ObjectID.new doc = {} doc['dbref'] = DBRef.new('namespace', oid) bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) assert_equal 'namespace', doc2['dbref'].namespace assert_equal oid, doc2['dbref'].object_id end def test_symbol doc = {'sym' => :foo} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) assert_equal :foo, doc2['sym'] end def test_binary bin = Binary.new 'binstring'.each_byte { |b| bin.put(b) } doc = {'bin' => bin} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) bin2 = doc2['bin'] assert_kind_of Binary, bin2 assert_equal 'binstring', bin2.to_s assert_equal Binary::SUBTYPE_BYTES, bin2.subtype end def test_binary_type bin = Binary.new([1, 2, 3, 4, 5], Binary::SUBTYPE_USER_DEFINED) doc = {'bin' => bin} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) bin2 = doc2['bin'] assert_kind_of Binary, bin2 assert_equal [1, 2, 3, 4, 5], bin2.to_a assert_equal Binary::SUBTYPE_USER_DEFINED, bin2.subtype end def test_binary_byte_buffer bb = ByteBuffer.new 5.times { |i| bb.put(i + 1) } doc = {'bin' => bb} bson = BSON.serialize(doc) doc2 = BSON.deserialize(bson) bin2 = doc2['bin'] assert_kind_of Binary, bin2 assert_equal [1, 2, 3, 4, 5], bin2.to_a assert_equal Binary::SUBTYPE_BYTES, bin2.subtype end def test_put_id_first val = OrderedHash.new val['not_id'] = 1 val['_id'] = 2 roundtrip = BSON.deserialize(BSON.serialize(val).to_a) assert_kind_of OrderedHash, roundtrip assert_equal '_id', roundtrip.keys.first val = {'a' => 'foo', 'b' => 'bar', :_id => 42, 'z' => 'hello'} roundtrip = BSON.deserialize(BSON.serialize(val).to_a) assert_kind_of OrderedHash, roundtrip assert_equal '_id', roundtrip.keys.first end def test_nil_id doc = {"_id" => nil} assert_equal doc, BSON.deserialize(bson = BSON.serialize(doc).to_a) end def test_timestamp val = {"test" => [4, 20]} assert_equal val, BSON.deserialize([0x13, 0x00, 0x00, 0x00, 0x11, 0x74, 0x65, 0x73, 0x74, 0x00, 0x04, 0x00, 0x00, 0x00, 0x14, 0x00, 0x00, 0x00, 0x00]) end def test_overflow doc = {"x" => 2**75} assert_raise RangeError do bson = BSON.serialize(doc) end doc = {"x" => 9223372036854775} assert_equal doc, BSON.deserialize(BSON.serialize(doc).to_a) doc = {"x" => 9223372036854775807} assert_equal doc, BSON.deserialize(BSON.serialize(doc).to_a) doc["x"] = doc["x"] + 1 assert_raise RangeError do bson = BSON.serialize(doc) end doc = {"x" => -9223372036854775} assert_equal doc, BSON.deserialize(BSON.serialize(doc).to_a) doc = {"x" => -9223372036854775808} assert_equal doc, BSON.deserialize(BSON.serialize(doc).to_a) doc["x"] = doc["x"] - 1 assert_raise RangeError do bson = BSON.serialize(doc) end end def test_do_not_change_original_object val = OrderedHash.new val['not_id'] = 1 val['_id'] = 2 assert val.keys.include?('_id') BSON.serialize(val) assert val.keys.include?('_id') val = {'a' => 'foo', 'b' => 'bar', :_id => 42, 'z' => 'hello'} assert val.keys.include?(:_id) BSON.serialize(val) assert val.keys.include?(:_id) end # note we only test for _id here because in the general case we will # write duplicates for :key and "key". _id is a special case because # we call has_key? to check for it's existance rather than just iterating # over it like we do for the rest of the keys. thus, things like # HashWithIndifferentAccess can cause problems for _id but not for other # keys. rather than require rails to test with HWIA directly, we do this # somewhat hacky test. def test_no_duplicate_id dup = {"_id" => "foo", :_id => "foo"} one = {"_id" => "foo"} assert_equal BSON.serialize(one).to_a, BSON.serialize(dup).to_a end end