Raymond Hettinger | c8d952d | 2009-05-27 06:50:31 +0000 | [diff] [blame] | 1 | from collections import OrderedDict |
Ezio Melotti | 66f2ea0 | 2013-08-08 15:03:45 +0300 | [diff] [blame] | 2 | from test.test_json import PyTest, CTest |
Benjamin Peterson | e3bfe19 | 2015-02-01 17:53:53 -0500 | [diff] [blame] | 3 | from test.support import bigaddrspacetest |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 4 | |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 5 | |
| 6 | CASES = [ |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 7 | ('/\\"\ucafe\ubabe\uab98\ufcde\ubcda\uef4a\x08\x0c\n\r\t`1~!@#$%^&*()_+-=[]{}|;:\',./<>?', '"/\\\\\\"\\ucafe\\ubabe\\uab98\\ufcde\\ubcda\\uef4a\\b\\f\\n\\r\\t`1~!@#$%^&*()_+-=[]{}|;:\',./<>?"'), |
| 8 | ('\u0123\u4567\u89ab\ucdef\uabcd\uef4a', '"\\u0123\\u4567\\u89ab\\ucdef\\uabcd\\uef4a"'), |
| 9 | ('controls', '"controls"'), |
| 10 | ('\x08\x0c\n\r\t', '"\\b\\f\\n\\r\\t"'), |
| 11 | ('{"object with 1 member":["array with 1 element"]}', '"{\\"object with 1 member\\":[\\"array with 1 element\\"]}"'), |
| 12 | (' s p a c e d ', '" s p a c e d "'), |
| 13 | ('\U0001d120', '"\\ud834\\udd20"'), |
| 14 | ('\u03b1\u03a9', '"\\u03b1\\u03a9"'), |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 15 | ("`1~!@#$%^&*()_+-={':[,]}|;.</>?", '"`1~!@#$%^&*()_+-={\':[,]}|;.</>?"'), |
| 16 | ('\x08\x0c\n\r\t', '"\\b\\f\\n\\r\\t"'), |
| 17 | ('\u0123\u4567\u89ab\ucdef\uabcd\uef4a', '"\\u0123\\u4567\\u89ab\\ucdef\\uabcd\\uef4a"'), |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 18 | ] |
| 19 | |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 20 | class TestEncodeBasestringAscii: |
| 21 | def test_encode_basestring_ascii(self): |
| 22 | fname = self.json.encoder.encode_basestring_ascii.__name__ |
Christian Heimes | 9054000 | 2008-05-08 14:29:10 +0000 | [diff] [blame] | 23 | for input_string, expect in CASES: |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 24 | result = self.json.encoder.encode_basestring_ascii(input_string) |
Ezio Melotti | b3aedd4 | 2010-11-20 19:04:17 +0000 | [diff] [blame] | 25 | self.assertEqual(result, expect, |
Benjamin Peterson | c6b607d | 2009-05-02 12:36:44 +0000 | [diff] [blame] | 26 | '{0!r} != {1!r} for {2}({3!r})'.format( |
| 27 | result, expect, fname, input_string)) |
Raymond Hettinger | c8d952d | 2009-05-27 06:50:31 +0000 | [diff] [blame] | 28 | |
| 29 | def test_ordered_dict(self): |
| 30 | # See issue 6105 |
| 31 | items = [('one', 1), ('two', 2), ('three', 3), ('four', 4), ('five', 5)] |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 32 | s = self.dumps(OrderedDict(items)) |
Raymond Hettinger | c8d952d | 2009-05-27 06:50:31 +0000 | [diff] [blame] | 33 | self.assertEqual(s, '{"one": 1, "two": 2, "three": 3, "four": 4, "five": 5}') |
Raymond Hettinger | bcf6f92 | 2009-05-27 09:58:34 +0000 | [diff] [blame] | 34 | |
| 35 | def test_sorted_dict(self): |
| 36 | items = [('one', 1), ('two', 2), ('three', 3), ('four', 4), ('five', 5)] |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 37 | s = self.dumps(dict(items), sort_keys=True) |
Raymond Hettinger | bcf6f92 | 2009-05-27 09:58:34 +0000 | [diff] [blame] | 38 | self.assertEqual(s, '{"five": 5, "four": 4, "one": 1, "three": 3, "two": 2}') |
Ezio Melotti | 6b60fb9 | 2011-05-14 06:47:51 +0300 | [diff] [blame] | 39 | |
| 40 | |
| 41 | class TestPyEncodeBasestringAscii(TestEncodeBasestringAscii, PyTest): pass |
Benjamin Peterson | e3bfe19 | 2015-02-01 17:53:53 -0500 | [diff] [blame] | 42 | class TestCEncodeBasestringAscii(TestEncodeBasestringAscii, CTest): |
| 43 | @bigaddrspacetest |
| 44 | def test_overflow(self): |
Benjamin Peterson | 1572944 | 2015-02-02 17:47:07 -0500 | [diff] [blame] | 45 | size = (2**32)//6 + 1 |
| 46 | s = "\x00"*size |
Benjamin Peterson | e3bfe19 | 2015-02-01 17:53:53 -0500 | [diff] [blame] | 47 | with self.assertRaises(OverflowError): |
| 48 | self.json.encoder.encode_basestring_ascii(s) |