| 1 | See first commit of https://github.com/Iotic-Labs/py-ubjson/pull/19
|
| 2 |
|
| 3 | ---
|
| 4 | test/test.py | 25 ++++++++++---------------
|
| 5 | 1 file changed, 10 insertions(+), 15 deletions(-)
|
| 6 |
|
| 7 | diff --git a/test/test.py b/test/test.py
|
| 8 | index c3920c4..67a9fbc 100644
|
| 9 | |
| 10 | |
| 11 | @@ -464,23 +464,18 @@ class TestEncodeDecodePlain(TestCase): # pylint: disable=too-many-public-method
|
| 12 | return (self.assertRaisesRegexp if PY2 else self.assertRaisesRegex)(*args, **kwargs)
|
| 13 |
|
| 14 | def test_recursion(self):
|
| 15 | - old_limit = getrecursionlimit()
|
| 16 | - setrecursionlimit(200)
|
| 17 | - try:
|
| 18 | - obj = current = []
|
| 19 | - for _ in range(getrecursionlimit() * 2):
|
| 20 | - new_list = []
|
| 21 | - current.append(new_list)
|
| 22 | - current = new_list
|
| 23 | + obj = current = []
|
| 24 | + for _ in range(getrecursionlimit() * 30):
|
| 25 | + new_list = []
|
| 26 | + current.append(new_list)
|
| 27 | + current = new_list
|
| 28 |
|
| 29 | - with self.assert_raises_regex(RuntimeError, 'recursion'):
|
| 30 | - self.ubjdumpb(obj)
|
| 31 | + with self.assert_raises_regex(RuntimeError, 'recursion'):
|
| 32 | + self.ubjdumpb(obj)
|
| 33 |
|
| 34 | - raw = ARRAY_START * (getrecursionlimit() * 2)
|
| 35 | - with self.assert_raises_regex(RuntimeError, 'recursion'):
|
| 36 | - self.ubjloadb(raw)
|
| 37 | - finally:
|
| 38 | - setrecursionlimit(old_limit)
|
| 39 | + raw = ARRAY_START * (getrecursionlimit() * 30)
|
| 40 | + with self.assert_raises_regex(RuntimeError, 'recursion'):
|
| 41 | + self.ubjloadb(raw)
|
| 42 |
|
| 43 | def test_encode_default(self):
|
| 44 | def default(obj):
|
| 45 | --
|
| 46 | 2.52.0
|
| 47 |
|