bpo-46542: test_json uses support.infinite_recursion() (GH-30972)

Fix test_json tests checking for RecursionError: modify these tests
to use support.infinite_recursion().
(cherry picked from commit e7a6285f1b)

Co-authored-by: Victor Stinner <vstinner@python.org>
This commit is contained in:
Miss Islington (bot) 2022-01-27 17:04:56 -08:00 committed by GitHub
parent 7c8d1f0238
commit 8cd82461f9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 6 deletions

View file

@ -1,3 +1,4 @@
from test import support
from test.test_json import PyTest, CTest
@ -69,11 +70,14 @@ class TestRecursion:
# test that loading highly-nested objects doesn't segfault when C
# accelerations are used. See #12017
with self.assertRaises(RecursionError):
self.loads('{"a":' * 100000 + '1' + '}' * 100000)
with support.infinite_recursion():
self.loads('{"a":' * 100000 + '1' + '}' * 100000)
with self.assertRaises(RecursionError):
self.loads('{"a":' * 100000 + '[1]' + '}' * 100000)
with support.infinite_recursion():
self.loads('{"a":' * 100000 + '[1]' + '}' * 100000)
with self.assertRaises(RecursionError):
self.loads('[' * 100000 + '1' + ']' * 100000)
with support.infinite_recursion():
self.loads('[' * 100000 + '1' + ']' * 100000)
def test_highly_nested_objects_encoding(self):
# See #12051
@ -81,9 +85,11 @@ class TestRecursion:
for x in range(100000):
l, d = [l], {'k':d}
with self.assertRaises(RecursionError):
self.dumps(l)
with support.infinite_recursion():
self.dumps(l)
with self.assertRaises(RecursionError):
self.dumps(d)
with support.infinite_recursion():
self.dumps(d)
def test_endless_recursion(self):
# See #12051
@ -93,7 +99,8 @@ class TestRecursion:
return [o]
with self.assertRaises(RecursionError):
EndlessJSONEncoder(check_circular=False).encode(5j)
with support.infinite_recursion():
EndlessJSONEncoder(check_circular=False).encode(5j)
class TestPyRecursion(TestRecursion, PyTest): pass