#
# Pythomnic3k project
# (c) 2005-2012, Dmitry Dvoinikov <dmitry@targeted.org>
+# (c) 2012-2013 Softmotions Ltd <info@softmotions.com>
# Distributed under BSD license
#
###############################################################################
###############################################################################
-if __name__ == "__main__": # self-test
-
- from random import shuffle
-
- try:
- import expected
- except ImportError:
- print("warning: module expected.py cannot be imported, exception tests are skipped")
-
- class expected:
- def __init__(self, *args, **kwargs):
- pass
-
- def __enter__(self):
- pass
-
- def __exit__(self, t, v, tb):
- return True
- else:
- from expected import expected
-
- ###################################
-
- print("self-testing module bson.py:")
-
- ###################################
-
- print("utilities: ", end="")
-
- assert cstrify("") == b"\x00"
- assert cstrify("foo") == b"foo\x00"
- assert cstrify(u"абв") == b"\xd0\xb0\xd0\xb1\xd0\xb2\x00"
-
- print("ok")
-
- ###################################
-
- print("exact byte sequences: ", end="")
-
- def test_type(t, v, b):
- bio = BytesIO()
- t(v).serialize(bio)
- assert bio.getvalue() == b, bio.getvalue()
- bio.seek(0, 0)
- assert t.parse(bio).value == v
- assert bio.tell() == len(bio.getvalue())
-
- test_type(BSON_Double, 0.0, b"\x00\x00\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Double, 1.0, b"\x00\x00\x00\x00\x00\x00\xf0?")
- test_type(BSON_Double, -1.0, b"\x00\x00\x00\x00\x00\x00\xf0\xbf")
- test_type(BSON_Double, 1e308, b"\xa0\xc8\xeb\x85\xf3\xcc\xe1\x7f")
- test_type(BSON_Double, 1e-308, b"\xd2\xe8\x19x\xd60\x07\x00")
- test_type(BSON_Double, -1e308, b"\xa0\xc8\xeb\x85\xf3\xcc\xe1\xff")
- test_type(BSON_Double, -1e-308, b"\xd2\xe8\x19x\xd60\x07\x80")
-
- test_type(BSON_Int32, 0, b"\x00\x00\x00\x00")
- test_type(BSON_Int32, 1, b"\x01\x00\x00\x00")
- test_type(BSON_Int32, -1, b"\xff\xff\xff\xff")
- test_type(BSON_Int32, 2 ** 31 - 1, b"\xff\xff\xff\x7f")
- test_type(BSON_Int32, -2 ** 31, b"\x00\x00\x00\x80")
-
- test_type(BSON_Int64, 0, b"\x00\x00\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Int64, 1, b"\x01\x00\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Int64, -1, b"\xff\xff\xff\xff\xff\xff\xff\xff")
- test_type(BSON_Int64, 2 ** 63 - 1, b"\xff\xff\xff\xff\xff\xff\xff\x7f")
- test_type(BSON_Int64, -2 ** 63, b"\x00\x00\x00\x00\x00\x00\x00\x80")
-
- test_type(BSON_String, "", b"\x01\x00\x00\x00\x00")
- test_type(BSON_String, "foo", b"\x04\x00\x00\x00foo\x00")
- test_type(BSON_String, u"абв", b"\x07\x00\x00\x00\xd0\xb0\xd0\xb1\xd0\xb2\x00")
-
- test_type(BSON_Boolean, True, b"\x01")
- test_type(BSON_Boolean, False, b"\x00")
-
- test_type(BSON_ObjectId, b"\x01\x02\x03\x04\x05\x06\x07\x08\x09\x10\x11\x12",
- b"\x01\x02\x03\x04\x05\x06\x07\x08\x09\x10\x11\x12")
-
- test_type(BSON_Null, None, b"")
-
- test_type(BSON_JavaScript, "var i = 1;", b"\x0b\x00\x00\x00var i = 1;\x00")
- test_type(BSON_JavaScript, u"/* комментарий */",
- b"\x1d\x00\x00\x00/* \xd0\xba\xd0\xbe\xd0\xbc\xd0\xbc\xd0\xb5\xd0\xbd\xd1\x82\xd0\xb0\xd1\x80\xd0\xb8\xd0\xb9 */\x00")
-
- test_type(BSON_Timestamp, 1, b"\x01\x00\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Timestamp, -1, b"\xff\xff\xff\xff\xff\xff\xff\xff")
- test_type(BSON_Timestamp, 2 ** 63 - 1, b"\xff\xff\xff\xff\xff\xff\xff\x7f")
- test_type(BSON_Timestamp, -2 ** 63, b"\x00\x00\x00\x00\x00\x00\x00\x80")
-
- test_type(BSON_Datetime, datetime(1970, 1, 1, 0, 0, 0), b"\x00\x00\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Datetime, datetime(1970, 1, 1, 0, 0, 1), b"\xe8\x03\x00\x00\x00\x00\x00\x00")
- test_type(BSON_Datetime, datetime(2010, 12, 12, 15, 11, 43), b"\x98\x27\x23\xdb\x2c\x01\x00\x00")
-
- test_type(BSON_Symbol, "foo", b"\x04\x00\x00\x00foo\x00")
- test_type(BSON_Symbol, u"символ",
- b"\x0d\x00\x00\x00\xd1\x81\xd0\xb8\xd0\xbc\xd0\xb2\xd0\xbe\xd0\xbb\x00")
-
- test_type(BSON_JavaScriptWithScope, ("var i = j + 1;", BSON_Document({"j": BSON_Int32(0)})),
- b"\x23\x00\x00\x00\x0f\x00\x00\x00var i = j + 1;\x00\x0c\x00\x00\x00\x10j\x00\x00\x00\x00\x00\x00")
- test_type(BSON_JavaScriptWithScope, ("foo(s);", BSON_Document({"s": BSON_String(u"абв")})),
- b"\x23\x00\x00\x00\x08\x00\x00\x00foo(s);\x00\x13\x00\x00\x00\x02s\x00\x07\x00\x00\x00\xd0\xb0\xd0\xb1\xd0\xb2\x00\x00")
-
- test_type(BSON_Regex, ("", ""), b"\x00\x00")
- test_type(BSON_Regex, ("^$", "i"), b"^$\x00i\x00")
-
- test_type(BSON_Binary_Generic, b"value", b"\x05\x00\x00\x00\x00value")
- test_type(BSON_Binary_Function, b"value", b"\x05\x00\x00\x00\x01value")
- test_type(BSON_Binary_UUID, b"value", b"\x05\x00\x00\x00\x03value")
- test_type(BSON_Binary_MD5, b"value", b"\x05\x00\x00\x00\x05value")
- test_type(BSON_Binary_UserDefined, b"value", b"\x05\x00\x00\x00\x80value")
-
- test_type(BSON_Document, {}, b"\x05\x00\x00\x00\x00")
- test_type(BSON_Document, {"foo": BSON_Null(None)}, b"\x0a\x00\x00\x00\x0afoo\x00\x00")
-
- print("ok")
-
- ###################################
-
- print("forward type mapping: ", end="")
-
- assert py_to_bs(1.0) == BSON_Double(1.0)
- assert py_to_bs("foo") == BSON_String("foo")
- assert py_to_bs(True) == BSON_Boolean(True)
- dt = datetime.now();
- assert py_to_bs(dt) == BSON_Datetime(dt)
- assert py_to_bs(None) == BSON_Null(None)
- assert py_to_bs(2 ** 31 - 1) == BSON_Int32(2 ** 31 - 1)
- assert py_to_bs(2 ** 31) == BSON_Int64(2 ** 31)
-
- def py_to_bs_same(v):
- return py_to_bs(v) is v
-
- assert py_to_bs_same(BSON_ObjectId(b"\x00" * 12))
- assert py_to_bs_same(BSON_Regex(("^$", "i")))
- assert py_to_bs_same(BSON_JavaScript("var i = 1;"))
- assert py_to_bs_same(BSON_Symbol("class"))
- assert py_to_bs_same(BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"j": BSON_Int32(10)}))))
- assert py_to_bs_same(BSON_Timestamp(0x7fffffffffffffff))
-
- assert py_to_bs([]) == BSON_Array([])
- assert py_to_bs(
- [
- b"implicit" if PY3 else BytesIO("implicit"),
- BSON_Binary_Generic(b"generic"),
- BSON_Binary_Function(b"function"),
- BSON_Binary_UUID(b"uuid"),
- BSON_Binary_MD5(b"md5"),
- BSON_Binary_UserDefined(b"userdefined"),
- ]) ==\
- BSON_Array(
- [
- BSON_Binary_Generic(b"implicit"),
- BSON_Binary_Generic(b"generic"),
- BSON_Binary_Function(b"function"),
- BSON_Binary_UUID(b"uuid"),
- BSON_Binary_MD5(b"md5"),
- BSON_Binary_UserDefined(b"userdefined"),
- ])
-
- assert py_to_bs({}) == BSON_Document({})
- assert py_to_bs(
- {
- "array": [{}, {}],
- "document": {"key": []},
- }) ==\
- BSON_Document(
- {
- "array": BSON_Array([BSON_Document({}), BSON_Document({})]),
- "document": BSON_Document({"key": BSON_Array([])}),
- })
-
- with expected(BSON_ConversionError, "cannot implicitly convert int value 9223372036854775808"):
- py_to_bs(2 ** 63)
-
- with expected(BSON_ConversionError, "cannot implicitly convert tuple value \(\)"):
- py_to_bs(())
-
- print("ok")
-
- ###################################
-
- print("document key ordering: ", end="")
-
- class Foo(int):
- pass
-
- ks1 = list(range(10000))
- shuffle(ks1)
- d1 = odict((str(k), Foo(k)) for k in ks1)
- b = serialize_to_bytes(d1)
- d2 = parse_bytes(b)
- ks2 = list(d2.values())
- assert ks2 == ks1
-
- print("ok")
-
- ###################################
-
- print("reverse type mapping: ", end="")
-
- assert bs_to_py(BSON_Double(1.0)) == 1.0
- assert bs_to_py(BSON_String("foo")) == "foo"
- assert bs_to_py(BSON_Boolean(True)) == True
- dt = datetime.now()
- assert bs_to_py(BSON_Datetime(dt)) == dt
- assert bs_to_py(BSON_Null(None)) == None
- assert bs_to_py(BSON_Int32(2 ** 31 - 1)) == 2 ** 31 - 1
- assert bs_to_py(BSON_Int64(2 ** 31)) == 2 ** 31
-
- def bs_to_py_same(v):
- return bs_to_py(v) is v
-
- assert type(BSON_ObjectId(b"\x00" * 12)._py_value()).__name__ == 'str' if PY3 else 'unicode'
- assert bs_to_py_same(BSON_Regex(("^$", "")))
- assert bs_to_py_same(BSON_JavaScript("var i = 0;"))
- assert bs_to_py_same(BSON_Symbol("class"))
- assert bs_to_py_same(BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"j": -1}))))
- assert bs_to_py_same(BSON_Timestamp(0x7fffffffffffffff))
-
- assert bs_to_py(BSON_Array([])) == []
-
- assert bs_to_py(BSON_Array(
- [
- BSON_Binary_Generic(b"generic"),
- BSON_Binary_Function(b"function"),
- BSON_Binary_UUID(b"uuid"),
- BSON_Binary_MD5(b"md5"),
- BSON_Binary_UserDefined(b"userdefined"),
- ])) ==\
- [
- b"generic" if PY3 else BSON_Binary_Generic(b"generic"),
- BSON_Binary_Function(b"function"),
- BSON_Binary_UUID(b"uuid"),
- BSON_Binary_MD5(b"md5"),
- BSON_Binary_UserDefined(b"userdefined"),
- ]
-
- assert bs_to_py(BSON_Document({})) == {}
- assert bs_to_py(BSON_Document(
- {
- "array": BSON_Array([BSON_Document({}), BSON_Document({})]),
- "document": BSON_Document({"key": BSON_Array([])}),
- })) ==\
- {
- "array": [{}, {}],
- "document": {"key": []},
- }
-
- print("ok")
-
- ###################################
-
- print("deprecated quirks: ", end="")
-
- bio = BytesIO(b"\x05\x00\x00\x00\x02value")
- assert BSON_Binary.parse(bio) == BSON_Binary_Generic(b"value")
-
- with expected(BSON_ParsingError, "unknown type code 0x06 at offset 0"):
- print(BSON_Value.parse(b"\x06", BytesIO()))
-
- with expected(BSON_ParsingError, "unknown type code 0x0c at offset 0"):
- print(BSON_Value.parse(b"\x0c", BytesIO()))
-
- print("ok")
-
- ###################################
-
- print("specification samples: ", end="")
-
- assert serialize_to_bytes({"hello": "world"}) ==\
- b"\x16\x00\x00\x00\x02hello\x00\x06\x00\x00\x00world\x00\x00"
-
- assert parse_bytes(b"\x16\x00\x00\x00\x02hello\x00\x06\x00\x00\x00world\x00\x00") ==\
- {"hello": "world"}
-
- assert serialize_to_bytes({"BSON": ["awesome", 5.05, 1986]}) ==\
- b"1\x00\x00\x00\x04BSON\x00&\x00\x00\x00\x020\x00\x08\x00\x00\x00"\
- b"awesome\x00\x011\x00333333\x14@\x102\x00\xc2\x07\x00\x00\x00\x00"
-
- assert parse_bytes(b"1\x00\x00\x00\x04BSON\x00&\x00\x00\x00\x020\x00\x08\x00\x00\x00"
- b"awesome\x00\x011\x00333333\x14@\x102\x00\xc2\x07\x00\x00\x00\x00") ==\
- {"BSON": ["awesome", 5.05, 1986]}
-
- print("ok")
-
- ###################################
-
- print("single type values: ", end="")
-
- b = serialize_to_bytes({"double": 1.0})
- assert b == serialize_to_bytes({"double": BSON_Double(1.0)})
- assert parse_bytes(b) == {"double": 1.0}
-
- b = serialize_to_bytes({"string": u"абв"})
- assert b == serialize_to_bytes({"string": BSON_String(u"абв")})
- assert parse_bytes(b) == {"string": u"абв"}
-
- b = serialize_to_bytes({"boolean": True})
- assert b == serialize_to_bytes({"boolean": BSON_Boolean(True)})
- assert parse_bytes(b) == {"boolean": True}
-
- dt = datetime.now()
- b = serialize_to_bytes({"datetime": dt})
- assert b == serialize_to_bytes({"datetime": BSON_Datetime(dt)})
- r = parse_bytes(b)
- dt2 = r.pop("datetime")
- assert not r
- dt = dt.replace(microsecond=0)
- dt2 = dt2.replace(microsecond=0)
- assert dt == dt2
-
- b = serialize_to_bytes({"null": None})
- assert b == serialize_to_bytes({"null": BSON_Null(None)})
- assert parse_bytes(b) == {"null": None}
-
- b = serialize_to_bytes({"int32": 2 ** 31 - 1})
- assert b == serialize_to_bytes({"int32": BSON_Int32(2 ** 31 - 1)})
- assert parse_bytes(b) == {"int32": 2 ** 31 - 1}
-
- b = serialize_to_bytes({"int64": 2 ** 63 - 1})
- assert b == serialize_to_bytes({"int64": BSON_Int64(2 ** 63 - 1)})
- assert parse_bytes(b) == {"int64": 2 ** 63 - 1}
-
- bin = bytes([i for i in range(256)])
- b = serialize_to_bytes({"bytes": BytesIO(bin)})
- assert b == serialize_to_bytes({"bytes": BSON_Binary_Generic(bin)})
- assert parse_bytes(b) == {"bytes": bin if PY3 else BSON_Binary_Generic(bin)}
-
- b = serialize_to_bytes({"function": BSON_Binary_Function(b"function")})
- assert parse_bytes(b) == {"function": BSON_Binary_Function(b"function")}
-
- b = serialize_to_bytes({"uuid": BSON_Binary_UUID(b"uuid")})
- assert parse_bytes(b) == {"uuid": BSON_Binary_UUID(b"uuid")}
-
- b = serialize_to_bytes({"md5": BSON_Binary_MD5(b"md5")})
- assert parse_bytes(b) == {"md5": BSON_Binary_MD5(b"md5")}
-
- b = serialize_to_bytes({"userdefined": BSON_Binary_UserDefined(b"userdefined")})
- assert parse_bytes(b) == {"userdefined": BSON_Binary_UserDefined(b"userdefined")}
-
- b = serialize_to_bytes({"objectid": BSON_ObjectId(b"\xff" * 12)})
- assert parse_bytes(b) == {"objectid": BSON_ObjectId(b"\xff" * 12)._py_value()}
-
- b = serialize_to_bytes({"regex": BSON_Regex(("^$", "i"))})
- assert parse_bytes(b) == {"regex": BSON_Regex(("^$", "i"))}
-
- b = serialize_to_bytes({"javascript": BSON_JavaScript("var i = 1;")})
- assert parse_bytes(b) == {"javascript": BSON_JavaScript("var i = 1;")}
-
- b = serialize_to_bytes({"symbol": BSON_Symbol("class")})
- assert parse_bytes(b) == {"symbol": BSON_Symbol("class")}
-
- b = serialize_to_bytes(
- {"javascript_ws": BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"i": BSON_Int32(1)})))})
- assert parse_bytes(b) == {
- "javascript_ws": BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"i": BSON_Int32(1)})))}
-
- b = serialize_to_bytes({"timestamp": BSON_Timestamp(0)})
- assert parse_bytes(b) == {"timestamp": BSON_Timestamp(0)}
-
- b = serialize_to_bytes({"document": {}})
- assert parse_bytes(b) == {"document": {}}
-
- b = serialize_to_bytes({"array": []})
- assert parse_bytes(b) == {"array": []}
-
- print("ok")
-
- ###################################
-
- print("parsing errors: ", end="")
-
- with expected(BSON_ParsingError, "incorrect structure length at offset 4"):
- parse_bytes(b"\x00\x00\x00\x00")
-
- with expected(BSON_ParsingError, "unexpected end of stream"):
- parse_bytes(b"\x05\x00\x00\x00")
-
- with expected(BSON_ParsingError, "expected null terminated structure at offset 5"):
- parse_bytes(b"\x05\x00\x00\x00\xff")
-
- with expected(BSON_ParsingError, "unknown type code 0xff at offset 6"):
- parse_bytes(b"\x06\x00\x00\x00\xff\x00")
-
- with expected(BSON_ParsingError, "read beyond end of structure at offset 14"):
- parse_bytes(b"\x0e\x00\x00\x00\x02\x00\x04\x00\x00\x00foo\x00")
-
- with expected(BSON_ParsingError, "incorrect string length at offset 10"):
- parse_bytes(b"\x0a\x00\x00\x00\x02\x00\x00\x00\x00\x00")
-
- with expected(BSON_ParsingError, "expected null terminated string at offset 13"):
- parse_bytes(b"\x0d\x00\x00\x00\x02\x00\x03\x00\x00\x00foo")
-
- with expected(BSON_ParsingError, "incorrect structure length at offset 10"):
- parse_bytes(b"\x06\x00\x00\x00\x05\x00\x00\x00\x00\x00")
-
- with expected(BSON_ParsingError, "incorrect boolean value at offset 7"):
- parse_bytes(b"\x07\x00\x00\x00\x08\x00\xff")
-
- with expected(BSON_ParsingError, "incorrect structure length at offset 10"):
- parse_bytes(b"\x0a\x00\x00\x00\x0f\x00\x04\x00\x00\x00")
-
- with expected(BSON_ParsingError, "read beyond end of structure at offset 41"):
- parse_bytes(b"\x2a\x00\x00\x00\x0fjsws\x00\x1e\x00\x00\x00\x0b\x00\x00\x00"
- b"var i = j;\x00\x0c\x00\x00\x00\x10j\x00\n\x00\x00\x00\x00\x00")
-
- print("ok")
-
- ###################################
-
- print("all ok")
+#if __name__ == "__main__": # self-test
+#
+# from random import shuffle
+#
+# try:
+# import expected
+# except ImportError:
+# print("warning: module expected.py cannot be imported, exception tests are skipped")
+#
+# class expected:
+# def __init__(self, *args, **kwargs):
+# pass
+#
+# def __enter__(self):
+# pass
+#
+# def __exit__(self, t, v, tb):
+# return True
+# else:
+# from expected import expected
+#
+# ###################################
+#
+# print("self-testing module bson.py:")
+#
+# ###################################
+#
+# print("utilities: ", end="")
+#
+# assert cstrify("") == b"\x00"
+# assert cstrify("foo") == b"foo\x00"
+# assert cstrify(u"абв") == b"\xd0\xb0\xd0\xb1\xd0\xb2\x00"
+#
+# print("ok")
+#
+# ###################################
+#
+# print("exact byte sequences: ", end="")
+#
+# def test_type(t, v, b):
+# bio = BytesIO()
+# t(v).serialize(bio)
+# assert bio.getvalue() == b, bio.getvalue()
+# bio.seek(0, 0)
+# assert t.parse(bio).value == v
+# assert bio.tell() == len(bio.getvalue())
+#
+# test_type(BSON_Double, 0.0, b"\x00\x00\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Double, 1.0, b"\x00\x00\x00\x00\x00\x00\xf0?")
+# test_type(BSON_Double, -1.0, b"\x00\x00\x00\x00\x00\x00\xf0\xbf")
+# test_type(BSON_Double, 1e308, b"\xa0\xc8\xeb\x85\xf3\xcc\xe1\x7f")
+# test_type(BSON_Double, 1e-308, b"\xd2\xe8\x19x\xd60\x07\x00")
+# test_type(BSON_Double, -1e308, b"\xa0\xc8\xeb\x85\xf3\xcc\xe1\xff")
+# test_type(BSON_Double, -1e-308, b"\xd2\xe8\x19x\xd60\x07\x80")
+#
+# test_type(BSON_Int32, 0, b"\x00\x00\x00\x00")
+# test_type(BSON_Int32, 1, b"\x01\x00\x00\x00")
+# test_type(BSON_Int32, -1, b"\xff\xff\xff\xff")
+# test_type(BSON_Int32, 2 ** 31 - 1, b"\xff\xff\xff\x7f")
+# test_type(BSON_Int32, -2 ** 31, b"\x00\x00\x00\x80")
+#
+# test_type(BSON_Int64, 0, b"\x00\x00\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Int64, 1, b"\x01\x00\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Int64, -1, b"\xff\xff\xff\xff\xff\xff\xff\xff")
+# test_type(BSON_Int64, 2 ** 63 - 1, b"\xff\xff\xff\xff\xff\xff\xff\x7f")
+# test_type(BSON_Int64, -2 ** 63, b"\x00\x00\x00\x00\x00\x00\x00\x80")
+#
+# test_type(BSON_String, "", b"\x01\x00\x00\x00\x00")
+# test_type(BSON_String, "foo", b"\x04\x00\x00\x00foo\x00")
+# test_type(BSON_String, u"абв", b"\x07\x00\x00\x00\xd0\xb0\xd0\xb1\xd0\xb2\x00")
+#
+# test_type(BSON_Boolean, True, b"\x01")
+# test_type(BSON_Boolean, False, b"\x00")
+#
+# test_type(BSON_ObjectId, b"\x01\x02\x03\x04\x05\x06\x07\x08\x09\x10\x11\x12",
+# b"\x01\x02\x03\x04\x05\x06\x07\x08\x09\x10\x11\x12")
+#
+# test_type(BSON_Null, None, b"")
+#
+# test_type(BSON_JavaScript, "var i = 1;", b"\x0b\x00\x00\x00var i = 1;\x00")
+# test_type(BSON_JavaScript, u"/* комментарий */",
+# b"\x1d\x00\x00\x00/* \xd0\xba\xd0\xbe\xd0\xbc\xd0\xbc\xd0\xb5\xd0\xbd\xd1\x82\xd0\xb0\xd1\x80\xd0\xb8\xd0\xb9 */\x00")
+#
+# test_type(BSON_Timestamp, 1, b"\x01\x00\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Timestamp, -1, b"\xff\xff\xff\xff\xff\xff\xff\xff")
+# test_type(BSON_Timestamp, 2 ** 63 - 1, b"\xff\xff\xff\xff\xff\xff\xff\x7f")
+# test_type(BSON_Timestamp, -2 ** 63, b"\x00\x00\x00\x00\x00\x00\x00\x80")
+#
+# test_type(BSON_Datetime, datetime(1970, 1, 1, 0, 0, 0), b"\x00\x00\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Datetime, datetime(1970, 1, 1, 0, 0, 1), b"\xe8\x03\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_Datetime, datetime(2010, 12, 12, 15, 11, 43), b"\x98\x27\x23\xdb\x2c\x01\x00\x00")
+#
+# test_type(BSON_Symbol, "foo", b"\x04\x00\x00\x00foo\x00")
+# test_type(BSON_Symbol, u"символ",
+# b"\x0d\x00\x00\x00\xd1\x81\xd0\xb8\xd0\xbc\xd0\xb2\xd0\xbe\xd0\xbb\x00")
+#
+# test_type(BSON_JavaScriptWithScope, ("var i = j + 1;", BSON_Document({"j": BSON_Int32(0)})),
+# b"\x23\x00\x00\x00\x0f\x00\x00\x00var i = j + 1;\x00\x0c\x00\x00\x00\x10j\x00\x00\x00\x00\x00\x00")
+# test_type(BSON_JavaScriptWithScope, ("foo(s);", BSON_Document({"s": BSON_String(u"абв")})),
+# b"\x23\x00\x00\x00\x08\x00\x00\x00foo(s);\x00\x13\x00\x00\x00\x02s\x00\x07\x00\x00\x00\xd0\xb0\xd0\xb1\xd0\xb2\x00\x00")
+#
+# test_type(BSON_Regex, ("", ""), b"\x00\x00")
+# test_type(BSON_Regex, ("^$", "i"), b"^$\x00i\x00")
+#
+# test_type(BSON_Binary_Generic, b"value", b"\x05\x00\x00\x00\x00value")
+# test_type(BSON_Binary_Function, b"value", b"\x05\x00\x00\x00\x01value")
+# test_type(BSON_Binary_UUID, b"value", b"\x05\x00\x00\x00\x03value")
+# test_type(BSON_Binary_MD5, b"value", b"\x05\x00\x00\x00\x05value")
+# test_type(BSON_Binary_UserDefined, b"value", b"\x05\x00\x00\x00\x80value")
+#
+# test_type(BSON_Document, {}, b"\x05\x00\x00\x00\x00")
+# test_type(BSON_Document, {"foo": BSON_Null(None)}, b"\x0a\x00\x00\x00\x0afoo\x00\x00")
+#
+# print("ok")
+#
+# ###################################
+#
+# print("forward type mapping: ", end="")
+#
+# assert py_to_bs(1.0) == BSON_Double(1.0)
+# assert py_to_bs("foo") == BSON_String("foo")
+# assert py_to_bs(True) == BSON_Boolean(True)
+# dt = datetime.now();
+# assert py_to_bs(dt) == BSON_Datetime(dt)
+# assert py_to_bs(None) == BSON_Null(None)
+# assert py_to_bs(2 ** 31 - 1) == BSON_Int32(2 ** 31 - 1)
+# assert py_to_bs(2 ** 31) == BSON_Int64(2 ** 31)
+#
+# def py_to_bs_same(v):
+# return py_to_bs(v) is v
+#
+# assert py_to_bs_same(BSON_ObjectId(b"\x00" * 12))
+# assert py_to_bs_same(BSON_Regex(("^$", "i")))
+# assert py_to_bs_same(BSON_JavaScript("var i = 1;"))
+# assert py_to_bs_same(BSON_Symbol("class"))
+# assert py_to_bs_same(BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"j": BSON_Int32(10)}))))
+# assert py_to_bs_same(BSON_Timestamp(0x7fffffffffffffff))
+#
+# assert py_to_bs([]) == BSON_Array([])
+# assert py_to_bs(
+# [
+# b"implicit" if PY3 else BytesIO("implicit"),
+# BSON_Binary_Generic(b"generic"),
+# BSON_Binary_Function(b"function"),
+# BSON_Binary_UUID(b"uuid"),
+# BSON_Binary_MD5(b"md5"),
+# BSON_Binary_UserDefined(b"userdefined"),
+# ]) ==\
+# BSON_Array(
+# [
+# BSON_Binary_Generic(b"implicit"),
+# BSON_Binary_Generic(b"generic"),
+# BSON_Binary_Function(b"function"),
+# BSON_Binary_UUID(b"uuid"),
+# BSON_Binary_MD5(b"md5"),
+# BSON_Binary_UserDefined(b"userdefined"),
+# ])
+#
+# assert py_to_bs({}) == BSON_Document({})
+# assert py_to_bs(
+# {
+# "array": [{}, {}],
+# "document": {"key": []},
+# }) ==\
+# BSON_Document(
+# {
+# "array": BSON_Array([BSON_Document({}), BSON_Document({})]),
+# "document": BSON_Document({"key": BSON_Array([])}),
+# })
+#
+# with expected(BSON_ConversionError, "cannot implicitly convert int value 9223372036854775808"):
+# py_to_bs(2 ** 63)
+#
+# with expected(BSON_ConversionError, "cannot implicitly convert tuple value \(\)"):
+# py_to_bs(())
+#
+# print("ok")
+#
+# ###################################
+#
+# print("document key ordering: ", end="")
+#
+# class Foo(int):
+# pass
+#
+# ks1 = list(range(10000))
+# shuffle(ks1)
+# d1 = odict((str(k), Foo(k)) for k in ks1)
+# b = serialize_to_bytes(d1)
+# d2 = parse_bytes(b)
+# ks2 = list(d2.values())
+# assert ks2 == ks1
+#
+# print("ok")
+#
+# ###################################
+#
+# print("reverse type mapping: ", end="")
+#
+# assert bs_to_py(BSON_Double(1.0)) == 1.0
+# assert bs_to_py(BSON_String("foo")) == "foo"
+# assert bs_to_py(BSON_Boolean(True)) == True
+# dt = datetime.now()
+# assert bs_to_py(BSON_Datetime(dt)) == dt
+# assert bs_to_py(BSON_Null(None)) == None
+# assert bs_to_py(BSON_Int32(2 ** 31 - 1)) == 2 ** 31 - 1
+# assert bs_to_py(BSON_Int64(2 ** 31)) == 2 ** 31
+#
+# def bs_to_py_same(v):
+# return bs_to_py(v) is v
+#
+# assert type(BSON_ObjectId(b"\x00" * 12)._py_value()).__name__ == 'str' if PY3 else 'unicode'
+# assert bs_to_py_same(BSON_Regex(("^$", "")))
+# assert bs_to_py_same(BSON_JavaScript("var i = 0;"))
+# assert bs_to_py_same(BSON_Symbol("class"))
+# assert bs_to_py_same(BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"j": -1}))))
+# assert bs_to_py_same(BSON_Timestamp(0x7fffffffffffffff))
+#
+# assert bs_to_py(BSON_Array([])) == []
+#
+# assert bs_to_py(BSON_Array(
+# [
+# BSON_Binary_Generic(b"generic"),
+# BSON_Binary_Function(b"function"),
+# BSON_Binary_UUID(b"uuid"),
+# BSON_Binary_MD5(b"md5"),
+# BSON_Binary_UserDefined(b"userdefined"),
+# ])) ==\
+# [
+# b"generic" if PY3 else BSON_Binary_Generic(b"generic"),
+# BSON_Binary_Function(b"function"),
+# BSON_Binary_UUID(b"uuid"),
+# BSON_Binary_MD5(b"md5"),
+# BSON_Binary_UserDefined(b"userdefined"),
+# ]
+#
+# assert bs_to_py(BSON_Document({})) == {}
+# assert bs_to_py(BSON_Document(
+# {
+# "array": BSON_Array([BSON_Document({}), BSON_Document({})]),
+# "document": BSON_Document({"key": BSON_Array([])}),
+# })) ==\
+# {
+# "array": [{}, {}],
+# "document": {"key": []},
+# }
+#
+# print("ok")
+#
+# ###################################
+#
+# print("deprecated quirks: ", end="")
+#
+# bio = BytesIO(b"\x05\x00\x00\x00\x02value")
+# assert BSON_Binary.parse(bio) == BSON_Binary_Generic(b"value")
+#
+# with expected(BSON_ParsingError, "unknown type code 0x06 at offset 0"):
+# print(BSON_Value.parse(b"\x06", BytesIO()))
+#
+# with expected(BSON_ParsingError, "unknown type code 0x0c at offset 0"):
+# print(BSON_Value.parse(b"\x0c", BytesIO()))
+#
+# print("ok")
+#
+# ###################################
+#
+# print("specification samples: ", end="")
+#
+# assert serialize_to_bytes({"hello": "world"}) ==\
+# b"\x16\x00\x00\x00\x02hello\x00\x06\x00\x00\x00world\x00\x00"
+#
+# assert parse_bytes(b"\x16\x00\x00\x00\x02hello\x00\x06\x00\x00\x00world\x00\x00") ==\
+# {"hello": "world"}
+#
+# assert serialize_to_bytes({"BSON": ["awesome", 5.05, 1986]}) ==\
+# b"1\x00\x00\x00\x04BSON\x00&\x00\x00\x00\x020\x00\x08\x00\x00\x00"\
+# b"awesome\x00\x011\x00333333\x14@\x102\x00\xc2\x07\x00\x00\x00\x00"
+#
+# assert parse_bytes(b"1\x00\x00\x00\x04BSON\x00&\x00\x00\x00\x020\x00\x08\x00\x00\x00"
+# b"awesome\x00\x011\x00333333\x14@\x102\x00\xc2\x07\x00\x00\x00\x00") ==\
+# {"BSON": ["awesome", 5.05, 1986]}
+#
+# print("ok")
+#
+# ###################################
+#
+# print("single type values: ", end="")
+#
+# b = serialize_to_bytes({"double": 1.0})
+# assert b == serialize_to_bytes({"double": BSON_Double(1.0)})
+# assert parse_bytes(b) == {"double": 1.0}
+#
+# b = serialize_to_bytes({"string": u"абв"})
+# assert b == serialize_to_bytes({"string": BSON_String(u"абв")})
+# assert parse_bytes(b) == {"string": u"абв"}
+#
+# b = serialize_to_bytes({"boolean": True})
+# assert b == serialize_to_bytes({"boolean": BSON_Boolean(True)})
+# assert parse_bytes(b) == {"boolean": True}
+#
+# dt = datetime.now()
+# b = serialize_to_bytes({"datetime": dt})
+# assert b == serialize_to_bytes({"datetime": BSON_Datetime(dt)})
+# r = parse_bytes(b)
+# dt2 = r.pop("datetime")
+# assert not r
+# dt = dt.replace(microsecond=0)
+# dt2 = dt2.replace(microsecond=0)
+# assert dt == dt2
+#
+# b = serialize_to_bytes({"null": None})
+# assert b == serialize_to_bytes({"null": BSON_Null(None)})
+# assert parse_bytes(b) == {"null": None}
+#
+# b = serialize_to_bytes({"int32": 2 ** 31 - 1})
+# assert b == serialize_to_bytes({"int32": BSON_Int32(2 ** 31 - 1)})
+# assert parse_bytes(b) == {"int32": 2 ** 31 - 1}
+#
+# b = serialize_to_bytes({"int64": 2 ** 63 - 1})
+# assert b == serialize_to_bytes({"int64": BSON_Int64(2 ** 63 - 1)})
+# assert parse_bytes(b) == {"int64": 2 ** 63 - 1}
+#
+# bin = bytes([i for i in range(256)])
+# b = serialize_to_bytes({"bytes": BytesIO(bin)})
+# assert b == serialize_to_bytes({"bytes": BSON_Binary_Generic(bin)})
+# assert parse_bytes(b) == {"bytes": bin if PY3 else BSON_Binary_Generic(bin)}
+#
+# b = serialize_to_bytes({"function": BSON_Binary_Function(b"function")})
+# assert parse_bytes(b) == {"function": BSON_Binary_Function(b"function")}
+#
+# b = serialize_to_bytes({"uuid": BSON_Binary_UUID(b"uuid")})
+# assert parse_bytes(b) == {"uuid": BSON_Binary_UUID(b"uuid")}
+#
+# b = serialize_to_bytes({"md5": BSON_Binary_MD5(b"md5")})
+# assert parse_bytes(b) == {"md5": BSON_Binary_MD5(b"md5")}
+#
+# b = serialize_to_bytes({"userdefined": BSON_Binary_UserDefined(b"userdefined")})
+# assert parse_bytes(b) == {"userdefined": BSON_Binary_UserDefined(b"userdefined")}
+#
+# b = serialize_to_bytes({"objectid": BSON_ObjectId(b"\xff" * 12)})
+# assert parse_bytes(b) == {"objectid": BSON_ObjectId(b"\xff" * 12)._py_value()}
+#
+# b = serialize_to_bytes({"regex": BSON_Regex(("^$", "i"))})
+# assert parse_bytes(b) == {"regex": BSON_Regex(("^$", "i"))}
+#
+# b = serialize_to_bytes({"javascript": BSON_JavaScript("var i = 1;")})
+# assert parse_bytes(b) == {"javascript": BSON_JavaScript("var i = 1;")}
+#
+# b = serialize_to_bytes({"symbol": BSON_Symbol("class")})
+# assert parse_bytes(b) == {"symbol": BSON_Symbol("class")}
+#
+# b = serialize_to_bytes(
+# {"javascript_ws": BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"i": BSON_Int32(1)})))})
+# assert parse_bytes(b) == {
+# "javascript_ws": BSON_JavaScriptWithScope(("var i = j;", BSON_Document({"i": BSON_Int32(1)})))}
+#
+# b = serialize_to_bytes({"timestamp": BSON_Timestamp(0)})
+# assert parse_bytes(b) == {"timestamp": BSON_Timestamp(0)}
+#
+# b = serialize_to_bytes({"document": {}})
+# assert parse_bytes(b) == {"document": {}}
+#
+# b = serialize_to_bytes({"array": []})
+# assert parse_bytes(b) == {"array": []}
+#
+# print("ok")
+#
+# ###################################
+#
+# print("parsing errors: ", end="")
+#
+# with expected(BSON_ParsingError, "incorrect structure length at offset 4"):
+# parse_bytes(b"\x00\x00\x00\x00")
+#
+# with expected(BSON_ParsingError, "unexpected end of stream"):
+# parse_bytes(b"\x05\x00\x00\x00")
+#
+# with expected(BSON_ParsingError, "expected null terminated structure at offset 5"):
+# parse_bytes(b"\x05\x00\x00\x00\xff")
+#
+# with expected(BSON_ParsingError, "unknown type code 0xff at offset 6"):
+# parse_bytes(b"\x06\x00\x00\x00\xff\x00")
+#
+# with expected(BSON_ParsingError, "read beyond end of structure at offset 14"):
+# parse_bytes(b"\x0e\x00\x00\x00\x02\x00\x04\x00\x00\x00foo\x00")
+#
+# with expected(BSON_ParsingError, "incorrect string length at offset 10"):
+# parse_bytes(b"\x0a\x00\x00\x00\x02\x00\x00\x00\x00\x00")
+#
+# with expected(BSON_ParsingError, "expected null terminated string at offset 13"):
+# parse_bytes(b"\x0d\x00\x00\x00\x02\x00\x03\x00\x00\x00foo")
+#
+# with expected(BSON_ParsingError, "incorrect structure length at offset 10"):
+# parse_bytes(b"\x06\x00\x00\x00\x05\x00\x00\x00\x00\x00")
+#
+# with expected(BSON_ParsingError, "incorrect boolean value at offset 7"):
+# parse_bytes(b"\x07\x00\x00\x00\x08\x00\xff")
+#
+# with expected(BSON_ParsingError, "incorrect structure length at offset 10"):
+# parse_bytes(b"\x0a\x00\x00\x00\x0f\x00\x04\x00\x00\x00")
+#
+# with expected(BSON_ParsingError, "read beyond end of structure at offset 41"):
+# parse_bytes(b"\x2a\x00\x00\x00\x0fjsws\x00\x1e\x00\x00\x00\x0b\x00\x00\x00"
+# b"var i = j;\x00\x0c\x00\x00\x00\x10j\x00\n\x00\x00\x00\x00\x00")
+#
+# print("ok")
+#
+# ###################################
+#
+# print("all ok")
# EOF
projects / platform / upstream / ejdb.git / commitdiff