Test socket_recvall()

Author: borntyping

riemann_client/__init__.py

@@ -1,4 +1,4 @@
 """A Python Riemann client and command line tool"""
 
-__version__ = '5.1.0'
+__version__ = '5.1.1'
 __author__ = 'Sam Clements <sam.clements@datasift.com>'

riemann_client/tests/test_riemann_transport.py

@@ -5,6 +5,24 @@
 import riemann_client.riemann_pb2
 import riemann_client.transport
 
+from riemann_client.transport import socket_recvall
+
+
+class FakeSocket(object):
+        def __init__(self):
+            self.data = [b'hello', b'world', b'']
+
+        def recv(self, bufsize):
+            return self.data.pop(0)
+
+
+def test_socket_recvall():
+    assert socket_recvall(FakeSocket(), 10) == b'helloworld'
+
+
+def test_socket_recvall_short():
+    assert socket_recvall(FakeSocket(), 5) == b'hello'
+
 
 @py.test.fixture
 def tcp_transport():

setup.py

@@ -11,7 +11,7 @@
 
 setuptools.setup(
     name='riemann-client',
-    version='5.1.0',
+    version='5.1.1',
 
     author="Sam Clements",
     author_email="sam.clements@datasift.com",