blob: 66882d65f67086225516685a0c13da9eac0c643d (
plain)
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
47
|
See first commit of https://github.com/Iotic-Labs/py-ubjson/pull/19
---
test/test.py | 25 ++++++++++---------------
1 file changed, 10 insertions(+), 15 deletions(-)
diff --git a/test/test.py b/test/test.py
index c3920c4..67a9fbc 100644
--- a/test/test.py
+++ b/test/test.py
@@ -464,23 +464,18 @@ class TestEncodeDecodePlain(TestCase): # pylint: disable=too-many-public-method
return (self.assertRaisesRegexp if PY2 else self.assertRaisesRegex)(*args, **kwargs)
def test_recursion(self):
- old_limit = getrecursionlimit()
- setrecursionlimit(200)
- try:
- obj = current = []
- for _ in range(getrecursionlimit() * 2):
- new_list = []
- current.append(new_list)
- current = new_list
+ obj = current = []
+ for _ in range(getrecursionlimit() * 30):
+ new_list = []
+ current.append(new_list)
+ current = new_list
- with self.assert_raises_regex(RuntimeError, 'recursion'):
- self.ubjdumpb(obj)
+ with self.assert_raises_regex(RuntimeError, 'recursion'):
+ self.ubjdumpb(obj)
- raw = ARRAY_START * (getrecursionlimit() * 2)
- with self.assert_raises_regex(RuntimeError, 'recursion'):
- self.ubjloadb(raw)
- finally:
- setrecursionlimit(old_limit)
+ raw = ARRAY_START * (getrecursionlimit() * 30)
+ with self.assert_raises_regex(RuntimeError, 'recursion'):
+ self.ubjloadb(raw)
def test_encode_default(self):
def default(obj):
--
2.52.0
|
