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_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)