Method test_datetime_ms

test/test_json_util.py:257–318

(self)

        )

    def test_datetime_ms(self):
        # Test ISO8601 in-range
        dat_min: dict[str, Any] = {"x": DatetimeMS(0)}
        dat_max: dict[str, Any] = {"x": DatetimeMS(_MAX_UTC_MS)}
        opts = JSONOptions(datetime_representation=DatetimeRepresentation.ISO8601)

        self.assertEqual(
            dat_min["x"].as_datetime(CodecOptions(tz_aware=False)),
            json_util.loads(json_util.dumps(dat_min))["x"],
        )
        self.assertEqual(
            dat_max["x"].as_datetime(CodecOptions(tz_aware=False)),
            json_util.loads(json_util.dumps(dat_max))["x"],
        )

        # Test ISO8601 out-of-range
        dat_min = {"x": DatetimeMS(-1)}
        dat_max = {"x": DatetimeMS(_MAX_UTC_MS + 1)}

        self.assertEqual('{"x": {"$date": {"$numberLong": "-1"}}}', json_util.dumps(dat_min))
        self.assertEqual(
            '{"x": {"$date": {"$numberLong": "' + str(int(dat_max["x"])) + '"}}}',
            json_util.dumps(dat_max),
        )
        # Test legacy.
        opts = JSONOptions(
            datetime_representation=DatetimeRepresentation.LEGACY, json_mode=JSONMode.LEGACY
        )
        self.assertEqual('{"x": {"$date": -1}}', json_util.dumps(dat_min, json_options=opts))
        self.assertEqual(
            '{"x": {"$date": ' + str(int(dat_max["x"])) + "}}",
            json_util.dumps(dat_max, json_options=opts),
        )

        # Test regular.
        opts = JSONOptions(
            datetime_representation=DatetimeRepresentation.NUMBERLONG, json_mode=JSONMode.LEGACY
        )
        self.assertEqual(
            '{"x": {"$date": {"$numberLong": "-1"}}}', json_util.dumps(dat_min, json_options=opts)
        )
        self.assertEqual(
            '{"x": {"$date": {"$numberLong": "' + str(int(dat_max["x"])) + '"}}}',
            json_util.dumps(dat_max, json_options=opts),
        )

        # Test decode from datetime.datetime to DatetimeMS
        dat_min = {"x": datetime.datetime.min}
        dat_max = {"x": DatetimeMS(_MAX_UTC_MS).as_datetime(CodecOptions(tz_aware=False))}
        opts = JSONOptions(
            datetime_representation=DatetimeRepresentation.ISO8601,
            datetime_conversion=DatetimeConversion.DATETIME_MS,
        )

        self.assertEqual(
            DatetimeMS(dat_min["x"]),
            json_util.loads(json_util.dumps(dat_min), json_options=opts)["x"],
        )