def test_basic_get_empty(self):
def on_get_frame(*_):
channel.rpc.on_frame(specification.Basic.GetEmpty())
connection = FakeConnection(on_write=on_get_frame)
channel = Channel(9, connection, 1)
channel.set_state(Channel.OPEN)
basic = Basic(channel)
result = basic.get(queue='travis-ci')
self.assertFalse(channel.rpc._request)
self.assertFalse(channel.rpc._response)
self.assertIsNone(result)
def test_basic_get_empty(self):
def on_get_frame(*_):
channel.rpc.on_frame(specification.Basic.GetEmpty())
connection = FakeConnection(on_write=on_get_frame)
channel = Channel(9, connection, 1)
channel.set_state(Channel.OPEN)
basic = Basic(channel)
result = basic.get(queue='travis-ci')
self.assertFalse(channel.rpc._request)
self.assertFalse(channel.rpc._response)
self.assertIsNone(result)
def test_basic_get_to_dict(self):
message = self.message.encode('utf-8')
message_len = len(message)
def on_get_frame(*_):
channel.rpc.on_frame(specification.Basic.GetOk())
channel.rpc.on_frame(ContentHeader(body_size=message_len))
channel.rpc.on_frame(ContentBody(value=message))
connection = FakeConnection(on_write=on_get_frame)
channel = Channel(9, connection, 1)
channel.set_state(Channel.OPEN)
basic = Basic(channel)
result = basic.get(queue='travis-ci', to_dict=True)
self.assertFalse(channel.rpc._request)
self.assertFalse(channel.rpc._response)
self.assertEqual(result['body'], message)
def test_basic_get_to_dict(self):
message = self.message.encode('utf-8')
message_len = len(message)
def on_get_frame(*_):
channel.rpc.on_frame(specification.Basic.GetOk())
channel.rpc.on_frame(ContentHeader(body_size=message_len))
channel.rpc.on_frame(ContentBody(value=message))
connection = FakeConnection(on_write=on_get_frame)
channel = Channel(9, connection, 1)
channel.set_state(Channel.OPEN)
basic = Basic(channel)
result = basic.get(queue='travis-ci', to_dict=True)
self.assertFalse(channel.rpc._request)
self.assertFalse(channel.rpc._response)
self.assertEqual(result['body'], message)
