From 818b48f2d3d30abde008e2e9581a5f12c3eeb420 Mon Sep 17 00:00:00 2001 From: Joel Koshy Date: Thu, 18 Dec 2014 17:11:00 +0530 Subject: [PATCH] KAFKA-1823; Fix transient PartitionAssignorTest failure (triggered when there are no topics) --- .../test/scala/unit/kafka/consumer/PartitionAssignorTest.scala | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/core/src/test/scala/unit/kafka/consumer/PartitionAssignorTest.scala b/core/src/test/scala/unit/kafka/consumer/PartitionAssignorTest.scala index 24954de..543070f 100644 --- a/core/src/test/scala/unit/kafka/consumer/PartitionAssignorTest.scala +++ b/core/src/test/scala/unit/kafka/consumer/PartitionAssignorTest.scala @@ -187,10 +187,12 @@ private object PartitionAssignorTest extends Logging { // check for uniform assignment if (verifyAssignmentIsUniform) { val partitionCountForStream = partitionCountPerStream(globalAssignment) - val maxCount = partitionCountForStream.valuesIterator.max - val minCount = partitionCountForStream.valuesIterator.min - assertTrue("Scenario %s: assignment is not uniform (partition counts per stream are in the range [%d, %d])" - .format(scenario, minCount, maxCount), (maxCount - minCount) <= 1) + if (partitionCountForStream.nonEmpty) { + val maxCount = partitionCountForStream.valuesIterator.max + val minCount = partitionCountForStream.valuesIterator.min + assertTrue("Scenario %s: assignment is not uniform (partition counts per stream are in the range [%d, %d])" + .format(scenario, minCount, maxCount), (maxCount - minCount) <= 1) + } } } -- 2.1.2