def test_message_snappy(self):
msg = KafkaClient.create_snappy_message("one", "two", "three")
enc = KafkaClient.encode_message(msg)
(messages, read) = KafkaClient.read_message_set(enc)
self.assertEquals(len(messages), 3)
self.assertEquals(messages[0].payload, "one")
self.assertEquals(messages[1].payload, "two")
self.assertEquals(messages[2].payload, "three")
def test_message_snappy(self):
msg = KafkaClient.create_snappy_message("one", "two", "three")
enc = KafkaClient.encode_message(msg)
(messages, read) = KafkaClient.read_message_set(enc)
self.assertEquals(len(messages), 3)
self.assertEquals(messages[0].payload, "one")
self.assertEquals(messages[1].payload, "two")
self.assertEquals(messages[2].payload, "three")
def test_message_snappy_random(self):
for i in xrange(ITERATIONS):
n = random.randint(1, 10)
strings = [random_string() for j in range(n)]
msg = KafkaClient.create_snappy_message(*strings)
enc = KafkaClient.encode_message(msg)
(messages, read) = KafkaClient.read_message_set(enc)
self.assertEquals(len(messages), n)
for j in range(n):
self.assertEquals(messages[j].payload, strings[j])
def test_message_snappy_random(self):
for i in xrange(ITERATIONS):
n = random.randint(1, 10)
strings = [random_string() for j in range(n)]
msg = KafkaClient.create_snappy_message(*strings)
enc = KafkaClient.encode_message(msg)
(messages, read) = KafkaClient.read_message_set(enc)
self.assertEquals(len(messages), n)
for j in range(n):
self.assertEquals(messages[j].payload, strings[j])
def test_create_snappy(self):
msg = KafkaClient.create_snappy_message("testing")
self.assertEquals(msg.magic, 1)
self.assertEquals(msg.attributes, 2)
self.assertEquals(msg.crc, -62350868)
(messages, _) = KafkaClient.read_message_set(snappy_decode(msg.payload))
inner = messages[0]
self.assertEquals(inner.magic, 1)
self.assertEquals(inner.attributes, 0)
self.assertEquals(inner.payload, "testing")
self.assertEquals(inner.crc, -386704890)
def test_create_snappy(self):
msg = KafkaClient.create_snappy_message("testing")
self.assertEquals(msg.magic, 1)
self.assertEquals(msg.attributes, 2)
self.assertEquals(msg.crc, -62350868)
(messages,
_) = KafkaClient.read_message_set(snappy_decode(msg.payload))
inner = messages[0]
self.assertEquals(inner.magic, 1)
self.assertEquals(inner.attributes, 0)
self.assertEquals(inner.payload, "testing")
self.assertEquals(inner.crc, -386704890)
