test: add some tests for ProofPlan::get_table_references

spaceandtimelabs · Oct 27, 2024 · 464f3fd · 464f3fd
1 parent c7a9839
commit 464f3fd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/crates/proof-of-sql/src/sql/proof_plans/filter_exec_test.rs b/crates/proof-of-sql/src/sql/proof_plans/filter_exec_test.rs
@@ -153,6 +153,10 @@ fn we_can_correctly_fetch_all_the_referenced_columns() {
             )
         ])
     );
+
+    let ref_tables = provable_ast.get_table_references();
+
+    assert_eq!(ref_tables, IndexSet::from_iter([table_ref]));
 }
 
 #[test]

diff --git a/crates/proof-of-sql/src/sql/proof_plans/projection_exec_test.rs b/crates/proof-of-sql/src/sql/proof_plans/projection_exec_test.rs
@@ -102,6 +102,10 @@ fn we_can_correctly_fetch_all_the_referenced_columns() {
             ),
         ])
     );
+
+    let ref_tables = provable_ast.get_table_references();
+
+    assert_eq!(ref_tables, IndexSet::from_iter([table_ref]));
 }
 
 #[test]
-Original file line number
+Diff line change
@@ Expand Up / @@ -153,6 +153,10 @@ fn we_can_correctly_fetch_all_the_referenced_columns() { @@
                 )
             ])
         );
+        let ref_tables = provable_ast.get_table_references();
+        assert_eq!(ref_tables, IndexSet::from_iter([table_ref]));
     }
     #[test]
@@ Expand Down @@