port from SPARK-19017

nsyca · nsyca · commit 51f7fb92e47e · 2017-01-12T20:16:05.000-05:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/SQLQueryTestSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/SQLQueryTestSuite.scala
@@ -163,7 +163,12 @@ class SQLQueryTestSuite extends QueryTest with SharedSQLContext {
         s"-- Number of queries: ${outputs.size}\n\n\n" +
         outputs.zipWithIndex.map{case (qr, i) => qr.toString(i)}.mkString("\n\n\n") + "\n"
       }
-      stringToFile(new File(testCase.resultFile), goldenOutput)
+      val resultFile = new File(testCase.resultFile);
+      val parent = resultFile.getParentFile();
+      if (!parent.exists()) {
+        assert(parent.mkdirs(), "Could not create directory: " + parent)
+      }
+      stringToFile(resultFile, goldenOutput)
     }
 
     // Read back the golden file.

Original file line number	Diff line number	Diff line change
`@@ -163,7 +163,12 @@ class SQLQueryTestSuite extends QueryTest with SharedSQLContext {`
`163`	`163`	`s"-- Number of queries: ${outputs.size}\n\n\n" +`
`164`	`164`	`outputs.zipWithIndex.map{case (qr, i) => qr.toString(i)}.mkString("\n\n\n") + "\n"`
`165`	`165`	`}`
`166`		`- stringToFile(new File(testCase.resultFile), goldenOutput)`
	`166`	`+ val resultFile = new File(testCase.resultFile);`
	`167`	`+ val parent = resultFile.getParentFile();`
	`168`	`+ if (!parent.exists()) {`
	`169`	`+ assert(parent.mkdirs(), "Could not create directory: " + parent)`
	`170`	`+ }`
	`171`	`+ stringToFile(resultFile, goldenOutput)`
`167`	`172`	`}`
`168`	`173`
`169`	`174`	`// Read back the golden file.`