fix(bson): missing top level BSON class (#6)

s/lint

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-set -eux
+set -ex
 
 # format code
 if [[ $CI ]]; then

tests/test_mongoengine.py

@@ -6,7 +6,7 @@
 
 import mongoengine
 import pymongo
-from bson import ObjectId
+from bson import BSON, ObjectId
 from mongoengine import Document, EmbeddedDocument, QuerySet, fields
 from pymongo.collation import Collation, CollationStrength
 
@@ -173,6 +173,12 @@ def main() -> None:
     Post().reload()
 
 
+def test_bson() -> None:
+    doc = {"foo": "bar", "buzz": True}
+    data = BSON.encode(doc)
+    assert BSON.decode(data) == doc
+
+
 def test_pymongo() -> None:
     pymongo.ALL
     pymongo.ASCENDING

typings/bson/__init__.pyi

@@ -1,9 +1,22 @@
-from typing import Any
+from typing import Any, Mapping
 
 from bson.codec_options import CodecOptions
 from bson.objectid import ObjectId
 from bson.son import SON
 
 def decode_iter(data: bytes, codec_options: CodecOptions = ...) -> Any: ...
 
+class BSON(bytes):
+    @classmethod
+    def encode(
+        cls,
+        document: Mapping[str, Any],
+        check_keys: bool = ...,
+        codec_options: CodecOptions = ...,
+    ) -> BSON: ...
+    def decode(  # type: ignore [override]
+        self,
+        codec_options: CodecOptions = ...,
+    ) -> Any: ...
+
 __all__ = ["ObjectId", "SON", "decode_iter", "CodecOptions"]