def test_table_multi(self):
val = {
'foo': 7,
'bar': Decimal('123345.1234'),
'baz': 'this is some random string I typed',
'ubaz': u'And something in unicode',
'dday_aniv': datetime(1994, 6, 6),
'more': {
'abc': -123,
'def': 'hello world',
'now': datetime(2007, 11, 11, 21, 14, 31),
'qty': Decimal('-123.45'),
'blank': {},
'extra': {
'deeper': 'more strings',
'nums': -12345678,
},
}
}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_table_multi(self):
val = {
'foo': 7,
'bar': Decimal('123345.1234'),
'baz': 'this is some random string I typed',
'ubaz': u'And something in unicode',
'dday_aniv': datetime(1994, 6, 6),
'nothing': None,
'more': {
'abc': -123,
'def': 'hello world',
'now': datetime(2007, 11, 11, 21, 14, 31),
'qty': Decimal('-123.45'),
'blank': {},
'extra': {
'deeper': 'more strings',
'nums': -12345678,
},
}
}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_shortstr_unicode(self):
w = AMQPWriter()
w.write_shortstr(u'hello')
s = w.getvalue()
self.assertEqualBinary(s, '\x05hello')
r = AMQPReader(s)
self.assertEqual(r.read_shortstr(), u'hello')
def test_shortstr_unicode(self):
w = AMQPWriter()
w.write_shortstr(u'hello')
s = w.getvalue()
self.assertEqualBinary(s, '\x05hello')
r = AMQPReader(s)
self.assertEqual(r.read_shortstr(), u'hello')
def test_short(self):
for i in range(256):
val = randint(0, 65535)
w = AMQPWriter()
w.write_short(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_short(), val)
def test_short(self):
for i in range(256):
val = randint(0, 65535)
w = AMQPWriter()
w.write_short(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_short(), val)
def test_longlong(self):
for i in range(256):
val = randint(0, (2**64) - 1)
w = AMQPWriter()
w.write_longlong(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_longlong(), val)
def test_longlong(self):
for i in range(256):
val = randint(0, (2 ** 64) - 1)
w = AMQPWriter()
w.write_longlong(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_longlong(), val)
def test_empty_longstr(self):
w = AMQPWriter()
w.write_longstr('')
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x00')
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), '')
def test_octet(self):
for val in range(256):
w = AMQPWriter()
w.write_octet(val)
s = w.getvalue()
self.assertEqualBinary(s, chr(val))
r = AMQPReader(s)
self.assertEqual(r.read_octet(), val)
def test_octet(self):
for val in range(256):
w = AMQPWriter()
w.write_octet(val)
s = w.getvalue()
self.assertEqualBinary(s, chr(val))
r = AMQPReader(s)
self.assertEqual(r.read_octet(), val)
def test_empty_longstr(self):
w = AMQPWriter()
w.write_longstr('')
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x00')
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), '')
def test_longstr_unicode(self):
val = u'a' * 512
w = AMQPWriter()
w.write_longstr(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x02\x00' + ('a' * 512))
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), val)
def test_single_bit(self):
for val, check in [(True, '\x01'), (False, '\x00')]:
w = AMQPWriter()
w.write_bit(val)
s = w.getvalue()
self.assertEqualBinary(s, check)
r = AMQPReader(s)
self.assertEqual(r.read_bit(), val)
def test_array_from_tuple(self):
val = (1, 'foo')
w = AMQPWriter()
w.write_array(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x0DI\x00\x00\x00\x01S\x00\x00\x00\x03foo')
r = AMQPReader(s)
self.assertEqual(r.read_array(), list(val))
def test_table(self):
val = {'foo': 7}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x09\x03fooI\x00\x00\x00\x07')
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_table_empty(self):
val = {}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x00')
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_single_bit(self):
for val, check in [(True, '\x01'), (False, '\x00')]:
w = AMQPWriter()
w.write_bit(val)
s = w.getvalue()
self.assertEqualBinary(s, check)
r = AMQPReader(s)
self.assertEqual(r.read_bit(), val)
def test_table(self):
val = {'foo': 7}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x09\x03fooI\x00\x00\x00\x07')
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_longstr_unicode(self):
val = u'a' * 512
w = AMQPWriter()
w.write_longstr(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x02\x00' + ('a' * 512))
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), val)
def test_table_empty(self):
val = {}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
self.assertEqualBinary(s, '\x00\x00\x00\x00')
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_array_from_tuple(self):
val = (1, 'foo')
w = AMQPWriter()
w.write_array(val)
s = w.getvalue()
self.assertEqualBinary(
s, '\x00\x00\x00\x0DI\x00\x00\x00\x01S\x00\x00\x00\x03foo')
r = AMQPReader(s)
self.assertEqual(r.read_array(), list(val))
def test_array_from_list(self):
val = [1, 'foo', None]
w = AMQPWriter()
w.write_array(val)
s = w.getvalue()
self.assertEqualBinary(
s, '\x00\x00\x00\x0EI\x00\x00\x00\x01S\x00\x00\x00\x03fooV',
)
r = AMQPReader(s)
self.assertEqual(r.read_array(), val)
def test_array_from_list(self):
val = [1, 'foo', None]
w = AMQPWriter()
w.write_array(val)
s = w.getvalue()
self.assertEqualBinary(
s,
'\x00\x00\x00\x0EI\x00\x00\x00\x01S\x00\x00\x00\x03fooV',
)
r = AMQPReader(s)
self.assertEqual(r.read_array(), val)
def test_table_with_array(self):
val = {
'foo': 7,
'bar': Decimal('123345.1234'),
'baz': 'this is some random string I typed',
'blist': [1,2,3],
'nlist': [1, [2,3,4]],
'ndictl': {'nfoo': 8, 'nblist': [5,6,7] }
}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_table_invalid(self):
"""
Check that an un-serializable table entry raises a ValueError
"""
val = {'test': object()}
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_table, val)
def test_multiple_bits(self):
w = AMQPWriter()
w.write_bit(True)
w.write_bit(True)
w.write_bit(False)
w.write_bit(True)
s = w.getvalue()
self.assertEqualBinary(s, '\x0b')
r = AMQPReader(s)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
self.assertEqual(r.read_bit(), True)
def test_table_with_array(self):
val = {
'foo': 7,
'bar': Decimal('123345.1234'),
'baz': 'this is some random string I typed',
'blist': [1, 2, 3],
'nlist': [1, [2, 3, 4]],
'ndictl': {
'nfoo': 8,
'nblist': [5, 6, 7]
}
}
w = AMQPWriter()
w.write_table(val)
s = w.getvalue()
r = AMQPReader(s)
self.assertEqual(r.read_table(), val)
def test_multiple_bits2(self):
"""
Check bits mixed with non-bits
"""
w = AMQPWriter()
w.write_bit(True)
w.write_bit(True)
w.write_bit(False)
w.write_octet(10)
w.write_bit(True)
s = w.getvalue()
self.assertEqualBinary(s, '\x03\x0a\x01')
r = AMQPReader(s)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
self.assertEqual(r.read_octet(), 10)
self.assertEqual(r.read_bit(), True)
def test_multiple_bits(self):
w = AMQPWriter()
w.write_bit(True)
w.write_bit(True)
w.write_bit(False)
w.write_bit(True)
s = w.getvalue()
self.assertEqualBinary(s, '\x0b')
r = AMQPReader(s)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
self.assertEqual(r.read_bit(), True)
def test_multiple_bits3(self):
"""
Check bit groups that span multiple bytes
"""
w = AMQPWriter()
# Spit out 20 bits
for i in range(10):
w.write_bit(True)
w.write_bit(False)
s = w.getvalue()
self.assertEqualBinary(s, '\x55\x55\x05')
r = AMQPReader(s)
for i in range(10):
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
class test_Channel(Case):
def setUp(self):
self.args = AMQPWriter()
self.connection = Mock(name='connection')
self.connection.channels = defaultdict(lambda: None)
self.channel = NoOpenChannel(self.connection, channel_id=1)
def test_basic_nack(self, delivery_tag=3172312312):
self.args.write_longlong(delivery_tag)
self.args.write_bit(0)
self.args.write_bit(0)
with self.assertRaises(NotConfirmed):
self.channel._basic_nack(AMQPReader(self.args.getvalue()))
callback = Mock(name='callback')
self.channel.events['basic_nack'].add(callback)
self.channel._basic_nack(AMQPReader(self.args.getvalue()))
callback.assert_called_with(delivery_tag, False, False)
class test_Channel(Case):
def setUp(self):
self.args = AMQPWriter()
self.connection = Mock(name='connection')
self.connection.channels = defaultdict(lambda: None)
self.channel = NoOpenChannel(self.connection, channel_id=1)
def test_basic_nack(self, delivery_tag=3172312312):
self.args.write_longlong(delivery_tag)
self.args.write_bit(0)
self.args.write_bit(0)
with self.assertRaises(NotConfirmed):
self.channel._basic_nack(AMQPReader(self.args.getvalue()))
callback = Mock(name='callback')
self.channel.events['basic_nack'].add(callback)
self.channel._basic_nack(AMQPReader(self.args.getvalue()))
callback.assert_called_with(delivery_tag, False, False)
def test_multiple_bits3(self):
"""
Check bit groups that span multiple bytes
"""
w = AMQPWriter()
# Spit out 20 bits
for i in range(10):
w.write_bit(True)
w.write_bit(False)
s = w.getvalue()
self.assertEqualBinary(s, '\x55\x55\x05')
r = AMQPReader(s)
for i in range(10):
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
def test_octet_invalid2(self):
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_octet, 256)
def test_empty_writer(self):
w = AMQPWriter()
self.assertEqual(w.getvalue(), bytes())
def test_short_invalid2(self):
w = AMQPWriter()
self.assertRaises(ValueError, w.write_short, 65536)
def test_empty_writer(self):
w = AMQPWriter()
self.assertEqual(w.getvalue(), bytes())
def test_multiple_bits2(self):
"""
Check bits mixed with non-bits
"""
w = AMQPWriter()
w.write_bit(True)
w.write_bit(True)
w.write_bit(False)
w.write_octet(10)
w.write_bit(True)
s = w.getvalue()
self.assertEqualBinary(s, '\x03\x0a\x01')
r = AMQPReader(s)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), True)
self.assertEqual(r.read_bit(), False)
self.assertEqual(r.read_octet(), 10)
self.assertEqual(r.read_bit(), True)
def test_longlong_invalid2(self):
w = AMQPWriter()
self.assertRaises(ValueError, w.write_longlong, 2**64)
def test_long_shortstr(self):
w = AMQPWriter()
self.assertRaises(ValueError, w.write_shortstr, 'x' * 256)
def setUp(self):
self.args = AMQPWriter()
self.connection = Mock(name='connection')
self.connection.channels = defaultdict(lambda: None)
self.channel = NoOpenChannel(self.connection, channel_id=1)
def test_short_invalid2(self):
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_short, 65536)
def test_longlong_invalid2(self):
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_longlong, 2**64)
def test_long_shortstr(self):
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_shortstr, 'x' * 256)
def test_long_shortstr_unicode(self):
w = AMQPWriter()
self.assertRaises(FrameSyntaxError, w.write_shortstr, u'\u0100' * 128)
def setUp(self):
self.args = AMQPWriter()
self.connection = Mock(name='connection')
self.connection.channels = defaultdict(lambda: None)
self.channel = NoOpenChannel(self.connection, channel_id=1)
def test_octet_invalid2(self):
w = AMQPWriter()
self.assertRaises(ValueError, w.write_octet, 256)
