Spark: Cached ClickHouse table is not used by SQL (#274) (#275)

ClickHouse · Oct 27, 2023 · 5e83d0c · 5e83d0c
1 parent fed4cd2
commit 5e83d0c
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/...park-it/src/test/scala/org/apache/spark/sql/clickhouse/single/ClickHouseSingleSuite.scala b/...park-it/src/test/scala/org/apache/spark/sql/clickhouse/single/ClickHouseSingleSuite.scala
@@ -436,4 +436,19 @@ class ClickHouseSingleSuite extends SparkClickHouseSingleTest {
       createOrReplaceTable()
     }
   }
+
+  test("cache table") {
+    val db = "cache_db"
+    val tbl = "cache_tbl"
+
+    withSimpleTable(db, tbl, true) {
+      try {
+        spark.sql(s"CACHE TABLE $db.$tbl")
+        val cachedPlan = spark.sql(s"SELECT * FROM $db.$tbl").queryExecution.commandExecuted
+          .find(node => spark.sharedState.cacheManager.lookupCachedData(node).isDefined)
+        assert(cachedPlan.isDefined)
+      } finally
+        spark.sql(s"UNCACHE TABLE $db.$tbl")
+    }
+  }
 }