Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-11-07 04:50:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add test for max_buffer_size.

Browse source
This commit is contained in:
INADA Naoki 2012-07-20 02:02:37 +09:00
parent 7b1167044b
commit e5462ff72f
1 changed files with 17 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test/test_sequnpack.py
View file

@ -1,7 +1,8 @@
#!/usr/bin/env python #!/usr/bin/env python
# coding: utf-8 # coding: utf-8
from msgpack import Unpacker from msgpack import Unpacker, BufferFull
import nose
def test_foobar(): def test_foobar():
unpacker = Unpacker(read_size=3) unpacker = Unpacker(read_size=3)
@ -27,6 +28,19 @@ def test_foobar():
k += 1 k += 1
assert k == len(b'foobar') assert k == len(b'foobar')
if __name__ == '__main__':
test_foobar()
def test_maxbuffersize():
nose.tools.assert_raises(ValueError, Unpacker, read_size=5, max_buffer_size=3)
unpacker = Unpacker(read_size=3, max_buffer_size=3)
unpacker.feed(b'fo')
nose.tools.assert_raises(BufferFull, unpacker.feed, b'ob')
unpacker.feed(b'o')
assert ord('f') == next(unpacker)
unpacker.feed(b'b')
assert ord('o') == next(unpacker)
assert ord('o') == next(unpacker)
assert ord('b') == next(unpacker)
if __name__ == '__main__':
nose.main()