Method test_encode_subclass

test/test_json_util.py:588–621

(self)

            self.assertIsInstance(doc["d"], cls)

    def test_encode_subclass(self):
        cases: list[Tuple[Type, Any]] = [
            (int, (1,)),
            (int, (2 << 60,)),
            (float, (1.1,)),
            (Int64, (64,)),
            (Int64, (2 << 60,)),
            (str, ("str",)),
            (bytes, (b"bytes",)),
            (datetime.datetime, (2024, 1, 16)),
            (DatetimeMS, (1,)),
            (uuid.UUID, ("f47ac10b-58cc-4372-a567-0e02b2c3d479",)),
            (Binary, (b"1", USER_DEFINED_SUBTYPE)),
            (Code, ("code",)),
            (DBRef, ("coll", ObjectId())),
            (ObjectId, ("65a6dab5f98bc03906ee3597",)),
            (MaxKey, ()),
            (MinKey, ()),
            (Regex, ("pat",)),
            (Timestamp, (1, 1)),
            (Decimal128, ("0.5",)),
        ]
        allopts = [
            CANONICAL_JSON_OPTIONS.with_options(uuid_representation=STANDARD),
            RELAXED_JSON_OPTIONS.with_options(uuid_representation=STANDARD),
            LEGACY_JSON_OPTIONS.with_options(uuid_representation=STANDARD),
        ]
        for cls, args in cases:
            basic_obj = cls(*args)
            my_cls = type(f"My{cls.__name__}", (cls,), {})
            my_obj = my_cls(*args)
            for opts in allopts:
                expected_json = json_util.dumps(basic_obj, json_options=opts)
                self.assertEqual(json_util.dumps(my_obj, json_options=opts), expected_json)

    def test_encode_type_marker(self):
        # Assert that a custom subclass can be JSON encoded based on the _type_marker attribute.