[samza] branch master updated: Fix ZkLocalApplicationRunner regression introduced by PR #951 (#1002)

[shanthoosh]

This is an automated email from the ASF dual-hosted git repository.

shanthoosh pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/samza.git


The following commit(s) were added to refs/heads/master by this push:
     new cc9acb2  Fix ZkLocalApplicationRunner regression introduced by PR #951 
(#1002)
cc9acb2 is described below

commit cc9acb2ee7c64214e21bf46791c74d137594d940
Author: mynameborat <46942335+mynamebo...@users.noreply.github.com>
AuthorDate: Thu Apr 18 17:32:29 2019 -0700

    Fix ZkLocalApplicationRunner regression introduced by PR #951 (#1002)
---
 .../apache/samza/test/processor/TestZkLocalApplicationRunner.java | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git 
a/samza-test/src/test/java/org/apache/samza/test/processor/TestZkLocalApplicationRunner.java
 
b/samza-test/src/test/java/org/apache/samza/test/processor/TestZkLocalApplicationRunner.java
index ad0b94c..e7c3ef8 100644
--- 
a/samza-test/src/test/java/org/apache/samza/test/processor/TestZkLocalApplicationRunner.java
+++ 
b/samza-test/src/test/java/org/apache/samza/test/processor/TestZkLocalApplicationRunner.java
@@ -153,10 +153,16 @@ public class TestZkLocalApplicationRunner extends 
IntegrationTestHarness {
     zkUtils = new ZkUtils(zkKeyBuilder, zkClient, ZK_CONNECTION_TIMEOUT_MS, 
ZK_SESSION_TIMEOUT_MS, new NoOpMetricsRegistry());
     zkUtils.connect();
 
+    ImmutableMap<String, Integer> topicToPartitionCount = ImmutableMap.of(
+        inputSinglePartitionKafkaTopic, 1,
+        outputSinglePartitionKafkaTopic, 1,
+        inputKafkaTopic, 5,
+        outputKafkaTopic, 5);
+
     List<NewTopic> newTopics =
         ImmutableList.of(inputKafkaTopic, outputKafkaTopic, 
inputSinglePartitionKafkaTopic, outputSinglePartitionKafkaTopic)
             .stream()
-            .map(topic -> new NewTopic(topic, 5, (short) 1))
+            .map(topic -> new NewTopic(topic, 
topicToPartitionCount.get(topic), (short) 1))
             .collect(Collectors.toList());
 
     assertTrue("Encountered errors during test setup. Failed to create 
topics.", createTopics(newTopics));