feat: add test to check for ctx.enable_url_table()

ballista/client/tests/context_unsupported.rs

@@ -211,4 +211,45 @@ mod unsupported {
 
         assert_batches_eq!(expected, &result);
     }
+
+    /// looks like `ctx.enable_url_table()` changes session context id.
+    ///
+    /// Error returned:
+    /// ```
+    /// Failed to load SessionContext for session ID b5530099-63d1-43b1-9e11-87ac83bb33e5:
+    /// General error: No session for b5530099-63d1-43b1-9e11-87ac83bb33e5 found
+    /// ```
+    #[rstest]
+    #[case::standalone(standalone_context())]
+    #[case::remote(remote_context())]
+    #[tokio::test]
+    #[should_panic]
+    async fn should_execute_sql_show_with_url_table(
+        #[future(awt)]
+        #[case]
+        ctx: SessionContext,
+        test_data: String,
+    ) {
+        let ctx = ctx.enable_url_table();
+
+        let result = ctx
+            .sql(&format!("select string_col, timestamp_col from '{test_data}/alltypes_plain.parquet' where id > 4"))
+            .await
+            .unwrap()
+            .collect()
+            .await
+            .unwrap();
+
+        let expected = [
+            "+------------+---------------------+",
+            "| string_col | timestamp_col       |",
+            "+------------+---------------------+",
+            "| 31         | 2009-03-01T00:01:00 |",
+            "| 30         | 2009-04-01T00:00:00 |",
+            "| 31         | 2009-04-01T00:01:00 |",
+            "+------------+---------------------+",
+        ];
+
+        assert_batches_eq!(expected, &result);
+    }
 }