From e4db5e6d6634545b2f1b4384d4c682036d8d11cf Mon Sep 17 00:00:00 2001
From: Jason Paryani <github@jparyani.com>
Date: Sun, 9 Apr 2017 17:17:10 -0700
Subject: [PATCH] Add exception raising tests for `traversal_limit_in_words`

---
 test/test_serialization.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/test/test_serialization.py b/test/test_serialization.py
index 66a1d88..392bbd2 100644
--- a/test/test_serialization.py
+++ b/test/test_serialization.py
@@ -175,6 +175,11 @@ def test_from_bytes_traversal_limit(all_types):
     bld.init("structList", size)
     data = bld.to_bytes()
 
+    msg = all_types.TestAllTypes.from_bytes(data)
+    with pytest.raises(capnp.KjException):
+        for i in range(0, size):
+            msg.structList[i].uInt8Field == 0
+
     msg = all_types.TestAllTypes.from_bytes(data,
         traversal_limit_in_words=2**62)
     for i in range(0, size):
@@ -187,6 +192,11 @@ def test_from_bytes_packed_traversal_limit(all_types):
     bld.init("structList", size)
     data = bld.to_bytes_packed()
 
+    msg = all_types.TestAllTypes.from_bytes_packed(data)
+    with pytest.raises(capnp.KjException):
+        for i in range(0, size):
+            msg.structList[i].uInt8Field == 0
+
     msg = all_types.TestAllTypes.from_bytes_packed(data,
         traversal_limit_in_words=2**62)
     for i in range(0, size):