Moving sql test on core to EXPLAIN

clockworklabs · Jan 30, 2025 · 46082fa · 46082fa
1 parent c592668
commit 46082fa
Show file tree

Hide file tree

Showing 3 changed files with 329 additions and 338 deletions.
diff --git a/crates/core/src/db/relational_db.rs b/crates/core/src/db/relational_db.rs
@@ -1314,10 +1314,15 @@ fn default_row_count_fn(db: Identity) -> RowCountFn {
 #[cfg(any(test, feature = "test"))]
 pub mod tests_utils {
     use super::*;
+    use crate::sql::ast::{SchemaViewer, TableSchemaView};
     use core::ops::Deref;
     use durability::EmptyHistory;
+    use expect_test::Expect;
+    use spacetimedb_lib::identity::AuthCtx;
     use spacetimedb_lib::{bsatn::to_vec, ser::Serialize};
     use spacetimedb_paths::FromPathUnchecked;
+    use spacetimedb_physical_plan::plan::tests_utils::*;
+    use spacetimedb_physical_plan::printer::ExplainOptions;
     use tempfile::TempDir;
 
     /// A [`RelationalDB`] in a temporary directory.
@@ -1605,6 +1610,20 @@ pub mod tests_utils {
             Self(log)
         }
     }
+
+    pub fn expect_query<T: TableSchemaView + StateView>(tx: &T, sql: &str, expect: Expect) {
+        let auth = AuthCtx::for_testing();
+        let schema = SchemaViewer::new(tx, &auth);
+
+        check_query(&schema, ExplainOptions::default(), sql, expect)
+    }
+
+    pub fn expect_sub<T: TableSchemaView + StateView>(tx: &T, sql: &str, expect: Expect) {
+        let auth = AuthCtx::for_testing();
+        let schema = SchemaViewer::new(tx, &auth);
+
+        check_sub(&schema, ExplainOptions::default(), sql, expect)
+    }
 }
 
 #[cfg(test)]