aboutsummaryrefslogtreecommitdiff
path: root/sql/hive
diff options
context:
space:
mode:
authorWenchen Fan <wenchen@databricks.com>2016-03-18 23:16:14 +0800
committerCheng Lian <lian@databricks.com>2016-03-18 23:16:14 +0800
commit0acb32a3f19484d3ea3b60fccef596025c8a8f83 (patch)
treea5dcc5ce805b24bf10deb2d6d241b4af8c08439b /sql/hive
parent53f32a22daa40b713fef477054290d8adbeb6f71 (diff)
downloadspark-0acb32a3f19484d3ea3b60fccef596025c8a8f83.tar.gz
spark-0acb32a3f19484d3ea3b60fccef596025c8a8f83.tar.bz2
spark-0acb32a3f19484d3ea3b60fccef596025c8a8f83.zip
[SPARK-13972][SQ] hive tests should fail if SQL generation failed
## What changes were proposed in this pull request? Now we should be able to convert all logical plans to SQL string, if they are parsed from hive query. This PR changes the error handling to throw exceptions instead of just log. We will send new PRs for spotted bugs, and merge this one after all bugs are fixed. ## How was this patch tested? existing tests. Author: Wenchen Fan <wenchen@databricks.com> Closes #11782 from cloud-fan/test.
Diffstat (limited to 'sql/hive')
-rw-r--r--sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/HiveComparisonTest.scala71
1 files changed, 28 insertions, 43 deletions
diff --git a/sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/HiveComparisonTest.scala b/sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/HiveComparisonTest.scala
index d21bb573d4..019d752a13 100644
--- a/sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/HiveComparisonTest.scala
+++ b/sql/hive/src/test/scala/org/apache/spark/sql/hive/execution/HiveComparisonTest.scala
@@ -132,25 +132,7 @@ abstract class HiveComparisonTest
new java.math.BigInteger(1, digest.digest).toString(16)
}
- /** Used for testing [[SQLBuilder]] */
- private var numConvertibleQueries: Int = 0
- private var numTotalQueries: Int = 0
-
override protected def afterAll(): Unit = {
- logInfo({
- val percentage = if (numTotalQueries > 0) {
- numConvertibleQueries.toDouble / numTotalQueries * 100
- } else {
- 0D
- }
-
- s"""SQLBuilder statistics:
- |- Total query number: $numTotalQueries
- |- Number of convertible queries: $numConvertibleQueries
- |- Percentage of convertible queries: $percentage%
- """.stripMargin
- })
-
try {
TestHive.reset()
} finally {
@@ -412,32 +394,35 @@ abstract class HiveComparisonTest
if (containsCommands) {
originalQuery
} else {
- numTotalQueries += 1
+ val convertedSQL = try {
+ new SQLBuilder(originalQuery.analyzed, TestHive).toSQL
+ } catch {
+ case NonFatal(e) => fail(
+ s"""Cannot convert the following HiveQL query plan back to SQL query string:
+ |
+ |# Original HiveQL query string:
+ |$queryString
+ |
+ |# Resolved query plan:
+ |${originalQuery.analyzed.treeString}
+ """.stripMargin, e)
+ }
+
try {
- val sql = new SQLBuilder(originalQuery.analyzed, TestHive).toSQL
- numConvertibleQueries += 1
- logInfo(
- s"""
- |### Running SQL generation round-trip test {{{
- |${originalQuery.analyzed.treeString}
- |Original SQL:
- |$queryString
- |
- |Generated SQL:
- |$sql
- |}}}
- """.stripMargin.trim)
- new TestHive.QueryExecution(sql)
- } catch { case NonFatal(e) =>
- logInfo(
- s"""
- |### Cannot convert the following logical plan back to SQL {{{
- |${originalQuery.analyzed.treeString}
- |Original SQL:
- |$queryString
- |}}}
- """.stripMargin.trim)
- originalQuery
+ new TestHive.QueryExecution(convertedSQL)
+ } catch {
+ case NonFatal(e) => fail(
+ s"""Failed to analyze the converted SQL string:
+ |
+ |# Original HiveQL query string:
+ |$queryString
+ |
+ |# Resolved query plan:
+ |${originalQuery.analyzed.treeString}
+ |
+ |# Converted SQL query string:
+ |$convertedSQL
+ """.stripMargin, e)
}
}
}