zlmdb 25.10.1__cp314-cp314-manylinux_2_34_x86_64.whl

zlmdb/tests/orm/_schema_py2.py

@@ -0,0 +1,250 @@
+###############################################################################
+#
+# The MIT License (MIT)
+#
+# Copyright (c) typedef int GmbH
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+###############################################################################
+
+import random
+import uuid
+import datetime
+import zlmdb
+
+
+class User(object):
+    def __init__(self):
+        self.oid = None
+        self.name = None
+        self.authid = None
+        self.uuid = None
+        self.email = None
+        self.birthday = None
+        self.is_friendly = None
+        self.tags = None
+        self.ratings = {}
+        self.friends = []
+        self.referred_by = None
+        self.realm_oid = None
+        self.icecream = None
+        self.mrealm = None
+        self.mrealm_notnull = None
+
+    def __eq__(self, other):
+        if not isinstance(other, self.__class__):
+            return False
+        if other.oid != self.oid:
+            return False
+        if other.name != self.name:
+            return False
+        if other.authid != self.authid:
+            return False
+        if other.uuid != self.uuid:
+            return False
+        if other.email != self.email:
+            return False
+        if other.birthday != self.birthday:
+            return False
+        if other.is_friendly != self.is_friendly:
+            return False
+        if (self.tags and not other.tags) or (not self.tags and other.tags):
+            return False
+        if other.realm_oid != self.realm_oid:
+            return False
+        if other.icecream != self.icecream:
+            return False
+        if other.mrealm != self.mrealm:
+            return False
+        if other.mrealm_notnull != self.mrealm_notnull:
+            return False
+        return True
+
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
+    def marshal(self):
+        obj = {
+            "oid": self.oid,
+            "name": self.name,
+            "authid": self.authid,
+            "uuid": self.uuid.hex if self.uuid else None,
+            "email": self.email,
+            "birthday": {
+                "year": self.birthday.year if self.birthday else None,
+                "month": self.birthday.month if self.birthday else None,
+                "day": self.birthday.day if self.birthday else None,
+            },
+            "is_friendly": self.is_friendly,
+            "tags": self.tags,
+            "ratings": self.ratings,
+            "friends": self.friends,
+            "referred_by": self.referred_by,
+            "realm_oid": self.realm_oid,
+            "icecream": self.icecream,
+            "mrealm": self.mrealm.hex if self.mrealm else None,
+            "mrealm_notnull": self.mrealm_notnull.hex if self.mrealm_notnull else None,
+        }
+        return obj
+
+    @staticmethod
+    def parse(obj):
+        user = User()
+        user.oid = obj.get("oid", None)
+        user.name = obj.get("name", None)
+        user.authid = obj.get("authid", None)
+        if "uuid" in obj:
+            user.uuid = uuid.UUID(hex=obj["uuid"])
+        user.email = obj.get("email", None)
+        if "birthday" in obj:
+            b = obj["birthday"]
+            user.birthday = datetime.date(b.year, b.month, b.day)
+        user.is_friendly = obj.get("is_friendly", None)
+        user.tags = obj.get("tags", None)
+        user.ratings = obj.get("ratings", {})
+        user.friends = obj.get("friends", [])
+        user.referred_by = obj.get("referred_by", None)
+        user.realm_oid = obj.get("realm_oid", None)
+        user.icecream = obj.get("icecream", None)
+        if "mrealm" in obj and obj["mrealm"]:
+            user.mrealm = uuid.UUID(hex=obj["mrealm"])
+        if "mrealm_notnull" in obj and obj["mrealm_notnull"]:
+            user.mrealm_notnull = uuid.UUID(hex=obj["mrealm_notnull"])
+        return user
+
+    @staticmethod
+    def create_test_user(oid=None, realm_oid=None):
+        user = User()
+        if oid is not None:
+            user.oid = oid
+        else:
+            user.oid = random.randint(0, 9007199254740992)
+        user.name = "Test {}".format(user.oid)
+        user.authid = "test-{}".format(user.oid)
+        user.uuid = uuid.uuid4()
+        user.email = "{}@example.com".format(user.authid)
+        user.birthday = datetime.date(1950, 12, 24)
+        user.is_friendly = True
+        user.tags = ["geek", "sudoko", "yellow"]
+        for j in range(10):
+            user.ratings["test-rating-{}".format(j)] = random.random()
+        user.friends = [random.randint(0, 9007199254740992) for _ in range(10)]
+        user.referred_by = random.randint(0, 9007199254740992)
+        if realm_oid is not None:
+            user.realm_oid = realm_oid
+        else:
+            user.realm_oid = random.randint(0, 9007199254740992)
+        user.icecream = random.choice(["vanilla", "lemon", "strawberry"])
+        user.mrealm = uuid.uuid4()
+        user.mrealm_notnull = uuid.uuid4()
+        return user
+
+
+class Schema1(zlmdb.Schema):
+    def __init__(self):
+        super(Schema1, self).__init__()
+
+        self.tab_uuid_str = zlmdb.MapUuidString(slot=1)
+        self.tab_uuid_oid = zlmdb.MapUuidOid(slot=2)
+        self.tab_uuid_uuid = zlmdb.MapUuidUuid(slot=3)
+        self.tab_str_str = zlmdb.MapStringString(slot=4)
+        self.tab_str_oid = zlmdb.MapStringOid(slot=5)
+        self.tab_str_uuid = zlmdb.MapStringUuid(slot=6)
+        self.tab_oid_str = zlmdb.MapOidString(slot=7)
+        self.tab_oid_oid = zlmdb.MapOidOid(slot=8)
+        self.tab_oid_uuid = zlmdb.MapOidUuid(slot=9)
+        self.tab_uuid_json = zlmdb.MapUuidJson(
+            slot=10, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_uuid_cbor = zlmdb.MapUuidCbor(
+            slot=11, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_uuid_pickle = zlmdb.MapUuidPickle(slot=12)
+        self.tab_str_json = zlmdb.MapStringJson(
+            slot=20, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_str_cbor = zlmdb.MapStringCbor(
+            slot=21, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_str_pickle = zlmdb.MapStringPickle(slot=22)
+        self.tab_oid_json = zlmdb.MapOidJson(
+            slot=30, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_oid_cbor = zlmdb.MapOidCbor(
+            slot=31, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_oid_pickle = zlmdb.MapOidPickle(slot=32)
+
+
+class Schema2(zlmdb.Schema):
+    def __init__(self):
+        super(Schema2, self).__init__()
+        self.users = zlmdb.MapOidPickle(1)
+
+
+class Schema3(zlmdb.Schema):
+    def __init__(self):
+        super(Schema3, self).__init__()
+        self.users = zlmdb.MapStringPickle(1)
+
+
+class Schema4(zlmdb.Schema):
+    def __init__(self):
+        super(Schema4, self).__init__()
+        self.users = zlmdb.MapOidPickle(1)
+
+        self.idx_users_by_authid = zlmdb.MapStringOid(2)
+        self.users.attach_index(
+            "idx1", self.idx_users_by_authid, lambda user: user.authid, nullable=False
+        )
+
+        self.idx_users_by_email = zlmdb.MapStringOid(3)
+        self.users.attach_index(
+            "idx2", self.idx_users_by_email, lambda user: user.email, nullable=True
+        )
+
+        self.idx_users_by_realm = zlmdb.MapOidOidOid(4)
+        self.users.attach_index(
+            "idx3", self.idx_users_by_realm, lambda user: (user.realm_oid, user.oid)
+        )
+
+        self.idx_users_by_icecream = zlmdb.MapStringOidOid(5)
+        self.users.attach_index(
+            "idx4",
+            self.idx_users_by_icecream,
+            lambda user: (user.icecream, user.oid),
+            nullable=False,
+        )
+
+        self.idx_users_by_mrealm_authid = zlmdb.MapUuidStringOid(6)
+        self.users.attach_index(
+            "idx5",
+            self.idx_users_by_mrealm_authid,
+            lambda user: (user.mrealm, user.authid),
+            nullable=True,
+        )
+
+        self.idx_users_by_mrealm_notnull_authid = zlmdb.MapUuidStringOid(7)
+        self.users.attach_index(
+            "idx6",
+            self.idx_users_by_mrealm_notnull_authid,
+            lambda user: (user.mrealm_notnull, user.authid),
+            nullable=False,
+        )

zlmdb/tests/orm/_schema_py3.py

@@ -0,0 +1,307 @@
+###############################################################################
+#
+# The MIT License (MIT)
+#
+# Copyright (c) typedef int GmbH
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+###############################################################################
+
+import random
+import uuid
+import datetime
+from typing import Optional, List, Dict
+
+import zlmdb
+
+
+class User(object):
+    oid: int
+    name: str
+    authid: str
+    uuid: uuid.UUID
+    email: str
+    birthday: datetime.date
+    is_friendly: bool
+    tags: Optional[List[str]]
+    ratings: Dict[str, float]
+    friends: List[int]
+    referred_by: int
+    realm_oid: int
+    icecream: str
+    mrealm: uuid.UUID  # type:ignore
+    mrealm_notnull: uuid.UUID  # type:ignore
+
+    def __init__(self):
+        self.oid = None
+        self.name = None
+        self.authid = None
+        self.uuid = None
+        self.email = None
+        self.birthday = None
+        self.is_friendly = None
+        self.tags = None
+        self.ratings = {}
+        self.friends = []
+        self.referred_by = None
+        self.realm_oid = None
+        self.icecream = None
+        self.mrealm = None
+        self.mrealm_notnull = None
+
+    def __eq__(self, other):
+        if not isinstance(other, self.__class__):
+            return False
+        if other.oid != self.oid:
+            return False
+        if other.name != self.name:
+            return False
+        if other.authid != self.authid:
+            return False
+        if other.uuid != self.uuid:
+            return False
+        if other.email != self.email:
+            return False
+        if other.birthday != self.birthday:
+            return False
+        if other.is_friendly != self.is_friendly:
+            return False
+        if (self.tags and not other.tags) or (not self.tags and other.tags):
+            return False
+        if other.realm_oid != self.realm_oid:
+            return False
+        if other.icecream != self.icecream:
+            return False
+        if other.mrealm != self.mrealm:
+            return False
+        if other.mrealm_notnull != self.mrealm_notnull:
+            return False
+        return True
+
+    def __ne__(self, other):
+        return not self.__eq__(other)
+
+    def marshal(self):
+        obj = {
+            "oid": self.oid,
+            "name": self.name,
+            "authid": self.authid,
+            "uuid": self.uuid.hex if self.uuid else None,
+            "email": self.email,
+            "birthday": {
+                "year": self.birthday.year if self.birthday else None,
+                "month": self.birthday.month if self.birthday else None,
+                "day": self.birthday.day if self.birthday else None,
+            },
+            "is_friendly": self.is_friendly,
+            "tags": self.tags,
+            "ratings": self.ratings,
+            "friends": self.friends,
+            "referred_by": self.referred_by,
+            "realm_oid": self.realm_oid,
+            "icecream": self.icecream,
+            "mrealm": self.mrealm.hex if self.mrealm else None,
+            "mrealm_notnull": self.mrealm_notnull.hex if self.mrealm_notnull else None,
+        }
+        return obj
+
+    @staticmethod
+    def parse(obj):
+        user = User()
+        user.oid = obj.get("oid", None)
+        user.name = obj.get("name", None)
+        user.authid = obj.get("authid", None)
+        if "uuid" in obj:
+            user.uuid = uuid.UUID(hex=obj["uuid"])
+        user.email = obj.get("email", None)
+        if "birthday" in obj:
+            b = obj["birthday"]
+            user.birthday = datetime.date(b.year, b.month, b.day)
+        user.is_friendly = obj.get("is_friendly", None)
+        user.tags = obj.get("tags", None)
+        user.ratings = obj.get("ratings", {})
+        user.friends = obj.get("friends", [])
+        user.referred_by = obj.get("referred_by", None)
+        user.realm_oid = obj.get("realm_oid", None)
+        user.icecream = obj.get("icecream", None)
+        if "mrealm" in obj and obj["mrealm"]:
+            user.mrealm = uuid.UUID(hex=obj["mrealm"])
+        if "mrealm_notnull" in obj and obj["mrealm_notnull"]:
+            user.mrealm_notnull = uuid.UUID(hex=obj["mrealm_notnull"])
+        return user
+
+    @staticmethod
+    def create_test_user(oid=None, realm_oid=None):
+        user = User()
+        if oid is not None:
+            user.oid = oid
+        else:
+            user.oid = random.randint(0, 9007199254740992)
+        user.name = "Test {}".format(user.oid)
+        user.authid = "test-{}".format(user.oid)
+        user.uuid = uuid.uuid4()
+        user.email = "{}@example.com".format(user.authid)
+        user.birthday = datetime.date(1950, 12, 24)
+        user.is_friendly = True
+        user.tags = ["geek", "sudoko", "yellow"]
+        for j in range(10):
+            user.ratings["test-rating-{}".format(j)] = 1 / (
+                j + 1
+            )  # round(random.random(), 3)
+        user.friends = [random.randint(0, 9007199254740992) for _ in range(10)]
+        user.referred_by = random.randint(0, 9007199254740992)
+        if realm_oid is not None:
+            user.realm_oid = realm_oid
+        else:
+            user.realm_oid = random.randint(0, 9007199254740992)
+        user.icecream = random.choice(["vanilla", "lemon", "strawberry"])
+        user.mrealm = uuid.uuid4()
+        user.mrealm_notnull = uuid.uuid4()
+        return user
+
+
+class Schema1(zlmdb.Schema):
+    tab_uuid_str: zlmdb.MapUuidString
+    tab_uuid_oid: zlmdb.MapUuidOid
+    tab_uuid_uuid: zlmdb.MapUuidUuid
+    tab_str_str: zlmdb.MapStringString
+    tab_str_oid: zlmdb.MapStringOid
+    tab_str_uuid: zlmdb.MapStringUuid
+    tab_oid_str: zlmdb.MapOidString
+    tab_oid_oid: zlmdb.MapOidOid
+    tab_oid_uuid: zlmdb.MapOidUuid
+    tab_uuid_json: zlmdb.MapUuidJson
+    tab_uuid_cbor: zlmdb.MapUuidCbor
+    tab_uuid_pickle: zlmdb.MapUuidPickle
+    tab_str_json: zlmdb.MapStringJson
+    tab_str_cbor: zlmdb.MapStringCbor
+    tab_str_pickle: zlmdb.MapStringPickle
+    tab_oid_json: zlmdb.MapOidJson
+    tab_oid_cbor: zlmdb.MapOidCbor
+    tab_oid_pickle: zlmdb.MapOidPickle
+
+    def __init__(self):
+        self.tab_uuid_str = zlmdb.MapUuidString(slot=1)
+        self.tab_uuid_oid = zlmdb.MapUuidOid(slot=2)
+        self.tab_uuid_uuid = zlmdb.MapUuidUuid(slot=3)
+        self.tab_str_str = zlmdb.MapStringString(slot=4)
+        self.tab_str_oid = zlmdb.MapStringOid(slot=5)
+        self.tab_str_uuid = zlmdb.MapStringUuid(slot=6)
+        self.tab_oid_str = zlmdb.MapOidString(slot=7)
+        self.tab_oid_oid = zlmdb.MapOidOid(slot=8)
+        self.tab_oid_uuid = zlmdb.MapOidUuid(slot=9)
+        self.tab_uuid_json = zlmdb.MapUuidJson(
+            slot=10, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_uuid_cbor = zlmdb.MapUuidCbor(
+            slot=11, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_uuid_pickle = zlmdb.MapUuidPickle(slot=12)
+        self.tab_str_json = zlmdb.MapStringJson(
+            slot=20, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_str_cbor = zlmdb.MapStringCbor(
+            slot=21, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_str_pickle = zlmdb.MapStringPickle(slot=22)
+        self.tab_oid_json = zlmdb.MapOidJson(
+            slot=30, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_oid_cbor = zlmdb.MapOidCbor(
+            slot=31, marshal=(lambda o: o.marshal()), unmarshal=User.parse
+        )
+        self.tab_oid_pickle = zlmdb.MapOidPickle(slot=32)
+
+
+class Schema2(zlmdb.Schema):
+    users: zlmdb.MapOidPickle
+
+    def __init__(self):
+        self.users = zlmdb.MapOidPickle(1)
+
+
+class Schema3(zlmdb.Schema):
+    users: zlmdb.MapStringPickle
+
+    def __init__(self):
+        self.users = zlmdb.MapStringPickle(1)
+
+
+class Schema4(zlmdb.Schema):
+    users: zlmdb.MapOidPickle
+
+    idx_users_by_authid: zlmdb.MapStringOid
+
+    idx_users_by_email: zlmdb.MapStringOid
+
+    idx_users_by_realm: zlmdb.MapOidOidOid
+
+    idx_users_by_icecream: zlmdb.MapStringOidOid
+
+    idx_users_by_mrealm_authid: zlmdb.MapUuidStringOid
+
+    idx_users_by_mrealm_authid_notnull: zlmdb.MapUuidStringOid
+
+    def __init__(self):
+        super(Schema4, self).__init__()
+
+        self.users = zlmdb.MapOidPickle(1)
+
+        self.idx_users_by_authid = zlmdb.MapStringOid(2)
+        self.users.attach_index(
+            "idx1", self.idx_users_by_authid, lambda user: user.authid, nullable=False
+        )
+
+        self.idx_users_by_email = zlmdb.MapStringOid(3)
+        self.users.attach_index(
+            "idx2", self.idx_users_by_email, lambda user: user.email, nullable=True
+        )
+
+        self.idx_users_by_realm = zlmdb.MapOidOidOid(4)
+        self.users.attach_index(
+            "idx3",
+            self.idx_users_by_realm,
+            lambda user: (user.realm_oid, user.oid),
+            nullable=False,
+        )
+
+        self.idx_users_by_icecream = zlmdb.MapStringOidOid(5)
+        self.users.attach_index(
+            "idx4",
+            self.idx_users_by_icecream,
+            lambda user: (user.icecream, user.oid),
+            nullable=False,
+        )
+
+        self.idx_users_by_mrealm_authid = zlmdb.MapUuidStringOid(6)
+        self.users.attach_index(
+            "idx5",
+            self.idx_users_by_mrealm_authid,
+            lambda user: (user.mrealm, user.authid),
+            nullable=True,
+        )
+
+        self.idx_users_by_mrealm_notnull_authid = zlmdb.MapUuidStringOid(7)
+        self.users.attach_index(
+            "idx6",
+            self.idx_users_by_mrealm_notnull_authid,
+            lambda user: (user.mrealm_notnull, user.authid),
+            nullable=False,
+        )

zlmdb/tests/orm/_test_flatbuffers.py

@@ -0,0 +1,144 @@
+###############################################################################
+#
+# The MIT License (MIT)
+#
+# Copyright (c) typedef int GmbH
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+#
+###############################################################################
+
+import os
+import sys
+import random
+import logging
+
+import txaio
+
+txaio.use_twisted()
+
+try:
+    from tempfile import TemporaryDirectory
+except ImportError:
+    from backports.tempfile import TemporaryDirectory  # type:ignore
+
+import zlmdb  # noqa
+
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from _schema_fbs import User  # noqa
+
+
+class UsersSchema(zlmdb.Schema):
+    def __init__(self):
+        self.tab_oid_fbs = zlmdb.MapOidFlatBuffers(1, build=User.build, cast=User.cast)
+
+
+def test_pmap_flatbuffers_values():
+    with TemporaryDirectory() as dbpath:
+        logging.info("Using temporary directory {} for database".format(dbpath))
+
+        schema = UsersSchema()
+
+        with zlmdb.Database(dbpath) as db:
+            N = 100
+            stats = zlmdb.TransactionStats()
+
+            with db.begin(write=True, stats=stats) as txn:
+                for i in range(N):
+                    user = User.create_test_user()
+                    schema.tab_oid_fbs[txn, user.oid] = user
+
+            assert stats.puts == N
+            assert stats.dels == 0
+            stats.reset()
+
+            with db.begin() as txn:
+                cnt = schema.tab_oid_fbs.count(txn)
+
+            assert cnt == N
+
+
+def test_pmap_flatbuffers_count():
+    with TemporaryDirectory() as dbpath:
+        logging.info("Using temporary directory {} for database".format(dbpath))
+
+        schema = UsersSchema()
+
+        # max DB size is 100 MB
+        with zlmdb.Database(dbpath, maxsize=100 * (2**20)) as db:
+            oids = set()
+            oid_to_referred_by = {}
+
+            stats = zlmdb.TransactionStats()
+
+            # number of transactions
+            M = 5
+
+            # number of insert rows per transaction
+            N = 10000
+            for j in range(M):
+                with db.begin(write=True, stats=stats) as txn:
+                    for i in range(N):
+                        user = User.create_test_user()
+                        schema.tab_oid_fbs[txn, user.oid] = user
+                        oids.add(user.oid)
+                        oid_to_referred_by[user.oid] = user.referred_by
+
+                assert stats.puts == N
+                assert stats.dels == 0
+                duration_ns = stats.duration
+                duration_ms = int(duration_ns / 1000000.0)
+                rows_per_sec = int(
+                    round(float(stats.puts + stats.dels) * 1000.0 / float(duration_ms))
+                )
+                logging.info(
+                    "Transaction ended: puts={} / dels={} rows in {} ms, {} rows/sec".format(
+                        stats.puts, stats.dels, duration_ms, rows_per_sec
+                    )
+                )
+
+                stats.reset()
+
+            # count all rows
+            with db.begin() as txn:
+                cnt = schema.tab_oid_fbs.count(txn)
+
+            assert cnt == N * M
+
+            # retrieve
+            with db.begin() as txn:
+                for j in range(5):
+                    started = zlmdb.walltime()
+                    M = 100
+                    for i in range(M):
+                        for oid in random.sample(oids, N):
+                            user = schema.tab_oid_fbs[txn, oid]
+                            assert user
+                            assert user.referred_by == oid_to_referred_by.get(oid, None)
+                    duration_ns = zlmdb.walltime() - started
+                    duration_ms = int(duration_ns / 1000000.0)
+                    rows_per_sec = int(
+                        round(float(M * N) * 1000.0 / float(duration_ms))
+                    )
+                    logging.info(
+                        "Transaction ended: {} rows read in {} ms, {} rows/sec".format(
+                            M * N, duration_ms, rows_per_sec
+                        )
+                    )