def test_producer_round_robin_partitioner_random_start(self):
try:
RoundRobinPartitioner.set_random_start(True)
producer = Producer(self.client,
partitioner_class=RoundRobinPartitioner)
# Two partitions, so 1st and 3rd reqs should go to same part, but
# 2nd should go to different. Other than that, without statistical
# test, we can't say much... Partitioner tests should ensure that
# we really aren't always starting on a non-random partition...
resp1 = yield producer.send_messages(
self.topic, msgs=[self.msg("one"), self.msg("two")])
resp2 = yield producer.send_messages(
self.topic, msgs=[self.msg("three")])
resp3 = yield producer.send_messages(
self.topic, msgs=[self.msg("four"), self.msg("five")])
self.assertEqual(resp1.partition, resp3.partition)
self.assertNotEqual(resp1.partition, resp2.partition)
yield producer.stop()
finally:
RoundRobinPartitioner.set_random_start(False)
def test_producer_round_robin_partitioner_random_start(self):
try:
RoundRobinPartitioner.set_random_start(True)
producer = Producer(self.client,
partitioner_class=RoundRobinPartitioner)
# Two partitions, so 1st and 3rd reqs should go to same part, but
# 2nd should go to different. Other than that, without statistical
# test, we can't say much... Partitioner tests should ensure that
# we really aren't always starting on a non-random partition...
resp1 = yield producer.send_messages(
self.topic, msgs=[self.msg("one"), self.msg("two")])
resp2 = yield producer.send_messages(
self.topic, msgs=[self.msg("three")])
resp3 = yield producer.send_messages(
self.topic, msgs=[self.msg("four"), self.msg("five")])
self.assertEqual(resp1.partition, resp3.partition)
self.assertNotEqual(resp1.partition, resp2.partition)
yield producer.stop()
finally:
RoundRobinPartitioner.set_random_start(False)