From b293a2ac4b89298d9db08781da38463c4b261d1e Mon Sep 17 00:00:00 2001
From: Guozhang Wang <wangguoz@gmail.com>
Date: Mon, 11 Jul 2016 15:42:38 -0700
Subject: [PATCH] Check null in SmokeTestDriver

---
 .../org/apache/kafka/streams/smoketest/SmokeTestDriver.java   | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/streams/src/test/java/org/apache/kafka/streams/smoketest/SmokeTestDriver.java b/streams/src/test/java/org/apache/kafka/streams/smoketest/SmokeTestDriver.java
index 205ba4b85f6b6..b22d8a776701f 100644
--- a/streams/src/test/java/org/apache/kafka/streams/smoketest/SmokeTestDriver.java
+++ b/streams/src/test/java/org/apache/kafka/streams/smoketest/SmokeTestDriver.java
@@ -379,7 +379,7 @@ private static boolean verifyDif(Map<String, Integer> map, Map<String, Set<Integ
                 int min = getMin(entry.getKey());
                 int max = getMax(entry.getKey());
                 int expected = max - min;
-                if (expected != entry.getValue()) {
+                if (entry.getValue() == null || expected != entry.getValue()) {
                     System.out.println("fail: key=" + entry.getKey() + " dif=" + entry.getValue() + " expected=" + expected);
                     success = false;
                 }
@@ -455,7 +455,7 @@ private static boolean verifyAvg(Map<String, Double> map, Map<String, Set<Intege
                 int max = getMax(entry.getKey());
                 double expected = ((long) min + (long) max) / 2.0;
 
-                if (expected != entry.getValue()) {
+                if (entry.getValue() == null || expected != entry.getValue()) {
                     System.out.println("fail: key=" + entry.getKey() + " avg=" + entry.getValue() + " expected=" + expected);
                     success = false;
                 }