1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
|
import json
from collections import UserList, defaultdict
from datetime import datetime
from decimal import Decimal
from django.test import SimpleTestCase
from django.utils.json import normalize_json
class JSONNormalizeTestCase(SimpleTestCase):
def test_converts_json_types(self):
for test_case, expected in [
(None, "null"),
(True, "true"),
(False, "false"),
(2, "2"),
(3.0, "3.0"),
(1e23 + 1, "1e+23"),
("1", '"1"'),
(b"hello", '"hello"'),
([], "[]"),
(UserList([1, 2]), "[1, 2]"),
({}, "{}"),
({1: "a"}, '{"1": "a"}'),
({"foo": (1, 2, 3)}, '{"foo": [1, 2, 3]}'),
(defaultdict(list), "{}"),
(float("nan"), "NaN"),
(float("inf"), "Infinity"),
(float("-inf"), "-Infinity"),
]:
with self.subTest(test_case):
normalized = normalize_json(test_case)
# Ensure that the normalized result is serializable.
self.assertEqual(json.dumps(normalized), expected)
def test_bytes_decode_error(self):
with self.assertRaisesMessage(ValueError, "Unsupported value"):
normalize_json(b"\xff")
def test_encode_error(self):
for test_case in [self, any, object(), datetime.now(), set(), Decimal("3.42")]:
with (
self.subTest(test_case),
self.assertRaisesMessage(TypeError, "Unsupported type"),
):
normalize_json(test_case)
|