From: Sergey Matveev <stargrave@stargrave.org>
Date: Sat, 30 Nov 2024 16:14:27 +0000 (+0300)
Subject: Limit lists-strategy
X-Git-Tag: v0.0.0~290^2~37
X-Git-Url: http://www.git.cypherpunks.su/?a=commitdiff_plain;h=fbee5812d2d5f848f3be09b2ea72435365157c145549d6259634c914b6a2195b;p=keks.git

Limit lists-strategy
---

diff --git a/pyac/tests/test_list.py b/pyac/tests/test_list.py
index 6f3f969..9d386b6 100644
--- a/pyac/tests/test_list.py
+++ b/pyac/tests/test_list.py
@@ -16,7 +16,7 @@ class TestList(TestCase):
         encoded = dumps([])
         self.assertEqual(encoded, b"\x08\x00")
 
-    @given(lists(any_st))
+    @given(lists(any_st, max_size=4))
     def test_list_encode_non_empty(self, test_list: List) -> None:
         encoded = dumps(test_list)
         self.assertEqual(
@@ -29,14 +29,14 @@ class TestList(TestCase):
         self.assertEqual(decoded, [])
         self.assertEqual(remaining, b"")
 
-    @given(lists(any_st), binary(max_size=20))
+    @given(lists(any_st, max_size=4), binary(max_size=20))
     def test_list_decode_non_empty(self, test_list: List, junk: bytes) -> None:
         encoded = b"\x08" + b"".join(dumps(i) for i in test_list) + b"\x00" + junk
         decoded, remaining = loads(encoded)
         self.assertEqual(decoded, test_list)
         self.assertEqual(remaining, junk)
 
-    @given(lists(any_st))
+    @given(lists(any_st, max_size=4))
     def test_no_eoc(self, test_list: List) -> None:
         encoded = dumps(test_list)[:-1]
         with self.assertRaises(NotEnoughData) as cm: