diff --git a/sql/core/src/test/scala/org/apache/spark/sql/QueryGeneratorHelper.scala b/sql/core/src/test/scala/org/apache/spark/sql/QueryGeneratorHelper.scala index 01ce0c07e93c7..a17f3a353569f 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/QueryGeneratorHelper.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/QueryGeneratorHelper.scala @@ -200,7 +200,6 @@ trait QueryGeneratorHelper { /** * Case class that encapsulates a Query, built with its different clauses. - * @param comment A string that will be printed as a comment a line before the query string. */ case class Query( selectClause: SelectClause, @@ -215,10 +214,9 @@ trait QueryGeneratorHelper { def getOptionClauseString(optionalClause: Option[Clause]): String = if (optionalClause.isDefined) { " " + optionalClause.get.toString } else { "" } - val queryString = f"$selectClause $fromClause${getOptionClauseString(whereClause)}" + + f"$selectClause $fromClause${getOptionClauseString(whereClause)}" + f"${getOptionClauseString(groupByClause)}${getOptionClauseString(orderByClause)}" + f"${getOptionClauseString(limitClause)}" - queryString } } }