Fixes

datafusion-contrib · Mar 1, 2025 · d7ceb5d · d7ceb5d
1 parent 7d16e59
commit d7ceb5d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/execution/local.rs b/src/execution/local.rs
@@ -95,8 +95,8 @@ impl ExecutionContext {
         #[allow(unused_mut)]
         let mut session_ctx = SessionContext::new_with_state(session_state);
 
-        // #[cfg(feature = "functions-json")]
-        // datafusion_functions_json::register_all(&mut session_ctx)?;
+        #[cfg(feature = "functions-json")]
+        datafusion_functions_json::register_all(&mut session_ctx)?;
 
         // Register Parquet Metadata Function
         let session_ctx = session_ctx.enable_url_table();

diff --git a/tests/extension_cases/functions_json.rs b/tests/extension_cases/functions_json.rs
@@ -34,7 +34,7 @@ CREATE TABLE test_table (
 /// Ensure one of the functions `json_contains` function is properly registered
 #[tokio::test(flavor = "multi_thread")]
 async fn test_basic() {
-    let mut execution = TestExecution::new().await.with_setup(TEST_TABLE).await;
+    let execution = TestExecution::new().await.with_setup(TEST_TABLE).await;
 
     let actual = execution
         .run_and_format("SELECT id, json_contains(json_col, 'b') as json_contains FROM test_table")
@@ -56,7 +56,7 @@ async fn test_basic() {
 /// ensure the json operators like -> are properly registered
 #[tokio::test(flavor = "multi_thread")]
 async fn test_operators() {
-    let mut execution = TestExecution::new().await.with_setup(TEST_TABLE).await;
+    let execution = TestExecution::new().await.with_setup(TEST_TABLE).await;
 
     let actual = execution
         .run_and_format("SELECT id, json_col->'a' as json_col_a FROM test_table")