def _patched_read_table(self):
"""
Read an AMQP table, and return as a Python dictionary.
"""
self.bitcount = self.bits = 0
tlen = unpack('>I', self.input.read(4))[0]
table_data = AMQPReader(self.input.read(tlen))
result = {}
while table_data.input.tell() < tlen:
name = table_data.read_shortstr()
ftype = ord(table_data.input.read(1))
if ftype == 65: # 'A' これが新しく加わった!!
len = unpack('>I', table_data.input.read(4))[0] # len は読み捨てる
ftype = ord(table_data.input.read(1))
if ftype == 83: # 'S'
val = table_data.read_longstr()
elif ftype == 73: # 'I'
val = unpack('>i', table_data.input.read(4))[0]
elif ftype == 68: # 'D'
d = table_data.read_octet()
n = unpack('>i', table_data.input.read(4))[0]
val = Decimal(n) / Decimal(10 ** d)
elif ftype == 84: # 'T'
val = table_data.read_timestamp()
elif ftype == 70: # 'F'
val = table_data.read_table() # recurse
else:
raise ValueError('Unknown table item type: %s' % repr(ftype))
result[name] = val
return result
def _patched_read_table(self):
"""
Read an AMQP table, and return as a Python dictionary.
"""
self.bitcount = self.bits = 0
tlen = unpack('>I', self.input.read(4))[0]
table_data = AMQPReader(self.input.read(tlen))
result = {}
while table_data.input.tell() < tlen:
name = table_data.read_shortstr()
ftype = ord(table_data.input.read(1))
if ftype == 65: # 'A' これが新しく加わった!!
len = unpack('>I', table_data.input.read(4))[0] # len は読み捨てる
ftype = ord(table_data.input.read(1))
if ftype == 83: # 'S'
val = table_data.read_longstr()
elif ftype == 73: # 'I'
val = unpack('>i', table_data.input.read(4))[0]
elif ftype == 68: # 'D'
d = table_data.read_octet()
n = unpack('>i', table_data.input.read(4))[0]
val = Decimal(n) / Decimal(10**d)
elif ftype == 84: # 'T'
val = table_data.read_timestamp()
elif ftype == 70: # 'F'
val = table_data.read_table() # recurse
else:
raise ValueError('Unknown table item type: %s' % repr(ftype))
result[name] = val
return result
def test_empty_longstr(self):
w = AMQPWriter()
w.write_longstr('')
s = w.getvalue()
self.assertEqual(s, '\x00\x00\x00\x00')
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), '')
def test_empty_longstr(self):
w = AMQPWriter()
w.write_longstr('')
s = w.getvalue()
self.assertEqual(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.assertEqual(s, '\x00\x00\x02\x00' + ('a' * 512))
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), val)
def test_longstr_unicode(self):
val = u'a' * 512
w = AMQPWriter()
w.write_longstr(val)
s = w.getvalue()
self.assertEqual(s, '\x00\x00\x02\x00' + ('a' * 512))
r = AMQPReader(s)
self.assertEqual(r.read_longstr(), val)
