IQ.Lvbs/pycapnp
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Merge branch 'fix-mmap-buf' into madhava/python_310

Browse Source
This commit is contained in:
Madhava Jay
2022-05-24 10:42:57 +10:00
parent ea70dac5cb ea776a2f55
commit 26de61d847
2 changed files with 45 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
test/test_serialization.py
View File

@@ -46,8 +46,8 @@ def test_roundtrip_bytes(all_types):
test_regression.init_all_types(msg)
message_bytes = msg.to_bytes()
msg = all_types.TestAllTypes.from_bytes(message_bytes)
test_regression.check_all_types(msg)
with all_types.TestAllTypes.from_bytes(message_bytes) as msg:
test_regression.check_all_types(msg)
@pytest.mark.skipif(
@@ -77,8 +77,8 @@ def test_roundtrip_bytes_mmap(all_types):
f.seek(0)
memory = mmap.mmap(f.fileno(), length)
msg = all_types.TestAllTypes.from_bytes(memory)
test_regression.check_all_types(msg)
with all_types.TestAllTypes.from_bytes(memory) as msg:
test_regression.check_all_types(msg)
@pytest.mark.skipif(
@@ -91,15 +91,16 @@ def test_roundtrip_bytes_buffer(all_types):
b = msg.to_bytes()
v = memoryview(b)
try:
msg = all_types.TestAllTypes.from_bytes(v)
test_regression.check_all_types(msg)
with all_types.TestAllTypes.from_bytes(v) as msg:
test_regression.check_all_types(msg)
finally:
v.release()
def test_roundtrip_bytes_fail(all_types):
with pytest.raises(TypeError):
all_types.TestAllTypes.from_bytes(42)
with all_types.TestAllTypes.from_bytes(42) as msg:
pass
@pytest.mark.skipif(
@@ -232,14 +233,14 @@ 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
with all_types.TestAllTypes.from_bytes(data) as msg:
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):
assert msg.structList[i].uInt8Field == 0
with all_types.TestAllTypes.from_bytes(data, traversal_limit_in_words=2 ** 62) as msg:
for i in range(0, size):
assert msg.structList[i].uInt8Field == 0
def test_from_bytes_packed_traversal_limit(all_types):