aboutsummaryrefslogtreecommitdiff
path: root/exec/java-exec/src/test/java/org/apache/drill/test/TestBuilder.java
diff options
context:
space:
mode:
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.java11
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;
}