From e0860661b116718ba89b714d6218232f1bd6aa0d Mon Sep 17 00:00:00 2001
From: saydakov <saydakov@yahoo-inc.com>
Date: Thu, 28 Jan 2016 13:20:50 -0800
Subject: [PATCH] fixed createIndex

---
 .../aggregation/AggregationTestHelper.java     | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/processing/src/test/java/io/druid/query/aggregation/AggregationTestHelper.java b/processing/src/test/java/io/druid/query/aggregation/AggregationTestHelper.java
index f0cc8a320fa..2cd82085b0e 100644
--- a/processing/src/test/java/io/druid/query/aggregation/AggregationTestHelper.java
+++ b/processing/src/test/java/io/druid/query/aggregation/AggregationTestHelper.java
@@ -244,22 +244,20 @@ public class AggregationTestHelper
       index = new OnheapIncrementalIndex(minTimestamp, gran, metrics, deserializeComplexMetrics, maxRowCount);
       while (rows.hasNext()) {
         Object row = rows.next();
-        try {
-          if (row instanceof String && parser instanceof StringInputRowParser) {
-            //Note: this is required because StringInputRowParser is InputRowParser<ByteBuffer> as opposed to
-            //InputRowsParser<String>
-            index.add(((StringInputRowParser) parser).parse((String) row));
-          } else {
-            index.add(parser.parse(row));
-          }
-        }
-        catch (IndexSizeExceededException ex) {
+        if (!index.canAppendRow()) {
           File tmp = tempFolder.newFolder();
           toMerge.add(tmp);
           indexMerger.persist(index, tmp, null, new IndexSpec());
           index.close();
           index = new OnheapIncrementalIndex(minTimestamp, gran, metrics, deserializeComplexMetrics, maxRowCount);
         }
+        if (row instanceof String && parser instanceof StringInputRowParser) {
+          //Note: this is required because StringInputRowParser is InputRowParser<ByteBuffer> as opposed to
+          //InputRowsParser<String>
+          index.add(((StringInputRowParser) parser).parse((String) row));
+        } else {
+          index.add(parser.parse(row));
+        }
       }
 
       if (toMerge.size() > 0) {