1# Tests of json module.
2
3load("assert.star", "assert")
4load("json.star", "json")
5
6assert.eq(dir(json), ["decode", "encode", "indent"])
7
8# Some of these cases were inspired by github.com/nst/JSONTestSuite.
9
10## json.encode
11
12assert.eq(json.encode(None), "null")
13assert.eq(json.encode(True), "true")
14assert.eq(json.encode(False), "false")
15assert.eq(json.encode(-123), "-123")
16assert.eq(json.encode(12345*12345*12345*12345*12345*12345), "3539537889086624823140625")
17assert.eq(json.encode(float(12345*12345*12345*12345*12345*12345)), "3.539537889086625e+24")
18assert.eq(json.encode(12.345e67), "1.2345e+68")
19assert.eq(json.encode("hello"), '"hello"')
20assert.eq(json.encode([1, 2, 3]), "[1,2,3]")
21assert.eq(json.encode((1, 2, 3)), "[1,2,3]")
22assert.eq(json.encode(range(3)), "[0,1,2]") # a built-in iterable
23assert.eq(json.encode(dict(x = 1, y = "two")), '{"x":1,"y":"two"}')
24assert.eq(json.encode(dict(y = "two", x = 1)), '{"x":1,"y":"two"}') # key, not insertion, order
25assert.eq(json.encode(struct(x = 1, y = "two")), '{"x":1,"y":"two"}')  # a user-defined HasAttrs
26assert.eq(json.encode("��"[:1]), '"\\ufffd"') # invalid UTF-8 -> replacement char
27
28def encode_error(expr, error):
29    assert.fails(lambda: json.encode(expr), error)
30
31encode_error(float("NaN"), "json.encode: cannot encode non-finite float nan")
32encode_error({1: "two"}, "dict has int key, want string")
33encode_error(len, "cannot encode builtin_function_or_method as JSON")
34encode_error(struct(x=[1, {"x": len}]), # nested failure
35             'in field .x: at list index 1: in dict key "x": cannot encode...')
36encode_error(struct(x=[1, {"x": len}]), # nested failure
37             'in field .x: at list index 1: in dict key "x": cannot encode...')
38encode_error({1: 2}, 'dict has int key, want string')
39
40## json.decode
41
42assert.eq(json.decode("null"), None)
43assert.eq(json.decode("true"), True)
44assert.eq(json.decode("false"), False)
45assert.eq(json.decode("-123"), -123)
46assert.eq(json.decode("-0"), -0)
47assert.eq(json.decode("3539537889086624823140625"), 3539537889086624823140625)
48assert.eq(json.decode("3539537889086624823140625.0"), float(3539537889086624823140625))
49assert.eq(json.decode("3.539537889086625e+24"), 3.539537889086625e+24)
50assert.eq(json.decode("0e+1"), 0)
51assert.eq(json.decode("-0.0"), -0.0)
52assert.eq(json.decode(
53    "-0.000000000000000000000000000000000000000000000000000000000000000000000000000001"),
54    -0.000000000000000000000000000000000000000000000000000000000000000000000000000001)
55assert.eq(json.decode('[]'), [])
56assert.eq(json.decode('[1]'), [1])
57assert.eq(json.decode('[1,2,3]'), [1, 2, 3])
58assert.eq(json.decode('{"one": 1, "two": 2}'), dict(one=1, two=2))
59assert.eq(json.decode('{"foo\\u0000bar": 42}'), {"foo\x00bar": 42})
60assert.eq(json.decode('"\\ud83d\\ude39\\ud83d\\udc8d"'), "����")
61assert.eq(json.decode('"\\u0123"'), 'ģ')
62assert.eq(json.decode('"\x7f"'), "\x7f")
63
64def decode_error(expr, error):
65    assert.fails(lambda: json.decode(expr), error)
66
67decode_error('truefalse',
68             "json.decode: at offset 4, unexpected character 'f' after value")
69
70decode_error('"abc', "unclosed string literal")
71decode_error('"ab\\gc"', "invalid character 'g' in string escape code")
72decode_error("'abc'", "unexpected character '\\\\''")
73
74decode_error("1.2.3", "invalid number: 1.2.3")
75decode_error("+1", "unexpected character '\\+'")
76decode_error("-abc", "invalid number: -")
77decode_error("-", "invalid number: -")
78decode_error("-00", "invalid number: -00")
79decode_error("00", "invalid number: 00")
80decode_error("--1", "invalid number: --1")
81decode_error("-+1", "invalid number: -\\+1")
82decode_error("1e1e1", "invalid number: 1e1e1")
83decode_error("0123", "invalid number: 0123")
84decode_error("000.123", "invalid number: 000.123")
85decode_error("-0123", "invalid number: -0123")
86decode_error("-000.123", "invalid number: -000.123")
87decode_error("0x123", "unexpected character 'x' after value")
88
89decode_error('[1, 2 ', "unexpected end of file")
90decode_error('[1, 2, ', "unexpected end of file")
91decode_error('[1, 2, ]', "unexpected character ']'")
92decode_error('[1, 2, }', "unexpected character '}'")
93decode_error('[1, 2}', "got '}', want ',' or ']'")
94
95decode_error('{"one": 1', "unexpected end of file")
96decode_error('{"one" 1', "after object key, got '1', want ':'")
97decode_error('{"one": 1 "two": 2', "in object, got '\"', want ',' or '}'")
98decode_error('{"one": 1,', "unexpected end of file")
99decode_error('{"one": 1, }', "unexpected character '}'")
100decode_error('{"one": 1]', "in object, got ']', want ',' or '}'")
101
102def codec(x):
103    return json.decode(json.encode(x))
104
105# string round-tripping
106strings = [
107    "��", # U+1F63F CRYING_CAT_FACE
108    "��‍��", # CAT FACE + ZERO WIDTH JOINER + BUST IN SILHOUETTE
109]
110assert.eq(codec(strings), strings)
111
112# codepoints is a string with every 16-bit code point.
113codepoints = ''.join(['%c' % c for c in range(65536)])
114assert.eq(codec(codepoints), codepoints)
115
116# number round-tripping
117numbers = [
118    0, 1, -1, +1, 1.23e45, -1.23e-45,
119    3539537889086624823140625,
120    float(3539537889086624823140625),
121]
122assert.eq(codec(numbers), numbers)
123
124## json.indent
125
126s = json.encode(dict(x = 1, y = ["one", "two"]))
127
128assert.eq(json.indent(s), '''{
129	"x": 1,
130	"y": [
131		"one",
132		"two"
133	]
134}''')
135
136assert.eq(json.decode(json.indent(s)), {"x": 1, "y": ["one", "two"]})
137
138assert.eq(json.indent(s, prefix='¶', indent='–––'), '''{
139¶–––"x": 1,
140¶–––"y": [
141¶––––––"one",
142¶––––––"two"
143¶–––]
144¶}''')
145
146assert.fails(lambda: json.indent("!@#$%^& this is not json"), 'invalid character')
147---
148