test_rpc: Adopt to msgpack-python>=0.50

msgpack-python version 0.50 or later supports bytearray objects, this
patch fixes to adopt to this change.

Signed-off-by: IWASE Yusuke <iwase.yusuke0@gmail.com>
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
This commit is contained in:
IWASE Yusuke 2018-01-11 16:02:40 +09:00 committed by FUJITA Tomonori
parent c3b484c787
commit 81cdd4f0d7

View File

@ -138,13 +138,18 @@ class Test_rpc(unittest.TestCase):
assert result == obj
assert isinstance(result, numbers.Integral)
@raises(TypeError)
def test_0_call_bytearray(self):
c = rpc.Client(self._client_sock)
obj = bytearray(b'foo')
result = c.call('resp', [obj])
assert result == obj
assert isinstance(result, str)
# Note: msgpack-python version 0.50 or later supports bytearray
# objects, here ignores TypeError for the backward compatibility.
try:
result = c.call('resp', [obj])
except TypeError:
# Case with msgpack-python version 0.4.x or earlier.
return
self.assertEqual(obj, result)
self.assertIsInstance(result, six.binary_type)
def test_1_shutdown_wr(self):
# test if the server shutdown on disconnect