strange, registering with json_table seems to break datafusion

dadepo · Feb 3, 2024 · 3a36962 · 3a36962
1 parent d273b9b
commit 3a36962
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/common/test_utils.rs b/src/common/test_utils.rs
@@ -99,7 +99,7 @@ pub fn set_up_json_data_test() -> Result<SessionContext> {
     // declare a new context
     let ctx = SessionContext::new();
     // declare a table in memory.
-    ctx.register_batch("json_table", batch)?;
+    ctx.register_batch("json_values_table", batch)?;
 
     // data for json_type
     let schema = Arc::new(Schema::new(vec![

diff --git a/src/sqlite/json_udfs.rs b/src/sqlite/json_udfs.rs
@@ -133,7 +133,7 @@ mod tests {
         let ctx = register_udfs_for_test()?;
         let df = ctx
             .sql(
-                r#"select index, json(json_data) as col_result FROM json_table ORDER BY index ASC"#,
+                r#"select index, json(json_data) as col_result FROM json_values_table ORDER BY index ASC"#,
             )
             .await?;
 
@@ -170,7 +170,7 @@ mod tests {
         let ctx = register_udfs_for_test()?;
         let df = ctx
             .sql(
-                r#"select index, json(' { "this" : "is", "a": [ "test" ]  ') as col_result FROM json_table ORDER BY index ASC"#,
+                r#"select index, json(' { "this" : "is", "a": [ "test" ]  ') as col_result FROM json_values_table ORDER BY index ASC"#,
             )
             .await?;
 
@@ -190,7 +190,7 @@ mod tests {
     #[tokio::test]
     async fn test_json_valid() -> Result<()> {
         let ctx = register_udfs_for_test()?;
-        let df = ctx.sql(r#"select index, json_valid(json_data) as col_result FROM json_table ORDER BY index ASC"#).await?;
+        let df = ctx.sql(r#"select index, json_valid(json_data) as col_result FROM json_values_table ORDER BY index ASC"#).await?;
 
         let batches = df.clone().collect().await?;
 
@@ -225,7 +225,7 @@ mod tests {
         let ctx = register_udfs_for_test()?;
         let df = ctx
             .sql(
-                r#"select index, json_type(json_data) as col_result FROM json_table ORDER BY index ASC"#,
+                r#"select index, json_type(json_data) as col_result FROM json_values_table ORDER BY index ASC"#,
             )
             .await?;