diff options
Diffstat (limited to 'exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java')
-rw-r--r-- | exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java b/exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java index 37e05cd7c..6c4cdadb4 100644 --- a/exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java +++ b/exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java @@ -30,6 +30,7 @@ import java.util.HashMap; import java.util.List; import java.util.Map; +import org.apache.commons.lang3.StringUtils; import org.apache.commons.lang3.tuple.Pair; import org.apache.drill.common.expression.LogicalExpression; import org.apache.drill.common.expression.SchemaPath; @@ -70,7 +71,7 @@ public class TestBuilder { protected Map<SchemaPath, TypeProtos.MajorType> baselineTypeMap; // queries to run before the baseline or test queries, can be used to set options private String baselineOptionSettingQueries; - private String testOptionSettingQueries; + private String testOptionSettingQueries = ""; // two different methods are available for comparing ordered results, the default reads all of the records // into giant lists of objects, like one giant on-heap batch of 'vectors' // this flag enables the other approach which iterates through a hyper batch for the test query results and baseline @@ -110,7 +111,9 @@ public class TestBuilder { this.approximateEquality = approximateEquality; this.baselineTypeMap = baselineTypeMap; this.baselineOptionSettingQueries = baselineOptionSettingQueries; - this.testOptionSettingQueries = testOptionSettingQueries; + this.testOptionSettingQueries = StringUtils.isNotEmpty(testOptionSettingQueries) + ? testOptionSettingQueries.concat(" ; ") + : testOptionSettingQueries; this.highPerformanceComparison = highPerformanceComparison; this.expectedNumBatches = expectedNumBatches; } @@ -205,12 +208,12 @@ public class TestBuilder { */ public TestBuilder optionSettingQueriesForTestQuery(String queries) { - testOptionSettingQueries = queries; + testOptionSettingQueries += queries.concat(" ; "); return this; } public TestBuilder optionSettingQueriesForTestQuery(String query, Object... args) throws Exception { - testOptionSettingQueries = String.format(query, args); + testOptionSettingQueries += String.format(query, args).concat(" ; "); return this; } |