def test_basic_send(self):
conn = kiddiepool.KiddieConnection()
conn.host = 'localhost'
conn.port = 9092
conn.validate().AndReturn(False)
conn.connect('localhost', 9092).AndReturn(True)
payload1 = kafkap.encode_produce_request('topic1',
kafkap.DEFAULT_PARTITION,
['message1'])
conn.sendall(payload1)
conn.validate().AndReturn(True)
payload2 = kafkap.encode_produce_request('topic2',
kafkap.DEFAULT_PARTITION,
['message2'])
conn.sendall(payload2)
self.mox.ReplayAll()
# Must set connection_factory since KafkaPool.connection_factory is
# still the unmocked (real) class
p = kiddiepool.KiddiePool(
['localhost:9092'],
max_size=1,
connection_factory=kiddiepool.KiddieConnection)
c = kafkap.KafkaClient(p)
c.send('message1', 'topic1')
c.sendmulti(['message2'], 'topic2')
def test_request_encoding(self):
tests = (
(('1', -1, ['a']), ('\x00\x00\x00\x17\x00\x00\x00\x011\xff\xff\xff'
'\xff\x00\x00\x00\n\x00\x00\x00\x06\x00\xe8\xb7\xbeCa')),
(('2', 99, ['a', 'b', 'c']), ('\x00\x00\x00+\x00\x00\x00\x012\x00'
'\x00\x00c\x00\x00\x00\x1e\x00\x00\x00\x06\x00\xe8\xb7\xbeCa'
'\x00\x00\x00\x06\x00q\xbe\xef\xf9b\x00\x00\x00\x06\x00\x06'
'\xb9\xdfoc')),
)
for args, bits in tests:
self.assertEqual(kafkap.encode_produce_request(*args), bits)
def test_basic_send(self):
conn = kiddiepool.KiddieConnection()
conn.host = "localhost"
conn.port = 9092
conn.validate().AndReturn(False)
conn.connect("localhost", 9092).AndReturn(True)
payload1 = kafkap.encode_produce_request("topic1", kafkap.DEFAULT_PARTITION, ["message1"])
conn.sendall(payload1)
conn.validate().AndReturn(True)
payload2 = kafkap.encode_produce_request("topic2", kafkap.DEFAULT_PARTITION, ["message2"])
conn.sendall(payload2)
self.mox.ReplayAll()
# Must set connection_factory since KafkaPool.connection_factory is
# still the unmocked (real) class
p = kiddiepool.KiddiePool(["localhost:9092"], max_size=1, connection_factory=kiddiepool.KiddieConnection)
c = kafkap.KafkaClient(p)
c.send("message1", "topic1")
c.sendmulti(["message2"], "topic2")
def test_request_encoding(self):
tests = (
(('1', -1, ['a']),
('\x00\x00\x00\x17\x00\x00\x00\x011\xff\xff\xff'
'\xff\x00\x00\x00\n\x00\x00\x00\x06\x00\xe8\xb7\xbeCa')),
(('2', 99, ['a', 'b', 'c']),
('\x00\x00\x00+\x00\x00\x00\x012\x00'
'\x00\x00c\x00\x00\x00\x1e\x00\x00\x00\x06\x00\xe8\xb7\xbeCa'
'\x00\x00\x00\x06\x00q\xbe\xef\xf9b\x00\x00\x00\x06\x00\x06'
'\xb9\xdfoc')),
)
for args, bits in tests:
self.assertEqual(kafkap.encode_produce_request(*args), bits)
def test_basic_send(self):
conn = kiddiepool.KiddieConnection()
conn.validate().AndReturn(False)
conn.connect('localhost', 9092).AndReturn(True)
payload1 = kafkap.encode_produce_request(
'topic1', kafkap.DEFAULT_PARTITION, ['message1'])
conn.sendall(payload1)
conn.validate().AndReturn(True)
payload2 = kafkap.encode_produce_request(
'topic2', kafkap.DEFAULT_PARTITION, ['message2'])
conn.sendall(payload2)
self.mox.ReplayAll()
# Must set connection_factory since KafkaPool.connection_factory is
# still the unmocked (real) class
p = kiddiepool.KiddiePool(['localhost:9092'], max_size=1,
connection_factory=kiddiepool.KiddieConnection)
c = kafkap.KafkaClient(p)
c.send('message1', 'topic1')
c.sendmulti(['message2'], 'topic2')
def test_request_encoding(self):
tests = (
(
("1", -1, ["a"]),
(
"\x00\x00\x00\x17\x00\x00\x00\x011\xff\xff\xff"
"\xff\x00\x00\x00\n\x00\x00\x00\x06\x00\xe8\xb7\xbeCa"
),
),
(
("2", 99, ["a", "b", "c"]),
(
"\x00\x00\x00+\x00\x00\x00\x012\x00"
"\x00\x00c\x00\x00\x00\x1e\x00\x00\x00\x06\x00\xe8\xb7\xbeCa"
"\x00\x00\x00\x06\x00q\xbe\xef\xf9b\x00\x00\x00\x06\x00\x06"
"\xb9\xdfoc"
),
),
)
for args, bits in tests:
self.assertEqual(kafkap.encode_produce_request(*args), bits)