def test_basic_get_content_body(self): message = b'Hello World!' body = ContentBody(value=message) channel = Channel(0, FakeConnection(), 360) channel.set_state(Channel.OPEN) basic = Basic(channel) uuid = channel.rpc.register_request([body.name]) channel.rpc.on_frame(body) self.assertEqual(basic._get_content_body(uuid, len(message)), message)
def test_basic_get_content_body_break_on_none_value(self): body = ContentBody(value=None) channel = Channel(0, FakeConnection(), 360) channel.set_state(Channel.OPEN) basic = Basic(channel) uuid = channel.rpc.register_request([body.name]) channel.rpc.on_frame(body) self.assertEqual(basic._get_content_body(uuid, 10), b'')
def test_get_content_body(self): message = b'Hello World!' body = ContentBody(value=message) channel = Channel(0, FakeConnection(), 360) channel.set_state(Channel.OPEN) basic = Basic(channel) uuid = channel.rpc.register_request([body.name]) channel.rpc.on_frame(body) self.assertEqual(basic._get_content_body(uuid, len(message)), message)
def test_basic_get_content_body(self): body = ContentBody(value=self.message.encode('utf-8')) channel = Channel(0, FakeConnection(), 360) channel.set_state(Channel.OPEN) basic = Basic(channel) uuid = channel.rpc.register_request([body.name]) channel.rpc.on_frame(body) self.assertEqual(basic._get_content_body(uuid, len(self.message)), self.message.encode('utf-8'))